liuxiaolong
2020-08-17 bae4af47f77a195a12a0437584d667465e826e12
service/msgPush.go
@@ -46,45 +46,65 @@
   }()
}
//解绑所有与该别名绑定的cid
//func UnbindAlias(alias string) (bool, error) {
//   appId := beego.AppConfig.String("pushAppId")
//   baseUrl := beego.AppConfig.String("pushBaseUrl") + appId
//   retryTimes := 0
//ReTry:
//   token := getCacheToken()
//   if token == "" {
//      return false, errors.New("token is nil")
//   }
//   url := baseUrl + "/user/alias/"+alias
//   header := map[string]string {
//      "token": token,
//   }
//   b,err := util.DoDeleteRequest(url, util.CONTENT_TYPE_UTF8_JSON, nil, header)
//   if err !=nil {
//      fmt.Println("DoDelete err:", err)
//      return false, err
//   }
//   var result PushResult
//   err = json.Unmarshal(b, &result)
//   if err != nil {
//      fmt.Println("unmarshal err:", err)
//      return false, err
//   }
//   if result.Code == 0 { //解绑成功
//      return true, nil
//   } else if result.Code == 10001 { //token过期
//      if retryTimes <=3 {
//         newToken, err := RefreshToken()
//         if err == nil {
//            updateToken(newToken)
//            retryTimes++
//            goto ReTry
//         }
//      }
//   }
//   return false, errors.New(result.Msg)
//}
//批量解绑别名
func UnbindAlias(cid string) (bool, error) {
   var uc models.UserClient
   ucList := uc.GetByCid(cid)
   if ucList ==nil {
      return true,errors.New("ucList is nil")
   }
   var aliasArr []string
   for _,als := range ucList {
      aliasArr = append(aliasArr, als.PhoneNum)
   }
   appId := beego.AppConfig.String("pushAppId")
   baseUrl := beego.AppConfig.String("pushBaseUrl") + appId
   retryTimes := 0
ReTry:
   token := getCacheToken()
   if token == "" {
      return false, errors.New("token is nil")
   }
   url := baseUrl + "/user/alias"
   header := map[string]string {
      "token": token,
   }
   caArr := make([]map[string]string, 0)
   for _,as := range aliasArr {
      caArr = append(caArr, map[string]string{
         "cid": cid,
         "alias": as,
      })
   }
   reqBody := map[string]interface{}{
      "data_list": caArr,
   }
   b,err := util.DoDeleteRequest(url, util.CONTENT_TYPE_UTF8_JSON, reqBody, header)
   if err !=nil {
      fmt.Println("DoDelete err:", err)
      return false, err
   }
   var result PushResult
   err = json.Unmarshal(b, &result)
   if err != nil {
      fmt.Println("unmarshal err:", err)
      return false, err
   }
   if result.Code == 0 { //解绑成功
      return true, nil
   } else if result.Code == 10001 { //token过期
      if retryTimes <=3 {
         newToken, err := RefreshToken()
         if err == nil {
            updateToken(newToken)
            retryTimes++
            goto ReTry
         }
      }
   }
   return false, errors.New(result.Msg)
}
//一个别名最多允许绑定10个cid
func BindAlias(cid, alias string) (bool, error) {
@@ -238,40 +258,85 @@
      }
   }
   if len(pushUserM) == 0 {
      return true,nil
      return true,errors.New("len(pushUserM) == 0")
   }
   carPersonM := make(map[string]string)
   var csv CarService
   carPersonM := make(map[string]string) //以车牌号为key,value是hik的personId
   csv := NewCarService()
   carPersons := csv.GetVehicleListByPerson("")
   if carPersons != nil {
      for _, cp := range carPersons {
         carPersonM[cp.PlateNo] = cp.PersonId
      }
   }
   delPersonIdM := make(map[string]string)
   spaceNos := csv.FindSpaceNo("")
   for _,sn := range spaceNos {
      if sn.State == 1 && sn.PlateNo != "" { //已经把车停到停车场的车主,不再推送消息
         if _,ok := carPersonM[sn.PlateNo];ok {
         if pId,ok := carPersonM[sn.PlateNo];ok {
            delPersonIdM[pId] = pId
            delete(carPersonM, sn.PlateNo)
         }
      }
   }
   var uc models.UserClient
   for _,personId := range carPersonM {
      if phoneNum,ok := pushUserM[personId]; ok { //此人已注册到系统,并且车不在停车库内
         aliasArr = append(aliasArr, phoneNum)
         if _,in := delPersonIdM[personId];!in {
            if uc.Exist(phoneNum) {
               aliasArr = append(aliasArr, phoneNum)
            }
         }
      }
   }
   if len(aliasArr) == 0 {
      fmt.Println("没有推送目标,aliasArr is empty")
      return true, nil
   lenAS := len(aliasArr)
   if  lenAS == 0 {
      return true, errors.New("aliasArr is empty")
   }
   cResult, taskId, ce := createPushMsg(title, msg)
   fmt.Println("createPushMsg taskId:", taskId, "cResult:",cResult, "err:", ce)
   if !cResult {
      fmt.Println("createPushMsg taskId:", taskId, "err:", ce)
      return false, errors.New("创建推送前置消息失败")
   }
   //alias 单次推送长度上限是200
   pushTime := 1
   if lenAS > 200 {
      pushTime = lenAS / 200
      if lenAS % 200 >0 {
         pushTime++
      }
   }
   isSuccess := false
   for i:=0;i<pushTime;i++ {
      var curAliasArr []string
      start := i*200
      end := (i+1)*200 -1
      if i<pushTime-1 {
         curAliasArr = aliasArr[start:end]
      } else {
         //取剩余所有
         curAliasArr = aliasArr[start:]
      }
      if len(curAliasArr) >0 {
         flag, e := doPush(taskId, curAliasArr)
         if e != nil {
            fmt.Println("doPush err:", e)
         }
         if flag {
            isSuccess = true
         }
      }
   }
   if isSuccess {
      return true, nil
   }
   return false, errors.New("推送失败")
}
func doPush(taskId string, aliasArr []string) (bool,error) {
   appId := beego.AppConfig.String("pushAppId")
   baseUrl := beego.AppConfig.String("pushBaseUrl") + appId
   retryTimes := 0
@@ -286,7 +351,7 @@
         "alias": aliasArr,
      },
      "taskid": taskId,
      "is_async": true,
      "is_async": false,
   }
   header := map[string]string {
      "token": token,
@@ -316,7 +381,6 @@
   } else {
      fmt.Println("推送结果:", result)
   }
   return false, errors.New("推送失败")
}
@@ -335,7 +399,7 @@
      return true,nil
   }
   carPersonM := make(map[string]string)
   var csv CarService
   csv := NewCarService()
   carPersons := csv.GetVehicleListByPerson("")
   if carPersons != nil {
      for _, cp := range carPersons {
@@ -343,11 +407,14 @@
      }
   }
   spaceNos := csv.FindSpaceNo("")
   var uc models.UserClient
   for _,sn := range spaceNos {
      if sn.State == 1 && sn.PlateNo != "" { //已经把车停到停车场的车主,不再推送消息
         if personId,ok := carPersonM[sn.PlateNo];ok {
            if phoneNum,ok := pushUserM[personId]; ok { //此人已注册到系统,并且车不在停车库内
               aliasArr = append(aliasArr, phoneNum)
               if uc.Exist(phoneNum) {
                  aliasArr = append(aliasArr, phoneNum)
               }
            }
         }
      }
@@ -364,52 +431,7 @@
      return false, errors.New("创建推送前置消息失败")
   }
   appId := beego.AppConfig.String("pushAppId")
   baseUrl := beego.AppConfig.String("pushBaseUrl") + appId
   retryTimes := 0
ReTry:
   token := getCacheToken()
   if token == "" {
      return false, errors.New("token is nil")
   }
   url := baseUrl+"/push/list/alias"
   reqBody := map[string]interface{} {
      "audience":map[string]interface{}{
         "alias": aliasArr,
      },
      "taskid": taskId,
      "is_async": true,
   }
   header := map[string]string {
      "token": token,
   }
   b, err := util.DoPostRequest(url, util.CONTENT_TYPE_UTF8_JSON, reqBody, nil, header)
   if err !=nil {
      fmt.Println("DoPost err:", err)
      return false, err
   }
   var result PushResult
   err = json.Unmarshal(b, &result)
   if err != nil {
      fmt.Println("unmarshal err:", err)
      return false, err
   }
   if result.Code == 0 {
      return true, nil
   } else if result.Code == 10001 { //token过期
      if retryTimes <=3 {
         newToken, err := RefreshToken()
         if err == nil {
            updateToken(newToken)
            retryTimes++
            goto ReTry
         }
      }
   } else {
      fmt.Println("推送结果:", result)
   }
   return false, errors.New("推送失败")
   return doPush(taskId, aliasArr)
}
func PushAll(title string, msg string) (bool,error) {