httpClient文件上传formdata

不念不忘少年蓝@ 2023-03-02 14:49 21阅读 0赞

/**
         * 上传文件
         *
         * @param urlStr
         * @param textMap
         * @param fileMap
         * @param contentType 没有传入文件类型默认采用application/octet-stream
         *                    contentType非空采用filename匹配默认的图片类型
         * @return 返回response数据
         */
        public static String formUploadFile(String urlStr, Map<String, String> textMap,
                                            Map<String, File> fileMap, String contentType) {
            String res = "";
            HttpURLConnection conn = null;
            // boundary就是request头和上传文件内容的分隔符
            String BOUNDARY = "-----------YTKJ|FILE|BOUNDARY--------";
            try {
                URL url = new URL(urlStr);
                conn = (HttpURLConnection) url.openConnection();
                conn.setConnectTimeout(5000);
                conn.setReadTimeout(30000);
                conn.setDoOutput(true);
                conn.setDoInput(true);
                conn.setUseCaches(false);
                conn.setRequestMethod("POST");
                conn.setRequestProperty("Connection", "Keep-Alive");
                conn.setRequestProperty("User-Agent",
                        "Mozilla/5.0 (Windows; U; Windows NT 6.1; zh-CN; rv:1.9.2.6)");
                conn.setRequestProperty("Content-Type",
                        "multipart/form-data; boundary=" + BOUNDARY);
                String AccountPassword = "admin:admin";
                String basicAuth = "Basic " + DatatypeConverter.printBase64Binary(AccountPassword.getBytes());
                conn.setRequestProperty("Authorization", basicAuth);
                // post请求缓存设为false
                conn.setRequestProperty("Cache-Control", "no-cache");
                // 设置该HttpURLConnection实例是否自动执行重定向
                conn.setInstanceFollowRedirects(true);
                OutputStream out = new DataOutputStream(conn.getOutputStream());
                // text
                if (textMap != null) {
                    StringBuffer strBuf = new StringBuffer();
                    Iterator iter = textMap.entrySet().iterator();
                    while (iter.hasNext()) {
                        Map.Entry entry = (Map.Entry) iter.next();
                        String inputName = (String) entry.getKey();
                        String inputValue = (String) entry.getValue();
                        if (inputValue == null) {
                            continue;
                        }
                        strBuf.append("\r\n").append("--").append(BOUNDARY)
                                .append("\r\n");
                        strBuf.append("Content-Disposition: form-data; name=\""
                                + inputName + "\"\r\n\r\n");
                        strBuf.append(inputValue);
                    }
                    out.write(strBuf.toString().getBytes());
                }
                // file
                if (fileMap != null) {
                    Iterator iter = fileMap.entrySet().iterator();
                    while (iter.hasNext()) {
                        Map.Entry entry = (Map.Entry) iter.next();
                        String inputName = (String) entry.getKey();
                        File inputValue = (File) entry.getValue();
                        if (inputValue == null) {
                            continue;
                        }
                        File file = inputValue;
                        String filename = file.getName();
    
                        //没有传入文件类型，同时根据文件获取不到类型，默认采用application/octet-stream
                        contentType = new MimetypesFileTypeMap().getContentType(file);
                        //contentType非空采用filename匹配默认的图片类型
                        if (!"".equals(contentType)) {
                            if (filename.endsWith(".png")) {
                                contentType = "image/png";
                            } else if (filename.endsWith(".jpg") || filename.endsWith(".jpeg") || filename.endsWith(".jpe")) {
                                contentType = "image/jpeg";
                            } else if (filename.endsWith(".gif")) {
                                contentType = "image/gif";
                            } else if (filename.endsWith(".ico")) {
                                contentType = "image/image/x-icon";
                            }
                        }
                        if (contentType == null || "".equals(contentType)) {
                            contentType = "application/octet-stream";
                        }
                        StringBuffer strBuf = new StringBuffer();
                        strBuf.append("\r\n").append("--").append(BOUNDARY)
                                .append("\r\n");
                        strBuf.append("Content-Disposition: form-data; name=\""
                                + inputName + "\"; filename=\"" + filename
                                + "\"\r\n");
                        strBuf.append("Content-Type:" + contentType + "\r\n\r\n");
                        out.write(strBuf.toString().getBytes());
                        DataInputStream in = new DataInputStream(
                                new FileInputStream(file));
                        int bytes = 0;
                        byte[] bufferOut = new byte[1024];
                        while ((bytes = in.read(bufferOut)) != -1) {
                            out.write(bufferOut, 0, bytes);
                        }
                        in.close();
                    }
                }
                byte[] endData = ("\r\n--" + BOUNDARY + "--\r\n").getBytes();
                out.write(endData);
                out.flush();
                out.close();
                int responseCode = conn.getResponseCode();
                String responseMessage = conn.getResponseMessage();
                log.info("responseCode={}|responseMessage={}", responseCode, responseMessage);
                if (responseCode != 200) {
                    // 读取返回数据
                    StringBuffer strBuf = new StringBuffer();
                    BufferedReader reader = new BufferedReader(new InputStreamReader(
                            conn.getErrorStream()));
                    String line = null;
                    while ((line = reader.readLine()) != null) {
                        strBuf.append(line).append("\n");
                    }
                    res = strBuf.toString();
                    reader.close();
                    reader = null;
                } else {
    
                    // 读取返回数据
                    StringBuffer strBuf = new StringBuffer();
                    BufferedReader reader = new BufferedReader(new InputStreamReader(
                            conn.getInputStream()));
                    String line = null;
                    while ((line = reader.readLine()) != null) {
                        strBuf.append(line).append("\n");
                    }
                    res = strBuf.toString();
                    reader.close();
                    reader = null;
                }
    
            } catch (Exception e) {
                System.out.println("发送POST请求出错。" + urlStr);
                e.printStackTrace();
                throw new RuntimeException("httpClient url connect error!");
            } finally {
                if (conn != null) {
                    conn.disconnect();
                    conn = null;
                }
            }
            return res;
        }