init...

2022-03-01 13:50:13 +08:00
parent 8eab9943e9
commit d2cabd7d58
113 changed files with 7925 additions and 0 deletions
@@ -0,0 +1,133 @@
+package dao
+
+import (
+	models2 "cutego/core/entity"
+	"cutego/pkg/common"
+	"cutego/pkg/config"
+	"cutego/pkg/constant"
+	redisTool "cutego/pkg/redispool"
+	"fmt"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/go-xorm/xorm"
+	"time"
+)
+
+// X 全局DB
+var (
+	SqlDB   *xorm.Engine
+	RedisDB *redisTool.RedisClient
+)
+
+func initDatabase() {
+	var err error
+	// 配置mysql数据库
+	ds := config.AppEnvConfig.DataSource
+	jdbc := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local",
+		ds.Username,
+		ds.Password,
+		ds.Host,
+		ds.Port,
+		ds.Database,
+		ds.Charset,
+	)
+	SqlDB, _ = xorm.NewEngine(ds.DbType, jdbc)
+	if err != nil {
+		common.FatalfLog("db error: %#v\n", err.Error())
+	}
+	err = SqlDB.Ping()
+	if err != nil {
+		common.FatalfLog("db connect error: %#v\n", err.Error())
+	}
+	SqlDB.SetMaxIdleConns(ds.MaxIdleSize)
+	SqlDB.SetMaxOpenConns(ds.MaxOpenSize)
+	timer := time.NewTicker(time.Minute * 30)
+	go func(x *xorm.Engine) {
+		for _ = range timer.C {
+			err = x.Ping()
+			if err != nil {
+				common.FatalfLog("db connect error: %#v\n", err.Error())
+			}
+		}
+	}(SqlDB)
+	SqlDB.ShowSQL(true)
+}
+func initRedis() {
+	// 配置redis数据库
+	RedisDB = redisTool.NewRedis()
+}
+func init() {
+	initDatabase()
+	initRedis()
+	cacheInitDataToRedis()
+}
+
+// 初始化缓存数据
+func cacheInitDataToRedis() {
+	initDict()
+	initConfig()
+}
+
+func initDict() {
+	// 查询字典类型数据
+	dictTypeDao := new(DictTypeDao)
+	typeAll := dictTypeDao.SelectAll()
+	// 所有字典数据
+	d := new(DictDataDao)
+	listData := d.GetDiceDataAll()
+	for _, dictType := range typeAll {
+		dictData := make([]map[string]interface{}, 0)
+		for _, data := range *listData {
+			if dictType.DictType == data.DictType {
+				dictData = append(dictData, map[string]interface{}{
+					"dictCode":  data.DictCode,
+					"dictSort":  data.DictSort,
+					"dictLabel": data.DictLabel,
+					"dictValue": data.DictValue,
+					"isDefault": data.IsDefault,
+					"remark":    data.Remark,
+				})
+			}
+		}
+		RedisDB.SET(constant.RedisConst{}.GetRedisDictKey()+dictType.DictType, common.StructToJson(dictData))
+	}
+}
+
+func initConfig() {
+	// 查询配置数据存入到缓存中
+	configDao := new(ConfigDao)
+	configSession := configDao.sql(SqlDB.NewSession())
+	configs := make([]*models2.SysConfig, 0)
+	err := configSession.Find(&configs)
+	if err != nil {
+		common.ErrorLog(err)
+		return
+	}
+	for _, sysConfig := range configs {
+		RedisDB.SET(constant.RedisConst{}.GetRedisConfigKey()+sysConfig.ConfigKey, common.StructToJson(map[string]interface{}{
+			"configId":    sysConfig.ConfigId,
+			"configName":  sysConfig.ConfigName,
+			"configKey":   sysConfig.ConfigKey,
+			"configValue": sysConfig.ConfigValue,
+			"configType":  sysConfig.ConfigType,
+			"remark":      sysConfig.Remark,
+		}))
+	}
+}
+
+// 通用: 根据ID更新
+// @Param idColumnName 字段名称
+// @Param idColumnValue 字段值
+// @Param bean 更新的bean内容
+// @Return
+func CustomUpdateById(idColumnName string, idColumnValue interface{}, bean interface{}) int64 {
+	session := SqlDB.NewSession()
+	session.Begin()
+	update, err := session.Where(idColumnName+" = ?", idColumnValue).Update(&bean)
+	if err != nil {
+		common.ErrorLog(err)
+		session.Rollback()
+		return 0
+	}
+	session.Commit()
+	return update
+}