package com.gkhy.safePlatform.equipment.controller;



import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.gkhy.safePlatform.equipment.entity.EquipmentTypeMng;
import com.gkhy.safePlatform.equipment.service.EquipmentTypeMngService;
import org.springframework.web.bind.annotation.*;
import com.gkhy.safePlatform.commons.query.PageQuery;
import com.gkhy.safePlatform.commons.utils.PageUtils;
import com.gkhy.safePlatform.commons.vo.ResultVO;
import com.gkhy.safePlatform.commons.enums.ResultCodes;
import com.gkhy.safePlatform.equipment.model.dto.req.EquipmentTypeMngQueryCriteria;

import java.util.Arrays;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import java.io.Serializable;
import java.util.List;

/**
 * 设备设施类型管理(EquipmentTypeMng)表控制层
 *
 * @author xurui
 * @since 2022-07-19 08:35:02
 */
@RestController
@RequestMapping("equipmentTypeMng")
public class EquipmentTypeMngController {
    /**
     * 服务对象
     */
    @Resource
    private EquipmentTypeMngService equipmentTypeMngService;

    /**
     * 分页查询所有数据
     *
     * @param pageQuery 查询实体
     * @return 所有数据
     */
    @PostMapping(value = "/page/list")
    public ResultVO selectAll(@RequestBody PageQuery<EquipmentTypeMngQueryCriteria> pageQuery){
		PageUtils.checkCheck(pageQuery.getPageIndex(), pageQuery.getPageSize());
		return this.equipmentTypeMngService.queryAll(pageQuery);
    }


    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping(value = "/selectOne/{id}")
    public ResultVO selectOne(@PathVariable Serializable id) {
        return new ResultVO<>(ResultCodes.OK,this.equipmentTypeMngService.getById(id));
    }

	/**
     * 新增或者修改数据
     *
     * @param equipmentTypeMng 实体对象
     * @return 修改结果
     */
    @PostMapping(value = "/addOrUpdate")
    public ResultVO update(@RequestBody EquipmentTypeMng equipmentTypeMng) {
        if (equipmentTypeMng.getId() == null) {
            return new ResultVO<>(ResultCodes.OK,equipmentTypeMngService.save(equipmentTypeMng));
        } else {
            equipmentTypeMngService.update(equipmentTypeMng,new UpdateWrapper<EquipmentTypeMng>().eq("id",equipmentTypeMng.getId()));
            return new ResultVO<>(ResultCodes.OK);
        }
    }

    /**
     * 删除数据
     *
     * @param ids 主键结合
     * @return 删除结果
     */
    @GetMapping(value = "/delete")
    public ResultVO delete(String ids) {
	List<String> idList = Arrays.stream(ids.split(","))
                .collect(Collectors.toList());
        this.equipmentTypeMngService.removeByIds(idList);
        return new ResultVO<>(ResultCodes.OK);
    }


    /**
     * 设备设施类型-树状数据结构
     *
     * @return 所有数据
     */
    @GetMapping(value = "/treeData")
    public ResultVO treeData(){
        return new ResultVO<>(ResultCodes.OK,this.equipmentTypeMngService.treeData(0L));
    }
}