170 lines
3.2 KiB
Go
170 lines
3.2 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/flytam/filenamify"
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var DefaultConfig = &Config{}
|
|
|
|
type (
|
|
Config struct {
|
|
Systems []System `json:"System" mapstructure:"System"`
|
|
}
|
|
System struct {
|
|
Name string
|
|
Db string
|
|
Jobs []Job
|
|
}
|
|
|
|
Job struct {
|
|
Name string
|
|
Tasks []Task
|
|
File string
|
|
Size int //文件最大行数
|
|
}
|
|
|
|
Task struct {
|
|
PK string `mapstructure:"pk"`
|
|
Sql string
|
|
Timestamp bool
|
|
Elt string
|
|
Order string
|
|
}
|
|
)
|
|
|
|
func (c Config) GetSystem(name string) (System, error) {
|
|
for _, s := range c.Systems {
|
|
if s.Name == name {
|
|
return s, nil
|
|
}
|
|
}
|
|
return System{}, errors.New("没有找到相关配置:" + name)
|
|
}
|
|
|
|
func (s System) GetJob(name string) (Job, error) {
|
|
for _, j := range s.Jobs {
|
|
if j.Name == name {
|
|
return j, nil
|
|
}
|
|
}
|
|
return Job{}, errors.New("没有找到相关配置:" + name)
|
|
}
|
|
|
|
func (j Job) GetFileName(params map[string]interface{}) string {
|
|
|
|
m := regexp.MustCompile("({[a-zA-Z0-9]+})")
|
|
//替换文件名参数
|
|
fileName := m.ReplaceAllFunc([]byte(j.File), func(b []byte) []byte {
|
|
field := string(b[1 : len(b)-1])
|
|
|
|
if val, ok := params[field]; ok {
|
|
return []byte(toString(val, "20060102"))
|
|
}
|
|
return b
|
|
})
|
|
|
|
//安全命名
|
|
path, name := filepath.Split(string(fileName))
|
|
name, err := filenamify.Filenamify(name, filenamify.Options{Replacement: "-"})
|
|
if err != nil {
|
|
log.Printf("不安全的文件名：%s", err.Error())
|
|
}
|
|
return path + name
|
|
}
|
|
|
|
func (t Task) GetSql(params map[string]interface{}) string {
|
|
sql := t.Sql
|
|
|
|
m := regexp.MustCompile("({[a-zA-Z0-9]+})")
|
|
if strings.Trim(t.Elt, " ") != "" {
|
|
sql = sql + " where " + t.Elt
|
|
build := m.ReplaceAllFunc([]byte(sql), func(b []byte) []byte {
|
|
field := string(b[1 : len(b)-1])
|
|
|
|
if val, ok := params[field]; ok {
|
|
return []byte(toString(val, t.Timestamp))
|
|
}
|
|
return b
|
|
})
|
|
|
|
sql = string(build)
|
|
}
|
|
|
|
if strings.Trim(t.Order, " ") != "" {
|
|
sql = sql + " order by " + t.Order
|
|
}
|
|
|
|
return sql
|
|
}
|
|
|
|
func toString(parm interface{}, timestamp interface{}) string {
|
|
switch p := parm.(type) {
|
|
case time.Time:
|
|
switch t := timestamp.(type) {
|
|
case bool:
|
|
if t {
|
|
return strconv.FormatInt(p.Unix(), 10)
|
|
} else {
|
|
return p.Format("2006-01-02 15:04:05")
|
|
}
|
|
case string:
|
|
return p.Format(t)
|
|
default:
|
|
return ""
|
|
}
|
|
|
|
case string:
|
|
return p
|
|
case int:
|
|
return strconv.Itoa(p)
|
|
case int32:
|
|
return strconv.FormatInt(int64(p), 10)
|
|
default:
|
|
return fmt.Sprint(p)
|
|
}
|
|
}
|
|
|
|
func LoadConfig(path string) *Config {
|
|
var c Config
|
|
|
|
viper.AddConfigPath(path) //设置读取的文件路径
|
|
viper.SetConfigName("Config") //设置读取的文件名
|
|
viper.SetConfigType("yaml") //设置文件的类型
|
|
|
|
//尝试进行配置读取
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
fmt.Println("请将config.yml.example拷贝为config.yaml")
|
|
panic(err)
|
|
}
|
|
//v := viper.GetViper()
|
|
//fmt.Println(v)
|
|
if err := viper.Unmarshal(&c); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
DefaultConfig = &c
|
|
return &c
|
|
}
|
|
|
|
func GetJob(conf *Config, sys, job string) (Job, string, error) {
|
|
s, err := conf.GetSystem(sys)
|
|
if err != nil {
|
|
return Job{}, "", err
|
|
}
|
|
|
|
j, err := s.GetJob(job)
|
|
if err != nil {
|
|
return Job{}, "", err
|
|
}
|
|
return j, s.Db, nil
|
|
}
|