Tools.scala

package com.hikcreate.data.util

import scala.collection.mutable
import scala.collection.JavaConverters._
import com.alibaba.fastjson.{JSON, JSONArray, JSONObject}
import com.hikcreate.data.client.IgniteClient
import com.hikcreate.data.common.Logging
import com.hikcreate.data.constant.Const
import com.hikcreate.ignite.domain.basic.{AlarmTypeInfo, EnterpriseInfo, VehicleInfo}
import org.apache.ignite.cache.query.SqlQuery
import scalaj.http.Http

object Tools extends Logging{

  def addLocation(json:JSONObject): Array[JSONObject] = {
    val gnss = json.getJSONArray("gnssList")
    json.remove("gnssList")
    val map = mutable.HashMap.empty[String,Any]
    json.keySet().asScala.foreach(key=>map.put(key,json.get(key)))
    (0 until gnss.size()).map{ index=>
      val addJson = gnss.getJSONObject(index)
      addJson.putAll(map.asJava)
      addJson
    }.toArray
  }

  def getAddressAndLocationCode(lon:Double,lat:Double):(String,String) = {
    val json = new JSONObject()
    val arr = new JSONArray()
    val lonAndLat = new JSONObject()
    lonAndLat.put("longitude",lon)
    lonAndLat.put("latitude",lat)
    arr.add(lonAndLat)
    json.put("locations",arr)
    val response = Http(Const.areaCodeAndAddressUrl)
      .postData(json.toJSONString)
      .header("content-type","application/json")
      //.charset("ISO-8859-1")
      .timeout(connTimeoutMs = 8000, readTimeoutMs = 8000)
      .asString
    if(response.code == 200){
      val body = JSON.parseObject(response.body)
      val item = body.getJSONObject("result").getJSONArray("regeoItems").getJSONObject(0)
      val address = item.getString("formattedAddress")
      val locationCode =  item.getJSONObject("addressComponent").getString("adcode")
      (address,locationCode)
    }else{
      throw new RuntimeException("http请求城市区编码出错")
    }
  }

  //根据经纬度获取地区编码
  def getLocationCode(lon:Double,lat:Double): (String,String,String) = {
    val locationCode = getAddressAndLocationCode(lon,lat)._2
    val provinceCode = locationCode.substring(0,2)
    val cityCode = locationCode.substring(2,4)
    val areaCode = locationCode.substring(4,6)
    (provinceCode,cityCode,areaCode)
  }

  def getInfoContentJsonobj(infoStr:String):JSONObject = {
    val jsonStr=("{\""+infoStr.replace(":=","\":\"").replace(";","\",\"")+"\"}").replace(",\"\"}","}")
    val jSONObject = JSON.parseObject(jsonStr)
    jSONObject
  }

  //根据企业编码得到企业信息
  def getEnterpriseInfo(enterpriseCode:String): Option[EnterpriseInfo] = {
    val vehicleInfoSql = new SqlQuery[Long,EnterpriseInfo](classOf[EnterpriseInfo],"enterpriseCode = ?")
    val vehicleInfos = IgniteClient.basicEnterpriseInfo.query(vehicleInfoSql.setArgs(enterpriseCode)).getAll
    if(vehicleInfos.size() == 1 ){
      val vehicleInfo = vehicleInfos.get(0).getValue
      Some(vehicleInfo)
    }else{
      None
    }
  }

  //根据车牌号和车牌号码得到车辆信息
  def getVehicleInfo(plateNum:String,plateColor:String): Option[VehicleInfo] = {
    val vehicleInfoSql = new SqlQuery[Long,VehicleInfo](classOf[VehicleInfo],"PLATENUM = ? AND PLATECOLOR= ? ")
    val vehicleInfos = IgniteClient.basicVehicleInfo.query(vehicleInfoSql.setArgs(plateNum,plateColor)).getAll
    if(vehicleInfos.size() == 1 ){
      val vehicleInfo = vehicleInfos.get(0).getValue
      Some(vehicleInfo)
    }else{
      None
    }
  }

  //根据报警类型和事件类型 得到 平台报警等级code 和 平台报警类型名称
  def getAlarmInfo(warnType:String,eventType:String): Option[AlarmTypeInfo] = {
    val alarmInfoSql = new SqlQuery[Long,AlarmTypeInfo](classOf[AlarmTypeInfo],"warningBigType = ? AND warningSmallType = ? ")
    val alarmInfos = IgniteClient.basicAlarmTypeInfo.query(alarmInfoSql.setArgs(warnType,eventType)).getAll
    if(alarmInfos.size() == 1 ){
      val alarmInfo = alarmInfos.get(0).getValue
      Some(alarmInfo)
    }else{
      None
    }
  }
}