diff --git a/codes_v2.go b/codes_v2.go index d7a53c7..f8f4fa9 100644 --- a/codes_v2.go +++ b/codes_v2.go @@ -77,7 +77,7 @@ func NewCodes2(op ...Codes2Option) (*Codes2, error) { tempDir: filepath.Join(DefaultDataDir, "temp"), spec: "10 0 9 * * *", updateKey: "codes", - retry: 3, + retry: DefaultRetry, dial: NewRangeDial(Hosts), dialOption: nil, m: maps.NewGeneric[string, *CodeModel](), @@ -117,7 +117,7 @@ func NewCodes2(op ...Codes2Option) (*Codes2, error) { // 定时更新 cr := cron.New(cron.WithSeconds()) _, err = cr.AddFunc(cs.spec, func() { - for i := 0; i < 3; i++ { + for i := 0; i == 0 || i < cs.retry; i++ { if err := cs.Update(); err != nil { logs.Err(err) <-time.After(time.Minute * 5) diff --git a/manage.go b/manage.go index 6532877..0c3cd3d 100644 --- a/manage.go +++ b/manage.go @@ -11,6 +11,7 @@ import ( const ( DefaultClients = 1 + DefaultRetry = 3 DefaultDataDir = "./data" DefaultDatabaseDir = "./data/database" )