服务端与服务端之间的文件上传和下载

末蓝、 2023-02-16 12:03 7阅读 0赞

## 服务端与服务端之间的文件上传和下载 ##

*  HttpClient文件上传、下载工具类
    
        /** * Copyright (C), 2020-06-11 * FileName: HttpClientFileUtil * Author: heyanbo * Date: 2020/6/11 12:33 * Description: HttpClient文件上传和下载 */
        package com.github.code.utils;
        
        import org.apache.http.HttpEntity;
        import org.apache.http.HttpResponse;
        import org.apache.http.client.config.RequestConfig;
        import org.apache.http.client.methods.CloseableHttpResponse;
        import org.apache.http.client.methods.HttpGet;
        import org.apache.http.client.methods.HttpPost;
        import org.apache.http.entity.ContentType;
        import org.apache.http.entity.mime.HttpMultipartMode;
        import org.apache.http.entity.mime.MultipartEntityBuilder;
        import org.apache.http.impl.client.CloseableHttpClient;
        import org.apache.http.impl.client.HttpClients;
        import org.apache.http.util.EntityUtils;
        import org.slf4j.Logger;
        import org.slf4j.LoggerFactory;
        import org.springframework.web.multipart.MultipartFile;
        
        import java.io.File;
        import java.io.FileOutputStream;
        import java.io.IOException;
        import java.io.InputStream;
        import java.nio.charset.Charset;
        import java.util.Map;
        import java.util.Objects;
        
        /** * <功能简要> <br> * <HttpClient文件上传和下载> * * @Author heyanbo * @createTime 2020/6/11 12:33 * @since */
        public class HttpClientFileUtil { 
        
            private static final Logger LOGGER = LoggerFactory.getLogger(HttpClientFileUtil.class);
        
            public static final int CONNECTION_TIMEOUT = 35000;
            public static final int CONNECTION_REQUEST_TIMEOUT = 35000;
            public static final int SOCKET_TIMEOUT = 60000;
            public static final String UTF_8 = "UTF-8";
        
        
            public static String upload(String token, String url, MultipartFile file, String fileParamName, Map<String, String> otherParams){ 
                CloseableHttpClient httpClient = HttpClients.createDefault();
                String result = "";
                String fileName = file.getOriginalFilename();
                try { 
                    HttpPost httpPost = new HttpPost(url);
                    RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(CONNECTION_TIMEOUT).setConnectionRequestTimeout(CONNECTION_REQUEST_TIMEOUT).setSocketTimeout(SOCKET_TIMEOUT).build();
                    httpPost.setConfig(requestConfig);
                    httpPost.setHeader("DataEncoding", UTF_8);
                    httpPost.setHeader("token", token);
                    MultipartEntityBuilder builder = MultipartEntityBuilder.create();
                    builder.setCharset(Charset.forName("utf-8"));
                    //解决返回中文乱码问题
                    builder.setMode(HttpMultipartMode.BROWSER_COMPATIBLE);
                    builder.addBinaryBody("file", file.getInputStream(), ContentType.MULTIPART_FORM_DATA, fileName);
                    //模拟浏览器表单提交
                    for (Map.Entry<String, String> entry : otherParams.entrySet()){ 
                        builder.addTextBody(entry.getKey(), entry.getValue());
                    }
        
                    HttpEntity entity = builder.build();
                    httpPost.setEntity(entity);
                    HttpResponse response = httpClient.execute(httpPost);
                    HttpEntity responseEntity = response.getEntity();
                    if (Objects.nonNull(responseEntity)){ 
                        //将返回值转化为字符串
                        result = EntityUtils.toString(responseEntity, Charset.forName(UTF_8));
                    }
                } catch (Exception e) { 
                    LOGGER.error("上传文件异常，异常信息:"+e);
                }finally { 
                    try { 
                        httpClient.close();
                    }catch (IOException e){ 
                        e.printStackTrace();
                    }
                }
                return result;
            }
        
            public static void downLoad(String url, Map<String, String> map, String remoteFileName, String localFileName){ 
                CloseableHttpClient httpClient = HttpClients.createDefault();
                map.put("fileName", remoteFileName);
                String downUrl = getUrl(url, map);
                String fileUrl = downUrl.replace("downLoad=false", "downLoad=true");
                LOGGER.info("downLoad url is :"+fileUrl);
                HttpGet httpGet = new HttpGet(fileUrl);
                httpGet.setHeader("fileName", remoteFileName);
        
                RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(CONNECTION_TIMEOUT).setConnectionRequestTimeout(CONNECTION_REQUEST_TIMEOUT).setSocketTimeout(SOCKET_TIMEOUT).build();
                httpGet.setConfig(requestConfig);
                CloseableHttpResponse httpResponse = null;
                FileOutputStream outputStream = null;
                InputStream inputStream = null;
        
                try { 
                    httpResponse = httpClient.execute(httpGet);
                    HttpEntity entity = httpResponse.getEntity();
                    File storeFile = new File(localFileName);
                    inputStream = httpResponse.getEntity().getContent();
        
                    byte[] buff = new byte[30*1024*1024];
                    int len = 0;
                    while ((len = inputStream.read(buff)) > 0){ 
                        outputStream.write(buff, 0, len);
                        outputStream.flush();
                    }
                } catch (Exception e) { 
                    LOGGER.error("get下载文件请求错误："+e);
                }finally { 
                    if (inputStream != null){ 
                        try { 
                            inputStream.close();
                        }catch (IOException e){ 
                            LOGGER.error("输入流关闭失败:"+e);
                        }
                    }
                    if (outputStream != null){ 
                        try { 
                            outputStream.close();
                        }catch (IOException e){ 
                            LOGGER.error("输出流关闭失败:"+e);
                        }
                    }
        
                    if (httpResponse != null){ 
                        try { 
                            httpResponse.close();
                        } catch (IOException e) { 
                            LOGGER.error("httpResponse关闭失败："+e);
                        }
                    }
                    if (httpClient != null){ 
                        try { 
                            httpClient.close();
                        }catch (IOException e){ 
                            LOGGER.error("httpClient连接关闭失败："+e);
                        }
                    }
                }
        
            }
        
            public static String getUrl(String url, Map<String, String> map){ 
                int i=0;
                for (Map.Entry<String, String> entry:map.entrySet()){ 
                    if (i==0){ 
                        url+= "?"+entry.getKey()+"="+entry.getValue();
                    }else { 
                        url+= "&"+entry.getKey()+"="+entry.getValue();
                    }
                    i++;
                }
                return url;
            }
        }
 *  文件上传服务端接口示例
    
        @PostMapping("/upload")
            @ResponseBody
            public String fileUpload(@RequestParam(value = "file") MultipartFile file,
                                     @RequestParam(value = "tenantName")String tenantName,
                                     @RequestParam(value = "chunk")Integer chunk,
                                     @RequestParam(value = "chunks")Integer chunks){ 
                if (Objects.isNull(file)){ 
                    return "文件上传失败，请重新选择文件";
                }
                return fileManageService.upload(file);
            }
 *  文件下载服务端接口示例
    
     *  Controller接口
        
            @GetMapping("/download")
                @ResponseBody
                public String fileDownload(@RequestParam(value = "fileName") String fileName, HttpServletResponse response){ 
                    if (Objects.isNull(fileName)){ 
                        return "文件下载失败，请选择文件要下载的文件";
                    }
                    return fileManageService.download(fileName, response);
                }
     *  serviceImpl实现类
        
            @Override
                public void download(String fileName, HttpServletResponse response)throws CustomException{ 
                    if (StringUtils.isBlank(fileName)){ 
                        throw new CustomException("参数不能为空");
                    }
            
                    File file = new File(saveAddr + fileName);
                    if (!file.exists()){ 
                        throw new CustomException("文件不存在");
                    }
            
                    BufferedInputStream bufferedInputStream = null;
                    OutputStream outputStream = null;
                    FileInputStream fileInputStream = null;
                    try { 
                        fileInputStream = new FileInputStream(file);
                        bufferedInputStream = new BufferedInputStream(fileInputStream);
                        response.setCharacterEncoding("utf-8");
                        response.setContentType("application/octet-stream");
                        response.setContentLength((int)file.length());
                        response.setHeader("Accept-Ranges", fileName);
                        response.addHeader("fileName", fileName);
                        outputStream = response.getOutputStream();
                        int len;
                        byte[] buffer = new byte[2048];
                        while ((len = bufferedInputStream.read(buffer)) != -1){ 
                            outputStream.write(buffer, 0, len);
                        }
                        outputStream.flush();
                    }catch (Exception e) { 
                        LOGGER.error("文件下载失败", e.getMessage());
                    }finally { 
                        try { 
                            if (bufferedInputStream != null){ 
                                bufferedInputStream.close();
                            }
                            if (outputStream !=null){ 
                                outputStream.close();
                            }
                            if (fileInputStream != null){ 
                                fileInputStream.close();
                            }
                        }catch (IOException e){ 
                            LOGGER.error("文件流关闭失败", e.getMessage());
                        }
                        file.delete();
                    }
                }