/*
|
* Copyright (c) 2018-2025, ztt All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the pig4cloud.com developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: ztt
|
*/
|
|
package com.chinaztt.mes.equipment.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
|
import com.chinaztt.mes.equipment.dto.EquipmentDTO;
|
import com.chinaztt.mes.equipment.entity.Equipment;
|
import com.chinaztt.mes.equipment.excel.EquipmentData;
|
import com.chinaztt.mes.equipment.excel.EquipmentUploadListener;
|
import com.chinaztt.mes.equipment.service.EquipmentService;
|
import com.chinaztt.mes.equipment.vo.FactoryVO;
|
import com.chinaztt.ztt.common.core.util.R;
|
import com.chinaztt.ztt.common.log.annotation.SysLog;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.AllArgsConstructor;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
|
/**
|
* 工厂信息表
|
*
|
* @author fenglang
|
* @date 2020-08-18 14:19:03
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/equipment")
|
@Api(value = "equipment", tags = "设备信息表管理")
|
public class EquipmentController {
|
|
private final EquipmentService equipmentService;
|
|
|
/**
|
* 分页查询
|
*
|
* @param page 分页对象
|
* @param equipmentDTO 工厂信息表
|
* @return
|
*/
|
@ApiOperation(value = "分页查询", notes = "分页查询")
|
|
@GetMapping("/page")
|
@PreAuthorize("@pms.hasPermission('equipment_equipment_view')")
|
public R getEquipmentPage(Page page, EquipmentDTO equipmentDTO) {
|
return R.ok(equipmentService.getEquipmentPage(page, QueryWrapperUtil.gen(equipmentDTO)));
|
}
|
|
/**
|
* 通过工作站查询设备
|
*
|
* @param id 分页对象
|
* @return
|
*/
|
@ApiOperation(value = "通过工作站查询设备", notes = "通过工作站查询设备")
|
|
@GetMapping("/page/{id}")
|
public R getEquipmentPageByWorkstation(@PathVariable("id") Long id) {
|
return R.ok(equipmentService.getEquipmentPageByWorkstation(id));
|
}
|
|
/**
|
* 通过id查询工厂信息表
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id查询", notes = "通过id查询")
|
@GetMapping("/{id}")
|
@PreAuthorize("@pms.hasPermission('equipment_equipment_view')")
|
public R getById(@PathVariable("id") Long id) {
|
return R.ok(equipmentService.getById(id));
|
}
|
|
/**
|
* 新增工厂信息表
|
*
|
* @param equipment 工厂信息表
|
* @return R
|
*/
|
@ApiOperation(value = "新增设备信息表", notes = "新增设备信息表")
|
@SysLog("新增设备信息表")
|
@PostMapping
|
@PreAuthorize("@pms.hasPermission('equipment_equipment_add')")
|
public R save(@RequestBody Equipment equipment) {
|
return R.ok(equipmentService.save(equipment));
|
}
|
|
/**
|
* excel上传
|
*
|
* @return
|
*/
|
@PostMapping("/upload")
|
public R upload(@RequestParam("file") MultipartFile file) {
|
try {
|
List<FactoryVO> factoryVOS = equipmentService.getFactorys();
|
Map<String, FactoryVO> map = factoryVOS.stream().collect(Collectors.toMap(FactoryVO::getFactoryName, factoryVO -> factoryVO, (k1, k2) -> k1));
|
EasyExcel.read(file.getInputStream(), EquipmentData.class, new EquipmentUploadListener(equipmentService, map)).sheet().doRead();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
return R.ok();
|
}
|
|
/**
|
* 修改工厂信息表
|
*
|
* @param equipment 工厂信息表
|
* @return R
|
*/
|
@ApiOperation(value = "修改设备信息表", notes = "修改设备信息表")
|
@SysLog("修改设备信息表")
|
@PutMapping
|
@PreAuthorize("@pms.hasPermission('equipment_equipment_edit')")
|
public R updateById(@RequestBody Equipment equipment) {
|
return R.ok(equipmentService.updateById(equipment));
|
}
|
|
/**
|
* 通过id删除工厂信息表
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id删除设备信息表", notes = "通过id删除设备信息表")
|
@SysLog("通过id删除工厂信息表")
|
@DeleteMapping("/{id}")
|
@PreAuthorize("@pms.hasPermission('equipment_equipment_del')")
|
public void removeById(@PathVariable Long id) {
|
equipmentService.deleteEquipment(id);
|
}
|
}
|
