init: initial commit

config/config.go

@@ -0,0 +1,19 @@
+package config
+
+type Config struct {
+	JWT    JWT    `mapstructure:"jwt" json:"jwt" yaml:"jwt"`
+	System System `mapstructure:"system" json:"system" yaml:"system"`
+	Mysql  Mysql  `mapstructure:"mysql" json:"mysql" yaml:"mysql"`
+	Pgsql  Pgsql  `mapstructure:"pgsql" json:"pgsql" yaml:"pgsql"`
+	Sqlite Sqlite `mapstructure:"sqlite" json:"sqlite" yaml:"sqlite"`
+	Redis  Redis  `mapstructure:"redis" json:"redis" yaml:"redis"`
+	Zap    Zap    `mapstructure:"zap" json:"zap" yaml:"zap"`
+
+	Minio          Minio          `mapstructure:"minio" json:"minio" yaml:"minio"`
+	RocketMQConfig RocketMQConfig `mapstructure:"rocket-mq" json:"rocket-mq" yaml:"rocket-mq"`
+	TencentCOS     TencentCOS     `mapstructure:"tencent-cos" json:"tencent-cos" yaml:"tencent-cos"`
+
+	MiniProgram    MiniProgram    `mapstructure:"mini-program" json:"mini-program" yaml:"mini-program"`          //小程序
+	ServiceAccount ServiceAccount `mapstructure:"service-account" json:"service-account" yaml:"service-account"` //服务号
+	WechatPay      WechatPay      `mapstructure:"wechat-pay" json:"wechat-pay" yaml:"wechat-pay"`                //微信支付
+}

config/config_db.go

@@ -0,0 +1,8 @@
+package config
+
+type DB struct {
+	Host     string `mapstructure:"host" json:"host" yaml:"host"`
+	Port     int    `mapstructure:"port" json:"port" yaml:"port"`
+	User     string `mapstructure:"user" json:"user" yaml:"user"`
+	Password string `mapstructure:"password" json:"password" yaml:"password"`
+}

config/config_jwt.go

@@ -0,0 +1,8 @@
+package config
+
+type JWT struct {
+	SigningKey  string `mapstructure:"signing-key" json:"signing-key" yaml:"signing-key"`    // jwt签名
+	ExpiresTime string `mapstructure:"expires-time" json:"expires-time" yaml:"expires-time"` // 过期时间
+	BufferTime  string `mapstructure:"buffer-time" json:"buffer-time" yaml:"buffer-time"`    // 缓冲时间
+	Issuer      string `mapstructure:"issuer" json:"issuer" yaml:"issuer"`                   // 签发者
+}

config/config_redis.go

@@ -0,0 +1,10 @@
+package config
+
+type Redis struct {
+	Name         string   `mapstructure:"name" json:"name" yaml:"name"`                         // 代表当前实例的名字
+	Addr         string   `mapstructure:"addr" json:"addr" yaml:"addr"`                         // 服务器地址:端口
+	Password     string   `mapstructure:"password" json:"password" yaml:"password"`             // 密码
+	DB           int      `mapstructure:"db" json:"db" yaml:"db"`                               // 单实例模式下redis的哪个数据库
+	Cluster      bool     `mapstructure:"cluster" json:"cluster" yaml:"cluster"`                // 是否使用集群模式
+	ClusterAddrs []string `mapstructure:"clusterAddrs" json:"clusterAddrs" yaml:"clusterAddrs"` // 集群模式下的节点地址列表
+}

config/db_list.go

