aboutsummaryrefslogtreecommitdiff
path: root/backend/internal/ibd
diff options
context:
space:
mode:
authorGravatar Anshul Gupta <ansg191@anshulg.com> 2024-08-07 19:48:19 -0700
committerGravatar Anshul Gupta <ansg191@anshulg.com> 2024-08-07 19:48:52 -0700
commita439618cdc8168bad617d04875697b572f3ed41d (patch)
treeb6d867a2d65713deb2e8c2daa25a9df2dd23308b /backend/internal/ibd
parent55231eb4cfb5b5ce246a68ef456619cdb65d8af1 (diff)
downloadibd-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.go18
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)
}