soc/soc-modules/soc-modules-iot/src/main/java/com/xunmei/iot/controller/IotAlarmRuleController.java

package com.xunmei.iot.controller;

import cn.hutool.core.lang.Pair;
import com.xunmei.common.core.web.domain.AjaxResult;
import com.xunmei.common.core.web.page.TableDataInfo;
import com.xunmei.iot.dto.alarm.IotAlarmRuleDto;
import com.xunmei.iot.dto.alarm.IotAlarmRuleQueryDto;
import com.xunmei.iot.service.IotAlarmRuleService;
import com.xunmei.iot.vo.alarm.IotInitAlarmRuleVo;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;

@RestController
@RequestMapping("/iot_alarm")
public class IotAlarmRuleController {

    @Autowired
    private IotAlarmRuleService iotAlarmRuleService;

    @ApiOperation(value = "新增界面初始化数据；")
    @GetMapping(value = "/getAddAlarmRule")
    public AjaxResult getAddAlarmRule() {
        try {
            IotInitAlarmRuleVo data = iotAlarmRuleService.getAddInitAlarmRuleData();
            return AjaxResult.success(data);
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.error(e.getMessage());
        }
    }

    @ApiOperation(value = "修改界面")
    @GetMapping(value = "/getRuleById/{ruleId}")
    public AjaxResult getRuleById(@PathVariable Long ruleId) {
        try {
            IotInitAlarmRuleVo alarmRuleData = iotAlarmRuleService.getAlarmRuleData(ruleId);
            return AjaxResult.success(alarmRuleData);
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.error(e.getMessage());
        }
    }

    @ApiOperation(value = "查询报警规则")
    @PostMapping(value = "/getRules")
    public TableDataInfo<IotInitAlarmRuleVo> getRules(@RequestBody IotAlarmRuleQueryDto args) {
        try {
//            ruleName = ruleName.trim();
//            if(ruleName.equals("undefined")) ruleName = "";
            TableDataInfo<IotInitAlarmRuleVo> alarmRuleData = iotAlarmRuleService.getAlarmRuleData(args);
            return alarmRuleData;
        } catch (Exception e) {
            e.printStackTrace();
            TableDataInfo<IotInitAlarmRuleVo> rh = new TableDataInfo<>();
            rh.setCode(500);
            rh.setMsg("获取失败");
            return rh;
        }
    }


    @ApiOperation(value = "保存或者更新告警规则")
    @PostMapping(value = "/saveOrUpdateAlarmRule")
    public AjaxResult saveOrUpdateAlarmRule(@RequestBody IotAlarmRuleDto request) {
        try {
            iotAlarmRuleService.saveOrUpdateAlarmRule(request);
            return AjaxResult.success();
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.error(e.getMessage());
        }
    }

    @ApiOperation(value = "修改界面")
    @GetMapping(value = "/deleteRule/{ruleId}")
    public AjaxResult deleteRuleById(@PathVariable Long ruleId) {
        try {
            iotAlarmRuleService.deleteAlarmRule(ruleId);
            return AjaxResult.success();
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.error(e.getMessage());
        }
    }

    @ApiOperation(value = "获取机构下所有主机信息")
    @GetMapping(value = "/getConnectHostOrg/{orgId}")
    public AjaxResult getConnectHostOrg(@PathVariable Long orgId) {
        List<Pair<String, String>> data = iotAlarmRuleService.getConnectHostOrg(orgId);
        return AjaxResult.success(data);
    }
}