move feed update to separate function

This commit is contained in:
filifa 2024-07-13 17:58:11 -05:00
parent 0f32f74431
commit 7d29b22e88
1 changed files with 22 additions and 13 deletions

33
main.go
View File

@ -74,6 +74,26 @@ func extractPatches(diffPatch []byte) ([]jsonpatch.Patch, error) {
return patches, nil return patches, nil
} }
func updateFeed(feedResp []byte, gamePk string, ts string, updateId string) ([]byte, error) {
diffPatchResp, err := statsapi.RequestDiffPatch(gamePk, ts, updateId)
patches, err := extractPatches(diffPatchResp)
if err != nil {
feedResp, err = statsapi.RequestFeed(gamePk)
if err != nil {
return feedResp, err
}
} else {
for _, patch := range patches {
feedResp, err = patch.Apply(feedResp)
if err != nil {
return feedResp, err
}
}
}
return feedResp, err
}
func main() { func main() {
gamePk := getGamePk() gamePk := getGamePk()
@ -109,21 +129,10 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
diffPatchResp, err := statsapi.RequestDiffPatch(gamePk, ts, p.UpdateId) feedResp, err = updateFeed(feedResp, gamePk, ts, p.UpdateId)
patches, err := extractPatches(diffPatchResp)
if err != nil {
feedResp, err = statsapi.RequestFeed(gamePk)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
} else {
for _, patch := range patches {
feedResp, err = patch.Apply(feedResp)
if err != nil {
log.Fatal(err)
}
}
}
err = json.Unmarshal(feedResp, &feed) err = json.Unmarshal(feedResp, &feed)
if err != nil { if err != nil {