upload a file with okhttp

thangpt2019 · thangpt2019 · commit 97dcd2770887 · 2016-11-21T22:04:19.000+07:00
diff --git a/Book-RESTful-Service/README.md b/Book-RESTful-Service/README.md
@@ -26,4 +26,5 @@ This RESTful ws provides API for all below sub projects:
 ###[Java REST Client Using Apache CXF Proxy based API](http://howtoprogram.xyz/2016/07/15/java-rest-client-using-apache-cxf-proxy-based-api/)
 ###[Java REST Client Using Netflix Feign](http://howtoprogram.xyz/2016/07/18/java-rest-client-using-netflix-feign/)
 ###[Java REST Client Using Unirest Java API](http://howtoprogram.xyz/2016/07/27/java-rest-client-using-unirest-java-api/)
-###[Java REST Client Using Retrofit 2](http://howtoprogram.xyz/2016/10/17/java-rest-client-using-retrofit-2/)
+###[Java REST Client Using Retrofit 2](http://howtoprogram.xyz/2016/10/17/java-rest-client-using-retrofit-2/)
+###[Upload a File with OkHttp](http://howtoprogram.xyz/2016/11/21/upload-file-okhttp/)
diff --git a/Book-RESTful-Service/src/main/java/com/howtoprogram/book/FileUploadController.java b/Book-RESTful-Service/src/main/java/com/howtoprogram/book/FileUploadController.java
@@ -0,0 +1,53 @@
+package com.howtoprogram.book;
+
+import java.util.Iterator;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+import org.springframework.web.multipart.MultipartHttpServletRequest;
+
+import com.howtoprogram.service.StorageService;
+
+@RestController
+@RequestMapping(value = "/v1")
+public class FileUploadController {
+
+	@Autowired
+	private StorageService storageService;
+
+	@RequestMapping(value = "/upload", method = RequestMethod.POST)
+	public ResponseEntity uploadFile(@RequestParam(value = "file") MultipartFile multipartFile) {
+		// stores the file to disk
+		storageService.store(multipartFile);
+		return new ResponseEntity<>("{}", HttpStatus.OK);
+	}
+
+	@RequestMapping(value = "/uploads", method = RequestMethod.POST)
+	public ResponseEntity uploadFile(MultipartHttpServletRequest request) {
+
+		try {
+			Iterator<String> itr = request.getFileNames();
+			while (itr.hasNext()) {
+				String uploadedFile = itr.next();
+				MultipartFile file = request.getFile(uploadedFile);
+				String mimeType = file.getContentType();
+				String filename = file.getOriginalFilename();
+				byte[] bytes = file.getBytes();
+				System.out.println(mimeType);
+				System.out.println(filename);
+
+			}
+		} catch (Exception e) {
+			return new ResponseEntity<>("{}", HttpStatus.INTERNAL_SERVER_ERROR);
+		}
+
+		return new ResponseEntity<>("{}", HttpStatus.OK);
+	}
+
+}
diff --git a/Book-RESTful-Service/src/main/java/com/howtoprogram/service/FileSystemStorageService.java b/Book-RESTful-Service/src/main/java/com/howtoprogram/service/FileSystemStorageService.java
@@ -0,0 +1,28 @@
+package com.howtoprogram.service;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+@Service
+public class FileSystemStorageService implements StorageService {
+	private final Path rootLocation = Paths.get(".");
+
+	@Override
+	public void store(MultipartFile file) {
+		try {
+			if (file.isEmpty()) {
+				throw new StorageException(
+				        "Failed to store empty file " + file.getOriginalFilename());
+			}
+			Files.copy(file.getInputStream(),
+			        this.rootLocation.resolve(file.getOriginalFilename()));
+		} catch (IOException e) {
+			throw new StorageException("Failed to store file " + file.getOriginalFilename(), e);
+		}
+	}
+}
diff --git a/Book-RESTful-Service/src/main/java/com/howtoprogram/service/StorageException.java b/Book-RESTful-Service/src/main/java/com/howtoprogram/service/StorageException.java
@@ -0,0 +1,12 @@
+package com.howtoprogram.service;
+
+public class StorageException extends RuntimeException {
+
+	public StorageException(String message) {
+		super(message);
+	}
+
+	public StorageException(String message, Throwable cause) {
+		super(message, cause);
+	}
+}
diff --git a/Book-RESTful-Service/src/main/java/com/howtoprogram/service/StorageService.java b/Book-RESTful-Service/src/main/java/com/howtoprogram/service/StorageService.java
@@ -0,0 +1,9 @@
+package com.howtoprogram.service;
+
+import org.springframework.web.multipart.MultipartFile;
+
+public interface StorageService {
+
+	void store(MultipartFile file);
+
+}
diff --git a/okhttp-examples/pom.xml b/okhttp-examples/pom.xml
@@ -20,6 +20,11 @@
 			<artifactId>okhttp</artifactId>
 			<version>3.4.1</version>
 		</dependency>
-
+		<dependency>
+			<groupId>junit</groupId>
+			<artifactId>junit</artifactId>
+			<version>4.8.1</version>
+			<scope>test</scope>
+		</dependency>
 	</dependencies>
 </project>
diff --git a/okhttp-examples/src/main/java/com/howtoprogram/okhttp/upload/UploadService.java b/okhttp-examples/src/main/java/com/howtoprogram/okhttp/upload/UploadService.java
@@ -0,0 +1,34 @@
+package com.howtoprogram.okhttp.upload;
+
+import java.io.File;
+import java.io.IOException;
+
+import okhttp3.MediaType;
+import okhttp3.MultipartBody;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.RequestBody;
+import okhttp3.Response;
+
+public class UploadService {
+
+	private static final MediaType MEDIA_TYPE_PNG = MediaType.parse("image/png");
+
+	public void uploadImage(File image, String imageName) throws IOException {
+
+		OkHttpClient client = new OkHttpClient();
+		RequestBody requestBody = new MultipartBody.Builder().setType(MultipartBody.FORM)
+		        .addFormDataPart("file", imageName, RequestBody.create(MEDIA_TYPE_PNG, image))
+		        .build();
+
+		Request request = new Request.Builder().url("http://localhost:8080/v1/upload")
+		        .post(requestBody).build();
+
+		Response response = client.newCall(request).execute();
+		if (!response.isSuccessful()) {
+			throw new IOException("Unexpected code " + response);
+		}
+
+	}
+
+}
diff --git a/okhttp-examples/src/test/java/com/howtoprogram/okhttp/upload/UploadServiceTest.java b/okhttp-examples/src/test/java/com/howtoprogram/okhttp/upload/UploadServiceTest.java
@@ -0,0 +1,18 @@
+package com.howtoprogram.okhttp.upload;
+
+import java.io.File;
+import java.io.IOException;
+
+import org.junit.Test;
+
+public class UploadServiceTest {
+
+	@Test
+	public void testUploadImage() throws IOException {
+		File file = new File("d:\\Tmp\\abc.png");
+		UploadService uploadService = new UploadService();
+		uploadService.uploadImage(file, "abc.png");
+
+	}
+
+}
