Springboot&计算机类考研交流平台

package com.example.controller; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollectionUtil; import cn.hutool.core.io.IoUtil; import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; import cn.hutool.crypto.SecureUtil; import cn.hutool.json.JSONObject; import cn.hutool.poi.excel.ExcelUtil; import cn.hutool.poi.excel.ExcelWriter; import com.example.common.Result; import com.example.common.ResultCode; import com.example.entity.XuexiaoxinxiInfo; import com.example.dao.XuexiaoxinxiInfoDao; import com.example.service.XuexiaoxinxiInfoService; import com.example.exception.CustomException; import com.example.common.ResultCode; import com.example.vo.EchartsData; import com.example.vo.XuexiaoxinxiInfoVo; import com.github.pagehelper.PageHelper; import com.github.pagehelper.PageInfo; import com.example.service.*; import org.springframework.web.bind.annotation.*; import org.springframework.beans.factory.annotation.Value; import cn.hutool.core.util.StrUtil; import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; import javax.servlet.ServletOutputStream; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.IOException; import java.util.*; import java.util.stream.Collectors; @RestController @RequestMapping(value = "/xuexiaoxinxiInfo") public class XuexiaoxinxiInfoController { @Resource private XuexiaoxinxiInfoService xuexiaoxinxiInfoService; @Resource private XuexiaoxinxiInfoDao xuexiaoxinxiInfoDao; @PostMapping public Result<XuexiaoxinxiInfo> add(@RequestBody XuexiaoxinxiInfoVo xuexiaoxinxiInfo) { //mixmajixami xuexiaoxinxiInfoService.add(xuexiaoxinxiInfo); return Result.success(xuexiaoxinxiInfo); } //youtixing1 //youtixing2 @GetMapping("/getByDiqu") public Result<List<Map<String,String>>> qidu() { return Result.success(xuexiaoxinxiInfoService.findByDiqu()); } @DeleteMapping("/{id}") public Result delete(@PathVariable Long id) { xuexiaoxinxiInfoService.delete(id); return Result.success(); } @PutMapping public Result update(@RequestBody XuexiaoxinxiInfoVo xuexiaoxinxiInfo) { xuexiaoxinxiInfoService.update(xuexiaoxinxiInfo); return Result.success(); } //@PutMapping("/update2") // public Result update2(@RequestBody XuexiaoxinxiInfoVo xuexiaoxinxiInfo) { // xuexiaoxinxiInfoService.update2(xuexiaoxinxiInfo); // return Result.success(); // } @GetMapping("/{id}") public Result<XuexiaoxinxiInfo> detail(@PathVariable Long id) { XuexiaoxinxiInfo xuexiaoxinxiInfo = xuexiaoxinxiInfoService.findById(id); return Result.success(xuexiaoxinxiInfo); } @GetMapping("/changeStatus/{id}") public Result<XuexiaoxinxiInfo> changeStatus(@PathVariable Long id) { xuexiaoxinxiInfoService.changeStatus(id); return Result.success(); } @GetMapping public Result<List<XuexiaoxinxiInfoVo>> all() { return Result.success(xuexiaoxinxiInfoService.findAll()); } @GetMapping("/page/{name}") public Result<PageInfo<XuexiaoxinxiInfoVo>> page(@PathVariable String name, @RequestParam(defaultValue = "1") Integer pageNum, @RequestParam(defaultValue = "5") Integer pageSize, HttpServletRequest request) { return Result.success(xuexiaoxinxiInfoService.findPage(name, pageNum, pageSize, request)); } @GetMapping("/pageqt/{name}") public Result<PageInfo<XuexiaoxinxiInfoVo>> pageqt(@PathVariable String name, @RequestParam(defaultValue = "1") Integer pageNum, @RequestParam(defaultValue = "8") Integer pageSize, HttpServletRequest request) { return Result.success(xuexiaoxinxiInfoService.findPageqt(name, pageNum, pageSize, request)); } // @PostMapping("/register") // public Result<XuexiaoxinxiInfo> register(@RequestBody XuexiaoxinxiInfo xuexiaoxinxiInfo) { // if (StrUtil.isBlank(xuexiaoxinxiInfo.getName()) || StrUtil.isBlank(xuexiaoxinxiInfo.getPassword())) { // throw new CustomException(ResultCode.PARAM_ERROR); // } // return Result.success(xuexiaoxinxiInfoService.add(xuexiaoxinxiInfo)); // } /** * 批量通过excel添加信息 * @param file excel文件 * @throws IOException */ @PostMapping("/upload") public Result upload(MultipartFile file) throws IOException { List<XuexiaoxinxiInfo> infoList = ExcelUtil.getReader(file.getInputStream()).readAll(XuexiaoxinxiInfo.class); if (!CollectionUtil.isEmpty(infoList)) { // 处理一下空数据 List<XuexiaoxinxiInfo> resultList = infoList.stream().filter(x -> ObjectUtil.isNotEmpty(x.getXuexiaomingcheng())).collect(Collectors.toList()); for (XuexiaoxinxiInfo info : resultList) { xuexiaoxinxiInfoService.add(info); } } return Result.success(); } //yoxutonxgjitu @GetMapping("/getExcelModel") public void getExcelModel(HttpServletResponse response) throws IOException { // 1. 生成excel Map<String, Object> row = new LinkedHashMap<>(); row.put("tupian", "A图片"); row.put("xuexiaomingcheng", "A学校名称"); row.put("leixing", "A类型"); row.put("chengshi", "A城市"); row.put("zhuanye", "A专业"); row.put("luqufenshu", "A录取分数"); row.put("yuanxiaojieshao", "A院校介绍"); row.put("status", "是"); row.put("level", "xuexiaoxinxi"); List<Map<String, Object>> list = CollUtil.newArrayList(row); // 2. 写excel ExcelWriter writer = ExcelUtil.getWriter(true); writer.write(list, true); response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8"); response.setHeader("Content-Disposition","attachment;filename=xuexiaoxinxiInfoModel.xlsx"); ServletOutputStream out = response.getOutputStream(); writer.flush(out, true); writer.close(); IoUtil.close(System.out); } @GetMapping("/getExcel") public void getExcel(HttpServletResponse response) throws IOException { // 1. 生成excel Map<String, Object> row = new LinkedHashMap<>(); row.put("tupian", "A图片"); row.put("xuexiaomingcheng", "A学校名称"); row.put("leixing", "A类型"); row.put("chengshi", "A城市"); row.put("zhuanye", "A专业"); row.put("luqufenshu", "A录取分数"); row.put("yuanxiaojieshao", "A院校介绍"); row.put("status", "是"); row.put("level", "权限"); List<Map<String, Object>> list = CollUtil.newArrayList(row); List<Map<String, Object>> daochuexcellist = xuexiaoxinxiInfoDao.daochuexcel(); Map<String, Double> typeMap = new HashMap<>(); for (Map<String, Object> map : daochuexcellist) { list.add(map); } // 2. 写excel ExcelWriter writer = ExcelUtil.getWriter(true); writer.write(list, true); response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8"); response.setHeader("Content-Disposition","attachment;filename=xuexiaoxinxiInfo.xlsx"); ServletOutputStream out = response.getOutputStream(); writer.flush(out, true); writer.close(); IoUtil.close(System.out); } private void getPieData(String name, List<EchartsData> pieList, Map<String, Double> dataMap) { EchartsData pieData = new EchartsData()