diff options
author | 2024-08-07 19:48:19 -0700 | |
---|---|---|
committer | 2024-08-07 19:48:52 -0700 | |
commit | a439618cdc8168bad617d04875697b572f3ed41d (patch) | |
tree | b6d867a2d65713deb2e8c2daa25a9df2dd23308b /backend/internal/ibd | |
parent | 55231eb4cfb5b5ce246a68ef456619cdb65d8af1 (diff) | |
download | ibd-trader-a439618cdc8168bad617d04875697b572f3ed41d.tar.gz ibd-trader-a439618cdc8168bad617d04875697b572f3ed41d.tar.zst ibd-trader-a439618cdc8168bad617d04875697b572f3ed41d.zip |
Rewrite `LetterRating` to match protobuf def
Also changes DB schema to accept strings instead of integers.
Diffstat (limited to 'backend/internal/ibd')
-rw-r--r-- | backend/internal/ibd/stockinfo.go | 18 |
1 files changed, 3 insertions, 15 deletions
diff --git a/backend/internal/ibd/stockinfo.go b/backend/internal/ibd/stockinfo.go index 9caa956..1e3b96f 100644 --- a/backend/internal/ibd/stockinfo.go +++ b/backend/internal/ibd/stockinfo.go @@ -207,23 +207,11 @@ func processSmartRating(node *html.Node, ratings *database.Ratings) error { } ratings.RelStr = uint8(ratingValue) case "Group RS Rating": - ratingValue, err := database.LetterRatingFromString(ratingValueStr) - if err != nil { - return fmt.Errorf("failed to parse Group RS Rating: %w", err) - } - ratings.GroupRelStr = ratingValue + ratings.GroupRelStr = database.LetterRatingFromString(ratingValueStr) case "SMR Rating": - ratingValue, err := database.LetterRatingFromString(ratingValueStr) - if err != nil { - return fmt.Errorf("failed to parse SMR Rating: %w", err) - } - ratings.SMR = ratingValue + ratings.SMR = database.LetterRatingFromString(ratingValueStr) case "Acc/Dis Rating": - ratingValue, err := database.LetterRatingFromString(ratingValueStr) - if err != nil { - return fmt.Errorf("failed to parse Acc/Dis Rating: %w", err) - } - ratings.AccDis = ratingValue + ratings.AccDis = database.LetterRatingFromString(ratingValueStr) default: return fmt.Errorf("unknown rating name: %s", ratingName) } |