fix: 初验针对修改
This commit is contained in:
@@ -1,12 +1,11 @@
|
||||
package impl
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"errors"
|
||||
|
||||
"git.apinb.com/bsm-sdk/core/cache/redis"
|
||||
"git.apinb.com/bsm-sdk/core/logger"
|
||||
"git.apinb.com/ops/logs/internal/config"
|
||||
"git.apinb.com/ops/logs/internal/models"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
@@ -19,13 +18,22 @@ func NewImpl() {
|
||||
RedisService = newRedisCache(config.Spec.Cache)
|
||||
DBService = newDatabase(config.Spec.Databases)
|
||||
logger.New(nil)
|
||||
}
|
||||
|
||||
// Close 关闭数据库和 Redis 连接。
|
||||
func Close() error {
|
||||
var closeErrors []error
|
||||
if DBService != nil {
|
||||
if err := DBService.AutoMigrate(models.GetAllModels()...); err != nil {
|
||||
panic(fmt.Sprintf("logs migrate: %v", err))
|
||||
}
|
||||
if err := models.InitData(DBService); err != nil {
|
||||
panic(fmt.Sprintf("logs init data: %v", err))
|
||||
if sqlDB, err := DBService.DB(); err != nil {
|
||||
closeErrors = append(closeErrors, err)
|
||||
} else if err := sqlDB.Close(); err != nil {
|
||||
closeErrors = append(closeErrors, err)
|
||||
}
|
||||
}
|
||||
if RedisService != nil && RedisService.Client != nil {
|
||||
if err := RedisService.Client.Close(); err != nil {
|
||||
closeErrors = append(closeErrors, err)
|
||||
}
|
||||
}
|
||||
return errors.Join(closeErrors...)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user