算力管理,记录查询

qixiaoning

2025-07-24 343a89f9fa20d2d142469b2a4531e16ce03d3525

 chanmanage-service/models/db.go

@@ -1,13 +1,15 @@
package models

import (
   "vamicro/config"

   "basic.com/valib/logger.git"
   "github.com/jinzhu/gorm"
   _ "github.com/jinzhu/gorm/dialects/sqlite"
   "basic.com/valib/logger.git"
   "vamicro/config"
)

var db *gorm.DB
var db2 *gorm.DB
var err error

// Init creates a connection to mysql database and
@@ -20,13 +22,21 @@
   db.LogMode(true)
   //db.SetLogger(&DbLogger{})
   db.AutoMigrate(&PollConfig{})

   db2, err = gorm.Open(config.DBconf.Name, "../config/camera-service.db")
   if err != nil {
      logger.Debug("db open error ", err)
   }
   db2.LogMode(true)
   db2.AutoMigrate(&PollConfig{})

}

//GetDB ...
// GetDB ...
func GetDB() *gorm.DB {
   return db
}

func CloseDB() {
   db.Close()
}
}