zwxi преди 1 година
родител
ревизия
10728b3c41
променени са 2 файла, в които са добавени 9 реда и са изтрити 7 реда
  1. 9 6
      services/comein/comein.go
  2. 0 1
      services/task.go

+ 9 - 6
services/comein/comein.go

@@ -318,7 +318,7 @@ func InitAddWhiteList() (err error) {
 	var count int
 	for i, v := range whiteList {
 		request.Whitelist = append(request.Whitelist, v)
-		if count < 1000 && (len(whiteList)-i-1) > 0 {
+		if count < 10000 && (len(whiteList)-i-1) > 0 {
 			count++
 			continue
 		}
@@ -337,15 +337,17 @@ func InitAddWhiteList() (err error) {
 		}
 
 		// 发起POST请求
-		resp, err := utils.MakePostRequest(postUrl, jsonData, headers)
-		if err != nil {
+		resp, e := utils.MakePostRequest(postUrl, jsonData, headers)
+		if e != nil {
+			err = e
 			utils.FileLog.Info("comein addWhitelist Err:", err.Error(), ";url:", postUrl)
 			return
 		}
 		defer resp.Body.Close()
 
-		body, err := ioutil.ReadAll(resp.Body)
-		if err != nil {
+		body, e := ioutil.ReadAll(resp.Body)
+		if e != nil {
+			err = e
 			fmt.Println("reponse error", err)
 			return
 		}
@@ -406,7 +408,7 @@ func GetSignature(opt, timestamp string) (signature string) {
 	return
 }
 
-func InitComeinWhiteList() {
+func InitComeinWhiteList()(err error) {
 	list := make([]*models.ComeinWhiteUser, 0)
 
 	realList, err := models.GetUserMobileFormal()
@@ -440,6 +442,7 @@ func InitComeinWhiteList() {
 		return
 	}
 	fmt.Println("添加随机手机号成功")
+	return
 }
 
 // SyncWhiteList 同步进门财经白名单定时任务

+ 0 - 1
services/task.go

@@ -7,7 +7,6 @@ import (
 	"github.com/beego/beego/v2/task"
 	"hongze/hongze_task/models"
 	"hongze/hongze_task/services/alarm_msg"
-	"hongze/hongze_task/services/comein"
 	"hongze/hongze_task/services/company"
 	"hongze/hongze_task/services/company_contract"
 	"hongze/hongze_task/services/data"