fix(ls): rate limit foreign index TRs

Author: smallfish06

internal/ls/client.go

@@ -310,7 +310,7 @@ func (c *Client) trLimiter(trCD string) *ratelimit.Limiter {
 
 func lsTRRateLimit(trCD string) (float64, int, bool) {
 	switch strings.ToLower(strings.TrimSpace(trCD)) {
-	case TRStockChart, TROverseasStockChart:
+	case TRStockChart, TROverseasStockChart, TRForeignIndexHistory, TRForeignIndexQuote:
 		return 1, 1, true
 	case TROverseasStockQuote, "g3102", TROverseasStockInstrument, "g3106", TROverseasStockMaster:
 		return 10, 1, true

internal/ls/client_test.go

@@ -295,6 +295,16 @@ func TestLSTRRateLimit_UsesDocumentedChartQuotas(t *testing.T) {
 		t.Fatalf("g3204 limit = rps %v burst %d ok %v, want 1/1/true", rps, burst, ok)
 	}
 
+	rps, burst, ok = lsTRRateLimit(TRForeignIndexQuote)
+	if !ok || rps != 1 || burst != 1 {
+		t.Fatalf("t3521 limit = rps %v burst %d ok %v, want 1/1/true", rps, burst, ok)
+	}
+
+	rps, burst, ok = lsTRRateLimit(TRForeignIndexHistory)
+	if !ok || rps != 1 || burst != 1 {
+		t.Fatalf("t3518 limit = rps %v burst %d ok %v, want 1/1/true", rps, burst, ok)
+	}
+
 	rps, burst, ok = lsTRRateLimit(TROverseasStockQuote)
 	if !ok || rps != 10 || burst != 1 {
 		t.Fatalf("g3101 limit = rps %v burst %d ok %v, want 10/1/true", rps, burst, ok)

internal/ls/paths.go

@@ -21,21 +21,24 @@ const (
 	PathStockOrder      = "/stock/order"
 	PathStockMisc       = "/stock/etc"
 	PathStockItemSearch = "/stock/item-search"
+	PathStockInvestInfo = "/stock/investinfo"
 
 	PathOverseasStockMarket = "/overseas-stock/market-data"
 	PathOverseasStockChart  = "/overseas-stock/chart"
 )
 
 const (
-	TRStockQuote      = "t1102"
-	TRStockChart      = "t8410"
-	TRStockBalance    = "t0424"
-	TRStockOrder      = "CSPAT00601"
-	TRStockMaster     = "t8436"
-	TRRealtimeKOSPI   = "S3_"
-	TRRealtimeKOSDAQ  = "K3_"
-	TRRealtimeUnified = "US3"
-	TRRealtimeNXT     = "NS3"
+	TRStockQuote          = "t1102"
+	TRStockChart          = "t8410"
+	TRStockBalance        = "t0424"
+	TRStockOrder          = "CSPAT00601"
+	TRStockMaster         = "t8436"
+	TRForeignIndexHistory = "t3518"
+	TRForeignIndexQuote   = "t3521"
+	TRRealtimeKOSPI       = "S3_"
+	TRRealtimeKOSDAQ      = "K3_"
+	TRRealtimeUnified     = "US3"
+	TRRealtimeNXT         = "NS3"
 
 	TROverseasStockQuote      = "g3101"
 	TROverseasStockInstrument = "g3104"