add pg conf

src/model/dao/init.go

@@ -0,0 +1,7 @@
+package dao
+
+func Init() {
+	DBInit()
+
+	RedisInit()
+}

src/model/dao/mdb.go

@@ -1,47 +1,21 @@
 package dao
 
 import (
-	"github.com/assimon/luuu/config"
-	"github.com/assimon/luuu/util/log"
-	"github.com/gookit/color"
+	"strings"
+
 	"github.com/spf13/viper"
-	"gorm.io/driver/mysql"
+
 	"gorm.io/gorm"
-	"gorm.io/gorm/logger"
-	"gorm.io/gorm/schema"
-	"time"
 )
 
 var Mdb *gorm.DB
 
-// MysqlInit 数据库初始化
-func MysqlInit() {
-	var err error
-	Mdb, err = gorm.Open(mysql.Open(config.MysqlDns), &gorm.Config{
-		NamingStrategy: schema.NamingStrategy{
-			TablePrefix:   viper.GetString("mysql_table_prefix"),
-			SingularTable: true,
-		},
-		Logger: logger.Default.LogMode(logger.Error),
-	})
-	if err != nil {
-		panic(err)
+func DBInit() {
+	dbType := viper.GetString("db_type")
+	if strings.EqualFold(dbType, "postgres") {
+		PostgreSQLInit()
+	} else {
+		MysqlInit()
 	}
-	if config.AppDebug {
-		Mdb = Mdb.Debug()
-	}
-	sqlDB, err := Mdb.DB()
-	if err != nil {
-		color.Red.Printf("[store_db] mysql get DB,err=%s\n", err)
-		panic(err)
-	}
-	sqlDB.SetMaxIdleConns(viper.GetInt("mysql_max_idle_conns"))
-	sqlDB.SetMaxOpenConns(viper.GetInt("mysql_max_open_conns"))
-	sqlDB.SetConnMaxLifetime(time.Hour * time.Duration(viper.GetInt("mysql_max_life_time")))
-	err = sqlDB.Ping()
-	if err != nil {
-		color.Red.Printf("[store_db] mysql connDB err:%s", err.Error())
-		panic(err)
-	}
-	log.Sugar.Debug("[store_db] mysql connDB success")
+	MdbTableInit()
 }

src/model/dao/mdb_mysql.go

@@ -0,0 +1,47 @@
+package dao
+
+import (
+	"time"
+
+	"github.com/assimon/luuu/config"
+	"github.com/assimon/luuu/util/log"
+	"github.com/gookit/color"
+	"github.com/spf13/viper"
+	"gorm.io/driver/mysql"
+
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
+)
+
+// MysqlInit 数据库初始化
+func MysqlInit() {
+	var err error
+	Mdb, err = gorm.Open(mysql.Open(config.MysqlDns), &gorm.Config{
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   viper.GetString("mysql_table_prefix"),
+			SingularTable: true,
+		},
+		Logger: logger.Default.LogMode(logger.Error),
+	})
+	if err != nil {
+		panic(err)
+	}
+	if config.AppDebug {
+		Mdb = Mdb.Debug()
+	}
+	sqlDB, err := Mdb.DB()
+	if err != nil {
+		color.Red.Printf("[store_db] mysql get DB,err=%s\n", err)
+		panic(err)
+	}
+	sqlDB.SetMaxIdleConns(viper.GetInt("mysql_max_idle_conns"))
+	sqlDB.SetMaxOpenConns(viper.GetInt("mysql_max_open_conns"))
+	sqlDB.SetConnMaxLifetime(time.Hour * time.Duration(viper.GetInt("mysql_max_life_time")))
+	err = sqlDB.Ping()
+	if err != nil {
+		color.Red.Printf("[store_db] mysql connDB err:%s", err.Error())
+		panic(err)
+	}
+	log.Sugar.Debug("[store_db] mysql connDB success")
+}

src/model/dao/mdb_postgres.go

@@ -0,0 +1,59 @@
+package dao
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/assimon/luuu/config"
+	"github.com/assimon/luuu/util/log"
+	"github.com/gookit/color"
+	"github.com/spf13/viper"
+
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
+)
+
+// PostgreSQLInit 数据库初始化
+func PostgreSQLInit() {
+	var err error
+	user := viper.GetString("postgres_user")
+	pass := viper.GetString("postgres_passwd")
+	host := viper.GetString("postgres_host")
+	port := viper.GetString("postgres_port")
+	dbname := viper.GetString("postgres_database")
+	dsn := fmt.Sprintf("user=%s password=%s host=%s port=%s dbname=%s sslmode=disable TimeZone=Asia/Shanghai",
+		user, pass, host, port, dbname,
+	)
+	Mdb, err = gorm.Open(postgres.Open(dsn),
+		&gorm.Config{
+			NamingStrategy: schema.NamingStrategy{
+				TablePrefix:   viper.GetString("postgres_table_prefix"),
+				SingularTable: true,
+			},
+			Logger: logger.Default.LogMode(logger.Error),
+		},
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	if config.AppDebug {
+		Mdb = Mdb.Debug()
+	}
+	sqlDB, err := Mdb.DB()
+	if err != nil {
+		color.Red.Printf("[store_db] postgres get DB,err=%s\n", err)
+		panic(err)
+	}
+	sqlDB.SetMaxIdleConns(viper.GetInt("postgres_max_idle_conns"))
+	sqlDB.SetMaxOpenConns(viper.GetInt("postgres_max_open_conns"))
+	sqlDB.SetConnMaxLifetime(time.Hour * time.Duration(viper.GetInt("postgres_max_life_time")))
+	err = sqlDB.Ping()
+	if err != nil {
+		color.Red.Printf("[store_db] postgres connDB err:%s", err.Error())
+		panic(err)
+	}
+	log.Sugar.Debug("[store_db] postgres connDB success")
+}

src/model/dao/mdb_table_init.go

@@ -0,0 +1,26 @@
+package dao
+
+import (
+	"sync"
+
+	"github.com/assimon/luuu/model/mdb"
+	"github.com/gookit/color"
+)
+
+var once sync.Once
+
+// 自动建表
+func MdbTableInit() {
+	once.Do(func() {
+		if err := Mdb.AutoMigrate(&mdb.Orders{}); err != nil {
+			color.Red.Printf("[store_db] AutoMigrate DB(Orders),err=%s\n", err)
+			// panic(err)
+			return
+		}
+		if err := Mdb.AutoMigrate(&mdb.WalletAddress{}); err != nil {
+			color.Red.Printf("[store_db] AutoMigrate DB(Orders),err=%s\n", err)
+			// panic(err)
+			return
+		}
+	})
+}