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) }