com.glaf.core.web.springmvc.MxDiskFileUploadJsonController.java Source code

Introduction

Here is the source code for com.glaf.core.web.springmvc.MxDiskFileUploadJsonController.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.glaf.core.web.springmvc;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Random;
import java.util.Set;

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

import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import com.alibaba.fastjson.JSONObject;
import com.glaf.core.base.DataFile;
import com.glaf.core.config.CustomProperties;
import com.glaf.core.config.SystemProperties;
import com.glaf.core.domain.BlobItemEntity;
import com.glaf.core.security.LoginContext;
import com.glaf.core.service.IBlobService;
import com.glaf.core.util.FileUtils;
import com.glaf.core.util.RequestUtils;

@Controller("/diskFileUploadJson")
@RequestMapping("/diskFileUploadJson")
public class MxDiskFileUploadJsonController {

    public final static String sp = System.getProperty("file.separator");

    protected IBlobService blobService;

    private String getError(String message) throws Exception {
        JSONObject object = new JSONObject();
        object.put("error", 1);
        object.put("message", message);
        return object.toString();
    }

    @javax.annotation.Resource
    public void setBlobService(IBlobService blobService) {
        this.blobService = blobService;
    }

    @RequestMapping
    public void upload(HttpServletRequest request, HttpServletResponse response) throws Exception {
        response.setContentType("text/html; charset=UTF-8");
        String businessKey = request.getParameter("businessKey");
        String serviceKey = request.getParameter("serviceKey");
        // ?
        LoginContext loginContext = RequestUtils.getLoginContext(request);
        String savePath = SystemProperties.getAppPath() + "/upload/" + loginContext.getUser().getId() + "/";
        // ?URL
        String saveUrl = request.getContextPath() + "/upload/" + loginContext.getUser().getId() + "/";
        if (StringUtils.isNotEmpty(serviceKey)) {
            saveUrl = saveUrl + serviceKey + "/";
        }
        // ???
        String[] fileTypes = new String[] { "gif", "jpg", "jpeg", "png", "bmp", "swf" };
        // ?
        long maxSize = 10240000;

        String allowSize = CustomProperties.getString("upload.maxSize");
        if (StringUtils.isEmpty(allowSize)) {
            allowSize = SystemProperties.getString("upload.maxSize");
        }

        if (StringUtils.isNotEmpty(allowSize) && StringUtils.isNumeric(allowSize)) {
            maxSize = Long.parseLong(allowSize);
        }

        // 
        File uploadDir = new File(savePath);
        try {
            if (!uploadDir.exists()) {
                FileUtils.mkdirs(savePath);
            }
        } catch (Exception ex) {
        }

        if (!uploadDir.isDirectory()) {
            response.getWriter().write(getError("?"));
            return;
        }
        // ??
        if (!uploadDir.canWrite()) {
            response.getWriter().write(getError("??"));
            return;
        }

        MultipartHttpServletRequest req = (MultipartHttpServletRequest) request;
        Map<String, MultipartFile> fileMap = req.getFileMap();
        Set<Entry<String, MultipartFile>> entrySet = fileMap.entrySet();
        for (Entry<String, MultipartFile> entry : entrySet) {
            MultipartFile mFile = entry.getValue();
            if (mFile.getOriginalFilename() != null && mFile.getSize() > 0) {
                // ?
                if (mFile.getSize() > maxSize) {
                    response.getWriter().write(getError("??"));
                    return;
                }
                String fileName = mFile.getOriginalFilename();
                // ??
                String fileExt = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase();
                if (!Arrays.<String>asList(fileTypes).contains(fileExt)) {
                    response.getWriter().write(getError("??????"));
                    return;
                }
                SimpleDateFormat df = new SimpleDateFormat("yyyyMMddHHmmss");
                String newFileName = df.format(new Date()) + "_" + new Random().nextInt(10000) + "." + fileExt;
                try {
                    DataFile dataFile = new BlobItemEntity();
                    dataFile.setBusinessKey(businessKey);
                    dataFile.setCreateBy(loginContext.getActorId());
                    dataFile.setCreateDate(new Date());
                    dataFile.setFileId(newFileName);
                    dataFile.setLastModified(System.currentTimeMillis());
                    dataFile.setName(fileName);
                    if (StringUtils.isNotEmpty(serviceKey)) {
                        dataFile.setServiceKey(serviceKey);
                    } else {
                        dataFile.setServiceKey("IMG_" + loginContext.getActorId());
                    }
                    dataFile.setFilename(fileName);
                    dataFile.setType(fileExt);
                    dataFile.setSize(mFile.getSize());
                    dataFile.setStatus(1);
                    blobService.insertBlob(dataFile);

                    FileUtils.save(savePath + sp + newFileName, mFile.getInputStream());

                } catch (Exception ex) {
                    ex.printStackTrace();
                    response.getWriter().write(getError("?"));
                    return;
                }

                JSONObject object = new JSONObject();
                object.put("error", 0);
                object.put("url", saveUrl + newFileName);
                response.getWriter().write(object.toString());
            }
        }
    }

}