From 17c56cfddfd4f4bb3b1e30ac4464e3f45ae5eedd Mon Sep 17 00:00:00 2001
From: bryanqiu <bryanqiu@tencent.com>
Date: Mon, 1 Jul 2024 13:43:56 +0800
Subject: [PATCH] dev

---
 database.go | 51 ++++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 44 insertions(+), 7 deletions(-)

diff --git a/database.go b/database.go
index ad381dd..4bd22c4 100644
--- a/database.go
+++ b/database.go
@@ -1,11 +1,48 @@
 package database
 
+import (
+	"fmt"
+	"strconv"
+	"strings"
+)
+
 type Config struct {
-    Host            string `toml:"host"`             //数据库名称
-    Port            int    `toml:"port"`             //数据库名称
-    Debug           bool   `toml:"debug"`            //调试开关（会在日志打印SQL)
-    Dbname          string `toml:"dbname"`           //数据库名称
-    Username        string `toml:"username"`         //数据库用户名
-    Password        string `toml:"password"`         //数据库连接密码
-    ExtraParameters string `toml:"extra_parameters"` //数据库连接扩展参数
+	Type            string `toml:"type"`             //数据库类型 mysql or pgsql
+	Host            string `toml:"host"`             //数据库名称
+	Port            int    `toml:"port"`             //数据库名称
+	Debug           bool   `toml:"debug"`            //调试开关（会在日志打印SQL)
+	Dbname          string `toml:"dbname"`           //数据库名称
+	Username        string `toml:"username"`         //数据库用户名
+	Password        string `toml:"password"`         //数据库连接密码
+	ExtraParameters string `toml:"extra_parameters"` //数据库连接扩展参数
+}
+
+func GetDsn(dbcfg *Config) (dsn string) {
+	switch dbcfg.Type {
+	case "mysql":
+		//dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
+		dsn = fmt.Sprintf(
+			"%s:%s@tcp(%s:%d)/%s?%s",
+			dbcfg.Username,
+			dbcfg.Password,
+			dbcfg.Host,
+			dbcfg.Port,
+			dbcfg.Dbname,
+			dbcfg.ExtraParameters,
+		)
+	case "pgsql":
+		//dsn := "host=localhost user=gorm password=gorm dbname=gorm port=9920 sslmode=disable TimeZone=Asia/Shanghai"
+		arrConfStr := []string{
+			"host=" + dbcfg.Host,
+			"port=" + strconv.Itoa(dbcfg.Port),
+			"user=" + dbcfg.Username,
+			"password=" + dbcfg.Password,
+			"dbname=" + dbcfg.Dbname,
+			dbcfg.ExtraParameters,
+		}
+		dsn = strings.Join(arrConfStr, " ")
+	default:
+		panic("DATABASE TYPE '" + dbcfg.Type + "' NOT SUPPORT (only mysql or pgsql)")
+	}
+	return dsn
 }