adressevedligehold/fileupload/FileUploadServlet.java

package dk.daoas.adressevedligehold.fileupload;

import java.io.File;
import java.io.IOException;
import java.util.List;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import dk.daoas.adressevedligehold.tasks.TaskManager;


/*
 * TODO: understøt indlæsning af FD
 * TODO: understøt indlæsning af NS
 */

/**
 * Servlet implementation class FileUploadServlet
 */
@WebServlet("/FileUploadServlet")
public class FileUploadServlet extends HttpServlet {
        private static final long serialVersionUID = 1L;
       
    public FileUploadServlet() {
        super();
    }

    /*
     * https://commons.apache.org/proper/commons-fileupload/using.html
     */
        protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
                // Create a factory for disk-based file items
                DiskFileItemFactory factory = new DiskFileItemFactory();

                // Configure a repository (to ensure a secure temp location is used)
                ServletContext servletContext = this.getServletConfig().getServletContext();
                File repository = (File) servletContext.getAttribute("javax.servlet.context.tempdir");
                
                System.out.println( "Path: " + repository.getPath() );
                
                factory.setRepository(repository);

                // Create a new file upload handler
                ServletFileUpload upload = new ServletFileUpload(factory);

                try {
                        // Parse the request
                        List<FileItem> items = upload.parseRequest(request);
                        
                        if ( items.size() == 0) {
                                System.out.println("No file uploaded");
                                return;
                        }
                        
                        if ( items.size() >1) {
                                System.out.println("Uploaded " + items.size() + " items");
                        }

                        
                        String distributor = null;
                        FileItem item = null;
                        boolean writeback = false;
                        
                        for (FileItem i : items) {
                                
                                
                                if (i.isFormField()) {
                                        System.out.println("FormField: " + i.getFieldName() + "=" + i.getString() );
                                        
                                        switch (i.getFieldName().toLowerCase()) {
                                        case "distributor":
                                                distributor = i.getString();
                                                break;
                                        case "writeback":
                                                writeback = Boolean.parseBoolean( i.getString() );
                                                break;
                                        default:
                                                System.out.println("Unknown form field: " + i.getFieldName() );
                                        }
                                        
                                } else {
                                        System.out.println("received " + i.getSize() + " bytes in " + i.getName() + " " + i.getContentType());
                                        item = i;
                                }                               
                        }
                        
                        if (item == null) {
                                System.out.println("No file uploaded");
                                return;
                        }
                        if (distributor == null || distributor.equals("")) {
                                item.delete();
                                return;
                        }
                        
        
                        distributor = distributor.toUpperCase();
                        
                        AddressSource source = null;
                        
                        switch (distributor) {
                        case "BK":
                                source = new AddressSourceBK( item );
                                break;
                        case "DAO":
                                source = new AddressSourceDAO( item );
                                break;
                        case "FD":
                                source = new AddressSourceFD( item );
                                break;
                        case "NS":
                                source = new AddressSourceNS( item );
                                break;
                        default:
                                System.out.println("Unknown distributor: " + distributor);
                                item.delete();
                                return;
                        }
                        
                        TaskManager taskManager = (TaskManager) this.getServletContext().getAttribute(TaskManager.TASKMANAGER);
                        AddressTask task = new AddressTask(source, writeback);
                        taskManager.submitTask(task);
                        
                                                
                }
                catch (FileUploadException fue) {
                        System.out.println("UploadException: " + fue.getMessage() );
                        
                }
                catch (Exception e) {
                        throw new ServletException(e);
                }
                
