feat/rankings: optimize Steam ID comparison (#236)

1 files changed, 28 insertions, 3 deletions

diff --git a/rankings/models.go b/rankings/models.go
index 1b349b0..0e15d34 100644
--- a/rankings/models.go
+++ b/rankings/models.go
@@ -1,5 +1,11 @@
 package main
 
+import (
+        "encoding/json"
+        "fmt"
+        "strconv"
+)
+
 type Record struct {
         MapID    int    `json:"id"`
         MapName  string `json:"name"`
@@ -25,15 +31,34 @@ type LeaderboardEntries struct {
         Entry []LeaderboardEntry `xml:"entry"`
 }
 
+type SteamID int64
+
+func (m SteamID) MarshalJSON() ([]byte, error) {
+        return json.Marshal(strconv.FormatInt(int64(m), 10))
+}
+
+func (id *SteamID) UnmarshalJSON(data []byte) error {
+        var s string
+        if err := json.Unmarshal(data, &s); err == nil {
+                n, err := strconv.ParseInt(s, 10, 64)
+                if err != nil {
+                        return err
+                }
+                *id = SteamID(n)
+                return nil
+        }
+        return fmt.Errorf("invalid type for SteamID: %s", data)
+}
+
 type LeaderboardEntry struct {
-        SteamID string `xml:"steamid"`
-        Score   int    `xml:"score"`
+        SteamID SteamID `xml:"steamid"`
+        Score   int     `xml:"score"`
 }
 
 type Player struct {
         Username          string        `json:"user_name"`
         AvatarLink        string        `json:"avatar_link"`
-        SteamID           string        `json:"steam_id"`
+        SteamID           SteamID       `json:"steam_id"`
         Entries           []PlayerEntry `json:"-"`
         SpScoreCount      int           `json:"sp_score"`
         MpScoreCount      int           `json:"mp_score"`