package com.doumee.api.business; 
 | 
  
 | 
import com.doumee.api.BaseController; 
 | 
import com.doumee.core.annotation.excel.ExcelExporter; 
 | 
import com.doumee.core.annotation.pr.PreventRepeat; 
 | 
import com.doumee.core.model.ApiResponse; 
 | 
import com.doumee.core.model.LoginUserInfo; 
 | 
import com.doumee.core.model.PageWrap; 
 | 
import com.doumee.core.model.PageData; 
 | 
import com.doumee.core.utils.Constants; 
 | 
import com.doumee.dao.business.dto.DispatchUnitQueryDTO; 
 | 
import com.doumee.dao.business.dto.SaveDispatchUnitDTO; 
 | 
import com.doumee.dao.business.model.DispatchUnit; 
 | 
import com.doumee.service.business.DispatchUnitService; 
 | 
import io.swagger.annotations.Api; 
 | 
import io.swagger.annotations.ApiOperation; 
 | 
import org.apache.shiro.SecurityUtils; 
 | 
import org.apache.shiro.authz.annotation.RequiresPermissions; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
import org.springframework.web.bind.annotation.*; 
 | 
import javax.servlet.http.HttpServletResponse; 
 | 
  
 | 
import java.util.ArrayList; 
 | 
import java.util.List; 
 | 
  
 | 
/** 
 | 
 * @author 江蹄蹄 
 | 
 * @date 2024/01/16 10:03 
 | 
 */ 
 | 
@Api(tags = "派遣单位信息表") 
 | 
@RestController 
 | 
@RequestMapping("/business/dispatchUnit") 
 | 
public class DispatchUnitController extends BaseController { 
 | 
  
 | 
    @Autowired 
 | 
    private DispatchUnitService dispatchUnitService; 
 | 
  
 | 
    @PreventRepeat 
 | 
    @ApiOperation("新建") 
 | 
    @PostMapping("/create") 
 | 
    @RequiresPermissions("business:dispatchunit:create") 
 | 
    public ApiResponse create(@RequestBody SaveDispatchUnitDTO saveDispatchUnitDTO) { 
 | 
        return ApiResponse.success(dispatchUnitService.create(saveDispatchUnitDTO)); 
 | 
    } 
 | 
  
 | 
    @PreventRepeat 
 | 
    @ApiOperation("添加方案工种") 
 | 
    @PostMapping("/createSolution") 
 | 
    @RequiresPermissions("business:dispatchunit:create") 
 | 
    public ApiResponse createSolution(@RequestBody SaveDispatchUnitDTO saveDispatchUnitDTO) { 
 | 
        return ApiResponse.success(dispatchUnitService.createSolution(saveDispatchUnitDTO)); 
 | 
    } 
 | 
  
 | 
  
 | 
    @ApiOperation("修改信息") 
 | 
    @PostMapping("/editDispatchUnit") 
 | 
    @RequiresPermissions("business:dispatchunit:update") 
 | 
    public ApiResponse editDispatchUnit(@RequestBody SaveDispatchUnitDTO saveDispatchUnitDTO) { 
 | 
        dispatchUnitService.editDispatchUnit(saveDispatchUnitDTO); 
 | 
        return ApiResponse.success(null); 
 | 
    } 
 | 
  
 | 
  
 | 
    @ApiOperation("根据ID删除") 
 | 
    @GetMapping("/delete/{id}") 
 | 
    @RequiresPermissions("business:dispatchunit:delete") 
 | 
    public ApiResponse deleteById(@PathVariable Integer id) { 
 | 
        dispatchUnitService.deleteById(id); 
 | 
        return ApiResponse.success(null); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("批量删除") 
 | 
    @GetMapping("/delete/batch") 
 | 
    @RequiresPermissions("business:dispatchunit:delete") 
 | 
    public ApiResponse deleteByIdInBatch(@RequestParam String ids) { 
 | 
        String [] idArray = ids.split(","); 
 | 
        List<Integer> idList = new ArrayList<>(); 
 | 
        for (String id : idArray) { 
 | 
            idList.add(Integer.valueOf(id)); 
 | 
        } 
 | 
        dispatchUnitService.deleteByIdInBatch(idList); 
 | 
        return ApiResponse.success(null); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("根据ID修改") 
 | 
    @PostMapping("/updateById") 
 | 
    @RequiresPermissions("business:dispatchunit:update") 
 | 
    public ApiResponse updateById(@RequestBody DispatchUnit dispatchUnit) { 
 | 
        dispatchUnitService.updateById(dispatchUnit); 
 | 
        return ApiResponse.success(null); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("分页查询") 
 | 
    @PostMapping("/page") 
 | 
    @RequiresPermissions("business:dispatchunit:query") 
 | 
    public ApiResponse<PageData<DispatchUnit>> findPage (@RequestBody PageWrap<DispatchUnit> pageWrap) { 
 | 
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); 
 | 
        pageWrap.getModel().setCompanyId(user.getCompanyId()); 
 | 
        return ApiResponse.success(dispatchUnitService.findPage(pageWrap)); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("导出Excel") 
 | 
    @PostMapping("/exportExcel") 
 | 
    @RequiresPermissions("business:dispatchunit:exportExcel") 
 | 
    public void exportExcel (@RequestBody PageWrap<DispatchUnit> pageWrap, HttpServletResponse response) { 
 | 
  
 | 
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); 
 | 
        pageWrap.getModel().setCompanyId(user.getCompanyId()); 
 | 
        ExcelExporter.build(DispatchUnit.class).export(dispatchUnitService.findPage(pageWrap).getRecords(), "派遣单位信息表", response); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("根据ID查询") 
 | 
    @GetMapping("/{id}") 
 | 
    @RequiresPermissions("business:dispatchunit:query") 
 | 
    public ApiResponse<DispatchUnit> findById(@PathVariable Integer id) { 
 | 
        return ApiResponse.success(dispatchUnitService.detailById(id)); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("数据列表") 
 | 
    @PostMapping("/findListByDTO") 
 | 
    @RequiresPermissions("business:dispatchunit:query") 
 | 
    public ApiResponse<List<DispatchUnit>> findListByDTO (@RequestBody DispatchUnitQueryDTO dispatchUnitQueryDTO) { 
 | 
        dispatchUnitQueryDTO.setDataType(Constants.TWO); 
 | 
        return ApiResponse.success(dispatchUnitService.findByDTO(dispatchUnitQueryDTO)); 
 | 
    } 
 | 
  
 | 
  
 | 
} 
 |