package com.tievd.jyz.handler.alg.common;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.tievd.jyz.cache.CarModelCache;
import com.tievd.jyz.cache.DeviceCache;
import com.tievd.jyz.entity.Device;
import com.tievd.jyz.entity.TrafficFlow;
import com.tievd.jyz.handler.alg.AlgHandleInterface;
import com.tievd.jyz.mqtt.dto.EventInfoDTO;
import com.tievd.jyz.service.ITrafficFlowService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.sql.Timestamp;
import java.time.LocalDate;
import java.util.ArrayList;
import java.util.List;

/**
 * 车流量算法处理类
 * @author timi
 */
@Slf4j
@Component
public class TrafficFlowAlgHandler implements AlgHandleInterface {

    /** 通用算法标识头 */
    private final String ALG_TAG[] = new String[]{"C10004"};

    @Autowired
    private ITrafficFlowService trafficFlowService;

    @Override
    public String[] getAlgTag(){
        return ALG_TAG;
    }

    /**
     * 处理入口
     * @param eventInfoDTO
     * @param sn
     * @param time
     */
    @Override
    public void handle(EventInfoDTO<JSONObject> eventInfoDTO, String sn,String time) {
        //当前事件唯一标识
        String eventCode = eventInfoDTO.getEventCode();
        //检测算法编码
        String algCode = eventInfoDTO.getAlgCode();
        //终端编码
        String cameraCode = eventInfoDTO.getCameraCode();
        //扩展属性
        JSONObject extend = new JSONObject();
        Object extendObj = eventInfoDTO.getExtend();
        if(extendObj instanceof String){
            extend = JSONObject.parseObject(extendObj.toString());
        }else if(extendObj instanceof JSONObject){
            extend = (JSONObject)extendObj;
        }
        log.info("车流量事件 eventCode:{},extend:{}",eventCode,extendObj);
        Device device = DeviceCache.getDeviceBySn(sn);
        if(ObjectUtil.isNull(device)){
            log.error("网关设备不存在,sn:{}",sn);
            return;
        }
        //车流量检测
        JSONArray carFlows = extend.getJSONArray("carFlow");
        List<TrafficFlow> flowList = new ArrayList<>();
        for (Object o : carFlows) {
            JSONObject flow = (JSONObject) o;
            Integer carCount = flow.getIntValue("carCount");
            String carModelCode = flow.getString("carModel");
            if (CarModelCache.contains(carModelCode)){
                log.info("包含不存在的车型：{}", carModelCode);
                continue;
            }
            
            TrafficFlow trafficFlow = new TrafficFlow()
                    .setCameraCode(cameraCode)
                    .setDeviceId(device.getId())
                    .setCarCount(carCount)
                    .setModelCode(carModelCode)
                    .setOrgCode(device.getOrgCode())
                    .setCaptureTime(new Timestamp(Long.parseLong(time)))
                    .setCaptureDay(LocalDate.now().toString())
                    .generateId();
            flowList.add(trafficFlow);
        }
        trafficFlowService.saveOrUpdateBatch(flowList);
    }
}