DiskFileUpload.java 2.44 KB
package com.xiniunet.erp.home.module.screen.api;

import com.alibaba.citrus.service.requestcontext.buffered.BufferedRequestContext;
import com.alibaba.citrus.turbine.dataresolver.Param;
import com.alibaba.fastjson.JSON;
import com.xiniunet.erp.foundation.helper.FoundationHelper;
import com.xiniunet.erp.tool.Constants;
import com.xiniunet.foundation.request.FolderFileCreateRequest;
import com.xiniunet.framework.base.BaseResponse;
import com.xiniunet.framework.exception.ErrorType;
import com.xiniunet.framework.security.Passport;
import com.xiniunet.framework.util.FilenameUtil;
import com.xiniunet.framework.util.auth.LocalData;
import org.apache.commons.fileupload.FileItem;
import org.springframework.beans.factory.annotation.Autowired;

import javax.servlet.http.HttpServletResponse;
import java.io.PrintWriter;

/**
 * Created by DEV001 on 2014/8/20.
 */
public class DiskFileUpload {

    @Autowired
    private HttpServletResponse response;

    @Autowired
    private BufferedRequestContext brc;

    @Autowired
    private FoundationHelper foundationHelper;

    public void execute(@Param("File") FileItem fileItem,
                        @Param("DiskId")Long diskId,
                        @Param("FolderId")Long folderId) throws Exception {
        BaseResponse baseResponse = new BaseResponse();
        try {
            if(fileItem == null){
                return;
            }
            // 必须关闭buffering,未完成的页面才会被显示在浏览器上。
            brc.setBuffering(false);

            // 设置content type,但不需要设置charset,框架会设置正确的charset。
            response.setContentType("text/plain");
            Passport passport = LocalData.getCurrentPassport();

            FolderFileCreateRequest createRequest = new FolderFileCreateRequest();
            createRequest.setDiskId(diskId);
            createRequest.setFileStream(fileItem.get());
            createRequest.setFolderId(folderId);
            createRequest.setFileName(fileItem.getName());
            createRequest.setFileExt(FilenameUtil.getExtName(fileItem.getName()));

            baseResponse = foundationHelper.createFolderFile(createRequest, passport);
        } catch (Exception e){
            baseResponse.addError(ErrorType.SYSTEM_ERROR, Constants.ERROR_MESSAGE_500);
            e.printStackTrace();
        }
	    String json = JSON.toJSONString(baseResponse);
	    PrintWriter out = response.getWriter();
	    out.println(json);
    }
}