v2.service.generic.library.utils.HttpStreamUtil.java Source code

Introduction

Here is the source code for v2.service.generic.library.utils.HttpStreamUtil.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package v2.service.generic.library.utils;

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.core.Response;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

/**
 *
 * @author lqshanshuo
 */
public class HttpStreamUtil {

    public static Response downLoad(HttpServletResponse response, String content, String defaultName) {
        if (defaultName == null || "".equalsIgnoreCase(defaultName)) {
            defaultName = "scripter_download.txt";
        }
        response.setContentType("application/octet-stream");
        response.setHeader("Content-Disposition", "attachment; filename=" + defaultName);
        try {
            OutputStream ostream = response.getOutputStream();
            byte[] buffer = content.getBytes("utf-8");
            ostream.write(buffer);
            ostream.flush();
            ostream.close();

            Response.ResponseBuilder builder = Response.ok();
            return builder.build();
        } catch (IOException ex) {
            return Response.status(Response.Status.INTERNAL_SERVER_ERROR).build();
        }
    }

    public static List<File> upLoad(HttpServletRequest request, String dir) {
        List<File> returnList = new ArrayList<>();
        if (!Misc.isNULL(dir)) {
            if ('/' != dir.charAt(0)) {
                dir = "/" + dir;
            }
        } else {
            dir = "/";
        }

        File cache_dir = new File("/tmp");
        File data_dir = new File(request.getSession().getServletContext().getRealPath("") + "/upload/" + dir);
        if (!cache_dir.isDirectory()) {
            Misc.mkdirs(cache_dir.getAbsolutePath());
        }
        if (!data_dir.isDirectory()) {
            Misc.mkdirs(data_dir.getAbsolutePath());
        }

        try {
            //FileItem  
            DiskFileItemFactory factory = new DiskFileItemFactory();
            //???100M
            factory.setSizeThreshold(100 * 1024 * 1024);
            //  
            factory.setRepository(cache_dir);
            //?  
            ServletFileUpload upload = new ServletFileUpload(factory);
            //?100M
            upload.setSizeMax(100 * 1024 * 1024);

            List /* FileItem */ items = upload.parseRequest(request);

            Iterator iter = items.iterator();
            while (iter.hasNext()) {
                FileItem item = (FileItem) iter.next();
                if (!item.isFormField()) {
                    //?  
                    String filename = item.getName();
                    if (filename.equals("") && item.getSize() == 0) {
                        break;
                    }
                    File uploadedFile = new File(data_dir + "/" + filename);
                    item.write(uploadedFile);
                    returnList.add(uploadedFile);
                    System.out.println(data_dir.getAbsolutePath() + "/" + filename);
                }
            }
        } catch (Exception e) {
            Logger.getLogger(HttpStreamUtil.class.getName()).log(Level.SEVERE, null, e);
        }
        return returnList;
    }
}

class Misc {

    public static File mkdirs(String dir) {

        boolean ret = false;
        File new_dir = new File(dir);
        if (!new_dir.isDirectory()) {
            ret = new_dir.mkdirs();
            if (!ret) {
                return null;
            }
        }
        return new_dir;
    }

    public static boolean isNULL(String obj) {
        if (null == obj || "".equals(obj)) {
            return true;
        }

        return false;
    }
}