Skip to content

Commit

Permalink
Reuse zstandard encoder
Browse files Browse the repository at this point in the history
  • Loading branch information
patapancakes committed Apr 22, 2024
1 parent 697c533 commit 1de3ce7
Showing 1 changed file with 5 additions and 10 deletions.
15 changes: 5 additions & 10 deletions api/savedata/update.go
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@ import (
"github.com/pagefaultgames/pokerogue-server/defs"
)

var zstdEncoder, _ = zstd.NewWriter(nil)

// /savedata/update - update save data
func Update(uuid []byte, slot int, save any) error {
err := db.UpdateAccountLastActivity(uuid)
Expand All @@ -29,13 +31,6 @@ func Update(uuid []byte, slot int, save any) error {
var filename string
var buf bytes.Buffer

zstdEncoder, err := zstd.NewWriter(&buf)
if err != nil {
return fmt.Errorf("failed to create zstd encoder: %s", err)
}

defer zstdEncoder.Close()

switch save := save.(type) {
case defs.SystemSaveData: // System
if save.TrainerId == 0 && save.SecretId == 0 {
Expand All @@ -53,7 +48,7 @@ func Update(uuid []byte, slot int, save any) error {

filename = "system"

err = gob.NewEncoder(zstdEncoder).Encode(save)
err = gob.NewEncoder(&buf).Encode(save)
if err != nil {
return fmt.Errorf("failed to serialize save: %s", err)
}
Expand All @@ -69,7 +64,7 @@ func Update(uuid []byte, slot int, save any) error {
filename += strconv.Itoa(slot)
}

err = gob.NewEncoder(zstdEncoder).Encode(save)
err = gob.NewEncoder(&buf).Encode(save)
if err != nil {
return fmt.Errorf("failed to serialize save: %s", err)
}
Expand All @@ -81,7 +76,7 @@ func Update(uuid []byte, slot int, save any) error {
return fmt.Errorf("tried to write empty save file")
}

err = os.WriteFile(fmt.Sprintf("userdata/%x/%s.pzs", uuid, filename), buf.Bytes(), 0644)
err = os.WriteFile(fmt.Sprintf("userdata/%x/%s.pzs", uuid, filename), zstdEncoder.EncodeAll(buf.Bytes(), nil), 0644)
if err != nil {
return fmt.Errorf("failed to write save to disk: %s", err)
}
Expand Down

0 comments on commit 1de3ce7

Please sign in to comment.