diff --git a/backend/data/stock_data_api.go b/backend/data/stock_data_api.go index cc05ce9..ada06e3 100644 --- a/backend/data/stock_data_api.go +++ b/backend/data/stock_data_api.go @@ -8,6 +8,7 @@ import ( "bytes" "context" "encoding/json" + "errors" "fmt" "github.com/PuerkitoBio/goquery" "github.com/chromedp/chromedp" @@ -291,6 +292,9 @@ func (receiver StockDataApi) GetStockCodeRealTimeData(StockCodes ...string) (*[] stockInfos := make([]StockInfo, 0) str := GB18030ToUTF8(resp.Body()) dataStr := strutil.SplitEx(str, "\n", true) + if len(dataStr) == 0 { + return &[]StockInfo{}, errors.New("获取股票信息失败,请检查股票代码是否正确") + } for _, data := range dataStr { //logger.SugaredLogger.Info(data) stockData, err := ParseFullSingleStockData(data) @@ -318,8 +322,8 @@ func (receiver StockDataApi) GetStockCodeRealTimeData(StockCodes ...string) (*[] func (receiver StockDataApi) Follow(stockCode string) string { logger.SugaredLogger.Infof("Follow %s", stockCode) stockInfos, err := receiver.GetStockCodeRealTimeData(stockCode) - if err != nil { - logger.SugaredLogger.Error(err.Error()) + if err != nil || len(*stockInfos) == 0 { + logger.SugaredLogger.Error(err) return "关注失败" } stockInfo := (*stockInfos)[0] diff --git a/frontend/src/components/stock.vue b/frontend/src/components/stock.vue index 896a02e..03b2051 100644 --- a/frontend/src/components/stock.vue +++ b/frontend/src/components/stock.vue @@ -231,12 +231,20 @@ function isTradingTime() { } function AddStock(){ + if (!data?.code) { + message.error("请输入有效股票代码"); + return; + } if (!stocks.value.includes(data.code)) { - stocks.value.push(data.code) Follow(data.code).then(result => { - message.success(result) + if(result==="关注成功"){ + stocks.value.push(data.code) + message.success(result) + monitor(); + }else{ + message.error(result) + } }) - monitor() }else{ message.error("已经关注了") }