/*
|
* 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.basic.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.chinaztt.mes.basic.entity.Division;
|
import com.chinaztt.mes.basic.entity.Staff;
|
import com.chinaztt.mes.basic.excel.DivisionData;
|
import com.chinaztt.mes.basic.excel.DivisionUploadListener;
|
import com.chinaztt.mes.basic.service.DivisionService;
|
import com.chinaztt.mes.basic.service.StaffService;
|
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
|
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;
|
|
|
/**
|
* 部门
|
*
|
* @author luh
|
* @date 2020-08-14 15:20:19
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/division")
|
@Api(value = "division", tags = "部门管理")
|
public class DivisionController {
|
|
private final DivisionService divisionService;
|
private final StaffService staffService;
|
|
/**
|
* 分页查询
|
*
|
* @param page 分页对象
|
* @param division 部门
|
* @return
|
*/
|
@ApiOperation(value = "分页查询", notes = "分页查询")
|
@GetMapping("/page")
|
@PreAuthorize("@pms.hasPermission('basic_division_view')")
|
public R getDivisionPage(Page page, Division division) {
|
return R.ok(divisionService.getDivisionPage(page, QueryWrapperUtil.gen(division)));
|
}
|
|
|
/**
|
* 通过id查询部门
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id查询", notes = "通过id查询")
|
@GetMapping("/{id}")
|
@PreAuthorize("@pms.hasPermission('basic_division_view')")
|
public R getById(@PathVariable("id") Long id) {
|
return R.ok(divisionService.getById(id));
|
}
|
|
/**
|
* 新增部门
|
*
|
* @param division 部门
|
* @return R
|
*/
|
@ApiOperation(value = "新增部门", notes = "新增部门")
|
@SysLog("新增部门")
|
@PostMapping
|
@PreAuthorize("@pms.hasPermission('basic_division_add')")
|
public R save(@RequestBody Division division) {
|
return R.ok(divisionService.save(division));
|
}
|
/**
|
* excel上传
|
*
|
* @return
|
*/
|
@PostMapping("/upload")
|
public R upload(@RequestParam("file") MultipartFile file) {
|
try {
|
EasyExcel.read(file.getInputStream(), DivisionData.class, new DivisionUploadListener(divisionService)).sheet().doRead();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
return R.ok();
|
}
|
/**
|
* 修改部门
|
*
|
* @param division 部门
|
*
|
* @return R
|
*/
|
@ApiOperation(value = "修改部门", notes = "修改部门")
|
@SysLog("修改部门")
|
@PutMapping
|
@PreAuthorize("@pms.hasPermission('basic_division_edit')")
|
public R updateById(@RequestBody Division division) {
|
return R.ok(divisionService.updateById(division));
|
}
|
|
/**
|
* 通过id删除部门
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id删除部门", notes = "通过id删除部门")
|
@SysLog("通过id删除部门")
|
@DeleteMapping("/{id}")
|
@PreAuthorize("@pms.hasPermission('basic_division_del')")
|
public R removeById(@PathVariable Long id) {
|
return R.ok(divisionService.removeById(id));
|
}
|
|
|
/**
|
* Description: 加载人员列表
|
*
|
* @param:
|
* @return: List<Staff>
|
* @author: luh
|
* @date: 2020/8/19 14:35
|
*/
|
@GetMapping("/staff")
|
public List<Staff> loadCompany() {
|
return staffService.list();
|
}
|
|
/**
|
* Description: 持久化部门实体（自定义）
|
*
|
* @param: division
|
* @return: R
|
* @author: luh
|
* @date: 2020/8/17 16:14
|
*/
|
@ApiOperation(value = "自定义更新部门", notes = "自定义更新部门")
|
@SysLog("自定义更新部门")
|
@PostMapping("/adddivision")
|
@PreAuthorize("@pms.hasPermission('basic_division_add')")
|
public R addDivision(@RequestBody Division division) {
|
if (division.getId().equals(0L)) {
|
//判断编号是否重复
|
Division divisionByFind=divisionService.getOne(Wrappers.<Division>lambdaQuery().eq(Division::getDivisionNo, division.getDivisionNo()));
|
if(divisionByFind!=null){
|
throw new RuntimeException("部门编号重复：" + divisionByFind.getDivisionName() + "-" + divisionByFind.getDivisionNo());
|
}else{
|
return R.ok(divisionService.save(division));
|
}
|
} else {
|
return R.ok(divisionService.updateById(division));
|
}
|
|
|
}
|
|
}
|
