seava.j4e.web.controller.upload.FileUploadController.java Source code

Introduction

Here is the source code for seava.j4e.web.controller.upload.FileUploadController.java

Source

/** 
 * DNet eBusiness Suite
 * Copyright: 2013 Nan21 Electronics SRL. All rights reserved.
 * Use is subject to license terms.
 */
package seava.j4e.web.controller.upload;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import seava.j4e.api.Constants;
import seava.j4e.api.descriptor.IUploadedFileDescriptor;
import seava.j4e.api.exceptions.BusinessException;
import seava.j4e.api.exceptions.ErrorCode;
import seava.j4e.api.service.IFileUploadService;
import seava.j4e.api.service.IFileUploadServiceFactory;
import seava.j4e.web.controller.AbstractBaseController;

import org.codehaus.jackson.map.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PathVariable;
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.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

@Controller
public class FileUploadController extends AbstractBaseController {

    private List<IFileUploadServiceFactory> fileUploadServiceFactories;

    final static Logger logger = LoggerFactory.getLogger(FileUploadController.class);

    /**
     * Generic file upload. Expects an uploaded file and a handler alias to
     * delegate the uploaded file processing.
     * 
     * @param handler
     *            spring bean alias of the
     *            {@link seava.j4e.api.service.IFileUploadService} which should
     *            process the uploaded file
     * @param file
     *            Uploaded file
     * @param request
     * @param response
     * @return
     * @throws Exception
     */
    @RequestMapping(value = "/{handler}", method = RequestMethod.POST)
    @ResponseBody
    public String fileUpload(@PathVariable("handler") String handler, @RequestParam("file") MultipartFile file,
            HttpServletRequest request, HttpServletResponse response) throws Exception {

        try {
            if (logger.isInfoEnabled()) {
                logger.info("Processing file upload request with-handler {} ", new Object[] { handler });
            }

            if (file.isEmpty()) {
                throw new BusinessException(ErrorCode.G_FILE_NOT_UPLOADED,
                        "Upload was not succesful. Try again please.");
            }

            this.prepareRequest(request, response);

            IFileUploadService srv = this.getFileUploadService(handler);
            Map<String, String> paramValues = new HashMap<String, String>();

            for (String p : srv.getParamNames()) {
                paramValues.put(p, request.getParameter(p));
            }

            IUploadedFileDescriptor fileDescriptor = new UploadedFileDescriptor();
            fileDescriptor.setContentType(file.getContentType());
            fileDescriptor.setOriginalName(file.getOriginalFilename());
            fileDescriptor.setNewName(file.getName());
            fileDescriptor.setSize(file.getSize());
            Map<String, Object> result = srv.execute(fileDescriptor, file.getInputStream(), paramValues);

            result.put("success", true);
            ObjectMapper mapper = getJsonMapper();
            return mapper.writeValueAsString(result);
        } catch (Exception e) {
            return this.handleManagedException(null, e, response);
        } finally {
            this.finishRequest();
        }

    }

    /**
     * Try to locate a file-upload service by its name (spring-bean alias)
     * 
     * @param dsName
     * @return
     * @throws Exception
     */
    protected IFileUploadService getFileUploadService(String name) throws Exception {
        IFileUploadService srv = null;

        if (this.getSettings().get(Constants.PROP_DEPLOYMENT).equals(Constants.PROP_DEPLOYMENT_JEE)) {
            srv = (IFileUploadService) this.getApplicationContext().getBean(name, IFileUploadService.class);
            return srv;
        } else {
            for (IFileUploadServiceFactory sf : getFileUploadServiceFactories()) {
                try {
                    srv = sf.create(name);
                    if (srv != null) {
                        return srv;
                    }
                } catch (NoSuchBeanDefinitionException e) {
                    // service not found in this factory, ignore
                }
            }
        }

        throw new BusinessException(ErrorCode.SRV_FILEUPLOAD_SRV_NOT_FOUND,
                "File upload service not found for name " + name + "!");
    }

    @SuppressWarnings("unchecked")
    public List<IFileUploadServiceFactory> getFileUploadServiceFactories() {
        if (this.fileUploadServiceFactories == null) {
            this.fileUploadServiceFactories = (List<IFileUploadServiceFactory>) this.getApplicationContext()
                    .getBean(Constants.SPRING_OSGI_FILE_UPLOAD_SERVICE_FACTORIES);
        }
        return this.fileUploadServiceFactories;
    }

    public void setFileUploadServiceFactories(List<IFileUploadServiceFactory> fileUploadServiceFactories) {

        this.fileUploadServiceFactories = fileUploadServiceFactories;
    }

}