灵狐GEO系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
guyue
2025-08-12 ff1fec7eae681e89e607fd441d2597ab67b5ed2a 
 src/main/java/com/linghu/controller/UserController.java


@@ -1,16 +1,47 @@


package com.linghu.controller;




import com.alibaba.excel.EasyExcel;


import com.alibaba.excel.ExcelWriter;


import com.alibaba.excel.write.handler.SheetWriteHandler;


import com.alibaba.excel.write.metadata.WriteSheet;


import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;


import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.extension.plugins.pagination.Page;


import com.linghu.listener.PlatformExcelListener;


import com.linghu.listener.UserExcelListener;


import com.linghu.model.common.ResponseResult;


import com.linghu.model.entity.Type;


import com.linghu.model.entity.Platform;


import com.linghu.model.entity.Sectionalization;


import com.linghu.model.entity.User;


import com.linghu.model.excel.ExcelDataWithRow;


import com.linghu.model.excel.PlatformExcel;


import com.linghu.model.excel.UserExcel;


import com.linghu.model.vo.UserPageVO;


import com.linghu.service.SectionalizationService;


import com.linghu.service.UserService;


import io.swagger.annotations.Api;


import io.swagger.annotations.ApiOperation;


import org.apache.poi.ss.usermodel.DataValidation;


import org.apache.poi.ss.usermodel.DataValidationConstraint;


import org.apache.poi.ss.usermodel.DataValidationHelper;


import org.apache.poi.ss.usermodel.Sheet;


import org.apache.poi.ss.util.CellRangeAddressList;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.web.bind.annotation.PostMapping;


import org.springframework.web.bind.annotation.RequestBody;


import org.springframework.web.bind.annotation.RequestMapping;


import org.springframework.web.bind.annotation.RestController;


import org.springframework.http.HttpHeaders;


import org.springframework.http.MediaType;


import org.springframework.http.ResponseEntity;


import org.springframework.util.StringUtils;


import org.springframework.web.bind.annotation.*;


import org.springframework.web.multipart.MultipartFile;




import javax.validation.Valid;


import java.io.ByteArrayOutputStream;


import java.io.IOException;


import java.time.LocalDateTime;


import java.util.*;


import java.util.regex.Pattern;


import java.util.stream.Collectors;




@RestController


@RequestMapping("/user")


@@ -19,34 +50,68 @@


    @Autowired


    private UserService userService;




/*


    @PostMapping


    @ApiOperation(value = "添加类型")


    public ResponseResult<User> add(@RequestBody User user) {


        boolean success = typeService.save(type);


        if (success) {


            return ResponseResult.success(type);


        }


        return ResponseResult.error("添加类型失败");


    }


*/






    /**


     * 新增用户


     */




    @PostMapping


    @ApiOperation(value = "添加")


    public ResponseResult add(@Valid @RequestBody User user) {


        return userService.addUser(user);




    }




    /**


     * 修改用户


     */


    @PutMapping


    @ApiOperation(value = "修改")


    public ResponseResult edit(@Valid @RequestBody User user) {


        return userService.editUser(user);




    }




    /**


     * 删除用户


     */




    @DeleteMapping("/{user_id}")


    @ApiOperation(value = "删除")


    public ResponseResult delete(@PathVariable("user_id") Integer user_id) {


        userService.removeById(user_id);


        return ResponseResult.success();




    }


    /**


     * 分页查询


     */


    @GetMapping


    @ApiOperation(value = "分页")


    public ResponseResult<Page<UserPageVO>> page(@RequestParam(value = "pageSize", required = false, defaultValue = "10")Integer pageSize,


                                                 @RequestParam(value = "pageNum", required = false,defaultValue = "1")Integer pageNum,


                                                 @RequestParam(value = "sectionalization_id",required = false)Integer sectionalization_id,


                                                 @RequestParam(value = "status" ,required = false)String status) {


        Page<UserPageVO> page = new Page<>(pageNum, pageSize);


        return ResponseResult.success( userService.getPage(page,sectionalization_id,status));


    }




    /**


     * 下载模板


     */




    @GetMapping("/downloadTemplate")


    @ApiOperation("下载模板")


    public ResponseEntity<byte[]> downloadTemplate() throws IOException {


        return userService.downloadUser();


    }






    /**


     * 导入文件


     */


    @PostMapping("/importUserExcel")


    @ApiOperation("导入用户数据")


    public ResponseResult importUserExcel(@RequestParam("file") MultipartFile file) {


        return userService.importUser(file);


    }


}