totole/services
1
0
Fork 0
mirror of https://github.com/kevin-DL/services.git synced 2026-01-11 19:04:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

add volume to stock history

Browse Source
This commit is contained in:
Asim Aslam
2021-06-20 17:43:21 +01:00
parent 466990f74e
commit 5db7872fa7
1 changed files with 65 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

131
stock/handler/stock.go
View File

@@ -9,40 +9,39 @@ import (
"regexp" "regexp"
"time" "time"
"github.com/patrickmn/go-cache"
"github.com/micro/micro/v3/service/errors" "github.com/micro/micro/v3/service/errors"
"github.com/micro/micro/v3/service/logger" "github.com/micro/micro/v3/service/logger"
pb "github.com/micro/services/stock/proto" pb "github.com/micro/services/stock/proto"
"github.com/patrickmn/go-cache"
) )
var ( var (
re = regexp.MustCompile(`\d{4}-\d{2}-\d{2}`) re = regexp.MustCompile(`\d{4}-\d{2}-\d{2}`)
) )
type Stock struct {
type Stock struct{ Api string
Api string Key string
Key string
Cache *cache.Cache Cache *cache.Cache
} }
type Quote struct { type Quote struct {
Symbol string Symbol string
Ask float64 Ask float64
Bid float64 Bid float64
Asize int32 Asize int32
Bsize int32 Bsize int32
Timestamp int64 Timestamp int64
} }
type History struct { type History struct {
Symbol string Symbol string
Open float64 Open float64
High float64 High float64
Low float64 Low float64
Close float64 Close float64
Volume int32 Volume int32
From string From string
} }
func (s *Stock) History(ctx context.Context, req *pb.HistoryRequest, rsp *pb.HistoryResponse) error { func (s *Stock) History(ctx context.Context, req *pb.HistoryRequest, rsp *pb.HistoryResponse) error {
@@ -56,26 +55,26 @@ func (s *Stock) History(ctx context.Context, req *pb.HistoryRequest, rsp *pb.His
uri := fmt.Sprintf("%shistory/stock/open-close?stock=%s&date=%s&apikey=%s", s.Api, req.Stock, req.Date, s.Key) uri := fmt.Sprintf("%shistory/stock/open-close?stock=%s&date=%s&apikey=%s", s.Api, req.Stock, req.Date, s.Key)
resp, err := http.Get(uri) resp, err := http.Get(uri)
if err != nil { if err != nil {
logger.Errorf("Failed to get history: %v\n", err) logger.Errorf("Failed to get history: %v\n", err)
return errors.InternalServerError("stock.history", "failed to get history") return errors.InternalServerError("stock.history", "failed to get history")
} }
defer resp.Body.Close() defer resp.Body.Close()
b, _ := ioutil.ReadAll(resp.Body) b, _ := ioutil.ReadAll(resp.Body)
if resp.StatusCode != 200 { if resp.StatusCode != 200 {
logger.Errorf("Failed to get history (non 200): %d %v\n", resp.StatusCode, string(b)) logger.Errorf("Failed to get history (non 200): %d %v\n", resp.StatusCode, string(b))
return errors.InternalServerError("stock.history", "failed to get history") return errors.InternalServerError("stock.history", "failed to get history")
} }
var respBody History var respBody History
if err := json.Unmarshal(b, &respBody); err != nil { if err := json.Unmarshal(b, &respBody); err != nil {
logger.Errorf("Failed to unmarshal history: %v\n", err) logger.Errorf("Failed to unmarshal history: %v\n", err)
return errors.InternalServerError("stock.history", "failed to get history") return errors.InternalServerError("stock.history", "failed to get history")
} }
rsp.Symbol = respBody.Symbol rsp.Symbol = respBody.Symbol
rsp.Open = respBody.Open rsp.Open = respBody.Open
@@ -83,7 +82,7 @@ func (s *Stock) History(ctx context.Context, req *pb.HistoryRequest, rsp *pb.His
rsp.High = respBody.High rsp.High = respBody.High
rsp.Low = respBody.Low rsp.Low = respBody.Low
rsp.Date = respBody.From rsp.Date = respBody.From
rsp.Volume = respBody.Volume
return nil return nil
} }
func (s *Stock) Quote(ctx context.Context, req *pb.QuoteRequest, rsp *pb.QuoteResponse) error { func (s *Stock) Quote(ctx context.Context, req *pb.QuoteRequest, rsp *pb.QuoteResponse) error {
@@ -93,33 +92,33 @@ func (s *Stock) Quote(ctx context.Context, req *pb.QuoteRequest, rsp *pb.QuoteRe
uri := fmt.Sprintf("%slast/stock/%s?apikey=%s", s.Api, req.Symbol, s.Key) uri := fmt.Sprintf("%slast/stock/%s?apikey=%s", s.Api, req.Symbol, s.Key)
resp, err := http.Get(uri) resp, err := http.Get(uri)
if err != nil { if err != nil {
logger.Errorf("Failed to get quote: %v\n", err) logger.Errorf("Failed to get quote: %v\n", err)
return errors.InternalServerError("stock.quote", "failed to get quote") return errors.InternalServerError("stock.quote", "failed to get quote")
} }
defer resp.Body.Close() defer resp.Body.Close()
b, _ := ioutil.ReadAll(resp.Body) b, _ := ioutil.ReadAll(resp.Body)
if resp.StatusCode != 200 { if resp.StatusCode != 200 {
logger.Errorf("Failed to get quote (non 200): %d %v\n", resp.StatusCode, string(b)) logger.Errorf("Failed to get quote (non 200): %d %v\n", resp.StatusCode, string(b))
return errors.InternalServerError("stock.quote", "failed to get quote") return errors.InternalServerError("stock.quote", "failed to get quote")
} }
var respBody Quote var respBody Quote
if err := json.Unmarshal(b, &respBody); err != nil { if err := json.Unmarshal(b, &respBody); err != nil {
logger.Errorf("Failed to unmarshal quote: %v\n", err) logger.Errorf("Failed to unmarshal quote: %v\n", err)
return errors.InternalServerError("stock.quote", "failed to get quote") return errors.InternalServerError("stock.quote", "failed to get quote")
} }
rsp.Symbol = respBody.Symbol rsp.Symbol = respBody.Symbol
rsp.AskPrice = respBody.Ask rsp.AskPrice = respBody.Ask
rsp.BidPrice = respBody.Bid rsp.BidPrice = respBody.Bid
rsp.AskSize = respBody.Asize rsp.AskSize = respBody.Asize
rsp.BidSize = respBody.Bsize rsp.BidSize = respBody.Bsize
rsp.Timestamp = time.Unix(0, respBody.Timestamp * int64(time.Millisecond)).UTC().Format(time.RFC3339Nano) rsp.Timestamp = time.Unix(0, respBody.Timestamp*int64(time.Millisecond)).UTC().Format(time.RFC3339Nano)
return nil return nil
} }
@@ -131,26 +130,26 @@ func (s *Stock) Price(ctx context.Context, req *pb.PriceRequest, rsp *pb.PriceRe
uri := fmt.Sprintf("%slast/trade/stock/%s?apikey=%s", s.Api, req.Symbol, s.Key) uri := fmt.Sprintf("%slast/trade/stock/%s?apikey=%s", s.Api, req.Symbol, s.Key)
resp, err := http.Get(uri) resp, err := http.Get(uri)
if err != nil { if err != nil {
logger.Errorf("Failed to get price: %v\n", err) logger.Errorf("Failed to get price: %v\n", err)
return errors.InternalServerError("stock.trade", "failed to get price") return errors.InternalServerError("stock.trade", "failed to get price")
} }
defer resp.Body.Close() defer resp.Body.Close()
b, _ := ioutil.ReadAll(resp.Body) b, _ := ioutil.ReadAll(resp.Body)
if resp.StatusCode != 200 { if resp.StatusCode != 200 {
logger.Errorf("Failed to get price (non 200): %d %v\n", resp.StatusCode, string(b)) logger.Errorf("Failed to get price (non 200): %d %v\n", resp.StatusCode, string(b))
return errors.InternalServerError("stock.quote", "failed to get price") return errors.InternalServerError("stock.quote", "failed to get price")
} }
var respBody map[string]interface{} var respBody map[string]interface{}
if err := json.Unmarshal(b, &respBody); err != nil { if err := json.Unmarshal(b, &respBody); err != nil {
logger.Errorf("Failed to unmarshal price: %v\n", err) logger.Errorf("Failed to unmarshal price: %v\n", err)
return errors.InternalServerError("stock.price", "failed to get price") return errors.InternalServerError("stock.price", "failed to get price")
} }
rsp.Symbol = req.Symbol rsp.Symbol = req.Symbol
rsp.Price = respBody["price"].(float64) rsp.Price = respBody["price"].(float64)