@@ -0,0 +1,52 @@
+package config
+
+import (
+	"gorm.io/gorm/logger"
+	"strings"
+)
+
+type DsnProvider interface {
+	Dsn() string
+}
+
+// Embeded 结构体可以压平到上一层，从而保持 config 文件的结构和原来一样
+// 见 playground: https://go.dev/play/p/KIcuhqEoxmY
+
+// GeneralDB 也被 Pgsql 和 Mysql 原样使用
+type GeneralDB struct {
+	Prefix       string `mapstructure:"prefix" json:"prefix" yaml:"prefix"`                         // 数据库前缀
+	Port         string `mapstructure:"port" json:"port" yaml:"port"`                               // 数据库端口
+	Config       string `mapstructure:"config" json:"config" yaml:"config"`                         // 高级配置
+	Dbname       string `mapstructure:"db-name" json:"db-name" yaml:"db-name"`                      // 数据库名
+	User         string `mapstructure:"user" json:"user" yaml:"user"`                               // 数据库账号
+	Password     string `mapstructure:"password" json:"password" yaml:"password"`                   // 数据库密码
+	Host         string `mapstructure:"host" json:"host" yaml:"host"`                               // 数据库地址
+	Engine       string `mapstructure:"engine" json:"engine" yaml:"engine" default:"InnoDB"`        // 数据库引擎，默认InnoDB
+	LogMode      string `mapstructure:"log-mode" json:"log-mode" yaml:"log-mode"`                   // 是否开启Gorm全局日志
+	MaxIdleConns int    `mapstructure:"max-idle-conns" json:"max-idle-conns" yaml:"max-idle-conns"` // 空闲中的最大连接数
+	MaxOpenConns int    `mapstructure:"max-open-conns" json:"max-open-conns" yaml:"max-open-conns"` // 打开到数据库的最大连接数
+	Singular     bool   `mapstructure:"singular" json:"singular" yaml:"singular"`                   // 是否开启全局禁用复数，true表示开启
+	LogZap       bool   `mapstructure:"log-zap" json:"log-zap" yaml:"log-zap"`                      // 是否通过zap写入日志文件
+}
+
+func (c GeneralDB) LogLevel() logger.LogLevel {
+	switch strings.ToLower(c.LogMode) {
+	case "silent", "Silent":
+		return logger.Silent
+	case "error", "Error":
+		return logger.Error
+	case "warn", "Warn":
+		return logger.Warn
+	case "info", "Info":
+		return logger.Info
+	default:
+		return logger.Info
+	}
+}
+
+type SpecializedDB struct {
+	Type      string `mapstructure:"type" json:"type" yaml:"type"`
+	AliasName string `mapstructure:"alias-name" json:"alias-name" yaml:"alias-name"`
+	GeneralDB `yaml:",inline" mapstructure:",squash"`
+	Disable   bool `mapstructure:"disable" json:"disable" yaml:"disable"`
+}

config/gorm_mysql.go

@@ -0,0 +1,9 @@
+package config
+
+type Mysql struct {
+	GeneralDB `yaml:",inline" mapstructure:",squash"`
+}
+
+func (m *Mysql) Dsn() string {
+	return m.User + ":" + m.Password + "@tcp(" + m.Host + ":" + m.Port + ")/" + m.Dbname + "?" + m.Config
+}

config/gorm_pgsql.go

@@ -0,0 +1,17 @@
+package config
+
+type Pgsql struct {
+	GeneralDB `yaml:",inline" mapstructure:",squash"`
+}
+
+// Dsn 基于配置文件获取 dsn
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (p *Pgsql) Dsn() string {
+	return "host=" + p.Host + " user=" + p.User + " password=" + p.Password + " dbname=" + p.Dbname + " port=" + p.Port + " " + p.Config
+}
+
+// LinkDsn 根据 dbname 生成 dsn
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (p *Pgsql) LinkDsn(dbname string) string {
+	return "host=" + p.Host + " user=" + p.User + " password=" + p.Password + " dbname=" + dbname + " port=" + p.Port + " " + p.Config
+}

config/gorm_sqlite.go

@@ -0,0 +1,13 @@
+package config
+
+import (
+	"path/filepath"
+)
+
+type Sqlite struct {
+	GeneralDB `yaml:",inline" mapstructure:",squash"`
+}
+
+func (s *Sqlite) Dsn() string {
+	return filepath.Join(s.Host, s.Dbname+".db")
+}

config/mini_program.go

@@ -0,0 +1,6 @@
+package config
+
+type MiniProgram struct {
+	AppId     string `mapstructure:"app-id" json:"app-id" yaml:"app-id"`
+	AppSecret string `mapstructure:"app-secret" json:"app-secret" yaml:"app-secret"`
+}

config/oss_minio.go

