From eba4eb850f0ecfb5839395aa125955ceaa2a454f Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 19 十月 2023 16:39:07 +0800 Subject: [PATCH] Merge branch 'feat-serf' --- pkg/sqlitex/sqlitex.go | 31 ++++++------------------------- 1 files changed, 6 insertions(+), 25 deletions(-) diff --git a/pkg/sqlitex/sqlitex.go b/pkg/sqlitex/sqlitex.go index 2c221a5..92fde9e 100644 --- a/pkg/sqlitex/sqlitex.go +++ b/pkg/sqlitex/sqlitex.go @@ -1,12 +1,8 @@ package sqlitex import ( - "go.uber.org/zap" - "gorm.io/driver/sqlite" - "gorm.io/gorm" - "gorm.io/gorm/logger" - "gorm.io/gorm/schema" - "moul.io/zapgorm2" + "github.com/jinzhu/gorm" + _ "github.com/jinzhu/gorm/dialects/sqlite" "time" ) @@ -22,33 +18,18 @@ var openDb *gorm.DB -func Init(conf *Conf, log *zap.Logger) error { - gConfig := &gorm.Config{ - PrepareStmt: true, - NamingStrategy: schema.NamingStrategy{ - TablePrefix: "", - SingularTable: true, - }, - DisableForeignKeyConstraintWhenMigrating: true, - } - dbLogger := zapgorm2.New(log).LogMode(logger.Info) - if !conf.LogMode { - dbLogger = dbLogger.LogMode(logger.Silent) - } - gConfig.Logger = dbLogger - db, err := gorm.Open(sqlite.Open(conf.Dsn), gConfig) +func Init(conf *Conf) error { + db, err := gorm.Open("sqlite3", conf.Dsn) if err != nil { return err } - sqlDb, err := db.DB() - if err != nil { - return err - } + sqlDb := db.DB() sqlDb.SetMaxIdleConns(int(conf.MaxIdleCon)) sqlDb.SetMaxOpenConns(int(conf.MaxOpenCon)) sqlDb.SetConnMaxLifetime(time.Duration(conf.ConnMaxLifeTimeSecond) * time.Second) sqlDb.SetConnMaxIdleTime(time.Duration(conf.ConnMaxIdleTimeSecond) * time.Second) openDb = db + db.LogMode(true) return nil } -- Gitblit v1.8.0