aboutsummaryrefslogtreecommitdiff
path: root/weed/filer/sqlite/sqlite_store.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/filer/sqlite/sqlite_store.go')
-rw-r--r--weed/filer/sqlite/sqlite_store.go74
1 files changed, 74 insertions, 0 deletions
diff --git a/weed/filer/sqlite/sqlite_store.go b/weed/filer/sqlite/sqlite_store.go
new file mode 100644
index 000000000..b32d86c11
--- /dev/null
+++ b/weed/filer/sqlite/sqlite_store.go
@@ -0,0 +1,74 @@
+package sqlite
+
+import (
+ "context"
+ "database/sql"
+ "fmt"
+
+ "github.com/chrislusf/seaweedfs/weed/filer"
+ "github.com/chrislusf/seaweedfs/weed/filer/abstract_sql"
+ "github.com/chrislusf/seaweedfs/weed/filer/mysql"
+ "github.com/chrislusf/seaweedfs/weed/util"
+ _ "modernc.org/sqlite"
+)
+
+func init() {
+ filer.Stores = append(filer.Stores, &SqliteStore{})
+}
+
+type SqliteStore struct {
+ abstract_sql.AbstractSqlStore
+}
+
+func (store *SqliteStore) GetName() string {
+ return "sqlite"
+}
+
+func (store *SqliteStore) Initialize(configuration util.Configuration, prefix string) (err error) {
+ dbFile := configuration.GetString(prefix + "dbFile")
+ createTable := `CREATE TABLE IF NOT EXISTS "%s" (
+ dirhash BIGINT,
+ name VARCHAR(1000),
+ directory TEXT,
+ meta BLOB,
+ PRIMARY KEY (dirhash, name)
+ ) WITHOUT ROWID;`
+ upsertQuery := `INSERT INTO "%s"(dirhash,name,directory,meta)VALUES(?,?,?,?)
+ ON CONFLICT(dirhash,name) DO UPDATE SET
+ directory=excluded.directory,
+ meta=excluded.meta;
+ `
+ return store.initialize(
+ dbFile,
+ createTable,
+ upsertQuery,
+ )
+}
+
+func (store *SqliteStore) initialize(dbFile, createTable, upsertQuery string) (err error) {
+
+ store.SupportBucketTable = true
+ store.SqlGenerator = &mysql.SqlGenMysql{
+ CreateTableSqlTemplate: createTable,
+ DropTableSqlTemplate: "drop table `%s`",
+ UpsertQueryTemplate: upsertQuery,
+ }
+
+ var dbErr error
+ store.DB, dbErr = sql.Open("sqlite", dbFile)
+ if dbErr != nil {
+ store.DB.Close()
+ store.DB = nil
+ return fmt.Errorf("can not connect to %s error:%v", dbFile, err)
+ }
+
+ if err = store.DB.Ping(); err != nil {
+ return fmt.Errorf("connect to %s error:%v", dbFile, err)
+ }
+
+ if err = store.CreateTable(context.Background(), abstract_sql.DEFAULT_TABLE); err != nil {
+ return fmt.Errorf("init table %s: %v", abstract_sql.DEFAULT_TABLE, err)
+ }
+
+ return nil
+}