up

backend/models/sql.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/glebarez/sqlite"
 	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 )
 
@@ -102,7 +103,13 @@ func DatabaseInit() error {
 		fmt.Println("mysql")
 		dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", DatabaseConfigs["user"].(string), DatabaseConfigs["pass"].(string), DatabaseConfigs["host"].(string), DatabaseConfigs["port"].(string), DatabaseConfigs["name"].(string))
 		DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	} else if DatabaseConfigs["type"].(string) == "pg" {
+		//postgresql init
+		fmt.Println("postgresql")
+		dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Shanghai", DatabaseConfigs["host"].(string), DatabaseConfigs["user"].(string), DatabaseConfigs["pass"].(string), DatabaseConfigs["name"].(string), DatabaseConfigs["port"].(string))
+		DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
 	}
+
 	if err != nil {
 		fmt.Println(err)
 		panic("数据库连接失败")