internal/storage/postgres.go

@@ -0,0 +1,115 @@
+package storage
+
+import (
+	"context"
+
+	"git.defalsify.org/vise.git/db"
+	postgres "git.defalsify.org/vise.git/db/postgres"
+	"git.defalsify.org/vise.git/lang"
+)
+
+var (
+	pdbC map[string]chan db.Db
+)
+
+type ThreadPostgresDb struct {
+	db      db.Db
+	connStr string
+}
+
+func NewThreadPostgresDb() *ThreadPostgresDb {
+	if pdbC == nil {
+		pdbC = make(map[string]chan db.Db)
+	}
+	return &ThreadPostgresDb{}
+}
+
+func (tpdb *ThreadPostgresDb) Connect(ctx context.Context, connStr string) error {
+	var ok bool
+	_, ok = pdbC[connStr]
+	if ok {
+		logg.WarnCtxf(ctx, "already registered thread postgres, skipping", "connStr", connStr)
+		return nil
+	}
+	postgresdb := postgres.NewPgDb().WithSchema("public")
+	err := postgresdb.Connect(ctx, connStr)
+	if err != nil {
+		return err
+	}
+	pdbC[connStr] = make(chan db.Db, 1)
+	pdbC[connStr] <- postgresdb
+	tpdb.connStr = connStr
+	return nil
+}
+
+func (tpdb *ThreadPostgresDb) reserve() {
+	if tpdb.db == nil {
+		tpdb.db = <-pdbC[tpdb.connStr]
+	}
+}
+
+func (tpdb *ThreadPostgresDb) release() {
+	if tpdb.db == nil {
+		return
+	}
+	pdbC[tpdb.connStr] <- tpdb.db
+	tpdb.db = nil
+}
+
+func (tpdb *ThreadPostgresDb) SetPrefix(pfx uint8) {
+	tpdb.reserve()
+	tpdb.db.SetPrefix(pfx)
+}
+
+func (tpdb *ThreadPostgresDb) SetSession(sessionId string) {
+	tpdb.reserve()
+	tpdb.db.SetSession(sessionId)
+}
+
+func (tpdb *ThreadPostgresDb) SetLanguage(lng *lang.Language) {
+	tpdb.reserve()
+	tpdb.db.SetLanguage(lng)
+}
+
+func (tpdb *ThreadPostgresDb) Safe() bool {
+	tpdb.reserve()
+	v := tpdb.db.Safe()
+	tpdb.release()
+	return v
+}
+
+func (tpdb *ThreadPostgresDb) Prefix() uint8 {
+	tpdb.reserve()
+	v := tpdb.db.Prefix()
+	tpdb.release()
+	return v
+}
+
+func (tpdb *ThreadPostgresDb) SetLock(typ uint8, locked bool) error {
+	tpdb.reserve()
+	err := tpdb.db.SetLock(typ, locked)
+	tpdb.release()
+	return err
+}
+
+func (tpdb *ThreadPostgresDb) Put(ctx context.Context, key []byte, val []byte) error {
+	tpdb.reserve()
+	err := tpdb.db.Put(ctx, key, val)
+	tpdb.release()
+	return err
+}
+
+func (tpdb *ThreadPostgresDb) Get(ctx context.Context, key []byte) ([]byte, error) {
+	tpdb.reserve()
+	v, err := tpdb.db.Get(ctx, key)
+	tpdb.release()
+	return v, err
+}
+
+func (tpdb *ThreadPostgresDb) Close() error {
+	tpdb.reserve()
+	close(pdbC[tpdb.connStr])
+	err := tpdb.db.Close()
+	tpdb.db = nil
+	return err
+}

internal/storage/storageservice.go

@@ -8,9 +8,10 @@ import (
 
 	"git.defalsify.org/vise.git/db"
 	fsdb "git.defalsify.org/vise.git/db/fs"
+	"git.defalsify.org/vise.git/logging"
 	"git.defalsify.org/vise.git/persist"
 	"git.defalsify.org/vise.git/resource"
-	"git.defalsify.org/vise.git/logging"
+	"git.grassecon.net/urdt/ussd/initializers"
 )
 
 var (
@@ -24,7 +25,7 @@ type StorageService interface {
 	EnsureDbDir() error
 }
 
-type MenuStorageService struct{
+type MenuStorageService struct {
 	dbDir         string
 	resourceDir   string
 	resourceStore db.Db
@@ -32,6 +33,19 @@ type MenuStorageService struct{
 	userDataStore db.Db
 }
 
+func buildConnStr() string {
+	host := initializers.GetEnv("DB_HOST", "localhost")
+	user := initializers.GetEnv("DB_USER", "postgres")
+	password := initializers.GetEnv("DB_PASSWORD", "")
+	dbName := initializers.GetEnv("DB_NAME", "")
+	port := initializers.GetEnv("DB_PORT", "5432")
+
+	return fmt.Sprintf(
+		"postgres://%s:%s@%s:%s/%s",
+		user, password, host, port, dbName,
+	)
+}
+
 func NewMenuStorageService(dbDir string, resourceDir string) *MenuStorageService {
 	return &MenuStorageService{
 		dbDir:       dbDir,
@@ -52,12 +66,27 @@ func (ms *MenuStorageService) GetPersister(ctx context.Context) (*persist.Persis
 }
 
 func (ms *MenuStorageService) GetUserdataDb(ctx context.Context) (db.Db, error) {
+	database, ok := ctx.Value("Database").(string)
+	if !ok {
+		fmt.Println("The database is not set")
+	}
+
+	if database == "postgres" {
+		ms.userDataStore = NewThreadPostgresDb()
+		connStr := buildConnStr()
+		err := ms.userDataStore.Connect(ctx, connStr)
+		if err != nil {
+			return nil, err
+		}
+	} else {
 		ms.userDataStore = NewThreadGdbmDb()
 		storeFile := path.Join(ms.dbDir, "userdata.gdbm")
 		err := ms.userDataStore.Connect(ctx, storeFile)
 		if err != nil {
 			return nil, err
 		}
+	}
+
 	return ms.userDataStore, nil
 }