70 lines
2.3 KiB
Go
70 lines
2.3 KiB
Go
package shumai
|
|
|
|
import (
|
|
"time"
|
|
|
|
"tyapi-server/internal/config"
|
|
"tyapi-server/internal/shared/external_logger"
|
|
)
|
|
|
|
// NewShumaiServiceWithConfig 使用 config 创建数脉服务
|
|
func NewShumaiServiceWithConfig(cfg *config.Config) (*ShumaiService, error) {
|
|
loggingConfig := external_logger.ExternalServiceLoggingConfig{
|
|
Enabled: cfg.Shumai.Logging.Enabled,
|
|
LogDir: cfg.Shumai.Logging.LogDir,
|
|
ServiceName: "shumai",
|
|
UseDaily: cfg.Shumai.Logging.UseDaily,
|
|
EnableLevelSeparation: cfg.Shumai.Logging.EnableLevelSeparation,
|
|
LevelConfigs: make(map[string]external_logger.ExternalServiceLevelFileConfig),
|
|
}
|
|
for k, v := range cfg.Shumai.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.Shumai.SignMethod == "md5" {
|
|
signMethod = SignMethodMD5
|
|
} else {
|
|
signMethod = SignMethodHMACMD5
|
|
}
|
|
timeout := 60 * time.Second
|
|
if cfg.Shumai.Timeout > 0 {
|
|
timeout = cfg.Shumai.Timeout
|
|
}
|
|
|
|
return NewShumaiService(
|
|
cfg.Shumai.URL,
|
|
cfg.Shumai.AppID,
|
|
cfg.Shumai.AppSecret,
|
|
signMethod,
|
|
timeout,
|
|
logger,
|
|
cfg.Shumai.AppID2, // 走政务接口使用这个
|
|
cfg.Shumai.AppSecret2, // 走政务接口使用这个
|
|
), nil
|
|
}
|
|
|
|
// NewShumaiServiceWithLogging 使用自定义日志配置创建数脉服务
|
|
func NewShumaiServiceWithLogging(url, appID, appSecret string, signMethod SignMethod, timeout time.Duration, loggingConfig external_logger.ExternalServiceLoggingConfig, appID2, appSecret2 string) (*ShumaiService, error) {
|
|
loggingConfig.ServiceName = "shumai"
|
|
logger, err := external_logger.NewExternalServiceLogger(loggingConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewShumaiService(url, appID, appSecret, signMethod, timeout, logger, appID2, appSecret2), nil
|
|
}
|
|
|
|
// NewShumaiServiceSimple 创建无数脉日志的数脉服务
|
|
func NewShumaiServiceSimple(url, appID, appSecret string, signMethod SignMethod, timeout time.Duration, appID2, appSecret2 string) *ShumaiService {
|
|
return NewShumaiService(url, appID, appSecret, signMethod, timeout, nil, appID2, appSecret2)
|
|
}
|