violet/ensign-crm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wip:sync image

Browse Source
This commit is contained in:
zzs 2024-09-27 18:33:21 +08:00
parent 99661122ab
commit a9f7feead3
3 changed files with 179 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
ensign-module-crm/ensign-module-crm-biz/src/main/java/com/ensign/crm/module/crm/controller/crm/ProxyController.java
View File

@ -1,5 +1,7 @@
package com.ensign.crm.module.crm.controller.crm;
import com.ensign.crm.framework.common.pojo.CommonResult;
import com.ensign.crm.module.crm.exception.AllKingdeeException;
import com.ensign.crm.module.crm.service.ProxyService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
@ -8,8 +10,8 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@ -39,4 +41,17 @@ public class ProxyController {
proxyService.doProxy(request, response);
}
@PostMapping(value = "/file/do/**")
@Operation(summary = "转发接口_文件")
@PreAuthorize("@ss.hasPermission('crm:proxy:all')")
public CommonResult<Object> proxyFile(HttpServletRequest request, MultipartFile file) throws IOException, URISyntaxException, AllKingdeeException {
return CommonResult.success(proxyService.doProxyFile(request, file));
}
@GetMapping("/file/read/do")
@PreAuthorize("@ss.hasPermission('crm:proxy:all')")
@Operation(summary = "获取图片")
public byte[] proxyRead(@RequestParam String fileId, @RequestParam boolean isTemp) throws IOException, URISyntaxException, AllKingdeeException {
return proxyService.dpGetImage(fileId, isTemp);
}
}

56
ensign-module-crm/ensign-module-crm-biz/src/main/java/com/ensign/crm/module/crm/service/ImageService.java Normal file
View File

@ -0,0 +1,56 @@
package com.ensign.crm.module.crm.service;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
/**
* @Description: TODO
* @Date: 2024/9/27 15:09
* @Created: by ZZSLL
*/
@Service
@Slf4j
public class ImageService {
@Resource
private RedisTemplate<String, byte[]> redisTemplate;
private final String CACHE_PREFIX = "kingdee:temp_file";
public byte[] downloadImage(String imageUrl) throws IOException {
CloseableHttpClient httpClient = HttpClients.createDefault();
HttpGet httpGet = new HttpGet(imageUrl);
CloseableHttpResponse response = httpClient.execute(httpGet);
try {
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
response.getEntity().writeTo(outputStream);
return outputStream.toByteArray();
} finally {
response.close();
httpClient.close();
}
}
public void saveImage(String key, byte[] imageData) {
redisTemplate.opsForValue().set(CACHE_PREFIX + key, imageData, 2, TimeUnit.HOURS);
}
public byte[] getImage(String key) {
return redisTemplate.opsForValue().get(CACHE_PREFIX + key);
}
}

124
ensign-module-crm/ensign-module-crm-biz/src/main/java/com/ensign/crm/module/crm/service/ProxyService.java
View File

