package com.zhongzheng.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.toolkit.StringUtils; import com.zhongzheng.bo.SysCertificateBo; import com.zhongzheng.bo.SysDictTypeBo; import com.zhongzheng.common.croe.BaseController; import com.zhongzheng.common.croe.LoginUser; import com.zhongzheng.common.filter.CustomException; import com.zhongzheng.common.filter.UserConstants; import com.zhongzheng.common.model.AjaxResult; import com.zhongzheng.common.model.TableDataInfo; import com.zhongzheng.common.model.TokenService; import com.zhongzheng.common.util.DateUtils; import com.zhongzheng.common.util.SecurityUtils; import com.zhongzheng.common.util.ServletUtils; import com.zhongzheng.common.util.ToolsUtils; import com.zhongzheng.domian.*; import com.zhongzheng.service.*; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import static net.sf.jsqlparser.util.validation.metadata.NamedObject.user; /** * 数据字典信息 * * @author zhongzheng */ @Api(tags ="我的客户管理") @RestController @RequestMapping("/system/company") public class SysCompanyController extends BaseController { @Autowired private ISysCompanyService companyService; @Autowired private ISysCustomerSeaService seaService; @Autowired private ISysCompanyContactsService companyContactsService; @Autowired private TokenService tokenService; @Autowired private ISysPersonService personService; @Autowired private ISysCertificateService certService; @Autowired private ICertificateMajorService majorService; @Autowired private ISysCertificateTypeService typeService; @ApiOperation("客户列表") @GetMapping("/list") public TableDataInfo list(SysCompany model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); model.setUserId(userId); return companyService.selectCompanyList(model); } @ApiOperation("新增企业") @PostMapping("/add") public AjaxResult add(@Validated @RequestBody SysCompany company) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); company.setUserId(userId); company.setSource("自有"); SysCustomerSea sea = seaService.selectCustomerSeaByName(company.getCompanyName()); if(sea!=null){ throw new CustomException("企业已存在公海中,请直接在公海中领取!"); } return toAjax(companyService.insertCompany(company)>0?1:0); } @ApiOperation("编辑企业") @PostMapping("/edit") public AjaxResult edit(@Validated @RequestBody SysCompany company) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); company.setUserId(userId); company.setSource("自有"); return toAjax(companyService.updateCompany(company,company.getUserId())); } @ApiOperation("删除人") @PostMapping("/del") public AjaxResult del(@Validated @RequestBody SysCompany entity) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); return toAjax(companyService.deleteCompanyById(entity.getCompanyId(),userId)); } @ApiOperation("客户联系人列表") @GetMapping("/contact/list") public TableDataInfo contact_list(SysCompanyContacts model) { return companyContactsService.selectCompanyContactsList(model); } @ApiOperation("新增企业联系人") @PostMapping("/contact/add") public AjaxResult contact_add(@Validated @RequestBody SysCompanyContacts entity) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); entity.setUserId(userId); return toAjax(companyContactsService.insertSysCompanyContacts(entity)>0?1:0); } @ApiOperation("编辑企业联系人") @PostMapping("/contact/edit") public AjaxResult contact_edit(@Validated @RequestBody SysCompanyContacts entity) { return toAjax(companyContactsService.updateSysCompanyContacts(entity)>0?1:0); } @ApiOperation("删除企业联系人") @PostMapping("/contact/del") public AjaxResult contact_del(@Validated @RequestBody SysCompanyContacts entity) { return toAjax(companyContactsService.deleteSysCompanyContactsById(entity.getContactId())); } @ApiOperation("员工列表") @GetMapping("/person/list") public TableDataInfo contact_list(SysPerson model) { return personService.selectPersonList(model); } @ApiOperation("新增企业员工") @PostMapping("/person/add") public AjaxResult person_add(@Validated @RequestBody SysPerson entity) { //entity.setTenantId((long)1); LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); entity.setCreator(userId); return toAjax(personService.insertPerson(entity)>0?1:0); } @ApiOperation("编辑企业员工") @PostMapping("/person/edit") public AjaxResult person_edit(@Validated @RequestBody SysPerson entity) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); return toAjax(personService.updatePerson(entity,userId)>0?1:0); } @ApiOperation("删除企业员工") @PostMapping("/person/del") public AjaxResult person_del(@Validated @RequestBody SysPerson entity) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); Long userId = loginUser.getUser().getUserId(); return toAjax(personService.deletePersonById(entity.getPersonId(),userId)); } @ApiOperation("员工证书列表") @GetMapping("/cert/list") public TableDataInfo cert_list(SysCertificateBo model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); model.setUserId(loginUser.getUser().getUserId()); //model.setUserId(1L); return certService.selectList(model); } @ApiOperation("企业安全生产许可证书列表") @GetMapping("/cert/safetyList") public TableDataInfo cert_safety_license_list(SysCertificateBo model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); model.setUserId(loginUser.getUser().getUserId()); model.setCertificateTypeId(1); //model.setUserId(1L); return certService.selectEnterpriseCerList(model); } @ApiOperation("企业资质证书列表") @GetMapping("/cert/qualificationList") public TableDataInfo cert_qualification_list(SysCertificateBo model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); model.setUserId(loginUser.getUser().getUserId()); model.setCertificateTypeId(3); //model.setUserId(1L); return certService.selectEnterpriseCerList(model); } @ApiOperation("添加证书") @PostMapping("/cert/add") public AjaxResult cert_add(@Validated @RequestBody SysCertificateBo model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); model.setUserId(loginUser.getUser().getUserId()); return toAjax(certService.insertCertificate(model)>0?1:0); } @ApiOperation("编辑证书") @PostMapping("/cert/edit") public AjaxResult cert_edit(@Validated @RequestBody SysCertificateBo model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); model.setUserId(loginUser.getUser().getUserId()); return toAjax(certService.updateCertificate(model,model.getUserId())); } @ApiOperation("删除证书") @PostMapping("/cert/del") public AjaxResult cert_del(@Validated @RequestBody SysCertificateBo model) { LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); model.setUserId(loginUser.getUser().getUserId()); return toAjax(certService.deleteCertificateById(model.getCertificateId(),model.getUserId())); } @ApiOperation("获取证书类型列表 parentId=2 个人证书") @GetMapping("/cert/certificate_type_list") public TableDataInfo certificate_type_list(Integer parentId) { return typeService.selectCertificateTypeListByCode(parentId); } @ApiOperation("获取专业列表") @GetMapping("/cert/major_list") public TableDataInfo cert_major_list(Integer certificateTypeId) { return majorService.selectCertificateMajorList(certificateTypeId); } @ApiOperation("证书预警列表") @GetMapping("/cert/warningList") public TableDataInfo cert_warning_list(SysCertificateBo model) { //LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest()); //model.setUserId(loginUser.getUser().getUserId()); model.setUserId(1L); model.setWarningState(1); return certService.selectList(model); } }