1
0
mirror of https://github.com/muety/wakapi.git synced 2023-08-10 21:12:56 +03:00

refactor: remove repeated code in readyPollTimer

This commit is contained in:
Edward 2023-07-14 00:33:55 +08:00
parent 7b0bbcefe6
commit 583ddcab7a
No known key found for this signature in database

View File

@ -162,22 +162,20 @@ func (w *WakatimeDumpImporter) Import(user *models.User, minFrom time.Time, maxT
// start polling for dump to be ready // start polling for dump to be ready
readyPollTimer, err = w.queue.DispatchEvery(func() { readyPollTimer, err = w.queue.DispatchEvery(func() {
u := *user u := *user
var ok bool
var dump *wakatime.DataDumpData
var err error
if dump_exist { if dump_exist {
ok, dump, err := getLatestDump(&u) ok, dump, err = getLatestDump(&u)
if err != nil {
onDumpFailed(err, &u)
} else if ok {
onDumpReady(dump, &u, out)
}
} else { } else {
ok, dump, err := checkDumpReady(datadumpData.Data.Id, &u) ok, dump, err = checkDumpReady(datadumpData.Data.Id, &u)
logbuch.Info("waiting for data dump '%s' for user '%s' to become downloadable (%.2f percent complete)", datadumpData.Data.Id, u.ID, dump.PercentComplete) }
logbuch.Info("waiting for data dump '%s' for user '%s' to become downloadable (%.2f percent complete)", dump.Id, u.ID, dump.PercentComplete)
if err != nil { if err != nil {
onDumpFailed(err, &u) onDumpFailed(err, &u)
} else if ok { } else if ok {
onDumpReady(dump, &u, out) onDumpReady(dump, &u, out)
} }
}
}, 10*time.Second) }, 10*time.Second)
return out, nil return out, nil