@@ -0,0 +1,11 @@
+package config
+
+type Minio struct {
+	Endpoint        string `mapstructure:"endpoint" json:"endpoint" yaml:"endpoint"`
+	AccessKeyId     string `mapstructure:"access-key-id" json:"access-key-id" yaml:"access-key-id"`
+	AccessKeySecret string `mapstructure:"access-key-secret" json:"access-key-secret" yaml:"access-key-secret"`
+	BucketName      string `mapstructure:"bucket-name" json:"bucket-name" yaml:"bucket-name"`
+	UseSSL          bool   `mapstructure:"use-ssl" json:"use-ssl" yaml:"use-ssl"`
+	BasePath        string `mapstructure:"base-path" json:"base-path" yaml:"base-path"`
+	BucketUrl       string `mapstructure:"bucket-url" json:"bucket-url" yaml:"bucket-url"`
+}

config/oss_tencent.go

@@ -0,0 +1,10 @@
+package config
+
+type TencentCOS struct {
+	Bucket     string `mapstructure:"bucket" json:"bucket" yaml:"bucket"`
+	Region     string `mapstructure:"region" json:"region" yaml:"region"`
+	SecretID   string `mapstructure:"secret-id" json:"secret-id" yaml:"secret-id"`
+	SecretKey  string `mapstructure:"secret-key" json:"secret-key" yaml:"secret-key"`
+	BaseURL    string `mapstructure:"base-url" json:"base-url" yaml:"base-url"`
+	PathPrefix string `mapstructure:"path-prefix" json:"path-prefix" yaml:"path-prefix"`
+}

config/rocket_mq.go

@@ -0,0 +1,12 @@
+package config
+
+type RocketMQConfig struct {
+	NameSpace     string `mapstructure:"name-space" json:"nameSpace" yaml:"name-space"`
+	Endpoint      string `mapstructure:"endpoint" json:"endpoint" yaml:"endpoint"`
+	ConsumerGroup string `mapstructure:"consumer-group" json:"consumerGroup" yaml:"consumer-group"`
+	AccessKey     string `mapstructure:"access-key" json:"accessKey" yaml:"access-key"`
+	SecretKey     string `mapstructure:"secret-key" json:"secretKey" yaml:"secret-key"`
+	Topic         string `mapstructure:"topic" json:"topic" yaml:"topic"`
+	EnableSSL     bool   `mapstructure:"enable-ssl" json:"enableSSL" yaml:"enable-ssl"`
+	LogEnabled    bool   `mapstructure:"log-enabled" json:"logEnabled" yaml:"log-enabled"`
+}

config/service_account.go

@@ -0,0 +1,6 @@
+package config
+
+type ServiceAccount struct {
+	AppId     string `mapstructure:"app-id" json:"app-id" yaml:"app-id"`
+	AppSecret string `mapstructure:"app-secret" json:"app-secret" yaml:"app-secret"`
+}

config/system.go

@@ -0,0 +1,9 @@
+package config
+
+type System struct {
+	Addr          int    `mapstructure:"addr" json:"addr" yaml:"addr"`
+	DbType        string `mapstructure:"db-type" json:"db-type" yaml:"db-type"`
+	RouterPrefix  string `mapstructure:"router-prefix" json:"router-prefix" yaml:"router-prefix"`
+	EnableCaptcha int    `mapstructure:"enable-captcha" json:"enable-captcha" yaml:"enable-captcha"`
+	OssType       string `mapstructure:"oss-type" json:"oss-type" yaml:"oss-type"`
+}

config/wechat_pay.go

@@ -0,0 +1,12 @@
+package config
+
+// WechatPay 微信支付
+type WechatPay struct {
+	MchId                      string `mapstructure:"mch-id" json:"mch-id" yaml:"mch-id"`
+	PublicKeyId                string `mapstructure:"public-key-id" json:"public-key-id" yaml:"public-key-id"`
+	MchCertificateSerialNumber string `mapstructure:"mch-certificate-serial-number" json:"mch-certificate-serial-number" yaml:"mch-certificate-serial-number"`
+	MchAPIv3Key                string `mapstructure:"mch-api-v3-key" json:"mch-api-v3-key" yaml:"mch-api-v3-key"`
+	PrivateKeyPath             string `mapstructure:"private-key-path" json:"private-key-path" yaml:"private-key-path"`
+	PublicKeyPath              string `mapstructure:"public-key-path" json:"public-key-path" yaml:"public-key-path"`
+	NotifyUrl                  string `mapstructure:"notify-url" json:"notify-url" yaml:"notify-url"`
+}

