Refactor database path handling in conf package

conf/conf.go

@@ -45,7 +45,7 @@ var TcpTimeout int
 var version string
 
 // 数据库位置
-var Db = "goForward.db"
+var Db string
 
 func init() {
 	if version != "" {

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"strings"
 	"sync"
 
 	"csz.net/goForward/conf"
@@ -66,4 +67,8 @@ func init() {
 	flag.StringVar(&conf.WebPass, "pass", "", "Web Password")
 	flag.IntVar(&conf.TcpTimeout, "tt", 60, "Tcp Timeout")
 	flag.Parse()
+	if !strings.HasSuffix(conf.Db, ".db") {
+		conf.Db += ".db"
+	}
+	sql.Once()
 }

sql/sql.go

@@ -16,7 +16,7 @@ import (
 // 定义数据库指针
 var db *gorm.DB
 
-func init() {
+func Once() {
 	var err error
 	var dbPath string
 	executablePath, err := os.Executable()