From 3c6585e38714d38530721ede83f9f44679d844a1 Mon Sep 17 00:00:00 2001 From: lash Date: Thu, 5 Sep 2024 04:07:45 +0100 Subject: [PATCH] Add missing user db create --- cmd/main.go | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/cmd/main.go b/cmd/main.go index 40dca71..7168d14 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -75,20 +75,27 @@ func getHandler(appFlags *asm.FlagParser, rs *resource.DbResource, pe *persist.P return ussdHandlers, nil } -func getDataPersister(dbDir string) (*persist.Persister, error) { +func getPersister(dbDir string) (*persist.Persister, error) { err = os.MkdirAll(dp, 0700) if err != nil { return nil, fmt.Errorf("state dir create exited with error: %v\n", err) } - dataStore := gdbmdb.NewGdbmDb() - dataStoreFile := path.Join(dbDir, "states.gdbm") - dataStore.Connect(ctx, dataStoreFile) - pr := persist.NewPersister(dataStore) + store := gdbmdb.NewGdbmDb() + storeFile := path.Join(dbDir, "states.gdbm") + store.Connect(ctx, storeFile) + pr := persist.NewPersister(store) return pr } +func getUserdataDb(dbDir string) { + store := gdbmdb.NewGdbmDb() + storeFile := path.Join(dbDir, "userdata.gdbm") + store.Connect(ctx, storeFile) + return store +} + func getResource(resourceDir string) (resource.Resource, error) { store := fsdb.NewFsDb() err = store.Connect(ctx, resourceDir) @@ -141,7 +148,7 @@ func main() { os.Exit(1) } - store, err := getUserDb(dataDir) + store, err := getUserdataDb(dataDir) if err != nil { fmt.Fprintf(os.Stderr, err.Error()) os.Exit(1)