finish

cmd/stock/main.go

@@ -46,8 +46,14 @@ func main() {
 		logx.Fatal(err)
 	}
 
+	timer.AddTicker(time.Second*1200, func() {
+		logx.Info("开始保存用户数据进MongoDB...")
+		user.SaveAll()
+		logx.Info("保存用户数进MongoDB结束！")
+
+	})
+
 	stockCodes := user.Codes(false)
-	fundCodes := user.Codes(true)
 
 	timer.RunEveryDay(9, 25, 0, func() {
 		if !stock.IsSellDay(time.Now()) {
@@ -80,12 +86,17 @@ func main() {
 			logx.Info("今天不是交易日！")
 			return
 		}
-		text := fund.FundsMsg(fundCodes...)
-		if text == "" {
-			logx.Errorf("收集基金数据为空！")
-			return
-		}
-		err := msg.Send(text)
+
+		fund.Clear()
+		user.ForEachUser(func(u module.IUser) bool {
+			codes := u.Codes(true)
+			stk := fund.NewFundArg(codes...)
+			err = wxgzh.Send(u.OpenID(), stk)
+			if err != nil {
+				logx.Error(err)
+			}
+			return true
+		})
 		if err != nil {
 			logx.Errorf("dingtalk.SendMsg(基金信息) err: %v", err)
 		} else {
@@ -93,9 +104,11 @@ func main() {
 		}
 	})
 
-	err = stock.Init(stockCodes...)
-	if err != nil {
-		logx.Fatal(err)
+	if len(stockCodes) > 0 {
+		err = stock.Init(stockCodes...)
+		if err != nil {
+			logx.Error(err)
+		}
 	}
 
 	ctx.Go(func(ctx contextx.Context) error {
@@ -152,20 +165,6 @@ func main() {
 					}
 					return true
 				})
-
-				//text := ss.Msg()
-				//if text == "" {
-				//	logx.Info("已更新数据，未超过阈值，无警告！")
-				//	continue
-				//}
-				//err = msg.Send(text)
-				//if err != nil {
-				//	count++
-				//	logx.Errorf("SendMsg ErrCount=%d err=%v", count, err)
-				//	if count > 10 {
-				//		return err
-				//	}
-				//}
 			}
 		}
 	})
@@ -185,7 +184,13 @@ func main() {
 		})
 	}, func(s *httpx.Server) {
 		s.Mode = "debug"
-		s.Port = 8567
+		s.Port = 16888
+	})
+
+	ctx.Go(func(ctx contextx.Context) error {
+		<-ctx.Done()
+		user.SaveAll()
+		return nil
 	})
 
 	err = ctx.Wait() // 等待停止信号