package org.jeecg.modules.dnc.utils;
|
|
import cn.hutool.crypto.SmUtil;
|
import com.alibaba.fastjson.JSONObject;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.http.HttpEntity;
|
import org.apache.http.client.config.RequestConfig;
|
import org.apache.http.client.methods.CloseableHttpResponse;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.FileEntity;
|
import org.apache.http.entity.StringEntity;
|
import org.apache.http.impl.client.CloseableHttpClient;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.http.protocol.HTTP;
|
import org.apache.http.util.EntityUtils;
|
import org.jeecg.modules.dnc.utils.file.SM3Util;
|
import org.jeecg.modules.dnc.dto.SysLogTypeObjectDto;
|
|
import java.io.File;
|
import java.net.URLEncoder;
|
|
/**
|
* @author clown
|
* * @date 2023/11/14
|
*/
|
@Slf4j
|
public class FileClient {
|
|
public static String getToken(String host, String port, String userName, String pwd,String addressToken) throws Exception, Throwable {
|
CloseableHttpClient client = null;
|
CloseableHttpResponse response = null;
|
try {
|
ObjectMapper objectMapper = new ObjectMapper();
|
JSONObject jsonObject = new JSONObject();
|
//构造请求参数appId和password
|
jsonObject.put("appId", userName);
|
//将密码通过SM3加密
|
jsonObject.put("password", SM3Util.encrypt(pwd));
|
String str = objectMapper.writeValueAsString(jsonObject);
|
StringEntity stringEntity = new StringEntity(str);
|
//构造http请求
|
String url = "http" + "://" + host + ":" + port + "/" + addressToken;
|
HttpPost httpPost = new HttpPost(url);
|
//设置Content-Type
|
httpPost.setHeader(HTTP.CONTENT_TYPE, "application/json");
|
//Content-length会在请求自动自动加上
|
//httpPost.setHeader(HTTP.CONTENT_LEN, stringEntity.getContentLength()+"");
|
//将构造好的参数放入请求中
|
httpPost.setEntity(stringEntity);
|
//设置请求超时时间
|
RequestConfig requestConfig = RequestConfig.custom()
|
.setConnectTimeout(5000).setConnectionRequestTimeout(5000)
|
.setSocketTimeout(5000).build();
|
httpPost.setConfig(requestConfig);
|
//发起请求
|
client = HttpClients.createDefault();
|
response = client.execute(httpPost);
|
//解析请求的response
|
if (response.getStatusLine().getStatusCode() == 200) {
|
HttpEntity entity = response.getEntity();
|
String result = EntityUtils.toString(entity, "UTF-8");
|
JSONObject jobject = JSONObject.parseObject(result);
|
int code = jobject.getIntValue("code");
|
String token = jobject.getString("token");
|
String message = jobject.getString("msg");
|
if (200 != code) {
|
throw new Exception("appAuth Error,code[" + code + "],message[" + message + "]");
|
} else {
|
//若请求成功，返回token
|
return token;
|
}
|
} else {
|
throw new Exception("appAuth Error:" + response.getStatusLine().toString());
|
}
|
} catch (Throwable e) {
|
throw e;
|
} finally {
|
if (response != null) {
|
try {
|
response.close();
|
} catch (Throwable t) {
|
}
|
}
|
if (client != null) {
|
try {
|
client.close();
|
} catch (Throwable t) {
|
}
|
}
|
}
|
}
|
|
|
/* public static void main(String[] args) {
|
try {
|
String token = getToken("127.0.0.1","8299","admin","123");
|
} catch (Throwable throwable) {
|
throwable.printStackTrace();
|
}
|
}*/
|
|
/**
|
*
|
* @param host
|
* @param port
|
* @param token
|
* @param remoteFilePath
|
* @param localFilePath
|
* @param addressUploadFile
|
* @return
|
* @throws Exception
|
* @throws Throwable
|
*/
|
public static String uploadFile(String host, String port, String token, String fileName,String remoteFilePath, String localFilePath,String addressUploadFile) throws Exception, Throwable {
|
CloseableHttpClient client = null;
|
CloseableHttpResponse response = null;
|
SysLogTypeObjectDto objectName = new SysLogTypeObjectDto();
|
objectName.setResult("失败");
|
try {
|
//拼接请求的url
|
String url = "http" + "://" + host + ":" + port + "/" + addressUploadFile;
|
//构造请求
|
HttpPost httpPost = new HttpPost(url);
|
//设置Content-Type为文件流格式
|
httpPost.setHeader(HTTP.CONTENT_TYPE, "application/octet-stream");
|
//设置header的请求参数
|
httpPost.addHeader("Accept", "*/*");
|
httpPost.addHeader("Accept-Encoding", "UTF-8");
|
httpPost.setHeader("Token", token);
|
//FileName需要经过URLEncoder加码，防止中文乱码
|
httpPost.setHeader("FileName", URLEncoder.encode(remoteFilePath, "UTF-8"));
|
/*文件大小*/
|
httpPost.setHeader("WenjianqIcaoren", "徐业植");//文件起草人
|
httpPost.setHeader("Shenpiren", "张世贵");//审批人
|
httpPost.setHeader("Miji", URLEncoder.encode("内部", "UTF-8"));
|
httpPost.addHeader("Baomiqixian","10,12,30");
|
httpPost.addHeader("Qianfaren", URLEncoder.encode("张世贵", "UTF-8"));//签发人
|
File file = new File(localFilePath);
|
if (file == null || !file.exists()){
|
httpPost.addHeader("Wenjiansanliezhi", URLEncoder.encode("sm3", "UTF-8"));
|
} else {
|
String sm3 = SmUtil.sm3(file);
|
httpPost.addHeader("Wenjiansanliezhi", URLEncoder.encode(sm3, "UTF-8"));
|
}
|
|
httpPost.addHeader("Beizhu", URLEncoder.encode("无", "UTF-8"));
|
//将对应上传的本地文件解析成文件流放入body
|
httpPost.setEntity(new FileEntity(new File(localFilePath)));
|
//根据文件大小设置超时时间
|
int timeout = (int) ((new File(localFilePath).length() / (1000)) * 2 + 2000);
|
//设置超时时间
|
RequestConfig requestConfig = RequestConfig.custom()
|
.setConnectTimeout(5000).setConnectionRequestTimeout(5000)
|
.setSocketTimeout(timeout).build();
|
httpPost.setConfig(requestConfig);
|
//请求构造完成，发起请求
|
client = HttpClients.createDefault();
|
response = client.execute(httpPost);
|
//解析response
|
if (response.getStatusLine().getStatusCode() == 200) {
|
HttpEntity entity = response.getEntity();
|
System.out.println(entity);
|
String result = EntityUtils.toString(entity, "UTF-8");
|
JSONObject jobject = JSONObject.parseObject(result);
|
int code = jobject.getIntValue("code");
|
String message = jobject.getString("msg");
|
if (200 != code) {
|
//throw new Exception("uploadFile Error,code[" + code + "],message[" + message + "]");
|
log.error( "code: "+ code +" message: " + message);
|
if (response != null) {
|
try {
|
response.close();
|
} catch (Throwable t) {
|
}
|
}
|
if (client != null) {
|
try {
|
client.close();
|
} catch (Throwable t) {
|
}
|
}
|
return "失败";
|
}
|
return "成功";
|
} else {
|
//throw new Exception("uploadFile Error:" + response.getStatusLine().toString());
|
log.error( "uploadFile Error:" + response.getStatusLine().toString());
|
objectName.setResult(response.getStatusLine().toString());
|
if (response != null) {
|
try {
|
response.close();
|
} catch (Throwable t) {
|
}
|
}
|
if (client != null) {
|
try {
|
client.close();
|
} catch (Throwable t) {
|
}
|
}
|
return "失败";
|
}
|
} catch (Throwable e) {
|
log.error( e.getMessage());
|
if (response != null) {
|
try {
|
response.close();
|
} catch (Throwable t) {
|
}
|
}
|
if (client != null) {
|
try {
|
client.close();
|
} catch (Throwable t) {
|
}
|
}
|
return "失败";
|
} finally {
|
|
if (response != null) {
|
try {
|
response.close();
|
} catch (Throwable t) {
|
}
|
}
|
if (client != null) {
|
try {
|
client.close();
|
} catch (Throwable t) {
|
}
|
}
|
return "成功";
|
}
|
}
|
|
}
|
