package pers.mengb.persys.servlet;

import javax.servlet.http.*;

import javax.servlet.*;

import java.io.*;

import java.io.File;

import java.io.IOException;

import java.io.PrintWriter;

import java.util.List;

import org.apache.commons.fileupload.FileItem;

import org.apache.commons.fileupload.FileUploadException;

import org.apache.commons.fileupload.disk.DiskFileItemFactory;

import org.apache.commons.fileupload.servlet.ServletFileUpload;

public class UploadServlet extends HttpServlet {

    private static final long serialVersionUID = 1L;

    protected void doPost(HttpServletRequest request, HttpServletResponse response)

            throws ServletException, IOException {

        /*

         * @author Administrator

         * 文件上传

         * 具体步骤：

         * 1）获得磁盘文件条目工厂 DiskFileItemFactory （要导包）

         * 2） 利用 request 获取 真实路径 ，供临时文件存储，和 最终文件存储 ，这两个存储位置可不同，也可相同

         * 3）对 DiskFileItemFactory 对象设置一些 属性

         * 4）高水平的API文件上传处理  ServletFileUpload upload = new ServletFileUpload(factory);

         * 目的是调用 parseRequest（request）方法  获得 FileItem 集合list ，

         *

         * 5）在 FileItem 对象中 获取信息，   遍历， 判断 表单提交过来的信息 是否是 普通文本信息  另做处理

         * 6）

         *    第一种. 用第三方 提供的  item.write( new File(path,filename) );  直接写到磁盘上

         *    第二种. 手动处理

         *

         */

            request.setCharacterEncoding("utf-8");    //设置编码

            DiskFileItemFactory factory = new DiskFileItemFactory();                 //获得磁盘文件条目工厂

            String path = this.getServletContext().getRealPath("uploadDir");           //获取文件需要上传到的路径

            //String path = request.getRealPath("/upload");    //该方法不推荐使用啦

                //如果没以下两行设置的话，上传大的 文件 会占用 很多内存，

                //设置暂时存放的 存储室 , 这个存储室，可以和 最终存储文件 的目录不同

                /*

                 * 原理 它是先存到 暂时存储室，然后在真正写到 对应目录的硬盘上，

                 * 按理来说 当上传一个文件时，其实是上传了两份，第一个是以 .tem 格式的

                 * 然后再将其真正写到 对应目录的硬盘上

                 */

                factory.setRepository(new File(path));

                factory.setSizeThreshold(1024*1024) ;                                //设置 缓存的大小，当上传文件的容量超过该缓存时，直接放到 暂时存储室

                ServletFileUpload upload = new ServletFileUpload(factory);           //高水平的API文件上传处理

                try {

                    List<FileItem> list = upload.parseRequest(request);

                    for(FileItem item : list)

                    {

                        String name = item.getFieldName();                            //获取表单的属性名字

                        //如果获取的 表单信息是普通的 文本 信息

                        if(item.isFormField())

                        {

                            //获取用户具体输入的字符串 ，名字起得挺好，因为表单提交过来的是 字符串类型的

                            String value = item.getString() ;

                            request.setAttribute(name, value);

                        }

                        //对传入的非 简单的字符串进行处理 ，比如说二进制的 图片，电影这些

                        else

                        {

                            /*

                             * 以下三步，主要获取 上传文件的名字

                             */

                            String value = item.getName() ;           //获取路径名

                            int start = value.lastIndexOf("\\");      //索引到最后一个反斜杠

                            String filename = value.substring(start+1);//截取 上传文件的 字符串名字，加1是 去掉反斜杠，

                            request.setAttribute(name, filename);

                            //真正写到磁盘上

                            //它抛出的异常 用exception 捕捉

                            //item.write( new File(path,filename) );//第三方提供的

                            //手动写的

                            OutputStream out = new FileOutputStream(new File(path,filename));

                            InputStream in = item.getInputStream() ;

                            int length = 0 ;

                            byte [] buf = new byte[1024] ;

                            System.out.println("获取上传文件的总共大小为："+item.getSize()+"字节");

                            // in.read(buf) 每次读到的数据存放在   buf 数组中

                            while( (length = in.read(buf) ) != -1)

                            {

                                //在   buf 数组中 取出数据 写到 （输出流）磁盘上

                                out.write(buf, 0, length);

                            }

                            in.close();

                            out.close();

                        }

                    }

                } catch (FileUploadException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

                catch (Exception e) {

                    // TODO Auto-generated catch block

                    //e.printStackTrace();

                }

                //request.getRequestDispatcher("filedemo.jsp").forward(request, response);

            }

}