Spring Cloud Feign的檔案上傳實現

程式猿DD_發表於2019-02-28
SpringCloud

在Spring Cloud封裝的Feign中並不直接支援傳檔案，但可以通過引入Feign的擴充套件包來實現，本來就來具體說說如何實現。

原文：http://blog.didispace.com/spring-cloud-starter-dalston-2-4/

服務提供方（接收檔案）

服務提供方的實現比較簡單，就按Spring MVC的正常實現方式即可，比如：

@EnableFeignClients
@EnableDiscoveryClient
@SpringBootApplication
public class Application {

    @RestController
    public class UploadController {

        @PostMapping(value = "/uploadFile", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
        public String handleFileUpload(@RequestPart(value = "file") MultipartFile file) {
            return file.getName();
        }

    }

    public static void main(String[] args) {
        new SpringApplicationBuilder(Application.class).web(true).run(args);
    }

}
複製程式碼

服務消費方（傳送檔案）

在服務消費方由於會使用Feign客戶端，所以在這裡需要在引入feign對錶單提交的依賴，具體如下：

<dependency>
   <groupId>io.github.openfeign.form</groupId>
   <artifactId>feign-form</artifactId>
   <version>3.0.3</version>
</dependency>
<dependency>
   <groupId>io.github.openfeign.form</groupId>
   <artifactId>feign-form-spring</artifactId>
   <version>3.0.3</version>
</dependency>
<dependency>
   <groupId>commons-fileupload</groupId>
   <artifactId>commons-fileupload</artifactId>
   <version>1.3.3</version>
</dependency>
複製程式碼

定義檔案上傳方的應用主類和FeignClient，假設服務提供方的服務名為eureka-feign-upload-server

@EnableFeignClients
@EnableDiscoveryClient
@SpringBootApplication
public class Application {

    public static void main(String[] args) {
        new SpringApplicationBuilder(Application.class).web(true).run(args);
    }

}

@FeignClient(value = "upload-server", configuration = UploadService.MultipartSupportConfig.class)
public interface UploadService {
 
    @PostMapping(value = "/uploadFile", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    String handleFileUpload(@RequestPart(value = "file") MultipartFile file);
 
    @Configuration
    class MultipartSupportConfig {
        @Bean
        public Encoder feignFormEncoder() {
            return new SpringFormEncoder();
        }
    }
 
}
複製程式碼

在啟動了服務提供方之後，嘗試在服務消費方編寫測試用例來通過上面定義的Feign客戶端來傳檔案，比如：

@Slf4j
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest
public class UploadTester {

    @Autowired
    private UploadService uploadService;

    @Test
    @SneakyThrows
    public void testHandleFileUpload() {

        File file = new File("upload.txt");
        DiskFileItem fileItem = (DiskFileItem) new DiskFileItemFactory().createItem("file",
                MediaType.TEXT_PLAIN_VALUE, true, file.getName());

        try (InputStream input = new FileInputStream(file); OutputStream os = fileItem.getOutputStream()) {
            IOUtils.copy(input, os);
        } catch (Exception e) {
            throw new IllegalArgumentException("Invalid file: " + e, e);
        }

        MultipartFile multi = new CommonsMultipartFile(fileItem);

        log.info(uploadService.handleFileUpload(multi));
    }

}
複製程式碼

完整示例：

讀者可以根據喜好選擇下面的兩個倉庫中檢視eureka-feign-upload-server和eureka-feign-upload-client兩個專案：

如果您對這些感興趣，歡迎star、follow、收藏、轉發給予支援！

相關文章