org.springframework.web.multipart.support.StandardMultipartHttpServletRequest Java Examples

/**
 * Create a new {@link MockMultipartHttpServletRequest} based on the
 * supplied {@code ServletContext} and the {@code MockMultipartFiles}
 * added to this builder.
 */
@Override
protected final MockHttpServletRequest createServletRequest(ServletContext servletContext) {

	MockMultipartHttpServletRequest request = new MockMultipartHttpServletRequest(servletContext);
	this.files.stream().forEach(request::addFile);
	this.parts.values().stream().flatMap(Collection::stream).forEach(request::addPart);

	if (!this.parts.isEmpty()) {
		new StandardMultipartHttpServletRequest(request)
				.getMultiFileMap().values().stream().flatMap(Collection::stream)
				.forEach(request::addFile);
	}

	return request;
}
 

/**
 * Create a new {@link MockMultipartHttpServletRequest} based on the
 * supplied {@code ServletContext} and the {@code MockMultipartFiles}
 * added to this builder.
 */
@Override
protected final MockHttpServletRequest createServletRequest(ServletContext servletContext) {

	MockMultipartHttpServletRequest request = new MockMultipartHttpServletRequest(servletContext);
	this.files.stream().forEach(request::addFile);
	this.parts.values().stream().flatMap(Collection::stream).forEach(request::addPart);

	if (!this.parts.isEmpty()) {
		new StandardMultipartHttpServletRequest(request)
				.getMultiFileMap().values().stream().flatMap(Collection::stream)
				.forEach(request::addFile);
	}

	return request;
}
 

@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler){
    Map<String,String[]> paramMaps = request.getParameterMap();
    Map<String,Object> params = CollectionUtil.newHashMap();
    paramMaps.forEach((k,v) -> {
        params.put(k,convert(v));
    });

    // 文件上传
    if(request instanceof AbstractMultipartHttpServletRequest){
        //CommonsMultipartResolver commonsMultipartResolver = new CommonsMultipartResolver(request.getServletContext());
        StandardMultipartHttpServletRequest multipartHttpServletRequest = (StandardMultipartHttpServletRequest)request;
        Map<String,MultipartFile> multipartFileMap = multipartHttpServletRequest.getFileMap();
        DataBoss.self().getPara().setFiles(multipartFileMap);
    }

    // 请求参数
    DataBoss.self().getPara().setData(params);
    // 路径参数
    DataBoss.self().getPara().setPathVars((Map)request.getAttribute(HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE));
    return true;
}
 

private void writeFilesAndAddToMap(HttpServletRequest request, Map<String, Object> variableMap, List<File> tempFileList) {
	
	// get multipart request
	StandardMultipartHttpServletRequest multipartHttpServletRequest = (StandardMultipartHttpServletRequest) request;

	// get file map from request
	Map<String, MultipartFile> fileMap = multipartHttpServletRequest.getFileMap();
	
	for (Entry<String, MultipartFile> fileMapEntry : fileMap.entrySet()) {

		// write file uploads to disk
		File file = writeMultipartFile(fileMapEntry.getValue());
		if (file != null) {
			
			// add to temp file list
			tempFileList.add(file);

			// add file paths to variable map
			variableMap.put(fileMapEntry.getKey(), file.getAbsolutePath());
		}
	}
}
 

private Collection<Part> getWireMockParts() {
	MockHttpServletRequest request = MockMvcRequestBuilders
			.request(this.result.getMethod(), this.result.getUriTemplate())
			.contentType(this.result.getRequestHeaders().getContentType())
			.content(this.result.getRequestBodyContent())
			.buildRequest(new MockServletContext());
	try {
		return new StandardMultipartHttpServletRequest(request).getParts().stream()
				.map(part -> partFromServletPart(part)).collect(Collectors.toList());
	}
	catch (Exception e) {
		throw new IllegalStateException(e);
	}
}
 

/**
 * 上传文件，如果是SpringBoot，配置如下：
 * # 上传单个文件最大允许
 * spring.servlet.multipart.max-file-size=10MB
 * # 每次请求最大允许
 * spring.servlet.multipart.max-request-size=100MB
 *
 * @return 自定义消息
 * @author jitwxs
 * @version 创建时间：2018年4月17日 下午4:05:26
 */
public static String upload(HttpServletRequest request) throws Exception {
    StandardMultipartHttpServletRequest req = (StandardMultipartHttpServletRequest) request;

    // 遍历普通参数（即formData的fileName和fileSize）
    Enumeration<String> names = req.getParameterNames();
    while (names.hasMoreElements()) {
        String key = names.nextElement();
        String val = req.getParameter(key);
        System.out.println("FormField：k=" + key + "v=" + val);
    }

    // 遍历文件参数（即formData的file）
    Iterator<String> iterator = req.getFileNames();
    if (iterator == null){
        return "未选择文件";
    }
    String result = "";
    while (iterator.hasNext()) {
        MultipartFile file = req.getFile(iterator.next());
        String fileNames = file.getOriginalFilename();
        // 文件名
        fileNames = new String(fileNames.getBytes("UTF-8"));
        //int split = fileNames.lastIndexOf(".");
        // 文件前缀
        //String fileName = fileNames.substring(0, split);
        // 文件后缀
        //String fileType = fileNames.substring(split + 1, fileNames.length());
        // 文件大小
        //Long fileSize = file.getSize();
        // 文件内容
        byte[] content = file.getBytes();

        File file1 = new File("D:\\test\\"+fileNames);

        FileUtils.writeByteArrayToFile(file1, content);

        FileOutputStream fos = new FileOutputStream(file1);
        fos.write(content);
        fos.flush();
        result = "D:\\test\\"+fileNames;
        System.out.println("write success");
    }
    return result;
}