Java tutorial
/* * JBoss, Home of Professional Open Source. * Copyright 2011, Red Hat, Inc., and individual contributors * as indicated by the @author tags. See the copyright.txt file in the * distribution for a full listing of individual contributors. * * This is free software; you can redistribute it and/or modify it * under the terms of the GNU Lesser General Public License as * published by the Free Software Foundation; either version 2.1 of * the License, or (at your option) any later version. * * This software is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this software; if not, write to the Free * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA * 02110-1301 USA, or see the FSF site: http://www.fsf.org. */ package org.gatein.management.gadget.mop.exportimport.server; import static org.gatein.management.gadget.mop.exportimport.server.ContainerRequestHandler.doInRequest; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.util.Collections; import java.util.HashMap; import java.util.Hashtable; import java.util.List; import java.util.Map; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.commons.fileupload.FileItem; import org.exoplatform.container.ExoContainer; import org.gatein.common.logging.Logger; import org.gatein.common.logging.LoggerFactory; import org.gatein.management.api.ContentType; import org.gatein.management.api.PathAddress; import org.gatein.management.api.controller.ManagedRequest; import org.gatein.management.api.controller.ManagedResponse; import org.gatein.management.api.controller.ManagementController; import org.gatein.management.api.operation.OperationNames; import gwtupload.server.UploadAction; import gwtupload.server.exceptions.UploadActionException; /** * {@code FileUploadServlet} * <p> * The file upload servlet based on GWT upload, used for import sites. * </p> * Created on Jan 3, 2011, 3:43:36 PM * * @author <a href="mailto:nbenothm@redhat.com">Nabil Benothman</a> * @version 1.0 */ public class FileUploadServlet extends UploadAction { private static final Logger log = LoggerFactory.getLogger(FileUploadServlet.class); private static final long serialVersionUID = 1L; private Hashtable<String, String> receivedContentTypes = new Hashtable<String, String>(); /** * Maintain a list with received files and their content types. */ private Hashtable<String, File> receivedFiles = new Hashtable<String, File>(); /** * Override executeAction to save the received files in a custom place and delete this items from session. */ @Override public String executeAction(HttpServletRequest request, List<FileItem> sessionFiles) throws UploadActionException { StringBuilder response = new StringBuilder("<response>\n"); int count = 0; for (FileItem item : sessionFiles) { // if (false == item.isFormField()) { if (!item.isFormField()) { count++; try { // Create a new file based on the remote file name in the client String saveName = item.getName().replaceAll("[\\\\/><\\|\\s\"'{}()\\[\\]]+", "_"); // Create a temporary file placed in the default system temp folder File file = File.createTempFile(saveName, ".zip"); item.write(file); // Save a list with the received files receivedFiles.put(item.getFieldName(), file); receivedContentTypes.put(item.getFieldName(), item.getContentType()); String importMode = request.getParameter("importMode"); // process the uploaded file processImport(request.getParameter("pc"), new FileInputStream(file), importMode); // Compose a xml message with the full file information which can be parsed in client side response.append("<file-").append(count).append("-field>").append(item.getFieldName()) .append("</file-").append(count).append("-field>\n"); response.append("<file-").append(count).append("-name>").append(item.getName()) .append("</file-").append(count).append("-name>\n"); response.append("<file-").append(count).append("-size>").append(item.getSize()) .append("</file-").append(count).append("-size>\n"); response.append("<file-").append(count).append("-type>").append(item.getContentType()) .append("</file-").append(count).append("type>\n"); } catch (Exception e) { throw new UploadActionException(e); } } } // Remove files from session because we have a copy of them removeSessionFileItems(request); // Send information of the received files to the client. return response.append("</response>\n").toString(); } /** * Get the content of an uploaded file. */ @Override public void getUploadedFile(HttpServletRequest request, HttpServletResponse response) throws IOException { String fieldName = request.getParameter(PARAM_SHOW); File f = receivedFiles.get(fieldName); if (f != null) { response.setContentType(receivedContentTypes.get(fieldName)); FileInputStream is = new FileInputStream(f); copyFromInputStreamToOutputStream(is, response.getOutputStream()); } else { renderXmlResponse(request, response, ERROR_ITEM_NOT_FOUND); } } /** * Remove a file when the user sends a delete request. */ @Override public void removeItem(HttpServletRequest request, String fieldName) throws UploadActionException { File file = receivedFiles.get(fieldName); receivedFiles.remove(fieldName); receivedContentTypes.remove(fieldName); if (file != null) { boolean deleted = file.delete(); if (!deleted) log.warn("Could not delete file " + file); } } private void processImport(final String containerName, final InputStream in, final String importMode) throws Exception { doInRequest(containerName, new ContainerCallback<Void>() { @Override public Void doInContainer(ExoContainer container) throws Exception { ManagementController controller = getComponent(container, ManagementController.class); Map<String, List<String>> attributes = new HashMap<String, List<String>>(1); attributes.put("importMode", Collections.singletonList(importMode)); ManagedRequest request = ManagedRequest.Factory.create(OperationNames.IMPORT_RESOURCE, PathAddress.pathAddress("mop"), attributes, in, ContentType.ZIP); ManagedResponse response = controller.execute(request); if (!response.getOutcome().isSuccess()) { throw new Exception(response.getOutcome().getFailureDescription()); } return null; } }); } }