liuxiaolong
2020-09-01 ca10e9d776509f54cb0251b7eae44cee74d94b2c
service/carService.go
@@ -7,7 +7,9 @@
   "encoding/json"
   "fmt"
   "github.com/astaxie/beego"
   "sort"
   "strconv"
   "strings"
)
type CarService struct {
   hikUtil *reqUtil.HikHttpUtil
@@ -142,6 +144,121 @@
   SpaceType             string       `json:"spaceType"`
}
func (sv *CarService) FindHikSpaceInfo() []SpaceNo {
   url := "/artemis/api/pms/v1/parkingSpace/spaceNo"
   reqBody := map[string]interface{} {
      "pageNo": 1,
      "pageSize": 1000,
   }
   pageResult := sv.getHikPageResult(url, reqBody)
   if pageResult != nil {
      rb, err := json.Marshal(pageResult.List)
      if err != nil {
         fmt.Println("marshal pageResult.List err:", err)
         return nil
      }
      var spaceList []SpaceNo
      err = json.Unmarshal(rb, &spaceList)
      if err != nil {
         fmt.Println("unmarshal spaceNos err:", err)
         return nil
      } else {
         var resultArr []SpaceNo
         for _,si := range spaceList {
            if si.State == 1 { //有车
               resultArr = append(resultArr, si)
            }
         }
         return resultArr
      }
   }
   return nil
}
type SpaceUser struct {
   SpaceNo    string          `json:"spaceNo"`
   PlateNo    string          `json:"plateNo"`
   Username    string          `json:"username"`
   PhoneNum    string          `json:"phoneNum"`
   CarType    string          `json:"carType"`
}
type SpaceUserArr []SpaceUser
func (arr SpaceUserArr) Len()int {
   return len(arr)
}
func (arr SpaceUserArr) Swap(i,j int) {
   arr[i],arr[j] = arr[j],arr[i]
}
func (arr SpaceUserArr) Less(i,j int) bool {
   sn1,_ := strconv.Atoi(arr[i].SpaceNo)
   sn2,_ := strconv.Atoi(arr[j].SpaceNo)
   return sn1 < sn2
}
func (sv *CarService) FindHikSpaceUser() SpaceUserArr {
   resultList := make(SpaceUserArr, 0)
   url := "/artemis/api/pms/v1/parkingSpace/spaceNo"
   reqBody := map[string]interface{} {
      "pageNo": 1,
      "pageSize": 1000,
   }
   pageResult := sv.getHikPageResult(url, reqBody)
   if pageResult != nil {
      rb, err := json.Marshal(pageResult.List)
      if err != nil {
         fmt.Println("marshal pageResult.List err:", err)
      } else {
         var spaceList []SpaceNo
         err = json.Unmarshal(rb, &spaceList)
         if err != nil {
            fmt.Println("unmarshal spaceNos err:", err)
         } else {
            if len(spaceList) > 0 { //当前有停留车
               personMap := sv.GetHikPersonMap()
               vehicleMap := sv.GetHikVehicleMap()
               for _,sp := range spaceList {
                  if sp.State == 1 {
                     su := SpaceUser{
                        SpaceNo: sp.SpaceNo,
                        PlateNo: sp.PlateNo,
                     }
                     if sp.PlateNo != "" {
                        r := []rune(sp.PlateNo)
                        ncStr := string(r[1:]) //把第一位汉字剔除掉
                        newK := strings.ReplaceAll(ncStr, "D", "*")
                        newK = strings.ReplaceAll(newK, "0", "*")
                        personId := ""
                        if v,exist := vehicleMap[newK]; exist {
                           personId = v.PersonId
                        }
                        if personId != "" {
                           su.CarType = "备案车"
                           if person, in := personMap[personId]; in {
                              su.Username = person.PersonName
                              su.PhoneNum = person.PhoneNo
                           }
                        } else {
                           su.CarType = "临时车"
                        }
                     } else {
                        su.PlateNo = "无车牌"
                     }
                     resultList = append(resultList, su)
                  }
               }
            }
         }
      }
   }
   sort.Sort(resultList)
   return resultList
}
func (sv *CarService) FindSpaceNo(userId string) models.PosResult {
   resultList := make(models.PosResult,0)
   var myPlateNosMap = make(map[string]string)
@@ -149,8 +266,12 @@
      vehicles := sv.GetVehicleListByPerson(userId)
      if vehicles != nil {
         for _,veh := range vehicles {
            myPlateNosMap[veh.PlateNo] = veh.PlateNo
            //myPlateNos = append(myPlateNos, veh.PlateNo)
            if veh.PlateNo != "" {
               r := []rune(veh.PlateNo)
               ncStr := string(r[1:]) //把第一位汉字剔除掉
               myPlateNosMap[ncStr] = ncStr
            }
            //myPlateNosMap[veh.PlateNo] = veh.PlateNo
         }
      }
   }