@ -14,9 +14,13 @@ import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.client.methods.RequestBuilder;
import org.apache.http.entity.StringEntity;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.http.HttpMethod;
@ -25,10 +29,13 @@ import org.springframework.http.client.ClientHttpResponse;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.stereotype.Service;
import org.springframework.util.StreamUtils;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
@ -68,12 +75,34 @@ public class ProxyService {
@Resource
private RedisTemplate<String, String> redisTemplate;
@Autowired
private ImageService imageService;
public void doProxy(HttpServletRequest request, HttpServletResponse response) throws IOException, URISyntaxException {
URI uri = new URI(request.getRequestURI());
String path = uri.getPath();
String query = request.getQueryString();
String target = initBasePath() + path.replace("/crm-api/proxy/do", "");
String accessToken;
try {
accessToken = initAccessToken();
log.info("Header: access_token: {}", accessToken);
} catch (AllKingdeeException e) {
log.error("获取accessToken时失败：{}", e.getMessage());
log.info("================request kingdee end================");
response.setStatus(HttpServletResponse.SC_OK);
response.setContentType("application/json");
response.setCharacterEncoding("UTF-8");
Map<String, Object> body = new HashMap<>();
body.put("status", false);
body.put("message", e.getMessage());
response.getWriter().println(JSONUtil.toJsonStr(body));
response.getWriter().flush();
return;
}
if (query != null && !query.isEmpty() && !query.equals("null")) {
target = target + "?" + query;
}
@ -102,25 +131,17 @@ public class ProxyService {
}
log.info("Header: {}: {}", headerName, arr);
delegate.getHeaders().addAll(headerName, arr);
}
try {
String accessToken = initAccessToken();
delegate.getHeaders().add("access_token", accessToken);
log.info("Header: access_token: {}", accessToken);
} catch (AllKingdeeException e) {
log.error("获取accessToken时失败：{}", e.getMessage());
log.info("================request kingdee end================");
response.setStatus(HttpServletResponse.SC_OK);
response.setContentType("application/json");
response.setCharacterEncoding("UTF-8");
Map<String, Object> body = new HashMap<>();
body.put("status", false);
body.put("message", e.getMessage());
response.getWriter().println(JSONUtil.toJsonStr(body));
response.getWriter().flush();
return;
}
StreamUtils.copy(request.getInputStream(), delegate.getBody());
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
byte[] buffer = new byte[4096];
int bytesRead;
while ((bytesRead = request.getInputStream().read(buffer)) != -1) {
byteArrayOutputStream.write(buffer, 0, bytesRead);
}
byte[] bytes = byteArrayOutputStream.toByteArray();
delegate.getBody().write(bytes);
// 执行远程调用
ClientHttpResponse clientHttpResponse = delegate.execute();
log.info("Response Status: {}", clientHttpResponse.getStatusCode().value());
@ -137,7 +158,60 @@ public class ProxyService {
StreamUtils.copy(clientHttpResponse.getBody(), response.getOutputStream());
}
public com.alibaba.fastjson.JSONObject doRequest(String endpoint, String methodName, Map<String, Object> body, Map<String, Object> header) throws AllKingdeeException, IOException, URISyntaxException {
public String doProxyFile(HttpServletRequest request, MultipartFile file) throws IOException, URISyntaxException, AllKingdeeException {
URI uri = new URI(request.getRequestURI());
String path = uri.getPath();
String target = initBasePath() + path.replace("/crm-api/proxy/file/do", "");
String accessToken = initAccessToken();
target = target + "?access_token=" + accessToken;
URI newUri = new URI(target);
CloseableHttpClient httpClient = HttpClients.createDefault();
HttpPost httpPost = new HttpPost(newUri);
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
builder.addPart("file", new FileBody(transferToFile(file)));
HttpEntity multipartEntity = builder.build();
httpPost.setEntity(multipartEntity);
HttpResponse kResponse = httpClient.execute(httpPost);
HttpEntity entity = kResponse.getEntity();
String jsonStr = EntityUtils.toString(entity, "UTF-8").trim();
JSONObject jsonObject = JSON.parseObject(jsonStr);
if (jsonObject.containsKey("url")) {
String url = jsonObject.getString("url");
String fileId = extractFileId(url);
String downloadUrl = url + "&access_token=" + initAccessToken();
byte[] bytes = imageService.downloadImage(downloadUrl);
imageService.saveImage(fileId, bytes);
return doConvertFileUrl(fileId, true);
}
return null;
}
private String extractFileId(String url) throws AllKingdeeException, URISyntaxException {
URI uri = new URI(url);
String query = uri.getQuery();
if (query != null) {
String[] queryParams = query.split("&");
for (String param : queryParams) {
String[] keyValue = param.split("=");
if (keyValue.length == 2 && keyValue[0].equals("id")) {
return keyValue[1];
}
}
}
throw new AllKingdeeException("获取file失败");
}
public File transferToFile(MultipartFile multipartFile) throws IOException {
String originalFilename = multipartFile.getOriginalFilename();
assert originalFilename != null;
String[] filename = originalFilename.split("\\.");
File file = File.createTempFile(filename[0], "." + filename[1]);
multipartFile.transferTo(file);
file.deleteOnExit();
return file;
}
public JSONObject doRequest(String endpoint, String methodName, Map<String, Object> body, Map<String, Object> header) throws AllKingdeeException, IOException, URISyntaxException {
log.info("================request kingdee start================");
HttpClient httpClient = HttpClients.createDefault();
@ -296,4 +370,18 @@ public class ProxyService {
return "";
}
public String doConvertFileUrl(String fileId, boolean isTemp) {
if (isTemp) {
return "/crm-api/proxy/file/read/do?fileId=" + fileId + "&isTemp=true";
}
return "";
}
public byte[] dpGetImage(String fileId, boolean isTemp) {
if (isTemp) {
return imageService.getImage(fileId);
}
return null;
}
}