Implemented file upload for documents resource.

lreimer · lreimer · commit 10d70e7e8059 · 2018-02-19T10:16:44.000+01:00
diff --git a/pom.xml b/pom.xml
@@ -14,6 +14,13 @@
             <scope>provided</scope>
         </dependency>
 
+        <dependency>
+            <groupId>org.glassfish.jersey.media</groupId>
+            <artifactId>jersey-media-multipart</artifactId>
+            <version>2.26</version>
+            <scope>provided</scope>
+        </dependency>
+
         <dependency>
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>
diff --git a/src/main/java/com/packtpub/javaee8/DocumentsResource.java b/src/main/java/com/packtpub/javaee8/DocumentsResource.java
@@ -1,20 +1,34 @@
 package com.packtpub.javaee8;
 
-import javax.enterprise.context.ApplicationScoped;
+import org.glassfish.jersey.media.multipart.FormDataContentDisposition;
+import org.glassfish.jersey.media.multipart.FormDataParam;
+
+import javax.enterprise.context.RequestScoped;
 import javax.servlet.ServletContext;
-import javax.ws.rs.GET;
-import javax.ws.rs.Path;
-import javax.ws.rs.Produces;
+import javax.ws.rs.*;
 import javax.ws.rs.core.Context;
+import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.Response;
+import javax.ws.rs.core.UriInfo;
 import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URI;
+import java.nio.file.FileSystems;
+import java.nio.file.Files;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
-@ApplicationScoped
+@RequestScoped
 @Path("documents")
 public class DocumentsResource {
 
+    private static final Logger LOGGER = Logger.getAnonymousLogger();
+
     @Context
     private ServletContext context;
+    @Context
+    private UriInfo uriInfo;
 
     @GET
     @Path("/me.jpg")
@@ -33,4 +47,24 @@ public Response gif() {
         String path = context.getRealPath("/magic.gif");
         return Response.ok(new File(path)).build();
     }
+
+    @POST
+    @Consumes(MediaType.MULTIPART_FORM_DATA)
+    public Response upload(@FormDataParam("file") InputStream inputStream,
+                           @FormDataParam("file") FormDataContentDisposition fileInfo) {
+        String fileName = fileInfo.getFileName();
+        saveFile(inputStream, fileName);
+
+        URI uri = uriInfo.getBaseUriBuilder().path(DocumentsResource.class).path(fileName).build();
+        return Response.created(uri).build();
+    }
+
+    private void saveFile(InputStream file, String name) {
+        try {
+            java.nio.file.Path path = FileSystems.getDefault().getPath("/tmp/", name);
+            Files.copy(file, path);
+        } catch (IOException ie) {
+            LOGGER.log(Level.WARNING, "Unable to save file.", ie);
+        }
+    }
 }
diff --git a/src/main/java/com/packtpub/javaee8/JAXRSConfiguration.java b/src/main/java/com/packtpub/javaee8/JAXRSConfiguration.java
@@ -1,12 +1,29 @@
 package com.packtpub.javaee8;
 
+import org.glassfish.jersey.media.multipart.MultiPartFeature;
+
 import javax.ws.rs.ApplicationPath;
 import javax.ws.rs.core.Application;
+import java.util.HashSet;
 import java.util.Set;
 
 /**
  * Configures a JAX-RS endpoint.
  */
 @ApplicationPath("api")
 public class JAXRSConfiguration extends Application {
+    @Override
+    public Set<Class<?>> getClasses() {
+        Set<Class<?>> classes = new HashSet<>();
+
+        classes.add(DocumentsResource.class);
+        classes.add(MultiPartFeature.class);
+
+        classes.add(HateosResource.class);
+        classes.add(JsonbResource.class);
+        classes.add(JsonpResource.class);
+        classes.add(VersionResource.class);
+
+        return classes;
+    }
 }
diff --git a/src/main/webapp/upload.html b/src/main/webapp/upload.html
@@ -0,0 +1,15 @@
+<!DOCTYPE html>
+<html>
+<head>
+    <meta charset="UTF-8">
+    <title>Building Web Services with Java EE 8</title>
+</head>
+<body>
+<h2>Upload file</h2>
+<form action="/content-service/api/documents" enctype="multipart/form-data" method="post">
+    <label>Select File</label><input type="file" name="file"/>
+    <br/>
+    <input type="submit" title="Upload"/>
+</form>
+</body>
+</html>
