提交kit/starter

Signed-off-by: lixxxww <941403820@qq.com>

kit/starter/gorm.go

@@ -0,0 +1,80 @@
+package starter
+
+import (
+	"database/sql"
+	"pandax/kit/logger"
+	"time"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+
+	gormlog "gorm.io/gorm/logger"
+
+	_ "github.com/lib/pq"
+)
+
+var Db *gorm.DB
+
+type DbGorm struct {
+	Type         string
+	Dsn          string
+	MaxIdleConns int
+	MaxOpenConns int
+}
+
+func (dg *DbGorm) GormInit() *gorm.DB {
+	switch dg.Type {
+	case "mysql":
+		Db = dg.GormMysql()
+	case "postgresql":
+		Db = dg.GormPostgresql()
+	}
+	return Db
+}
+func (dg *DbGorm) GormMysql() *gorm.DB {
+
+	logger.Log.Infof("连接mysql [%s]", dg.Dsn)
+	mysqlConfig := mysql.Config{
+		DSN:                       dg.Dsn, // DSN data source name
+		DefaultStringSize:         191,    // string 类型字段的默认长度
+		DisableDatetimePrecision:  true,   // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
+		DontSupportRenameIndex:    true,   // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
+		DontSupportRenameColumn:   true,   // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
+		SkipInitializeWithVersion: false,  // 根据版本自动配置
+	}
+	ormConfig := &gorm.Config{Logger: gormlog.Default.LogMode(gormlog.Silent)}
+	db, err := gorm.Open(mysql.New(mysqlConfig), ormConfig)
+	if err != nil {
+		logger.Log.Panicf("连接mysql失败! [%s]", err.Error())
+		return nil
+	}
+	sqlDB, _ := db.DB()
+	sqlDB.SetMaxIdleConns(dg.MaxIdleConns)
+	sqlDB.SetMaxOpenConns(dg.MaxOpenConns)
+	return db
+}
+
+func (dg *DbGorm) GormPostgresql() *gorm.DB {
+
+	db, err := sql.Open("postgres", dg.Dsn)
+	if err != nil {
+		logger.Log.Panicf("连接postgresql失败! [%s]", err.Error())
+	}
+	ormConfig := &gorm.Config{}
+	gormDb, err := gorm.Open(postgres.New(postgres.Config{
+		Conn: db,
+	}), ormConfig)
+	if err != nil {
+		logger.Log.Panicf("连接postgresql失败! [%s]", err.Error())
+	}
+	sqlDB, err := gormDb.DB()
+	// SetMaxIdleConns 设置空闲连接池中连接的最大数量
+	sqlDB.SetMaxIdleConns(dg.MaxIdleConns)
+	// SetMaxOpenConns 设置打开数据库连接的最大数量。
+	sqlDB.SetMaxOpenConns(dg.MaxOpenConns)
+	// SetConnMaxLifetime 设置了连接可复用的最大时间。
+	sqlDB.SetConnMaxLifetime(time.Hour)
+
+	return gormDb
+}

kit/starter/redis.go

@@ -0,0 +1,24 @@
+package starter
+
+import (
+	"fmt"
+	"pandax/kit/logger"
+
+	"github.com/go-redis/redis"
+)
+
+func ConnRedis(host, password string, db, port int) *redis.Client {
+	// 设置redis客户端
+	logger.Log.Infof("连接redis [%s:%d]", host, port)
+	rdb := redis.NewClient(&redis.Options{
+		Addr:     fmt.Sprintf("%s:%d", host, port),
+		Password: password, // no password set
+		DB:       db,       // use default DB
+	})
+	// 测试连接
+	_, e := rdb.Ping().Result()
+	if e != nil {
+		logger.Log.Panic(fmt.Sprintf("连接redis失败! [%s:%d]", host, port))
+	}
+	return rdb
+}

kit/starter/tdengine.go

@@ -0,0 +1,27 @@
+package starter
+
+import (
+	"database/sql"
+	"fmt"
+)
+
+type TaosDB struct {
+	Username string
+	Password string
+	Host     string
+	Port     int
+	Dbname   string
+	Config   string //配置
+	Db       *sql.DB
+}
+
+func (d *TaosDB) InitTdDB() error {
+	dsn := fmt.Sprintf("%s:%s@%s(%s:%d)/%s",
+		d.Username, d.Password, "http", d.Host, d.Port, d.Dbname)
+	if d.Config != "" {
+		dsn = fmt.Sprintf("%s&%s", dsn, d.Config)
+	}
+	var err error
+	d.Db, err = sql.Open("taosRestful", dsn)
+	return err
+}