Code/sms_water/src/main/java/com/huaxu/controller/SceneController.java

package com.huaxu.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.huaxu.common.FileUploadUtil;
import com.huaxu.dto.DeviceDto;
import com.huaxu.entity.SceneTypeEntity;
import com.huaxu.model.AjaxMessage;
import com.huaxu.model.LoginUser;
import com.huaxu.model.Pagination;
import com.huaxu.model.ResultStatus;
import com.huaxu.util.UserUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.validation.annotation.Validated;
import org.springframework.ui.ModelMap;

import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Date;
import org.springframework.web.bind.annotation.*;
import com.huaxu.entity.SceneEntity;
import com.huaxu.service.SceneService;
import org.springframework.web.multipart.MultipartFile;

/**
 * 场景信息页面控制器
 * @author WYY
 * @date 2020-11-17 21:15
 */
@RestController
@RequestMapping("/scene")
@Api(tags = "场景管理")
public class SceneController {
    @Autowired
    private SceneService sceneService;
    @Value("${SMS.sys_config_path}")
    private String baseDir;

    @ApiOperation(value = "分页查询场景")
    @RequestMapping(value = "/selectPage", method = RequestMethod.GET)
    @ResponseBody
    public AjaxMessage<Pagination<SceneEntity>> selectPage(@RequestParam Integer pageNum, @RequestParam Integer pageSize, @RequestParam(value = "sceneName", required = false) String sceneName,@RequestParam(required = false) Long companyOrgId,@RequestParam(required = false) Long deptOrgId) {
        IPage<SceneEntity> iPage = new Page<>(pageNum, pageSize);
        SceneEntity sceneEntity = new SceneEntity();
        sceneEntity.setSceneName(sceneName);
        sceneEntity.setCompanyOrgId(companyOrgId);
        sceneEntity.setDeptOrgId(deptOrgId);
        iPage = sceneService.findPage(iPage, sceneEntity);
        Pagination<SceneEntity> pages = new Pagination<>(iPage);
        return new AjaxMessage<>(ResultStatus.OK, pages);
    }

    @ApiOperation(value = "查询场景树")
    @RequestMapping(value = "/selectList", method = RequestMethod.POST)
    @ResponseBody
    public AjaxMessage<List<SceneEntity>> allList() {
        SceneEntity sceneEntity = new SceneEntity();
        List<SceneEntity> menuEntities = sceneService.findAllList(sceneEntity);
        return new AjaxMessage<>(ResultStatus.OK, menuEntities);
    }

    @ApiOperation(value = "查询场景树")
    @RequestMapping(value = "/selectListForParm", method = RequestMethod.POST)
    @ResponseBody
    public AjaxMessage<List<SceneEntity>> selectListForParm(@RequestParam(required = false) Long comId,@RequestParam(required = false) Long deptId) {
        if (comId != null) {
            SceneEntity sceneEntity = new SceneEntity();
            sceneEntity.setCompanyOrgId(comId);
            sceneEntity.setDeptOrgId(deptId);
            List<SceneEntity> menuEntities = sceneService.findAllList(sceneEntity);
            return new AjaxMessage<>(ResultStatus.OK, menuEntities);
        } else {
            return new AjaxMessage<>(ResultStatus.OK);
        }
    }

    /**
     * 新增
     */
    @ApiOperation(value = "场景图片")
    @RequestMapping(value = "addSceneImage", method = RequestMethod.POST)
    @ResponseBody
    public AjaxMessage<String> addUserPhoto(@ApiParam(value = "场景图片", required = true) @RequestParam("avatarfile") MultipartFile file) {
        String avatar = "";
        if (!file.isEmpty()) {
            try {
                avatar = FileUploadUtil.uploadWeb(baseDir, file);
            } catch (IOException e) {
                return new AjaxMessage<>(ResultStatus.ERROR, e.getMessage());
            }
        }
        return new AjaxMessage<>(ResultStatus.OK, avatar);
    }

