51 lines
1.6 KiB
Go
51 lines
1.6 KiB
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/go-redis/redis/v8"
|
|
"gorm.io/driver/clickhouse"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func InitMysql(username, password, address, port, dbName string) (*gorm.DB, error) {
|
|
dsn := fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8mb4&parseTime=True&loc=Local", username, password, address, port, dbName)
|
|
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
|
|
Logger: &dbLogger{},
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: true, // 使用单数表名
|
|
},
|
|
})
|
|
return db, err
|
|
}
|
|
|
|
func InitRedis(password, address, port string, dbName int) (*redis.Client, error) {
|
|
rdb := redis.NewClient(&redis.Options{Addr: fmt.Sprintf("%v:%v", address, port), Password: password, DB: dbName})
|
|
if rdb == nil {
|
|
return nil, fmt.Errorf("init redis fail")
|
|
}
|
|
return rdb, nil
|
|
}
|
|
|
|
func InitClickHouse(host, port, user, password, database string) (*gorm.DB, error) {
|
|
dsn := fmt.Sprintf("tcp://%v:%v?database=%v&username=%v&password=%v&read_timeout=10&write_timeout=20", host, port, database, user, password)
|
|
db, err := gorm.Open(clickhouse.Open(dsn), &gorm.Config{
|
|
Logger: &dbLogger{},
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: true, // 使用单数表名
|
|
},
|
|
})
|
|
return db, err
|
|
}
|
|
|
|
func AutoMigrateClickHouse[T any](db *gorm.DB, table *T) error {
|
|
// 带ClickHouse引擎的迁移
|
|
return db.Set("gorm:table_options", "ENGINE=MergeTree() ORDER BY tuple()").AutoMigrate(table)
|
|
}
|
|
|
|
func InitClickHouse2(host, port, user, password, database string) (*ClickHouseRepo, error) {
|
|
ch := NewClickHouseRepo(host, port, user, password, database)
|
|
return ch, ch.Open()
|
|
}
|