48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
package config
|
||
|
||
import (
|
||
"strconv"
|
||
|
||
"github.com/zeromicro/go-zero/rest"
|
||
)
|
||
|
||
type PostgresConfig struct {
|
||
Host string `yaml:"Host"`
|
||
Port int `yaml:"Port"`
|
||
User string `yaml:"User"`
|
||
Password string `yaml:"Password"`
|
||
DBName string `yaml:"DBName"`
|
||
SSLMode string `yaml:"SSLMode"`
|
||
}
|
||
|
||
type AuthConfig struct {
|
||
AccessSecret string `yaml:"AccessSecret"`
|
||
AccessExpire int64 `yaml:"AccessExpire"`
|
||
}
|
||
|
||
// InternalConfig 项目间内部 API 配置
|
||
type InternalConfig struct {
|
||
Token string `yaml:"Token"` // 写死的项目间 API token（X-API-TOKEN）
|
||
}
|
||
|
||
type Config struct {
|
||
rest.RestConf
|
||
Postgres PostgresConfig `yaml:"Postgres"`
|
||
Auth AuthConfig `yaml:"Auth"`
|
||
Internal InternalConfig `yaml:"Internal"`
|
||
Redis struct {
|
||
Host string `yaml:"Host"`
|
||
Type string `yaml:"Type"`
|
||
} `yaml:"Redis"`
|
||
}
|
||
|
||
// DSN 返回 PostgreSQL 连接串
|
||
func (p PostgresConfig) DSN() string {
|
||
return "host=" + p.Host +
|
||
" port=" + strconv.Itoa(p.Port) +
|
||
" user=" + p.User +
|
||
" password=" + p.Password +
|
||
" dbname=" + p.DBName +
|
||
" sslmode=" + p.SSLMode
|
||
}
|