config/zap.go

@@ -0,0 +1,81 @@
+package config
+
+import (
+	"go.uber.org/zap/zapcore"
+	"time"
+)
+
+type Zap struct {
+	Level         string `mapstructure:"level" json:"level" yaml:"level"`
+	Prefix        string `mapstructure:"prefix" json:"prefix" yaml:"prefix"`
+	Format        string `mapstructure:"format" json:"format" yaml:"format"`
+	Director      string `mapstructure:"director" json:"director"  yaml:"director"`
+	EncodeLevel   string `mapstructure:"encode-level" json:"encode-level" yaml:"encode-level"`
+	StacktraceKey string `mapstructure:"stacktrace-key" json:"stacktrace-key" yaml:"stacktrace-key"`
+	ShowLine      bool   `mapstructure:"show-line" json:"show-line" yaml:"show-line"`
+	LogInConsole  bool   `mapstructure:"log-in-console" json:"log-in-console" yaml:"log-in-console"`
+	RetentionDay  int    `mapstructure:"retention-day" json:"retention-day" yaml:"retention-day"`
+}
+
+// Levels 返回一个基于 Zap 实例中配置的日志级别的 zapcore.Level 切片。
+// 该切片从配置的日志级别开始，包含所有更高严重性级别，直到 FatalLevel。
+// 如果无法解析配置的日志级别，则默认使用 DebugLevel。
+//
+// 返回值:
+//   - []zapcore.Level: 包含从配置的日志级别（或解析失败时的 DebugLevel）到 FatalLevel 的所有日志级别的切片。
+func (c *Zap) Levels() []zapcore.Level {
+	// 初始化一个容量为 7 的空切片，用于存储日志级别。
+	levels := make([]zapcore.Level, 0, 7)
+
+	// 解析配置的日志级别。如果解析失败，则默认使用 DebugLevel。
+	level, err := zapcore.ParseLevel(c.Level)
+	if err != nil {
+		level = zapcore.DebugLevel
+	}
+
+	// 从解析的（或默认的）日志级别开始，迭代到 FatalLevel，并将每个级别追加到切片中 按照日志级别分片存储
+	for ; level <= zapcore.FatalLevel; level++ {
+		levels = append(levels, level)
+	}
+
+	// 返回填充好的日志级别切片
+	return levels
+}
+
+// Encoder 返回一个 zapcore.Encoder，用于编码日志记录。
+func (c *Zap) Encoder() zapcore.Encoder {
+	config := zapcore.EncoderConfig{
+		TimeKey:       "time",
+		NameKey:       "name",
+		LevelKey:      "level",
+		CallerKey:     "caller",
+		MessageKey:    "message",
+		StacktraceKey: c.StacktraceKey,
+		LineEnding:    zapcore.DefaultLineEnding,
+		EncodeTime: func(t time.Time, encoder zapcore.PrimitiveArrayEncoder) {
+			encoder.AppendString(c.Prefix + t.Format("2006-01-02 15:04:05"))
+		},
+		EncodeLevel:    c.LevelEncoder(),
+		EncodeCaller:   zapcore.FullCallerEncoder,
+		EncodeDuration: zapcore.SecondsDurationEncoder,
+	}
+	if c.Format == "json" {
+		return zapcore.NewJSONEncoder(config)
+	}
+	return zapcore.NewConsoleEncoder(config)
+}
+
+func (c *Zap) LevelEncoder() zapcore.LevelEncoder {
+	switch {
+	case c.EncodeLevel == "LowercaseLevelEncoder":
+		return zapcore.LowercaseLevelEncoder
+	case c.EncodeLevel == "LowercaseColorLevelEncoder":
+		return zapcore.LowercaseColorLevelEncoder
+	case c.EncodeLevel == "CapitalLevelEncoder":
+		return zapcore.CapitalLevelEncoder
+	case c.EncodeLevel == "CapitalColorLevelEncoder":
+		return zapcore.CapitalColorLevelEncoder
+	default:
+		return zapcore.LowercaseLevelEncoder
+	}
+}