f
This commit is contained in:
69
internal/infrastructure/external/shumai/shumai_factory.go
vendored
Normal file
69
internal/infrastructure/external/shumai/shumai_factory.go
vendored
Normal file
@@ -0,0 +1,69 @@
|
||||
package shumai
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"hyapi-server/internal/config"
|
||||
"hyapi-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)
|
||||
}
|
||||
Reference in New Issue
Block a user