jihe-dc/zc-business/src/main/java/com/zc/business/interfaces/OperationLogAspect.java

package com.zc.business.interfaces;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.entity.SysDept;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.ip.IpUtils;
import com.ruoyi.common.utils.spring.SpringUtils;
import com.ruoyi.system.service.ISysDeptService;
import com.zc.business.domain.DcDevice;
import com.zc.business.domain.DcOperLog;
import com.zc.business.service.IDcDeviceService;
import com.zc.business.service.IDcOperLogService;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.text.SimpleDateFormat;
import java.util.*;

@Aspect
@Component
public class OperationLogAspect {


    private final static Map<String, Object> DEVICEFUNCTIONNAME = new HashMap<String, Object>() {{
        put("1", new HashMap<String, Object>() {{
        }});
        put("2", new HashMap<String, Object>() {{
            put("11", "文件名发送");
            put("13", "文件发送");
            put("1B", "指定当前需要播放的清单");
        }});
        put("3", new HashMap<String, Object>() {{
        }});
        put("4", new HashMap<String, Object>() {{
        }});
        put("5", new HashMap<String, Object>() {{
            put("", "发布语音广播");
        }});
        put("6", new HashMap<String, Object>() {{
        }});
        put("7", new HashMap<String, Object>() {{
        }});
        put("8", new HashMap<String, Object>() {{
            put("01", "通信连接检查");
            put("0b", "配置装置参数");
            put("0d", "读取装置参数");
            put("0f", "读取装置状态数据");
        }});
        put("9", new HashMap<String, Object>() {{
        }});
        put("10", new HashMap<String, Object>() {{
            put("SETMD", "模式设定命令");
            put("ASKMD", "模式查询命令");
            put("SETTM", "工作时间设定命令");
            put("ASKTM", "工作时间查询命令");
            put("SETDF", "自定义模式设置");
            put("ASKDF", "自定义模式查询命令");
        }});
        put("11", new HashMap<String, Object>() {{
        }});
        put("12", new HashMap<String, Object>() {{
            put("51", "设备控制模式");
            put("52", "读取设备控制模式");
            put("30", "设置自动控制模式工作状态");
            put("3D", "读取自动控制模式工作状态");
            put("3C", "读取当前设备工作状态");
        }});
        put("13", new HashMap<String, Object>() {{
        }});
        put("14", new HashMap<String, Object>() {{
        }});
        put("15", new HashMap<String, Object>() {{
            put("A1", "查询数据");
            put("A2", "查询数据");
            put("A3", "查询数据");
            put("A4", "查询数据");
            put("A5", "查询数据");
            put("A6", "查询数据");
        }});
        put("16", new HashMap<String, Object>() {{
        }});
    }};


    @AfterReturning(pointcut = "@annotation(operationLog)", returning = "jsonResult")
    public void AfterReturning(JoinPoint joinPoint, OperationLog operationLog, Object jsonResult) throws Throwable {
        System.out.println("运行成功");
        around((ProceedingJoinPoint) joinPoint, operationLog, "0", jsonResult, null);
    }

    @AfterThrowing(value = "@annotation(operationLog)", throwing = "e")
    public void AfterThrowing(JoinPoint joinPoint, OperationLog operationLog, Exception e) throws Throwable {
        System.out.println("运行失败");
        around((ProceedingJoinPoint) joinPoint, operationLog, "1", null, e);
    }

