| 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.MemberQueryDTO; | 
| import com.doumee.dao.business.model.Member; | 
| import com.doumee.service.business.MemberService; | 
| 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/member") | 
| public class MemberController extends BaseController { | 
|   | 
|     @Autowired | 
|     private MemberService memberService; | 
|   | 
|     @PreventRepeat | 
|     @ApiOperation("新建") | 
|     @PostMapping("/create") | 
|     @RequiresPermissions("business:member:create") | 
|     public ApiResponse create(@RequestBody Member member) { | 
|         return ApiResponse.success(memberService.create(member)); | 
|     } | 
|   | 
|     @ApiOperation("根据ID删除") | 
|     @GetMapping("/delete/{id}") | 
|     @RequiresPermissions("business:member:delete") | 
|     public ApiResponse deleteById(@PathVariable Integer id) { | 
|         memberService.deleteById(id); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("批量删除") | 
|     @GetMapping("/delete/batch") | 
|     @RequiresPermissions("business:member: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)); | 
|         } | 
|         memberService.deleteByIdInBatch(idList); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("根据ID修改") | 
|     @PostMapping("/updateById") | 
|     @RequiresPermissions("business:member:update") | 
|     public ApiResponse updateById(@RequestBody Member member) { | 
|         memberService.updateById(member); | 
|         return ApiResponse.success(null); | 
|     } | 
|   | 
|     @ApiOperation("分页查询") | 
|     @PostMapping("/page") | 
|     @RequiresPermissions("business:member:query") | 
|     public ApiResponse<PageData<Member>> findPage (@RequestBody PageWrap<MemberQueryDTO> pageWrap) { | 
| //        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
| //        pageWrap.getModel().setCompanyId(user.getCompanyId()); | 
|         return ApiResponse.success(memberService.findPageForCompany(pageWrap)); | 
|     } | 
|   | 
|     @ApiOperation("导出Excel") | 
|     @PostMapping("/exportExcel") | 
|     @RequiresPermissions("business:member:exportExcel") | 
|     public void exportExcel (@RequestBody PageWrap<MemberQueryDTO> pageWrap, HttpServletResponse response) { | 
|       /*  LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|         pageWrap.getModel().setCompanyId(user.getCompanyId()); | 
|         List<Member> list = memberService.findPageForCompany(pageWrap).getRecords(); | 
|         for (int i = 0; i < list.size(); i++) { | 
|             Member member = list.get(i); | 
|             member.setSn(i+1); | 
|             member.setAge(Constants.getAgeByIdCard(member.getIdcardNo())); | 
|         }*/ | 
|         ExcelExporter.build(Member.class).exportWithFirstAndEnd( memberService.findPageForCompany(pageWrap).getRecords(), "员工列表详单","员工列表详单",null, response); | 
|     } | 
|     @ApiOperation("根据ID查询") | 
|     @GetMapping("/{id}") | 
|     @RequiresPermissions("business:member:query") | 
|     public ApiResponse findById(@PathVariable Integer id) { | 
|         return ApiResponse.success(memberService.findById(id)); | 
|     } | 
| } |