@@ -179,21 +300,17 @@
         return nil
      }
      for _,s := range spaceList {
         isMine := false
         if _,exist := myPlateNosMap[s.PlateNos]; exist {
            isMine = true
         }
         pi := models.PosInfo {
            SpaceNo: s.SpaceNo,
            PosNo: "",
            State: s.State,
            IsMine: isMine,
         }
         if s.PlateNo != "" {
            pi.PlateNo = s.PlateNo
         } else if s.PlateNos != "" {
            pi.PlateNo = s.PlateNos
         }
         pi.IsMine = isMyPlateNo(myPlateNosMap, pi.PlateNo)
         if v,ok := models.SpaceNo2Pos[s.SpaceNo];ok {
            pi.PosNo = v
         }
@@ -214,6 +331,35 @@
   }
   return resultList
}
//判断是不是自己的车牌号
//1.目前有误识别的问题,会把D识别成0,D和0不分
//2.汉字误识别的几率比较高
func isMyPlateNo(plateNoM map[string]string, targetPlateNo string) bool {
   if targetPlateNo != "" {
      r := []rune(targetPlateNo)
      nctPlateNo := string(r[1:])
      if _,exist := plateNoM[nctPlateNo];exist {
         return true
      } else {
         match := false
         for k,_ := range plateNoM {
            newK := strings.ReplaceAll(k, "D", "*")
            newK = strings.ReplaceAll(newK, "0", "*")
            tt := strings.ReplaceAll(nctPlateNo,"D", "*")
            tt = strings.ReplaceAll(tt,"0", "*")
            if newK == tt {
               match = true
               break
            }
         }
         return match
      }
   }
   return false
}
func (sv *CarService) BindCarSpace() bool {
@@ -248,6 +394,37 @@
   PicUri                      string             `json:"picUri"`
   ServerIndexCode             string             `json:"serverIndexCode"`
   PersonId                   string             `json:"personId"`
}
//以人员id为key
func (sv *CarService) GetHikPersonMap() map[string]HikPerson {
   m := make(map[string]HikPerson)
   personList := sv.GetHikPersonList()
   if personList != nil {
      for _,p := range personList {
         m[p.PersonId] = p
      }
   }
   return m
}
//以车牌号为Key
func (sv *CarService) GetHikVehicleMap() map[string] HikVehicle {
   m := make(map[string]HikVehicle)
   carPersons := sv.GetVehicleListByPerson("")
   if carPersons != nil {
      for _,veh := range carPersons {
         if veh.PlateNo != "" {
            r := []rune(veh.PlateNo)
            ncStr := string(r[1:]) //把第一位汉字剔除掉
            newK := strings.ReplaceAll(ncStr, "D", "*")
            newK = strings.ReplaceAll(newK, "0", "*")
            m[newK] = veh
         }
      }
   }
   return m
}
func (sv *CarService) GetHikPersonList() []HikPerson {
@@ -302,20 +479,32 @@
   return list
}
func SyncHikPerson() {
func SyncHikPerson() int {
   syncCount := 0
   sv := NewCarService()
   hikPersons := sv.GetHikPersonList()
   if hikPersons != nil {
   if hikPersons != nil && len(hikPersons) >0 {
      var u models.User
      uMap := u.GetAllMapByPhone()
      idMap := u.GetAllMap()
      hikPM := make(map[string]string)
      for _,hp := range hikPersons {
         if sp,ok := uMap[hp.PhoneNo];ok {
            if sp.Id != hp.PersonId {
               u.SyncHikPersonId(hp.PhoneNo, hp.PersonId)
         hikPM[hp.PersonId] = hp.PhoneNo
         if sp,ok := idMap[hp.PersonId];ok {
            if sp.PhoneNum != hp.PhoneNo {
               u.UpdatePhoneNum(hp.PhoneNo, hp.PersonId)
               syncCount++
            }
         }
      }
      for k,_ := range idMap {
         if _,in := hikPM[k];!in {
            u.Delete(k)
            syncCount++
         }
      }
   }
   return syncCount
}
func (sv *CarService) getHikPageResult(url string, reqBody map[string]interface{}) *vo.HikPageResult {
@@ -366,10 +555,13 @@
   reqBody := map[string]interface{} {
      "pageNo": 1,
      "pageSize": 1000,
      "personIds": personId,
   }
   if personId != "" {
      reqBody["personIds"] = personId
   }
   pageResult := sv.getHikPageResult(url, reqBody)
   if pageResult ==nil {
      fmt.Println("GetVehicleListByPerson pageResult is nil")
      return nil
   }
@@ -378,6 +570,8 @@
      var list []HikVehicle
      if err = json.Unmarshal(b, &list);err == nil {
         return list
      } else {
         fmt.Println("unmarshal err:", err)
      }
   }