center/dataease-dm
center/dataease-dm
13
0
Fork 0
forked from github/dataease
Code Pull Requests Activity

fix(X-Pack): 飞书发送定时报告内容包含远程图片会报错

Browse Source
This commit is contained in:
fit2cloud-chenyw 2024-04-29 11:30:38 +08:00
parent 382eec1f77
commit c83076bbd7
1 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
sdk/dataease-plugin-common/src/main/java/io/dataease/plugins/common/util/HttpClientUtil.java
View File

@ -30,7 +30,9 @@ import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.net.ProxySelector;
import java.nio.charset.StandardCharsets;
import java.security.cert.X509Certificate;
@ -285,4 +287,37 @@ public class HttpClientUtil {
throw new RuntimeException("HttpClient查询失败: " + e.getMessage());
}
}
public static byte[] downFromRemote(String url, HttpClientConfig config) {
HttpGet httpGet = new HttpGet(url);
CloseableHttpClient httpClient = buildHttpClient(url);
try {
httpGet.setConfig(config.buildRequestConfig());
Map<String, String> header = config.getHeader();
for (String key : header.keySet()) {
httpGet.addHeader(key, header.get(key));
}
HttpResponse response = httpClient.execute(httpGet);
InputStream inputStream = response.getEntity().getContent();
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
byte[] buffer = new byte[1024];
int bytesRead;
while ((bytesRead = inputStream.read(buffer)) != -1) {
outputStream.write(buffer, 0, bytesRead);
}
return outputStream.toByteArray();
} catch (Exception e) {
logger.error("HttpClient查询失败", e);
throw new RuntimeException("HttpClient查询失败: " + e.getMessage());
} finally {
try {
httpClient.close();
} catch (Exception e) {
logger.error("HttpClient关闭连接失败", e);
}
}
}
}