改善配置模块代码~

This commit is contained in:
2025-05-20 09:01:07 +08:00
parent a66f1c69b1
commit 67573e6ac2
9 changed files with 61 additions and 102 deletions

View File

@@ -13,11 +13,12 @@ import (
var err error
func Connect() {
switch config.DB.Type {
switch config.Default.DB.Type {
case "SQLite":
connectSQLite()
case "PostgreSQL":
connectPGSQL()
//connectPGSQL()
PGSQL()
default:
panic("DBType error")
}
@@ -25,7 +26,7 @@ func Connect() {
}
func connectSQLite() {
Main, err = gorm.Open(sqlite.Open(config.DB.Path), &gorm.Config{})
Main, err = gorm.Open(sqlite.Open(config.Default.DB.Path), &gorm.Config{})
if err != nil {
fmt.Printf("error in connecting to SQLite:")
fmt.Println(err)
@@ -35,7 +36,7 @@ func connectSQLite() {
}
func connectPGSQL() {
Main, err = gorm.Open(postgres.Open(config.DB.Path))
Main, err = gorm.Open(postgres.Open(config.Default.DB.Path), &gorm.Config{})
if err != nil {
panic(err)
}