                response.getWriter().write("OK");
        }

}
1	package dk.daoas.adressevedligehold.fileupload;
2
3	import java.io.File;
4	import java.io.IOException;
5	import java.util.List;
6
7	import javax.servlet.ServletContext;
8	import javax.servlet.ServletException;
9	import javax.servlet.annotation.WebServlet;
10	import javax.servlet.http.HttpServlet;
11	import javax.servlet.http.HttpServletRequest;
12	import javax.servlet.http.HttpServletResponse;
13
14	import org.apache.commons.fileupload.FileItem;
15	import org.apache.commons.fileupload.FileUploadException;
16	import org.apache.commons.fileupload.disk.DiskFileItemFactory;
17	import org.apache.commons.fileupload.servlet.ServletFileUpload;
18
19	import dk.daoas.adressevedligehold.tasks.TaskManager;
20
21
22	/*
23	* TODO: understøt indlæsning af FD
24	* TODO: understøt indlæsning af NS
25	*/
26
27	/**
28	* Servlet implementation class FileUploadServlet
29	*/
30	@WebServlet("/FileUploadServlet")
31	public class FileUploadServlet extends HttpServlet {
32	private static final long serialVersionUID = 1L;
33
34	public FileUploadServlet() {
35	super();
36	}
37
38	/*
39	* https://commons.apache.org/proper/commons-fileupload/using.html
40	*/
41	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
42	// Create a factory for disk-based file items
43	DiskFileItemFactory factory = new DiskFileItemFactory();
44
45	// Configure a repository (to ensure a secure temp location is used)
46	ServletContext servletContext = this.getServletConfig().getServletContext();
47	File repository = (File) servletContext.getAttribute("javax.servlet.context.tempdir");
48
49	System.out.println( "Path: " + repository.getPath() );
50
51	factory.setRepository(repository);
52
53	// Create a new file upload handler
54	ServletFileUpload upload = new ServletFileUpload(factory);
55
56	try {
57	// Parse the request
58	List<FileItem> items = upload.parseRequest(request);
59
60	if ( items.size() == 0) {
61	System.out.println("No file uploaded");
62	return;
63	}
64
65	if ( items.size() >1) {
66	System.out.println("Uploaded " + items.size() + " items");
67	}
68
69
70	String distributor = null;
71	FileItem item = null;
72	boolean writeback = false;
73
74	for (FileItem i : items) {
75
76
77	if (i.isFormField()) {
78	System.out.println("FormField: " + i.getFieldName() + "=" + i.getString() );
79
80	switch (i.getFieldName().toLowerCase()) {
81	case "distributor":
82	distributor = i.getString();
83	break;
84	case "writeback":
85	writeback = Boolean.parseBoolean( i.getString() );
86	break;
87	default:
88	System.out.println("Unknown form field: " + i.getFieldName() );
89	}
90
91	} else {
92	System.out.println("received " + i.getSize() + " bytes in " + i.getName() + " " + i.getContentType());
93	item = i;
94	}
95	}
96
97	if (item == null) {
98	System.out.println("No file uploaded");
99	return;
100	}
101	if (distributor == null \|\| distributor.equals("")) {
102	item.delete();
103	return;
104	}
105
106
107
108	distributor = distributor.toUpperCase();
109
110	AddressSource source = null;
111
112	switch (distributor) {
113	case "BK":
114	source = new AddressSourceBK( item );
115	break;
116	case "DAO":
117	source = new AddressSourceDAO( item );
118	break;
119	case "FD":
120	source = new AddressSourceFD( item );
121	break;
122	case "NS":
123	source = new AddressSourceNS( item );
124	break;
125	default:
126	System.out.println("Unknown distributor: " + distributor);
127	item.delete();
128	return;
129	}
130
131	TaskManager taskManager = (TaskManager) this.getServletContext().getAttribute(TaskManager.TASKMANAGER);
132	AddressTask task = new AddressTask(source, writeback);
133	taskManager.submitTask(task);
134
135
136
137	}
138	catch (FileUploadException fue) {
139	System.out.println("UploadException: " + fue.getMessage() );
140
141	}
142	catch (Exception e) {
143	throw new ServletException(e);
144	}
145
146	response.getWriter().write("OK");
147	}
148
149	}