| package com.doumee.api.business; | 
|   | 
| import com.doumee.config.DataSyncConfig; | 
| 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.PageWrap; | 
| import com.doumee.core.model.PageData; | 
| import com.doumee.core.utils.Constants; | 
| import com.doumee.dao.admin.request.UpdateCompanySortDTO; | 
| import com.doumee.dao.admin.response.CompanyDTO; | 
| import com.doumee.dao.business.model.Company; | 
| import com.doumee.service.business.CompanyService; | 
| import com.doumee.service.business.ext.ERPSyncService; | 
| import com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromHKServiceImpl; | 
| import io.swagger.annotations.Api; | 
| import io.swagger.annotations.ApiImplicitParam; | 
| import io.swagger.annotations.ApiImplicitParams; | 
| import io.swagger.annotations.ApiOperation; | 
| import org.apache.shiro.authz.annotation.RequiresPermissions; | 
| import org.springframework.beans.factory.annotation.Autowired; | 
| import org.springframework.validation.annotation.Validated; | 
| import org.springframework.web.bind.annotation.*; | 
| import javax.servlet.http.HttpServletResponse; | 
|   | 
| import java.util.ArrayList; | 
| import java.util.List; | 
|   | 
| import com.doumee.api.BaseController; | 
| /** | 
|  * @author 江蹄蹄 | 
|  * @date 2023/11/30 15:33 | 
|  */ | 
| @Api(tags = "企业信息表") | 
| @RestController | 
| @RequestMapping("/business/company") | 
| public class CompanyController extends BaseController { | 
|   | 
|     @Autowired | 
|     private CompanyService companyService; | 
|   | 
|     @Autowired | 
|     private ERPSyncService erpSyncService; | 
|     @Autowired | 
|     private HkSyncOrgUserFromHKServiceImpl hkSyncOrgUserFHKService; | 
|     @Autowired | 
|     private DataSyncConfig dataSyncConfig; | 
|   | 
|     @PreventRepeat | 
|     @ApiOperation("新建") | 
|     @PostMapping("/create") | 
|     @RequiresPermissions("business:company:create") | 
|     public ApiResponse create(@RequestBody Company company) { | 
|         return ApiResponse.success(companyService.create(company)); | 
|     } | 
|   | 
|     @PreventRepeat | 
|     @ApiOperation("创建劳务公司") | 
|     @PostMapping("/createLaborServices") | 
|     @RequiresPermissions("business:company:create") | 
|     public ApiResponse<Integer> createLaborServices(@RequestBody Company company){ | 
|         return ApiResponse.success(companyService.createLaborServices(company)); | 
|     } | 
|   | 
|     @ApiOperation("根据ID删除") | 
|     @GetMapping("/delete/{id}") | 
|     @RequiresPermissions("business:company:delete") | 
|     public ApiResponse deleteById(@PathVariable Integer id) { | 
|         companyService.deleteById(id,this.getLoginUser(null)); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("批量删除") | 
|     @GetMapping("/delete/batch") | 
|     @RequiresPermissions("business:company: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)); | 
|         } | 
|         companyService.deleteByIdInBatch(idList,this.getLoginUser(null)); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("根据ID修改") | 
|     @PostMapping("/updateById") | 
|     @RequiresPermissions("business:company:update") | 
|     public ApiResponse updateById(@RequestBody Company company) { | 
|         companyService.updateById(company); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("分页查询") | 
|     @PostMapping("/page") | 
|     @RequiresPermissions("business:company:query") | 
|     public ApiResponse<PageData<Company>> findPage (@RequestBody PageWrap<Company> pageWrap) { | 
|         return ApiResponse.success(companyService.findPage(pageWrap)); | 
|     } | 
|   | 
|     @ApiOperation("组织树查询") | 
|     @GetMapping("/tree") | 
|     @ApiImplicitParams({ | 
|             @ApiImplicitParam(name = "type", dataType = "Integer", value = "类型 0劳务公司 1内部组织", required = true), | 
|     }) | 
|     @RequiresPermissions("business:company:query") | 
|     public ApiResponse<List<Company>> tree (@RequestParam Integer type) { | 
|         return ApiResponse.success(companyService.companyTree(type)); | 
|     } | 
|   | 
|   | 
|     @ApiOperation("排序") | 
|     @PostMapping("/updateSort") | 
|     @RequiresPermissions("business:company:query") | 
|     public ApiResponse updateSort (@Validated @RequestBody UpdateCompanySortDTO dto) { | 
|         companyService.updateSort(dto); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|   | 
|     @ApiOperation("修改状态") | 
|     @PostMapping("/updateStatusById") | 
|     @RequiresPermissions("business:company:update") | 
|     public ApiResponse updateStatusById(@RequestBody Company company) { | 
|         companyService.updateStatusById(company); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("导出Excel") | 
|     @PostMapping("/exportExcel") | 
|     @RequiresPermissions("business:company:exportExcel") | 
|     public void exportExcel (@RequestBody PageWrap<Company> pageWrap, HttpServletResponse response) { | 
|         ExcelExporter.build(Company.class).export(companyService.findPage(pageWrap).getRecords(), "企业信息表", response); | 
|     } | 
|   | 
|     @ApiOperation("根据ID查询") | 
|     @GetMapping("/{id}") | 
|     @RequiresPermissions("business:company:query") | 
|     public ApiResponse findById(@PathVariable Integer id) { | 
|         return ApiResponse.success(companyService.findById(id)); | 
|     } | 
|   | 
|     @ApiOperation("根据分类ID查询") | 
|     @PostMapping("/findLaborServicesList") | 
|     @RequiresPermissions("business:company:query") | 
|     public ApiResponse<List<Company>> findLaborServicesList(Company query){ | 
|         Company company = new Company(); | 
|         company.setIsdeleted(Constants.ZERO); | 
|         company.setType(query.getType()); | 
|         company.setCategoryId(query.getCategoryId()); | 
|         return ApiResponse.success(companyService.findList(company)); | 
|     } | 
|   | 
|   | 
|     /** | 
|      * 查询部门及其子部门信息 | 
|      * @param type 0 查询所有信息 1 查询内部信息 | 
|      * @return | 
|      */ | 
|     @ApiOperation("根据类型查询组织信息") | 
|     @GetMapping("/findCompanyTreePage") | 
|     @RequiresPermissions("business:company:query") | 
|     public ApiResponse<List<CompanyDTO>> findCompanyTreePage(Integer type){ | 
|         return ApiResponse.success(companyService.findCompanyTreePage(type)); | 
|     } | 
|   | 
|   | 
|   | 
|     @ApiOperation("全量部门信息同步") | 
|     @PostMapping("/syncAll") | 
|     @RequiresPermissions("business:company:sync") | 
|     public ApiResponse sync(){ | 
|         if(Constants.formatIntegerNum(dataSyncConfig.getOrgUserDataOrigin()) == DataSyncConfig.origin.erp){ | 
|             return ApiResponse.success( erpSyncService.syncCompany(null)); | 
|         }else if(Constants.formatIntegerNum(dataSyncConfig.getOrgUserDataOrigin()) == DataSyncConfig.origin.hk){ | 
|              hkSyncOrgUserFHKService.syncOrgData(); | 
|             return ApiResponse.success("同步成功"); | 
|         } | 
|         return ApiResponse.failed("无效操作"); | 
|     } | 
| } |