gorm

configs/sqlite3.go

@@ -6,6 +6,8 @@ import (
 	"os"
 
 	_ "github.com/mattn/go-sqlite3"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
 )
 
 // 使用SQLite3初始化數據庫
@@ -103,3 +105,12 @@ func GetDB() (*sql.DB, error) {
 	}
 	return db, nil
 }
+
+// ORMDB 使用 GORM
+func ORMDB() (db *gorm.DB) {
+	db, err := gorm.Open(sqlite.Open("data/sqlite3.db"), &gorm.Config{})
+	if err != nil {
+		log.Println(err)
+	}
+	return db
+}