79 lines
1.7 KiB
Go
Executable File
79 lines
1.7 KiB
Go
Executable File
package storage
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"time"
|
|
|
|
"gitea.bel.blue/local/storage/resolve"
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
type SQLite struct {
|
|
*sql.DB
|
|
}
|
|
|
|
func NewSQLite(path string) (*SQLite, error) {
|
|
db, err := sql.Open("sqlite", path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ctx, can := context.WithTimeout(context.Background(), time.Minute)
|
|
defer can()
|
|
if err := db.PingContext(ctx); err != nil {
|
|
db.Close()
|
|
return nil, err
|
|
}
|
|
if _, err := db.ExecContext(ctx, `
|
|
CREATE TABLE IF NOT EXISTS data (
|
|
namespace TEXT,
|
|
key TEXT,
|
|
value TEXT
|
|
);
|
|
CREATE UNIQUE INDEX IF NOT EXISTS data_pk ON data (namespace, key);
|
|
`); err != nil {
|
|
db.Close()
|
|
return nil, err
|
|
}
|
|
return &SQLite{
|
|
DB: db,
|
|
}, nil
|
|
}
|
|
|
|
func (sqlite *SQLite) Get(key string, ns ...string) ([]byte, error) {
|
|
namespace := resolve.Namespace(ns)
|
|
row := sqlite.QueryRow(`SELECT value FROM data WHERE namespace=$1 AND key=$2`, namespace, key)
|
|
|
|
var value string
|
|
if err := row.Scan(&value); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return []byte(value), row.Err()
|
|
}
|
|
|
|
func (sqlite *SQLite) Set(key string, value []byte, ns ...string) error {
|
|
namespace := resolve.Namespace(ns)
|
|
_, err := sqlite.Exec(`INSERT INTO data (namespace, key, value) VALUES ($1, $2, $3)`, namespace, key, string(value))
|
|
return err
|
|
}
|
|
|
|
func (sqlite *SQLite) List(ns []string, limits ...string) ([]string, error) {
|
|
namespace := resolve.Namespace(ns)
|
|
rows, err := sqlite.Query(`SELECT key FROM data WHERE namespace=$1`, namespace)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := []string{}
|
|
for rows.Next() {
|
|
var a string
|
|
if err := rows.Scan(&a); err != nil {
|
|
return nil, err
|
|
}
|
|
result = append(result, a)
|
|
}
|
|
|
|
return result, rows.Err()
|
|
}
|