浏览代码

add database url to config file

zu1k 5 年之前
父节点
当前提交
65ad07f579
共有 3 个文件被更改,包括 9 次插入0 次删除
  1. 1 0
      config/config.go
  2. 3 0
      config/config.yaml
  3. 5 0
      internal/database/db.go

+ 1 - 0
config/config.go

@@ -14,6 +14,7 @@ var configFilePath = "config.yaml"
 
 type ConfigOptions struct {
 	Domain      string   `json:"domain" yaml:"domain"`
+	DatabaseUrl string   `json:"database_url" yaml:"database_url"`
 	CFEmail     string   `json:"cf_email" yaml:"cf_email"`
 	CFKey       string   `json:"cf_key" yaml:"cf_key"`
 	SourceFiles []string `json:"source-files" yaml:"source-files"`

+ 3 - 0
config/config.yaml

@@ -1,6 +1,9 @@
 # your domain
 domain: example.com
 
+# database url
+database_url: ""
+
 # cloudflare api
 cf_email: ""
 cf_key: ""

+ 5 - 0
internal/database/db.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"os"
 
+	"github.com/zu1k/proxypool/config"
+
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 )
@@ -12,6 +14,9 @@ var DB *gorm.DB
 
 func connect() (err error) {
 	dsn := "user=proxypool password=proxypool dbname=proxypool port=5432 sslmode=disable TimeZone=Asia/Shanghai"
+	if url := config.Config.DatabaseUrl; url != "" {
+		dsn = url
+	}
 	if url := os.Getenv("DATABASE_URL"); url != "" {
 		dsn = url
 	}