rest服務中通過resteasy方式上傳檔案

resteasy伺服器代碼

Control：

import com.mochasoft.latte.commons.log.BaseLogger;
import org.jboss.resteasy.plugins.providers.multipart.MultipartFormDataInput;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;

import javax.ws.rs.Consumes;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import java.io.IOException;
@Controller
@Path(value = "/file")
public class FileUploadRest extends BaseLogger {
    @Autowired
    private IFileUploadService fileUploadService;

    /**
     * 日志檔案上傳
     *
     * @param
     * @return
     * @throws IOException
     */
    @POST
    @Path("/upload")
    @Consumes("multipart/form-data")
    @Produces("application/json;charset=UTF-8")
    public Object uploadFile( MultipartFormDataInput input) throws IOException {
        return fileUploadService.uploadFile(input);
    }
}
           

ServiceIml:

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.jboss.resteasy.plugins.providers.multipart.InputPart;
import org.jboss.resteasy.plugins.providers.multipart.MultipartFormDataInput;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import javax.ws.rs.core.MultivaluedMap;
import java.io.*;
import java.util.*;

import static com.mochasoft.zjlms.manage.constants.ManageConstants.UPLOADED_FILE_PATH;

@Service
public class FileUploadServiceImpl extends BaseLogger implements IFileUploadService {
    String path=System.getProperty("user.dir")+"\\upload\\";

    /**
     * 日志資料上傳
     *
     * @param jsonData
     * @return
     */
    public Object uploadFile(MultipartFormDataInput input) {
        String fileName = "";
        try {
            //Get API input data
            Map<String, List<InputPart>> uploadForm = input.getFormDataMap();
            List<InputPart> inputParts = uploadForm.get("file");
            for (InputPart inputPart : inputParts) {
                try {
                    MultivaluedMap<String, String> header = inputPart.getHeaders();
                    fileName = getFileName(header);
                    //convert the uploaded file to inputstream
                    InputStream inputStream = inputPart.getBody(InputStream.class,null);
                    byte [] bytes = IOUtils.toByteArray(inputStream);
                    double fileSize = Double.parseDouble(bytes.length + "") /  / ;
                    if(fileSize > ){
                        return R.error(,"必須參數大小不能超過規定範圍");
                    }
                    //constructs upload file path
                    fileName = path + fileName;
                   // 截取上傳檔案字尾
                    String extName = fileName.substring(fileName.indexOf(".") + ).toLowerCase().trim();
                    // 判斷字尾
                    String fileType = "txt,log";
                    writeFile(bytes,fileName);
                    logger.info("日志檔案上傳成功: " + fileName);
                } catch (Exception e) {
                    logger.error("日志檔案上傳失敗--", e);
                    continue;
                }
            }
        } catch (Exception e) {
            logger.error("日志資料上傳異常--", e);
        }
        return R.ok();
    }

    /**
     * 寫檔案到本地目錄
     *
     * @param content
     * @param filename
     * @throws IOException
     */
    private void writeFile(byte[] content, String filename) throws IOException {
        File file = new File(filename);
        if (!file.exists()) {
            file.createNewFile();
        }
        FileOutputStream fop = new FileOutputStream(file);
        fop.write(content);
        fop.flush();
        fop.close();
    }
    /**
     * 擷取filename
     *
     * @param header
     * @throws IOException
     */
    private String getFileName(MultivaluedMap<String, String> header) {
        String[] contentDisposition = header.getFirst("Content-Disposition").split(";");
        for (String filename : contentDisposition) {
            if ((filename.trim().startsWith("filename"))) {
                String[] name = filename.split("=");
                String finalFileName = name[].trim().replaceAll("\"", "");
                return finalFileName;
            }
        }
        return "unknown";
    }