up

pkg/store/sqlite/sqlite.go

@@ -0,0 +1,24 @@
+package db
+
+import (
+	"log"
+	"os"
+
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+func NewDB() *gorm.DB {
+	if _, err := os.Stat("db"); os.IsNotExist(err) {
+		errDir := os.MkdirAll("db", 0755)
+		if errDir != nil {
+			log.Fatalln("Error creating directory:", err)
+		}
+	}
+	var err error
+	db, err = gorm.Open(sqlite.Open("./db/cat.db"), &gorm.Config{})
+	if err != nil {
+		panic("failed to connect database")
+	}
+
+}