init repo

internal/storage/dao/engine.go

@@ -0,0 +1,31 @@
+package dao
+
+import (
+	"fmt"
+
+	"gitea.com/bitwsd/document_ai/config"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+var DB *gorm.DB
+
+func InitDB(conf config.DatabaseConfig) {
+	dns := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Asia%%2FShanghai", conf.Username, conf.Password, conf.Host, conf.Port, conf.DBName)
+	db, err := gorm.Open(mysql.Open(dns), &gorm.Config{})
+	if err != nil {
+		panic(err)
+	}
+	sqlDB, err := db.DB()
+	if err != nil {
+		panic(err)
+	}
+	sqlDB.SetMaxIdleConns(10)
+	sqlDB.SetMaxOpenConns(100)
+	DB = db
+}
+
+func CloseDB() {
+	sqlDB, _ := DB.DB()
+	sqlDB.Close()
+}