67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package shujubao
|
|
|
|
import (
|
|
"time"
|
|
|
|
"tyapi-server/internal/config"
|
|
"tyapi-server/internal/shared/external_logger"
|
|
)
|
|
|
|
// NewShujubaoServiceWithConfig 使用配置创建数据宝服务
|
|
func NewShujubaoServiceWithConfig(cfg *config.Config) (*ShujubaoService, error) {
|
|
loggingConfig := external_logger.ExternalServiceLoggingConfig{
|
|
Enabled: cfg.Shujubao.Logging.Enabled,
|
|
LogDir: cfg.Shujubao.Logging.LogDir,
|
|
ServiceName: "shujubao",
|
|
UseDaily: cfg.Shujubao.Logging.UseDaily,
|
|
EnableLevelSeparation: cfg.Shujubao.Logging.EnableLevelSeparation,
|
|
LevelConfigs: make(map[string]external_logger.ExternalServiceLevelFileConfig),
|
|
}
|
|
for k, v := range cfg.Shujubao.Logging.LevelConfigs {
|
|
loggingConfig.LevelConfigs[k] = external_logger.ExternalServiceLevelFileConfig{
|
|
MaxSize: v.MaxSize,
|
|
MaxBackups: v.MaxBackups,
|
|
MaxAge: v.MaxAge,
|
|
Compress: v.Compress,
|
|
}
|
|
}
|
|
logger, err := external_logger.NewExternalServiceLogger(loggingConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var signMethod SignMethod
|
|
if cfg.Shujubao.SignMethod == "md5" {
|
|
signMethod = SignMethodMD5
|
|
} else {
|
|
signMethod = SignMethodHMACMD5
|
|
}
|
|
timeout := 60 * time.Second
|
|
if cfg.Shujubao.Timeout > 0 {
|
|
timeout = cfg.Shujubao.Timeout
|
|
}
|
|
|
|
return NewShujubaoService(
|
|
cfg.Shujubao.URL,
|
|
cfg.Shujubao.AppSecret,
|
|
signMethod,
|
|
timeout,
|
|
logger,
|
|
), nil
|
|
}
|
|
|
|
// NewShujubaoServiceWithLogging 使用自定义日志配置创建数据宝服务
|
|
func NewShujubaoServiceWithLogging(url, appSecret string, signMethod SignMethod, timeout time.Duration, loggingConfig external_logger.ExternalServiceLoggingConfig) (*ShujubaoService, error) {
|
|
loggingConfig.ServiceName = "shujubao"
|
|
logger, err := external_logger.NewExternalServiceLogger(loggingConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewShujubaoService(url, appSecret, signMethod, timeout, logger), nil
|
|
}
|
|
|
|
// NewShujubaoServiceSimple 创建无日志的数据宝服务
|
|
func NewShujubaoServiceSimple(url, appSecret string, signMethod SignMethod, timeout time.Duration) *ShujubaoService {
|
|
return NewShujubaoService(url, appSecret, signMethod, timeout, nil)
|
|
}
|