优化extend.GetBjCodes,直接输出代码

client.go

@@ -181,6 +181,11 @@ func (this *Client) handlerDealMessage(c *client.Client, msg ios.Acker) {
 
 }
 
+// SetTimeout 设置超时时间
+func (this *Client) SetTimeout(t time.Duration) {
+	this.Wait.SetTimeout(t)
+}
+
 // SendFrame 发送数据,并等待响应
 func (this *Client) SendFrame(f *protocol.Frame, cache ...any) (any, error) {
 	f.MsgID = atomic.AddUint32(&this.msgID, 1)
@@ -457,7 +462,12 @@ func (this *Client) GetHistoryTradeBefore(code string, w *Workday, before time.T
 	start := time.Date(resp.List[0].Time.Year(), resp.List[0].Time.Month(), 1, 0, 0, 0, 0, resp.List[0].Time.Location())
 	var res *protocol.TradeResp
 	w.Range(start, before, func(t time.Time) bool {
-		res, err = this.GetHistoryTradeDay(t.Format("20060102"), code)
+		for i := 0; i < 3; i++ {
+			res, err = this.GetHistoryTradeDay(t.Format("20060102"), code)
+			if err == nil {
+				break
+			}
+		}
 		if err != nil {
 			return false
 		}

extend/codes-bj.go

@@ -4,6 +4,14 @@ import (
 	"github.com/injoyai/tdx"
 )
 
-func GetBjCodes() ([]*tdx.BjCode, error) {
-	return tdx.GetBjCodes()
+func GetBjCodes() ([]string, error) {
+	cs, err := tdx.GetBjCodes()
+	if err != nil {
+		return nil, err
+	}
+	ls := []string(nil)
+	for _, v := range cs {
+		ls = append(ls, "bj"+v.Code)
+	}
+	return ls, nil
 }