    /**
     * 新增
     */
    @ApiOperation(value = "新增")
    @RequestMapping(value = "/add", method = RequestMethod.POST)
    @ResponseBody
    public AjaxMessage<Integer> addScene(@ApiParam(value = "场景", required = true) @RequestBody SceneEntity scene) {
        int result = sceneService.addScene(scene) ? 1 : 0;
        return new AjaxMessage<>(ResultStatus.OK, result);
    }


    /**
     * 按ID查询场景
     */
    @ApiOperation(value = "按ID查询场景")
    @RequestMapping(value = "/selectById", method = RequestMethod.GET)
    @ResponseBody
    public AjaxMessage<SceneEntity> selectById(@ApiParam(value = "ID", required = true) @RequestParam Long id) {
        SceneEntity scene = sceneService.findSceneById(id);
        return new AjaxMessage<>(ResultStatus.OK, scene);
    }

    /**
     * 修改保存场景信息
     */
    @RequestMapping(value = "/edit", method = RequestMethod.POST)
    @ResponseBody
    @ApiOperation(value = "修改保存场景信息")
    public AjaxMessage<Integer> editScene(@ApiParam(value = "场景", required = true) @RequestBody SceneEntity scene) {

        //SceneEntity cScene = sceneService.findSceneById(scene.getId());
        if(scene.getParentSceneId()!=null&&scene.getParentSceneId().equals(scene.getId()))
        {
            return new AjaxMessage<>(ResultStatus.SCENE_PARENT_ALREADY_EXISTS);
        }
        int result = sceneService.updateSceneById(scene) ? 1 : 0;
        return new AjaxMessage<>(ResultStatus.OK, result);
    }
    @RequestMapping(value = "/findByParentIdsLike", method = RequestMethod.GET)
    @ResponseBody
    @ApiOperation(value = "根据父级ID查询所有子节点及本身")
    public  List<Long> findByParentIdsLike(@ApiParam(value = "父级场景ID", required = false)@RequestParam(required = false) Long id) {
        List<Long> longs = sceneService.findByParentIdsLike(id);
        return longs;
    }
    /**
     * 删除
     */
    @ApiOperation(value = "删除")
    @RequestMapping(value = "/deleteByIds", method = RequestMethod.POST)
    @ResponseBody
    public AjaxMessage<Integer> del(@ApiParam(value = "场景ID", required = true) @RequestBody Long[] ids) {

        List<SceneEntity> scenes = sceneService.findByParentSceneIds(ids);
        if (scenes.size() > 0) {
            return new AjaxMessage<>(ResultStatus.EXSIT_IS_PARENT_ERROR);
        }
        int result = sceneService.delSceneByIds(ids) ? 1 : 0;
        return new AjaxMessage<>(ResultStatus.OK, result);
    }
    /**
     * 一级场景查询
     */
    @ApiOperation(value = "一级场景查询",notes = "sceneTypeName 为水源、水厂、泵站")
    @RequestMapping(value = "/selectByTypeName", method = RequestMethod.GET)
    @ResponseBody
    public AjaxMessage< List<SceneEntity>> selectByTypeName(@ApiParam(value = "一级场景类型名称", required = true) @RequestParam String sceneTypeName) {
        SceneEntity sceneEntity = new SceneEntity();
        sceneEntity.setSceneTypeName(sceneTypeName);
        List<SceneEntity> sceneEntities = sceneService.selectByTypeName(sceneEntity);
        return new AjaxMessage<>(ResultStatus.OK, sceneEntities);
    }
    @RequestMapping(value = "/findByParentId", method = RequestMethod.GET)
    @ResponseBody
    @ApiOperation(value = "根据父级ID查询所有子节点及本身(全部信息)")
    public  AjaxMessage<List<SceneEntity>> findByParentId(@ApiParam(value = "父级场景ID", required = false)@RequestParam(required = false) Long id) {
        List<SceneEntity> sceneEntities = sceneService.findByParentId(id);
        return new AjaxMessage<>(ResultStatus.OK, sceneEntities);
    }
}