impl sqlite

sqlite.go

@@ -0,0 +1,78 @@
+package storage
+
+import (
+	"context"
+	"database/sql"
+	"time"
+
+	"gitea.inhome.blapointe.com/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()
+}