feign文件上传（多文件，单个文件）

背景

最近有需求，对文件进行解析后，上传到fastdfs里面，之间涉及到多个服务的调用，需要用到feign进行多文件上传，网上参考了好多，要么只实现了单文件上传，多文件的时候就会报错，现在总结一下该问题

<spring-cloud.version>Greenwich.SR2</spring-cloud.version>
<spring-boot.version>2.1.6.RELEASE</spring-boot.version>
依赖

<dependency>
    <groupId>io.github.openfeign.form</groupId>
    <artifactId>feign-form</artifactId>
    <version>3.8.0</version>
</dependency>
<dependency>
    <groupId>io.github.openfeign.form</groupId>
    <artifactId>feign-form-spring</artifactId>
    <version>3.8.0</version>
</dependency>

 

FeignSpringFormEncoder 

import feign.RequestTemplate;
import feign.codec.EncodeException;
import feign.codec.Encoder;
import feign.form.FormEncoder;
import feign.form.MultipartFormContentProcessor;
import feign.form.spring.SpringManyMultipartFilesWriter;
import feign.form.spring.SpringSingleMultipartFileWriter;
import lombok.val;
import org.springframework.web.multipart.MultipartFile;

import java.lang.reflect.Type;
import java.util.Collections;
import java.util.Map;

import static feign.form.ContentType.MULTIPART;

/**
 * @auther xiaodao
 * @date 2019/9/18 17:08
 */
public class FeignSpringFormEncoder extends FormEncoder {
    /**
     * Constructor with the default Feign's encoder as a delegate.
     */
    public FeignSpringFormEncoder() {
        this(new Encoder.Default());
    }

    /**
     * Constructor with specified delegate encoder.
     *
     * @param delegate delegate encoder, if this encoder couldn't encode object.
     */
    public FeignSpringFormEncoder(Encoder delegate) {
        super(delegate);

        val processor = (MultipartFormContentProcessor) getContentProcessor(MULTIPART);
        processor.addFirstWriter(new SpringSingleMultipartFileWriter());
        processor.addFirstWriter(new SpringManyMultipartFilesWriter());
    }

    @Override

    public void encode(Object object, Type bodyType, RequestTemplate template) throws EncodeException {
        if (bodyType.equals(MultipartFile.class)) {
            MultipartFile file = (MultipartFile) object;
            Map<String, Object> data = Collections.singletonMap(file.getName(), object);
            super.encode(data, MAP_STRING_WILDCARD, template);
            return;
        } else if (bodyType.equals(MultipartFile[].class)) {
            MultipartFile[] file = (MultipartFile[]) object;
            if(file != null) {
                Map<String, Object> data = Collections.singletonMap(file.length==0?"":file[0].getName(), object);
                super.encode(data, MAP_STRING_WILDCARD, template);
                return;
            }
        }
        super.encode(object, bodyType, template);
    }

    private boolean isMultipartFileCollection(Object object) {
        if (!(object instanceof Iterable)) {
            return false;
        }
        val iterable = (Iterable<?>) object;
        val iterator = iterable.iterator();
        return iterator.hasNext() && iterator.next() instanceof MultipartFile;
    }

MultipartSupportConfig

import feign.codec.Encoder;
import org.springframework.beans.factory.ObjectFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.http.HttpMessageConverters;
import org.springframework.cloud.openfeign.support.SpringEncoder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class MultipartSupportConfig {

    @Autowired
    private ObjectFactory<HttpMessageConverters> messageConverters;

    @Bean
    public Encoder feignFormEncoder() {
        return new FeignSpringFormEncoder(new SpringEncoder(messageConverters));
    }

}

 

feign接口

import com.xiaodao.common.vo.RespVO;
import com.xiaodao.common.vo.RespVOBuilder;
import com.xiaodao.exam.MultipartSupportConfig;
import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestPart;
import org.springframework.web.multipart.MultipartFile;

@FeignClient(value = "file-service", path = "/file/v1", fallbackFactory = FileClientFallback.class, configuration = MultipartSupportConfig.class)
public interface FileClient {


    @RequestMapping(value = "/uploads", method = RequestMethod.POST, consumes = MediaType.MULTIPART_FORM_DATA_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    RespVO uploads(@RequestPart(value = "files") MultipartFile[] files);


    @RequestMapping(value = "/upload", method = RequestMethod.POST, consumes = MediaType.MULTIPART_FORM_DATA_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    RespVO upload(@RequestPart(value = "file") MultipartFile file);
}

class FileClientFallback implements FileClient {

    @Override
    public RespVO uploads(MultipartFile[] files) {
        return RespVOBuilder.failure("file-service 服務一常");
    }

    @Override
    public RespVO upload(MultipartFile file) {
        return RespVOBuilder.failure("file-service 服務一常");
    }

}

 

文件上传端（前台文件上传调用接口）

import com.xiaodao.common.vo.RespVO;
import com.xiaodao.exam.file.feign.FileClient;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;


/**
 * @Description : TODO
 * @Author : xiaodao
 **/

@RestController
@RequestMapping("/filetest")
@Api(value = "文件测试", tags = "文件测试")
@Slf4j
public class FileTestController {
    @Autowired
    private FileClient fileClient;

    @ApiOperation(value = "多文件上传", notes = "多文件上传")
    @ApiImplicitParams({@ApiImplicitParam(name = "files", value = "多个文件，", paramType = "formData", allowMultiple = true, required = true, dataType = "file")})
    @PostMapping(value = "/uploads", headers = "content-type=multipart/form-data")
    public RespVO uploads(@RequestParam(value = "files", required = true) MultipartFile[] files) {
        return fileClient.uploads(files);
    }

    @ApiOperation(value = "单个文件上传", notes = "单个文件上传")
    @ApiImplicitParams({})
    @PostMapping(value = "/upload", headers = "content-type=multipart/form-data")
    public RespVO upload(@RequestParam(value = "file", required = true) MultipartFile file) {
        return fileClient.upload(file);
    }
}

文件上传端（被feign调用的接口）

import com.xiaodao.common.vo.RespVO;
import com.xiaodao.common.vo.RespVOBuilder;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.util.*;

@RestController
@Api(value = "文件上传", tags = "文件上传")
@RequestMapping("/file/v1")
@Slf4j
public class FileController {

    @ApiOperation(value = "多文件上传", notes = "多文件上传")
    @ApiImplicitParams({@ApiImplicitParam(name = "files", value = "多个文件，", paramType = "formData", allowMultiple = true, required = true, dataType = "file")})
    @PostMapping(value = "/uploads", headers = "content-type=multipart/form-data")
    public RespVO uploads(@RequestParam(value = "files", required = true) MultipartFile[] files) {
        Map<String, Object> dataMap = new HashMap<String, Object>();
        String path = "";
        Arrays.stream(files).forEach(mf -> {
            log.error(mf.getOriginalFilename().substring(mf.getOriginalFilename().lastIndexOf(".") + 1));

        });
        return RespVOBuilder.success(files.length);
    }

    @ApiOperation(value = "单个文件上传", notes = "单个文件上传")
    @ApiImplicitParams({@ApiImplicitParam(name = "file", value = "单个文件，", paramType = "formData", allowMultiple = true, required = true, dataType = "file")})
    @PostMapping(value = "/upload", headers = "content-type=multipart/form-data")
    public RespVO uploads(@RequestParam(value = "file", required = true) MultipartFile file) {
        Map<String, Object> dataMap = new HashMap<String, Object>();
        return RespVOBuilder.success(file.getSize());
    }

}