218 lines
8.4 KiB
Java
218 lines
8.4 KiB
Java
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.RedisUtil;
|
||
import cn.celess.blog.util.ResponseUtil;
|
||
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.*;
|
||
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.*;
|
||
import java.net.HttpURLConnection;
|
||
import java.net.URL;
|
||
import java.nio.charset.StandardCharsets;
|
||
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<String, Long> countMap = new HashMap<>();
|
||
countMap.put("articleCount", countService.getArticleCount());
|
||
countMap.put("commentCount", countService.getCommentCount());
|
||
countMap.put("leaveMsgCount", countService.getLeaveMessageCount());
|
||
countMap.put("categoryCount", countService.getCategoriesCount());
|
||
countMap.put("tagCount", countService.getTagsCount());
|
||
countMap.put("visitorCount", countService.getVisitorCount());
|
||
return ResponseUtil.success(countMap);
|
||
}
|
||
|
||
|
||
/**
|
||
* 获取header的全部参数
|
||
*
|
||
* @param request HttpServletRequest
|
||
* @return Response
|
||
*/
|
||
@GetMapping("/headerInfo")
|
||
public Response headerInfo(HttpServletRequest request) {
|
||
Map<String, Object> map = new HashMap<>();
|
||
Enumeration<String> 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 ResponseUtil.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 ResponseUtil.success("验证成功");
|
||
} else {
|
||
request.getSession().removeAttribute("code");
|
||
return ResponseUtil.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", 0);
|
||
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() {
|
||
StringBuffer sb = new StringBuffer();
|
||
try {
|
||
//建立URL
|
||
URL url = new URL("https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=zh-CN");
|
||
|
||
//打开http
|
||
HttpURLConnection urlConnection = (HttpURLConnection) url.openConnection();
|
||
urlConnection.setDoInput(true);
|
||
urlConnection.setRequestMethod("GET");
|
||
urlConnection.connect();
|
||
|
||
//获得输入
|
||
InputStream inputStream = urlConnection.getInputStream();
|
||
InputStreamReader inputStreamReader = new InputStreamReader(inputStream, StandardCharsets.UTF_8);
|
||
BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
|
||
|
||
//将bufferReader的值给放到buffer里
|
||
String str = null;
|
||
while ((str = bufferedReader.readLine()) != null) {
|
||
sb.append(str);
|
||
}
|
||
//关闭bufferReader和输入流
|
||
bufferedReader.close();
|
||
inputStreamReader.close();
|
||
inputStream.close();
|
||
//断开连接
|
||
urlConnection.disconnect();
|
||
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
return ResponseUtil.failure(null);
|
||
}
|
||
|
||
JSONObject imageObj = JSONObject.fromObject(sb.toString());
|
||
JSONArray jsonArray = imageObj.getJSONArray("images");
|
||
String imageName = jsonArray.getJSONObject(0).getString("url");
|
||
return ResponseUtil.success("https://cn.bing.com" + imageName);
|
||
}
|
||
}
|