init

config/db.go

@@ -0,0 +1,26 @@
+package config
+
+import (
+	"fmt"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"log"
+)
+
+var DB *gorm.DB
+
+func InitDB() {
+	var err error
+	// "%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local"
+	dsn := fmt.Sprintf(
+		"%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		Conf.DB.User,
+		Conf.DB.Passwd,
+		Conf.DB.Host,
+		Conf.DB.Port,
+		Conf.DB.Database)
+	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	if err != nil {
+		log.Fatalf("无法连接到数据库: %v", err)
+	}
+}

config/readConfig.go

@@ -0,0 +1,31 @@
+package config
+
+import "github.com/BurntSushi/toml"
+
+type Config struct {
+	Server ServerConfig `toml:"server"`
+	DB     DBConfig     `toml:"db"`
+}
+
+type ServerConfig struct {
+	Port int `toml:"port"`
+}
+
+type DBConfig struct {
+	Host     string `toml:"host"`
+	Port     int    `toml:"port"`
+	Database string `toml:"database"`
+	User     string `toml:"user"`
+	Passwd   string `toml:"passwd"`
+}
+
+var Conf *Config
+
+func ReadConfig(path string) error {
+	var c Config
+	if _, err := toml.DecodeFile(path, &c); err != nil {
+		return err
+	}
+	Conf = &c
+	return nil
+}