alarm_daily统计任务，创建alarm年月分表任务，alarm相关接口更新

jinpengyong

2020-05-19 c49d3947b165e0868d48a4c969e55093b8d05705

alarm_daily统计任务，创建alarm年月分表任务，alarm相关接口更新

 src/main/java/com/moral/mapper/AlarmMapper.java

New file
@@ -0,0 +1,15 @@
package com.moral.mapper;

import java.util.List;
import java.util.Map;

import org.apache.ibatis.annotations.Param;

public interface AlarmMapper {

    List<Map<String, Object>> getAvgAlarmData(Map<String, Object> devices);

    int insertAlarmDaily(List<Map<String,Object>> list);

    void createTable(@Param("yearAndMonth") String yearAndMonth);
}

 src/main/java/com/moral/service/AlarmService.java

New file
@@ -0,0 +1,13 @@
package com.moral.service;

import java.util.List;
import java.util.Map;

public interface AlarmService {

    List<Map<String, Object>> getAvgAlarmData(Map<String,Object> devices);

    int insertAlarmDaily(List<Map<String,Object>> list);

    void createTable(String yearAndMonth);
}

 src/main/java/com/moral/service/impl/AlarmServiceImpl.java

New file
@@ -0,0 +1,35 @@
package com.moral.service.impl;

import java.util.List;
import java.util.Map;

import javax.annotation.Resource;

import org.springframework.stereotype.Service;

import com.moral.mapper.AlarmMapper;
import com.moral.service.AlarmService;


@Service
public class AlarmServiceImpl implements AlarmService {

    @Resource
    private AlarmMapper alarmMapper;

    @Override
    public List<Map<String, Object>> getAvgAlarmData(Map<String, Object> devices) {
        return alarmMapper.getAvgAlarmData(devices);
    }

    @Override
    public int insertAlarmDaily(List<Map<String, Object>> list) {
        return alarmMapper.insertAlarmDaily(list);
    }

    @Override
    public void createTable(String yearAndMonth) {
        alarmMapper.createTable(yearAndMonth);
    }

}

 src/main/java/com/moral/task/AlarmTableInsertTask.java

New file
@@ -0,0 +1,101 @@
package com.moral.task;

import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.Resource;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.CollectionUtils;

import com.alibaba.fastjson.JSONObject;
import com.moral.service.AlarmService;
import com.moral.service.DeviceService;
import com.moral.service.SensorService;
import com.xxl.job.core.biz.model.ReturnT;
import com.xxl.job.core.handler.annotation.XxlJob;

public class AlarmTableInsertTask {
    private static transient Logger logger = LoggerFactory.getLogger(AlarmTableInsertTask.class);
    @Resource
    private AlarmService alarmService;

    @Resource
    private SensorService sensorService;

    @Resource
    private DeviceService deviceService;

    @XxlJob("alarmDaily")
    public ReturnT insertAlarmDailyTable(String params) {
        LocalDateTime time = LocalDateTime.now();
        LocalDateTime endTime = time.truncatedTo(ChronoUnit.DAYS);
        LocalDateTime startTime = endTime.minusDays(1);
        List<String> sensorKeys = sensorService.getSensorKeys();
        List<String> macs = deviceService.getMacs();

        String year = startTime.getYear() + "";
        String month = null;
        int monthValue = startTime.getMonthValue();
        if (monthValue < 10) {
            month = "0" + monthValue;
        } else {
            month = monthValue + "";
        }
        String yearAndMonth = year + month;

        Map<String, Object> devices = new HashMap<>();
        devices.put("sensorKeys", sensorKeys);
        devices.put("start", startTime);
        devices.put("end", endTime);
        devices.put("macs", macs);
        devices.put("yearAndMonth",yearAndMonth);

        List<Map<String, Object>> dailyData = alarmService.getAvgAlarmData(devices);
        List<Map<String, Object>> dailyDataList = new ArrayList<>();
        for (Map<String, Object> deviceData : dailyData) {
            Map<String, Object> map = new HashMap<>();
            for (String sensorKey : sensorKeys) {
                if (deviceData.get(sensorKey) != null) {
                    map.put(sensorKey, deviceData.get(sensorKey).toString());
                }
            }
            JSONObject json = new JSONObject(map);
            deviceData.put("json", json);
        }

        for (Map<String, Object> dailyMap : dailyData) {
            Map<String, Object> map = new HashMap<>();
            String mac = dailyMap.get("mac").toString();
            String json = dailyMap.get("json").toString();
            String state = dailyMap.get("state").toString();
            map.put("mac", mac);
            map.put("json", json);
            map.put("state", state);
            map.put("time", startTime);
            dailyDataList.add(map);
        }
        if (!CollectionUtils.isEmpty(dailyDataList)) {
            alarmService.insertAlarmDaily(dailyDataList);
            return new ReturnT(200, "插入天表成功");
        }
        return new ReturnT(500, "插入天表失败");
    }


    @XxlJob("createAlarmSubTable")
    public void createAlarmSubTable(String params) {
        LocalDateTime time = LocalDateTime.now().plusMonths(1);;
        String year = time.getYear() + "";
        String month = time.getMonthValue() + "";
        if (time.getMonthValue() < 10) {
            month = "0" + time.getMonthValue();
        }
        alarmService.createTable(year + month);
    }
}

 src/main/resources/mapper/AlarmMapper.xml

New file
@@ -0,0 +1,44 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.moral.mapper.AlarmMapper">
    <select id="getAvgAlarmData" resultType="java.util.LinkedHashMap">
        SELECT
        a.mac,
        max(a.state) state
        <foreach collection="sensorKeys" open="," separator="," item="sensorKey">
            AVG(a.json->'$.${sensorKey}[0]') AS '${sensorKey}'
        </foreach>
        FROM
        alarm_${yearAndMonth} a
        WHERE
        a.time >= #{start}
        AND a.time <![CDATA[<]]> #{end}
        AND a.mac in
        <foreach collection="macs" open="(" separator="," close=")"
                 item="mac">
            #{mac}
        </foreach>
        group by
        a.mac
    </select>

    <insert id="insertAlarmDaily">
        insert into
        alarm_daily
        values
        <foreach collection="list" item="item" separator=",">
            (#{item.mac},#{item.json},#{item.state},#{item.time})
        </foreach>
    </insert>

    <update id="createTable">
        CREATE TABLE `alarm_${yearAndMonth}` (
        `mac` varchar(20) CHARACTER SET utf8 DEFAULT NULL,
        `json` json DEFAULT NULL,
        `state` int(2) DEFAULT NULL,
        `time` datetime DEFAULT NULL,
        KEY `_idx_time` (`time`) USING BTREE,
        KEY `_idx_mac` (`mac`) USING BTREE)
        ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci ROW_FORMAT=DYNAMIC;
    </update>
</mapper>