fix db column

zhangzengfei

2024-09-29 3737ab3dd0cc753be986638316c96cb3114601e4

 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
}