From 3737ab3dd0cc753be986638316c96cb3114601e4 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期日, 29 九月 2024 16:20:46 +0800
Subject: [PATCH] fix db column
---
db/database.go | 18 +++++++++++++-----
1 files changed, 13 insertions(+), 5 deletions(-)
diff --git a/db/database.go b/db/database.go
index f4afe69..ccbb054 100644
--- a/db/database.go
+++ b/db/database.go
@@ -1,10 +1,11 @@
package db
import (
+ "time"
+
"gorm.io/driver/mysql"
"gorm.io/gorm"
"ruleModelEngine/config"
- "time"
)
var DB *gorm.DB
@@ -20,20 +21,27 @@
// return nil
//}
-func ConnectDB() (error) {
- dsn := "root:c++java123@tcp(" + config.DataBase.Host + ":" + config.DataBase.Port + ")/" + config.DataBase.Name + "?charset=utf8mb4&parseTime=True&loc=Local"
+func ConnectDB() error {
+ dsn := config.DataBase.Username + ":" + config.DataBase.Password + "@tcp(" + config.DataBase.Host + ":" + config.DataBase.Port + ")/" + config.DataBase.Name + "?charset=utf8mb4&parseTime=True&loc=Local"
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
if err != nil {
return err
}
- sqlDb,err := db.DB()
- if err != nil{
+ sqlDb, err := db.DB()
+ if err != nil {
return err
}
sqlDb.SetMaxIdleConns(2)
sqlDb.SetMaxOpenConns(1)
sqlDb.SetConnMaxLifetime(time.Duration(120) * time.Second)
sqlDb.SetConnMaxIdleTime(time.Duration(1800) * time.Second)
+
+ db.AutoMigrate(
+ &PublicHouse{},
+ &PersonnelStatusRule{},
+ &Task{},
+ )
+
DB = db
return nil
}
--
Gitblit v1.8.0