From 5504abe3e0b0df07a31753c94d4cf1c03df5a88f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=A6=BE=E5=87=A0=E6=B5=B7?= Date: Sat, 18 Jul 2020 12:29:07 +0800 Subject: [PATCH] Other.java -> CommonController.java --- .../{Other.java => CommonController.java} | 380 +++++++++--------- 1 file changed, 190 insertions(+), 190 deletions(-) rename src/main/java/cn/celess/blog/controller/{Other.java => CommonController.java} (97%) diff --git a/src/main/java/cn/celess/blog/controller/Other.java b/src/main/java/cn/celess/blog/controller/CommonController.java similarity index 97% rename from src/main/java/cn/celess/blog/controller/Other.java rename to src/main/java/cn/celess/blog/controller/CommonController.java index 928a71e..39dca2c 100644 --- a/src/main/java/cn/celess/blog/controller/Other.java +++ b/src/main/java/cn/celess/blog/controller/CommonController.java @@ -1,190 +1,190 @@ -package cn.celess.blog.controller; - -import cn.celess.blog.enmu.ResponseEnum; -import cn.celess.blog.entity.Response; -import cn.celess.blog.entity.model.QiniuResponse; -import cn.celess.blog.exception.MyException; -import cn.celess.blog.service.CountService; -import cn.celess.blog.service.QiniuService; -import cn.celess.blog.util.HttpUtil; -import cn.celess.blog.util.RedisUtil; -import cn.celess.blog.util.VeriCodeUtil; -import net.sf.json.JSONArray; -import net.sf.json.JSONObject; -import org.slf4j.Logger; -import org.slf4j.LoggerFactory; -import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.http.MediaType; -import org.springframework.web.bind.annotation.GetMapping; -import org.springframework.web.bind.annotation.PostMapping; -import org.springframework.web.bind.annotation.RequestParam; -import org.springframework.web.bind.annotation.RestController; -import org.springframework.web.multipart.MultipartFile; - -import javax.imageio.ImageIO; -import javax.servlet.http.HttpServletRequest; -import javax.servlet.http.HttpServletResponse; -import java.awt.image.BufferedImage; -import java.io.IOException; -import java.io.OutputStream; -import java.util.Enumeration; -import java.util.HashMap; -import java.util.Map; -import java.util.concurrent.TimeUnit; - -/** - * @author : xiaohai - * @date : 2019/04/02 22:03 - */ -@RestController -public class Other { - public static final Logger logger = LoggerFactory.getLogger(Object.class); - - @Autowired - CountService countService; - @Autowired - QiniuService qiniuService; - @Autowired - RedisUtil redisUtil; - @Autowired - HttpServletRequest request; - - - @GetMapping("/counts") - public Response allCount() { - Map countMap = new HashMap<>(); - countMap.put("articleCount", countService.getArticleCount()); - countMap.put("commentCount", countService.getCommentCount()); - countMap.put("categoryCount", countService.getCategoriesCount()); - countMap.put("tagCount", countService.getTagsCount()); - countMap.put("visitorCount", countService.getVisitorCount()); - return Response.success(countMap); - } - - - /** - * 获取header的全部参数 - * - * @param request HttpServletRequest - * @return Response - */ - @GetMapping("/headerInfo") - public Response headerInfo(HttpServletRequest request) { - Map map = new HashMap<>(); - Enumeration headerNames = request.getHeaderNames(); - String str = null; - while (headerNames.hasMoreElements()) { - str = headerNames.nextElement(); - map.put(str, request.getHeader(str)); - } - map.put("sessionID", request.getSession().getId()); - map.put("request.getRemoteAddr()", request.getRemoteAddr()); - return Response.success(map); - } - - /** - * 返回验证码 - * - * @param response HttpServletResponse - * @throws IOException IOException - */ - @GetMapping(value = "/imgCode", produces = MediaType.IMAGE_PNG_VALUE) - public void getImg(HttpServletResponse response) throws IOException { - Object[] obj = VeriCodeUtil.createImage(); - request.getSession().setAttribute("code", obj[0]); - //将图片输出给浏览器 - BufferedImage image = (BufferedImage) obj[1]; - response.setContentType("image/png"); - OutputStream os = response.getOutputStream(); - ImageIO.write(image, "png", os); - os.close(); - } - - /** - * 验证 验证码的正确性 - * - * @param code 传进来的验证码 - * @param request HttpServletRequest - * @return Session中写入验证状态 - */ - @PostMapping("/verCode") - public Response verCode(@RequestParam("code") String code, HttpServletRequest request) { - request.getSession().setAttribute("verImgCodeStatus", false); - String codeStr = (String) request.getSession().getAttribute("code"); - if (code == null) { - throw new MyException(ResponseEnum.PARAMETERS_ERROR); - } - if (codeStr == null) { - throw new MyException(ResponseEnum.IMG_CODE_TIMEOUT); - } - code = code.toLowerCase(); - codeStr = codeStr.toLowerCase(); - if (code.equals(codeStr)) { - request.getSession().removeAttribute("code"); - request.getSession().setAttribute("verImgCodeStatus", true); - return Response.success("验证成功"); - } else { - request.getSession().removeAttribute("code"); - return Response.failure("验证失败，请重新获取验证码"); - } - } - - /** - * FIXME :: 单张图片多次上传的问题 - * editor.md图片上传的接口 - * FUCK !!! - * - * @param file 文件 - * @return - * @throws IOException - */ - @PostMapping("/imgUpload") - public void upload(HttpServletRequest request, HttpServletResponse response, @RequestParam("editormd-image-file") MultipartFile file) throws IOException { - JSONObject jsonObject = new JSONObject(); - String uploadTimesStr = redisUtil.get(request.getRemoteAddr() + "-ImgUploadTimes"); - int uploadTimes = 0; - if (uploadTimesStr != null) { - uploadTimes = Integer.parseInt(uploadTimesStr); - } - if (uploadTimes == 10) { - throw new MyException(ResponseEnum.FAILURE.getCode(), "上传次数已达10次，请2小时后在上传"); - } - request.setCharacterEncoding("utf-8"); - response.setContentType("text/html"); - if (file.isEmpty()) { - jsonObject.put("success", 0); - jsonObject.put("message", "上传失败，请选择文件"); - response.getWriter().println(jsonObject.toString()); - return; - } - String fileName = file.getOriginalFilename(); - String mime = fileName.substring(fileName.lastIndexOf(".")); - if (".png".equals(mime.toLowerCase()) || ".jpg".equals(mime.toLowerCase()) || - ".jpeg".equals(mime.toLowerCase()) || ".bmp".equals(mime.toLowerCase())) { - QiniuResponse qiniuResponse = qiniuService.uploadFile(file.getInputStream(), "img_" + System.currentTimeMillis() + mime); - jsonObject.put("success", 1); - jsonObject.put("message", "上传成功"); - jsonObject.put("url", "http://cdn.celess.cn/" + qiniuResponse.key); - response.getWriter().println(jsonObject.toString()); - redisUtil.setEx(request.getRemoteAddr() + "-ImgUploadTimes", uploadTimes + 1 + "", 2, TimeUnit.HOURS); - return; - } - jsonObject.put("success", 0); - jsonObject.put("message", "上传失败，请上传图片文件"); - response.getWriter().println(jsonObject.toString()); - } - - @GetMapping("/bingPic") - public Response bingPic() { - - JSONObject imageObj; - try { - imageObj = JSONObject.fromObject(HttpUtil.get("https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=zh-CN")); - } catch (IOException e) { - return Response.failure(null); - } - JSONArray jsonArray = imageObj.getJSONArray("images"); - String imageName = jsonArray.getJSONObject(0).getString("url"); - return Response.success("https://cn.bing.com" + imageName); - } -} +package cn.celess.blog.controller; + +import cn.celess.blog.enmu.ResponseEnum; +import cn.celess.blog.entity.Response; +import cn.celess.blog.entity.model.QiniuResponse; +import cn.celess.blog.exception.MyException; +import cn.celess.blog.service.CountService; +import cn.celess.blog.service.QiniuService; +import cn.celess.blog.util.HttpUtil; +import cn.celess.blog.util.RedisUtil; +import cn.celess.blog.util.VeriCodeUtil; +import net.sf.json.JSONArray; +import net.sf.json.JSONObject; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.http.MediaType; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.multipart.MultipartFile; + +import javax.imageio.ImageIO; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.awt.image.BufferedImage; +import java.io.IOException; +import java.io.OutputStream; +import java.util.Enumeration; +import java.util.HashMap; +import java.util.Map; +import java.util.concurrent.TimeUnit; + +/** + * @author : xiaohai + * @date : 2019/04/02 22:03 + */ +@RestController +public class CommonController { + public static final Logger logger = LoggerFactory.getLogger(Object.class); + + @Autowired + CountService countService; + @Autowired + QiniuService qiniuService; + @Autowired + RedisUtil redisUtil; + @Autowired + HttpServletRequest request; + + + @GetMapping("/counts") + public Response allCount() { + Map countMap = new HashMap<>(); + countMap.put("articleCount", countService.getArticleCount()); + countMap.put("commentCount", countService.getCommentCount()); + countMap.put("categoryCount", countService.getCategoriesCount()); + countMap.put("tagCount", countService.getTagsCount()); + countMap.put("visitorCount", countService.getVisitorCount()); + return Response.success(countMap); + } + + + /** + * 获取header的全部参数 + * + * @param request HttpServletRequest + * @return Response + */ + @GetMapping("/headerInfo") + public Response headerInfo(HttpServletRequest request) { + Map map = new HashMap<>(); + Enumeration headerNames = request.getHeaderNames(); + String str = null; + while (headerNames.hasMoreElements()) { + str = headerNames.nextElement(); + map.put(str, request.getHeader(str)); + } + map.put("sessionID", request.getSession().getId()); + map.put("request.getRemoteAddr()", request.getRemoteAddr()); + return Response.success(map); + } + + /** + * 返回验证码 + * + * @param response HttpServletResponse + * @throws IOException IOException + */ + @GetMapping(value = "/imgCode", produces = MediaType.IMAGE_PNG_VALUE) + public void getImg(HttpServletResponse response) throws IOException { + Object[] obj = VeriCodeUtil.createImage(); + request.getSession().setAttribute("code", obj[0]); + //将图片输出给浏览器 + BufferedImage image = (BufferedImage) obj[1]; + response.setContentType("image/png"); + OutputStream os = response.getOutputStream(); + ImageIO.write(image, "png", os); + os.close(); + } + + /** + * 验证 验证码的正确性 + * + * @param code 传进来的验证码 + * @param request HttpServletRequest + * @return Session中写入验证状态 + */ + @PostMapping("/verCode") + public Response verCode(@RequestParam("code") String code, HttpServletRequest request) { + request.getSession().setAttribute("verImgCodeStatus", false); + String codeStr = (String) request.getSession().getAttribute("code"); + if (code == null) { + throw new MyException(ResponseEnum.PARAMETERS_ERROR); + } + if (codeStr == null) { + throw new MyException(ResponseEnum.IMG_CODE_TIMEOUT); + } + code = code.toLowerCase(); + codeStr = codeStr.toLowerCase(); + if (code.equals(codeStr)) { + request.getSession().removeAttribute("code"); + request.getSession().setAttribute("verImgCodeStatus", true); + return Response.success("验证成功"); + } else { + request.getSession().removeAttribute("code"); + return Response.failure("验证失败，请重新获取验证码"); + } + } + + /** + * FIXME :: 单张图片多次上传的问题 + * editor.md图片上传的接口 + * FUCK !!! + * + * @param file 文件 + * @return + * @throws IOException + */ + @PostMapping("/imgUpload") + public void upload(HttpServletRequest request, HttpServletResponse response, @RequestParam("editormd-image-file") MultipartFile file) throws IOException { + JSONObject jsonObject = new JSONObject(); + String uploadTimesStr = redisUtil.get(request.getRemoteAddr() + "-ImgUploadTimes"); + int uploadTimes = 0; + if (uploadTimesStr != null) { + uploadTimes = Integer.parseInt(uploadTimesStr); + } + if (uploadTimes == 10) { + throw new MyException(ResponseEnum.FAILURE.getCode(), "上传次数已达10次，请2小时后在上传"); + } + request.setCharacterEncoding("utf-8"); + response.setContentType("text/html"); + if (file.isEmpty()) { + jsonObject.put("success", 0); + jsonObject.put("message", "上传失败，请选择文件"); + response.getWriter().println(jsonObject.toString()); + return; + } + String fileName = file.getOriginalFilename(); + String mime = fileName.substring(fileName.lastIndexOf(".")); + if (".png".equals(mime.toLowerCase()) || ".jpg".equals(mime.toLowerCase()) || + ".jpeg".equals(mime.toLowerCase()) || ".bmp".equals(mime.toLowerCase())) { + QiniuResponse qiniuResponse = qiniuService.uploadFile(file.getInputStream(), "img_" + System.currentTimeMillis() + mime); + jsonObject.put("success", 1); + jsonObject.put("message", "上传成功"); + jsonObject.put("url", "http://cdn.celess.cn/" + qiniuResponse.key); + response.getWriter().println(jsonObject.toString()); + redisUtil.setEx(request.getRemoteAddr() + "-ImgUploadTimes", uploadTimes + 1 + "", 2, TimeUnit.HOURS); + return; + } + jsonObject.put("success", 0); + jsonObject.put("message", "上传失败，请上传图片文件"); + response.getWriter().println(jsonObject.toString()); + } + + @GetMapping("/bingPic") + public Response bingPic() { + + JSONObject imageObj; + try { + imageObj = JSONObject.fromObject(HttpUtil.get("https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=zh-CN")); + } catch (IOException e) { + return Response.failure(null); + } + JSONArray jsonArray = imageObj.getJSONArray("images"); + String imageName = jsonArray.getJSONObject(0).getString("url"); + return Response.success("https://cn.bing.com" + imageName); + } +}