micro-document/pkg/db/db.go

package db

import (
	"fmt"
	"micro-document/config"
	"micro-document/internel/model"
	"time"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

var DocDB *gorm.DB

func DbInit() {
	username := config.CF.DocDB.User
	password := config.CF.DocDB.Password
	address := config.CF.DocDB.Host
	database := config.CF.DocDB.DbName
	if username != "" || password != "" || address != "" || database != "" {
		dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", username, password, address, database)
		dbq, err := gorm.Open(mysql.Open(dsn), &gorm.Config{PrepareStmt: true, DisableForeignKeyConstraintWhenMigrating: true, Logger: logger.Default.LogMode(logger.Warn)})
		if err != nil {
			panic(err)
		} else {
			sqlDB, err := dbq.DB()
			if err != nil {
				panic(err)
			} else {
				sqlDB.SetMaxIdleConns(20)
				sqlDB.SetMaxOpenConns(100)
				sqlDB.SetConnMaxLifetime(time.Hour)
			}
		}
		DocDB = dbq.Debug()
		DocDB.Set("gorm:table_options", "ENGINE=InnoDB default charset=utf8")
		DocDB.AutoMigrate(&model.Governance{}, &model.PressReleases{}, &model.SecFilings{})
	}
}