(fix): use ko data path when within container

kameshsampath · Apr 9, 2024 · 92eae87 · 92eae87
1 parent 6e260a8
commit 92eae87
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/cmd/server/kodata/.gitkeep b/cmd/server/kodata/.gitkeep
diff --git a/impl/todo_impl.go b/impl/todo_impl.go
@@ -29,18 +29,22 @@ func init() {
 	logger, _ := zap.NewDevelopment()
 	defer logger.Sync()
 	log = logger.Sugar()
-	cwd, _ := os.Getwd()
-	dataDir := path.Join(cwd, "data")
-	if _, err := os.Stat(dataDir); err != nil && os.IsNotExist(err) {
-		if err := os.MkdirAll(dataDir, os.ModePerm); err != nil {
-			log.Fatal("error creating data directory,%s", err)
+	//KO_DATA_PATH will be available only when running with ko container
+	dataDir, ok := os.LookupEnv("KO_DATA_PATH")
+	if !ok {
+		cwd, _ := os.Getwd()
+		dataDir = path.Join(cwd, "data")
+		if _, err := os.Stat(dataDir); err != nil && os.IsNotExist(err) {
+			if err := os.MkdirAll(dataDir, os.ModePerm); err != nil {
+				log.Fatal("error creating data directory,%s", err)
+			}
 		}
 	}
 	dbFile := path.Join(dataDir, "sequence.db")
 	log.Infof("Data File:%s", dbFile)
 	sc, err = utils.NewSequence(fmt.Sprintf("file://%s?cache=shared", dbFile))
 	if err != nil {
-		log.Fatal("error initializing sequence DB,%s", err)
+		log.Fatalf("error initializing sequence DB,%s", err)
 	}
 }