shinestar-server/src/main/java/com/jpsoft/smart/modules/base/controller/AlarmInfoController.java

package com.jpsoft.smart.modules.base.controller;

import com.github.pagehelper.Page;
import com.jpsoft.smart.modules.base.entity.AlarmInfo;
import com.jpsoft.smart.modules.base.service.AlarmInfoService;
import com.jpsoft.smart.modules.common.dto.MessageResult;
import com.jpsoft.smart.modules.common.dto.Sort;
import com.jpsoft.smart.modules.common.utils.PojoUtils;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.*;

@RestController
@RequestMapping("/base/alarmInfo")
public class AlarmInfoController {
    private Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    private AlarmInfoService alarmInfoService;

    @PostMapping("alarmNotify")
    public MessageResult<String> alarmNotify(String message){
        MessageResult<String> msgResult = new MessageResult<>();

        logger.info(message);

        return msgResult;
    }

    @ApiOperation(value="创建空记录")
    @GetMapping("create")
    public MessageResult<AlarmInfo> create(){
        MessageResult<AlarmInfo> msgResult = new MessageResult<>();

        AlarmInfo alarmInfo = new AlarmInfo();

        msgResult.setData(alarmInfo);
        msgResult.setResult(true);

        return msgResult;
    }
    
    @ApiOperation(value="添加信息")
    @PostMapping("add")
    public MessageResult<AlarmInfo> add(@RequestBody AlarmInfo alarmInfo,@RequestAttribute String subject){
        MessageResult<AlarmInfo> msgResult = new MessageResult<>();

        try {
            alarmInfo.setId(UUID.randomUUID().toString());
            alarmInfo.setDelFlag(false);
            alarmInfo.setCreateBy(subject);
            alarmInfo.setCreateTime(new Date());
            
            int affectCount = alarmInfoService.insert(alarmInfo);

            if (affectCount > 0) {
                msgResult.setResult(true);
                msgResult.setData(alarmInfo);
            } else {
                msgResult.setResult(false);
                msgResult.setMessage("数据库添加失败");
            }
        }
        catch(Exception ex){
            logger.error(ex.getMessage(),ex);

            msgResult.setResult(false);
            msgResult.setMessage(ex.getMessage());
        }

        return msgResult;
    }

    @ApiOperation(value="获取信息")
    @GetMapping("edit/{id}")
    public MessageResult<AlarmInfo> edit(@PathVariable("id") String id){
        MessageResult<AlarmInfo> msgResult = new MessageResult<>();

        try {
            AlarmInfo alarmInfo = alarmInfoService.get(id);

            if (alarmInfo != null) {
                msgResult.setResult(true);
                msgResult.setData(alarmInfo);
            } else {
                msgResult.setResult(false);
                msgResult.setMessage("数据库不存在该记录!");
            }
        }
        catch(Exception ex){
            logger.error(ex.getMessage(),ex);

            msgResult.setResult(false);
            msgResult.setMessage(ex.getMessage());
        }

        return msgResult;
    }

    @ApiOperation(value="更新用户")
    @PostMapping("update")
    public MessageResult<AlarmInfo> update(@RequestBody AlarmInfo alarmInfo,@RequestAttribute String subject){
        MessageResult<AlarmInfo> msgResult = new MessageResult<>();

        try {
            alarmInfo.setUpdateBy(subject);
            alarmInfo.setUpdateTime(new Date());
            
            int affectCount = alarmInfoService.update(alarmInfo);

            if (affectCount > 0) {
                msgResult.setResult(true);
                msgResult.setData(alarmInfo);
            } else {
                msgResult.setResult(false);
                msgResult.setMessage("数据库更新失败");
            }
        }
        catch(Exception ex){
            logger.error(ex.getMessage(),ex);

            msgResult.setResult(false);
            msgResult.setMessage(ex.getMessage());
        }

        return msgResult;
    }

	@ApiOperation(value="删除")
    @PostMapping("delete/{id}")
    public MessageResult<Integer> delete(@PathVariable("id") String id,@RequestAttribute String subject){
        MessageResult<Integer> msgResult = new MessageResult<>();

        try {
            AlarmInfo alarmInfo = alarmInfoService.get(id);
            alarmInfo.setDelFlag(true);
            alarmInfo.setUpdateBy(subject);
            alarmInfo.setUpdateTime(new Date());

            int affectCount = alarmInfoService.update(alarmInfo);

            if (affectCount > 0) {
                msgResult.setResult(true);
                msgResult.setData(affectCount);
            } else {
                msgResult.setResult(false);
                msgResult.setMessage("删除失败");
            }
        }
        catch(Exception ex){
            logger.error(ex.getMessage(),ex);

            msgResult.setResult(false);
            msgResult.setMessage(ex.getMessage());
        }

        return msgResult;
    }


    @ApiOperation(value="批量删除")
    @PostMapping("batchDelete")
    public MessageResult<Integer> batchDelete(@RequestBody List<String> idList,@RequestAttribute String subject){
        MessageResult<Integer> msgResult = new MessageResult<>();

        try {
            int affectCount = 0;

            for (String id : idList) {
                AlarmInfo alarmInfo = alarmInfoService.get(id);
                alarmInfo.setDelFlag(true);
                alarmInfo.setUpdateBy(subject);
                alarmInfo.setUpdateTime(new Date());

                affectCount += alarmInfoService.update(alarmInfo);
            }

            if (affectCount > 0) {
                msgResult.setResult(true);
                msgResult.setData(affectCount);
            } else {
                msgResult.setResult(false);
                msgResult.setMessage("删除失败");
            }
        }
        catch(Exception ex){
            logger.error(ex.getMessage(),ex);

            msgResult.setResult(false);
            msgResult.setMessage(ex.getMessage());
        }

        return msgResult;
    }

    @ApiOperation(value="列表")
    @RequestMapping(value = "pageList",method = RequestMethod.POST)
    public MessageResult<Map> pageList(
            String id,
            @RequestParam(value="type",defaultValue="") String type,
            @RequestParam(value="draw",defaultValue="1") int draw,
            @RequestParam(value="start",defaultValue="1") int start,
            @RequestParam(value="length",defaultValue="20") int length,
            @RequestAttribute String subject){

        //当前用户ID
        System.out.println(subject);

        MessageResult<Map> msgResult = new MessageResult<>();

        Map<String,Object> searchParams = new HashMap<>();

        List<Sort> sortList = new ArrayList<>();
        sortList.add(new Sort("id_","asc"));

        if (StringUtils.isNotEmpty(id)) {
            searchParams.put("id","%" + id + "%");
        }

        if (StringUtils.isNotEmpty(type)) {
            searchParams.put("type",type);
        }

        int pageNum = start / length;

        if (start % length != 0) {
            pageNum++;
        }

        int pageSize = length;

        Page<AlarmInfo> page = alarmInfoService.pageSearch(searchParams,pageNum,pageSize,sortList);

        msgResult.setResult(true);
        msgResult.setData(PojoUtils.pageWrapper(page));

        return msgResult;
    }
}