optimize

cmd/stock/main.go

@@ -11,7 +11,6 @@ import (
 	"net/http"
 	"stock/cfg"
 	"stock/fund"
-	"stock/module"
 	"stock/msg"
 	"stock/stock"
 	"stock/user"
@@ -89,7 +88,10 @@ func main() {
 		}
 
 		fund.Clear()
-		user.ForEachUser(func(u module.IUser) bool {
+		user.ForEachUser(func(u *user.User) bool {
+			if u.IsStop() {
+				return true
+			}
 			codes := u.Codes(true)
 			stk := fund.NewFundArg(codes...)
 			err = wxgzh.Send(u.OpenID(), stk)
@@ -153,7 +155,10 @@ func main() {
 					continue
 				}
 
-				user.ForEachUser(func(u module.IUser) bool {
+				user.ForEachUser(func(u *user.User) bool {
+					if u.IsStop() {
+						return true
+					}
 					codes := u.Codes(false)
 					stk, err := stock.GetStocks(codes...)
 					if err != nil {