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