    public void around(ProceedingJoinPoint joinPoint, OperationLog operationLog, String state, Object jsonResult, Exception exception) throws Throwable {

        ISysDeptService deptService = SpringUtils.getBean(ISysDeptService.class);
        IDcOperLogService dcOperLogService = SpringUtils.getBean(IDcOperLogService.class);
        IDcDeviceService dcDeviceService = SpringUtils.getBean(IDcDeviceService.class);

        try {
            DcOperLog operLog = new DcOperLog();
            String operUrl = operationLog.operUrl();
            LoginUser loginUser;
            try {

                loginUser = SecurityUtils.getLoginUser();
            } catch (Exception e) {
                loginUser = null;
            }
            Object[] pointArgs = joinPoint.getArgs();

            HttpServletRequest request;
            try {

                request = ServletUtils.getRequest();
            } catch (Exception e) {
                request = null;
            }

            AjaxResult proceed = (AjaxResult) joinPoint.proceed(pointArgs);
            //String jsonResultString = "";
            //JSONArray jsonArray = (JSONArray) proceed;
            //for (Object o : jsonArray) {
            //    JSONObject jsonObject = (JSONObject) o;
            //    if (Objects.equals(jsonObject.getString("code"), "200")) {
            //        JSONArray data = jsonObject.getJSONArray("data");
            //        for (Object datum : data) {
            //            JSONObject results = (JSONObject) datum;
            //            JSONObject result = results.getJSONObject("result");
            //            if (Objects.equals(result.getString("code"), "200")) {
            //                jsonResultString = "设备"+d+"";
            //            }
            //        }
            //    }
            //}

            SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            String date = df.format(new Date());
            String sta = "";
            if (Objects.equals(state, "0")&&Objects.equals(String.valueOf(proceed.get("code")), "200")) {
                //if (Objects.equals(String.valueOf(proceed.get("code")), "200")) {
                    operLog.setStatus(0);
                    sta = "成功";
                //} else {
                //    operLog.setStatus(1);
                //    sta = "失败";
                //}
            } else {
                operLog.setStatus(1);
                sta = "失败";
            }

            ArrayList<String> deviceIds = new ArrayList<>();
            ArrayList<String> deviceNames = new ArrayList<>();
            //ArrayList<String> functionIds = new ArrayList<>();
            String deviceType = "";
            StringBuilder remark = new StringBuilder("在" + date + "调用");
            if (operUrl.contains("batchFunctions")) {
                //批量调用

                //if (loginUser != null) {
                //    SysDept sysDept = deptService.selectDeptById(loginUser.getDeptId());
                //
                //    dcOperLog.setOperName(loginUser.getUsername());
                //    dcOperLog.setDeptName(sysDept.getDeptName());
                //    dcOperLog.setOperLocation(loginUser.getLoginLocation());
                //    dcOperLog.setOperType("2");
                //} else {
                //    dcOperLog.setOperType("1");
                //}
                //dcOperLog.setOperIp(IpUtils.getIpAddr(request));
                //dcOperLog.setOperTime(new Date());

                JSONObject pointArg = (JSONObject) pointArgs[0];
                JSONArray devices = pointArg.getJSONArray("devices");
                JSONArray functions = pointArg.getJSONArray("functions");

                for (int i = 0; i < devices.size(); i++) {

                    JSONObject device = (JSONObject) JSON.toJSON(devices.get(i));

                    LambdaQueryWrapper<DcDevice> lambdaQueryWrapper = new LambdaQueryWrapper<>();
                    lambdaQueryWrapper.eq(DcDevice::getId, device.get("id"));
                    DcDevice dcDevice = dcDeviceService.getById(device.getString("id"));

                    if (!Objects.isNull(dcDevice)) {
                        Long deviceId = dcDevice.getId();
                        String deviceName = dcDevice.getDeviceName();

                        deviceType = dcDevice.getDeviceType();

                        deviceIds.add(String.valueOf(deviceId));
                        deviceNames.add(deviceName);

                        remark.append(deviceName).append("的");

                        for (int i1 = 0; i1 < functions.size(); i1++) {

                            JSONObject function = (JSONObject) JSON.toJSON(functions.get(i1));

                            String hashMap = Objects.equals(deviceType, "5") ?
                                    "发布语音广播" :
                                    String.valueOf(
                                            ((HashMap<String, Object>) (DEVICEFUNCTIONNAME.get(deviceType)))
                                                    .get(function.getString("functionId")));

                            remark.append("调用").append(hashMap).append("功能").append(sta);

                            if (i1 < functions.size() - 1) {
                                remark.append("、");
                            }
                        }

                        if (i < (devices.size() - 1)) {
                            remark.append(",");
                        }

                    }
                }


                if (loginUser != null) {
                    SysDept sysDept = deptService.selectDeptById(loginUser.getDeptId());

                    operLog.setOperName(loginUser.getUsername());
                    operLog.setDeptName(sysDept.getDeptName());
                    operLog.setOperLocation(loginUser.getLoginLocation());
                    operLog.setOperType("2");
                } else {
                    operLog.setOperType("1");
                }
                operLog.setOperParam((Arrays.toString(pointArgs)));
                operLog.setDcDeviceId(deviceIds.toString());
                operLog.setDcDeviceName(deviceNames.toString());
                operLog.setOperIp(IpUtils.getIpAddr(request));
                operLog.setOperTime(new Date());
                operLog.setDcDeviceType(deviceType);
                operLog.setRemark(remark.toString());
                operLog.setJsonResult(String.valueOf(proceed));
                dcOperLogService.addDcOperLog(operLog);


            } else if (operUrl.contains("functions")) {
                LambdaQueryWrapper<DcDevice> lambdaQueryWrapper = new LambdaQueryWrapper<>();
                lambdaQueryWrapper.eq(DcDevice::getIotDeviceId, pointArgs[0]);
                DcDevice dcDevice = dcDeviceService.getById(Long.valueOf(String.valueOf(pointArgs[0])));
                if (!Objects.isNull(dcDevice)) {

                    Long deviceId = dcDevice.getId();
                    String deviceName = dcDevice.getDeviceName();

                    deviceType = dcDevice.getDeviceType();

                    deviceIds.add(String.valueOf(deviceId));
                    deviceNames.add(deviceName);

                    String functionId = String.valueOf(pointArgs[1]);
                    String hashMap = Objects.equals(deviceType, "5") ? "发布语音广播" : String.valueOf(((HashMap<String, Object>) (DEVICEFUNCTIONNAME.get(deviceType))).get(functionId));

                    remark.append(deviceName).append("的").append(hashMap).append("功能并且调用").append(sta).append("了");


                    if (loginUser != null) {
                        SysDept sysDept = deptService.selectDeptById(loginUser.getDeptId());
                        operLog.setOperName(loginUser.getUsername());
                        operLog.setDeptName(sysDept.getDeptName());
                        operLog.setOperLocation(loginUser.getLoginLocation());
                        operLog.setOperType("2");
                    } else {
                        operLog.setOperType("1");
                    }
                    operLog.setOperParam((Arrays.toString(pointArgs)));
                    operLog.setDcDeviceId(deviceIds.toString());
                    operLog.setDcDeviceName(deviceNames.toString());
                    operLog.setOperIp(IpUtils.getIpAddr(request));
                    operLog.setOperTime(new Date());
                    operLog.setDcDeviceType(deviceType);
                    operLog.setJsonResult(remark.toString());
                    dcOperLogService.addDcOperLog(operLog);


                    //
                    ////DcDevice dcDevice = list.get(0);
                    //operLog.setDcDeviceId(String.valueOf(dcDevice.getId()));
                    //operLog.setDcDeviceName(dcDevice.getDeviceName());
                    //deviceType = dcDevice.getDeviceType();
                    //operLog.setDcDeviceType(deviceType);
                    //if (loginUser != null) {
                    //    SysDept sysDept = deptService.selectDeptById(loginUser.getDeptId());
                    //
                    //    dcOperLog.setOperName(loginUser.getUsername());
                    //    dcOperLog.setDeptName(sysDept.getDeptName());
                    //    dcOperLog.setOperLocation(loginUser.getLoginLocation());
                    //    dcOperLog.setOperType("0");
                    //} else {
                    //    dcOperLog.setOperType("1");
                    //}
                    //dcOperLog.setOperIp(IpUtils.getIpAddr(request));
                    //dcOperLog.setOperParam(Arrays.toString(pointArgs));
                    //dcOperLog.setJsonResult("设备" + dcDevice.getDeviceName() + "在" + date + "调用了" + hashMap + "功能并且调用" + sta + "了");
                    ////dcOperLog.setJsonResult(dcDevice.getDeviceName() + "设备在" + date + "调用" + sta + "ID为" + dcDevice.getId() + "的设备功能标识为" + pointArgs[1].toString() + "的方法");
                    //dcOperLog.setOperTime(new Date());
                    //dcOperLogService.addDcOperLog(dcOperLog);
                    //
                    //
                    //String remark = "设备在" + date + "调用" + sta + "ID为" + deviceIds + "的设备功能标识为" + functionIds + "的方法";
                    //System.out.println(remark);
                    //System.out.println(deviceIds.toString());
                    //System.out.println(functionIds.toString());
                    //DcOperLog operLog = new DcOperLog();
                    //SysDept sysDept = deptService.selectDeptById(loginUser.getDeptId());
                    //operLog.setOperParam((Arrays.toString(pointArgs)));
                    //operLog.setOperLocation(loginUser.getLoginLocation());
                    //operLog.setOperName(loginUser.getUsername());
                    //operLog.setDeptName(sysDept.getDeptName());
                    //operLog.setDcDeviceId(deviceIds.toString());
                    //operLog.setOperType("2");
                    //operLog.setOperIp(IpUtils.getIpAddr(request));
                    //operLog.setOperTime(new Date());
                    //operLog.setDcDeviceType(deviceType);
                    //operLog.setJsonResult(remark);
                    //if (Objects.equals(state, "0")) {
                    //    operLog.setStatus(0);
                    //} else {
                    //    operLog.setStatus(1);
                    //}
                    //dcOperLogService.addDcOperLog(operLog);


                }
            }

        } catch (Exception e) {
            throw e;
        }
    }
}