package com.mm.controller;
|
|
import cn.hutool.crypto.SmUtil;
|
import com.mm.util.*;
|
import com.mm.vo.*;
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.tomcat.util.http.fileupload.IOUtils;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.method.HandlerMethod;
|
|
import javax.servlet.ServletInputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.InetAddress;
|
import java.net.URLDecoder;
|
import java.net.UnknownHostException;
|
|
|
@RestController
|
@RequestMapping("/sys/api")
|
public class ServiceMdcController {
|
|
@Value("${giveServiceTo.appIdCheck}")
|
private String appIdCheck;
|
|
@Value("${giveServiceTo.passwordCheck}")
|
private String passwordCheck;
|
@Value("${giveServiceTo.logIp}")
|
private String logIp;
|
@Value("${giveServiceTo.logPort}")
|
private String logPort;
|
|
@Value("${giveServiceTo.serverIp}")
|
private String serverIp;
|
|
@Value("${giveServiceTo.clientIp}")
|
private String clientIp;
|
@Value("${fileNumPath}")
|
private String fileNumPath;
|
|
|
@PostMapping(value = "/appAuth")
|
public TokenResp getToken(@RequestBody PostParams postParams) {
|
|
//获取请求头中传入的appId和password
|
String appId = postParams.getAppId();
|
String password = postParams.getPassword();
|
|
//用于校验的正确的appId和password（根据业务调整）
|
/*String appIdCheck = "admin";
|
String passwordCheck = "123";*/
|
|
if (appIdCheck.equals(appId)) {
|
//校验通过，生成token签名
|
if (SM3Util.verify(passwordCheck, password)) {
|
String token = JwTUtil.sign(appId, passwordCheck);
|
return new TokenResp("200", "认证成功", token);
|
} else {
|
return new TokenResp("101", "密码错误", null);
|
}
|
} else {
|
return new TokenResp("101", "账号错误", null);
|
}
|
|
// //校验appId和paaword，由于测试用的明文密码,SM3Util.verify()先将铭文密码SM3加密后再与传入password对比
|
// if (appIdCheck.equals(appId) && SM3Util.verify(passwordCheck, password)) {
|
// //校验通过，生成token签名
|
// String token = JwTUtil.sign(appId, passwordCheck);
|
// return new TokenResp("200", "认证成功", token);
|
// } else {
|
// //to-do认证失败
|
// return null;
|
// }
|
}
|
|
// public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
|
// //如果不是映射到方法直接通过
|
// if (!(handler instanceof HandlerMethod)) {
|
// return true;
|
// }
|
// //从 http 请求头中取出 token
|
// String token = request.getHeader("Token");
|
// if (token == null) {
|
// throw new RuntimeException("无 token ，请重新登陆");
|
// }
|
// /*//验证 token，JwtUtil为生成token和验证token的工具类
|
// JwTUtil.checkSign(token);
|
//
|
// //获取 token 中的 appId
|
// String appId = JwTUtil.getUserId(token);
|
// //获取 token 中的其他数据
|
// Map<String, Object> info = JwTUtil.getInfo(token);
|
// info.forEach((k, v) -> System.out.println(k + ":" + v));*/
|
// return true;
|
// }
|
|
@PostMapping(value = "/fileUpload")
|
public RespData uploadFiles(HttpServletRequest request) {
|
SysLogTypeObjectDto objectName = new SysLogTypeObjectDto();
|
objectName.setDateTime(DateUtil.format(DateUtil.getNow(), DateUtil.STR_DATE_TIME_SMALL));
|
|
FileDetail fileDetail = new FileDetail();
|
try {
|
request.setCharacterEncoding("UTF-8");
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
//获取FileName
|
if (request.getHeader("FileName") != null) {
|
try {
|
fileDetail.setFileName(URLDecoder.decode(request.getHeader("FileName"), "UTF-8"));
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
}
|
//获取Content-Length
|
if (request.getHeader("Content-Length") != null) {
|
try {
|
fileDetail.setContentLength(URLDecoder.decode(request.getHeader("Content-Length"), "UTF-8"));
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
}
|
//获取文件流
|
ServletInputStream inputStream = null;
|
try {
|
inputStream = request.getInputStream();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
|
if (inputStream == null) {
|
System.out.println("inputStream为空");
|
}
|
|
//操作文件流，上传文件
|
//fileService.uploadFile(inputStream,fileDetail.getFileName());
|
//DataInputStream in = new DataInputStream(inputStream);
|
FileOutputStream fileOut = null;
|
try {
|
fileOut = new FileOutputStream(fileDetail.getFileName());//这里可以改路径
|
IOUtils.copy(inputStream, fileOut);
|
fileOut.flush();
|
// fileOut.flush();
|
objectName.setResult("成功");
|
|
File file = new File(fileDetail.getFileName());
|
objectName.setFileSize(file.getName());
|
objectName.setDateTime(DateUtil.format(DateUtil.getNow(), DateUtil.STR_DATE_TIME_SMALL));
|
objectName.setFileSize(FileUtil.changeFileFormatKb(String.valueOf(file.length())));
|
objectName.setDestination(serverIp);//服务端IP
|
objectName.setSourceAddress(clientIp);//客户端
|
String sm3 = SmUtil.sm3(file);
|
objectName.setAbstract1(sm3);
|
|
//不记录数据量
|
|
String fileNum = FileNumUtil.readFileSum(fileNumPath);
|
if (StringUtils.isBlank(fileNum)) {
|
objectName.setFileNum(Integer.toString(1));
|
FileNumUtil.fileWriterSql(Integer.toString(1), fileNumPath);
|
} else {
|
Integer num = Integer.valueOf(fileNum) + 1;
|
objectName.setFileNum(Integer.toString(num));
|
FileNumUtil.fileWriterSql(Integer.toString(num), fileNumPath);
|
}
|
InetAddress address = null;
|
try {
|
address = InetAddress.getLocalHost();
|
String ip = address.getHostAddress();
|
objectName.setAddress(ip);
|
} catch (UnknownHostException e) {
|
objectName.setAddress("20.10.17.19");
|
}
|
try {
|
objectName.setTypes("Info");
|
SyslogClient.sendClient(logIp, Integer.valueOf(logPort), objectName.toString());
|
} catch (Exception e) {
|
|
}
|
|
} catch (IOException e) {
|
objectName.setResult("失败");
|
objectName.setTypes("error");
|
SyslogClient.sendClient(logIp, Integer.valueOf(logPort), objectName.toString());
|
e.printStackTrace();
|
} finally {
|
try {
|
if (inputStream != null) {
|
inputStream.close();
|
}
|
if (fileOut != null) {
|
fileOut.close();
|
}
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
return new RespData("200", "文件接收成功");
|
}
|
}
|
