kanade
9968887665
- 添加 .gitignore 忽略规则 - 实现应用配置加载与管理逻辑 - 添加默认配置文件 app.ini - 配置 Gitea CI/CD 工作流用于构建和部署 - 实现金蝶云客户端初始化功能 - 添加 RPC 插件支持 Consul 注册中心 - 实现部门数据获取及树形结构处理逻辑 - 添加通用工具函数库 - 初始化 Go 模块依赖管理 - 创建 Dockerfile 用于服务容器化部署
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
package ik3cloud
|
|
|
|
import (
|
|
"encoding"
|
|
"fmt"
|
|
"ik3cloud/app/config"
|
|
"ik3cloud/app/lib/logger"
|
|
"ik3cloud/app/lib/rpcplugin"
|
|
"io"
|
|
"path"
|
|
"reflect"
|
|
"time"
|
|
|
|
config2 "git.kumo.work/shama/service/config"
|
|
|
|
"github.com/rcrowley/go-metrics"
|
|
"github.com/shopspring/decimal"
|
|
"github.com/smallnest/rpcx/server"
|
|
"github.com/vmihailenco/msgpack/v5"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
"gopkg.in/natefinch/lumberjack.v2"
|
|
)
|
|
|
|
// RegisterBinaryExt registers a binary marshaler/unmarshaler for the given type.
|
|
// The type must be implemented both interfaces.
|
|
func RegisterBinaryExt[T encoding.BinaryMarshaler, _ interface {
|
|
encoding.BinaryUnmarshaler
|
|
*T
|
|
}](extID int8) {
|
|
var zero T
|
|
msgpack.RegisterExtEncoder(extID, zero, func(e *msgpack.Encoder, v reflect.Value) ([]byte, error) {
|
|
m := v.Interface().(encoding.BinaryMarshaler)
|
|
return m.MarshalBinary()
|
|
})
|
|
msgpack.RegisterExtDecoder(extID, zero, func(d *msgpack.Decoder, v reflect.Value, extLen int) error {
|
|
u := v.Addr().Interface().(encoding.BinaryUnmarshaler)
|
|
b := make([]byte, extLen)
|
|
if err := d.ReadFull(b); err != nil {
|
|
return err
|
|
}
|
|
return u.UnmarshalBinary(b)
|
|
})
|
|
}
|
|
|
|
// InitApp 初始化
|
|
func InitApp() (closes []func()) {
|
|
// 关闭资源
|
|
closes = []func(){}
|
|
// 初始化配置文件
|
|
config.InitConfig(&config.IniConfig{
|
|
ConfigPath: "config",
|
|
RunModelErrAllow: true,
|
|
})
|
|
|
|
var levels []zapcore.Level
|
|
for _, level := range config.Config.Log.Levels {
|
|
levels = append(levels, zapcore.Level(level))
|
|
}
|
|
// 初始化日志
|
|
logger.InitLogger(&logger.LoggerConfig{
|
|
Levels: levels,
|
|
ShowLine: config.Config.Log.ShowLine,
|
|
LogInConsole: config.Config.Log.LogInConsole,
|
|
EncodeLevel: logger.LowercaseColorLevelEncoder,
|
|
Prefix: config.Config.Log.Prefix,
|
|
Writer: func(conf *logger.LoggerConfig, filename string, level zapcore.Level) io.Writer {
|
|
maxAge := config.Config.Log.MaxAge
|
|
maxSize := config.Config.Log.MaxSize
|
|
maxBackups := config.Config.Log.MaxBackups
|
|
if level == zapcore.ErrorLevel {
|
|
maxAge = 0
|
|
maxSize = 0
|
|
maxBackups = 0
|
|
}
|
|
return &lumberjack.Logger{
|
|
Filename: path.Join(config.Config.Log.Director, filename+".log"),
|
|
MaxAge: maxAge, // days
|
|
Compress: true, // disabled by default
|
|
MaxSize: maxSize,
|
|
MaxBackups: maxBackups,
|
|
}
|
|
},
|
|
})
|
|
|
|
// 初始化rpc服务
|
|
config2.InitConfig(config2.Config{
|
|
RegistryServer: config.Config.Server.RegistryServer,
|
|
})
|
|
|
|
// 注册序列化
|
|
RegisterBinaryExt[decimal.Decimal](1)
|
|
return
|
|
}
|
|
|
|
// AddRegistryPlugin @Title 注册中心
|
|
func AddRegistryPlugin(s *server.Server) {
|
|
r := &rpcplugin.ConsulRegisterPlugin{
|
|
ServiceAddress: fmt.Sprintf("tcp@%s:%d", config.Config.Server.Host, config.Config.Server.Port),
|
|
ConsulServers: config.Config.Server.RegistryServer,
|
|
BasePath: config.Config.Server.BasePath,
|
|
Metrics: metrics.NewRegistry(),
|
|
UpdateInterval: time.Minute,
|
|
}
|
|
err := r.Start()
|
|
if err != nil {
|
|
logger.Logger.Fatal("注册中心初始化失败", zap.Error(err))
|
|
}
|
|
s.Plugins.Add(r)
|
|
}
|