public MetadataUploadService() {
   File servletTmpDir;
   try {
     // Get the temporary directory used by the servlet
     servletTmpDir = (File) getServletContext().getAttribute("javax.servlet.context.tempdir");
   } catch (Exception ex) {
     // Just use the default system temp dir (less secure)
     servletTmpDir = null;
   }
   // Create a disk file item factory for processing requests
   DiskFileItemFactory factory = new DiskFileItemFactory();
   if (servletTmpDir != null) {
     // Use the temporary directory for the servlet for large files
     factory.setRepository(servletTmpDir);
   }
   // Create the file uploader using this factory
   metadataUpload = new ServletFileUpload(factory);
 }
  /** @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response) */
  protected void doPost(HttpServletRequest request, HttpServletResponse response)
      throws ServletException, IOException {
    // TODO Auto-generated method stub
    String message = "File upload.";
    List<String> filenames = null;
    isMultipart = ServletFileUpload.isMultipartContent(request);
    HttpSession session = request.getSession(false);
    User loggedUser = null;
    if (session != null) loggedUser = (User) session.getAttribute("loggedUser");
    if (loggedUser == null) {
      request.getRequestDispatcher("login.jsp").forward(request, response);
    } else {
      Book book = new Book();
      if (isMultipart) {
        ServletContext context = this.getServletConfig().getServletContext();
        DiskFileItemFactory factory = new DiskFileItemFactory();
        // maximum size that will be stored in memory
        factory.setSizeThreshold(maxMemSize);
        // Location to save data that is larger than maxMemSize.
        File repoPath = (File) context.getAttribute("javax.servlet.context.tempdir");
        factory.setRepository(repoPath);

        // Create a new file upload handler
        ServletFileUpload upload = new ServletFileUpload(factory);
        // maximum file size to be uploaded.
        upload.setSizeMax(maxFileSize);

        try {
          List<FileItem> fileItems = upload.parseRequest(new ServletRequestContext(request));
          filePath = request.getServletContext().getRealPath("/") + "picture/";
          filenames = new ArrayList<String>();

          for (FileItem fi : fileItems) {
            if (!fi.isFormField()) {
              // Get the uploaded file parameters
              String fieldName = fi.getFieldName();
              if (fi.getName() != null && !fi.getName().equals("")) {
                String fileName = URLEncoder.encode(fi.getName(), "UTF-8");

                String contentType = fi.getContentType();

                boolean isInMemory = fi.isInMemory();
                long sizeInBytes = fi.getSize();
                file = new File(filePath + fileName);
                fi.write(file);
                book.setPictureURL("picture/" + fileName);
              }
            } else {
              String fieldName = fi.getFieldName();
              String fieldValue = fi.getString();
              if (fieldName.equals("title")) {
                if (fieldValue != null) {
                  book.setTitle(fieldValue);
                  ;
                }
              }
              if (fieldName.equals("authors")) book.setAuthors(fieldValue);
              if (fieldName.equals("publisher")) book.setPublisher(fieldValue);
              if (fieldName.equals("description"))
                if (fieldValue != null) {
                  book.setDescription(fieldValue);
                }
              if (fieldName.equals("genre")) book.setGenre(fieldValue);
              if (fieldName.equals("numberInStock"))
                book.setNumberInStock(Integer.parseInt(fieldValue));
              if (fieldName.equals("publicationYear"))
                book.setPublicationYear(Integer.parseInt(fieldValue));
              if (fieldName.equals("price")) book.setPrice(Double.parseDouble(fieldValue));
            }
          }
        } catch (Exception ex) {
          System.out.println(ex);
        }
      }

      BookDAO bookDao = new BookDAO();
      bookDao.add(book);

      response.sendRedirect("LoginBookPreviewViewController");
    }
  }
Exemplo n.º 3
0
  private void updateCover(
      HttpServletRequest request, HttpServletResponse response, final String method)
      throws FileUploadException, ServletException, IOException, IllegalAccessException,
          InvocationTargetException {
    List<String> types = Arrays.asList(".jpg", ".gif", ".png", ".jpeg"); // supported
    // type

    DiskFileItemFactory dfi = new DiskFileItemFactory(); // apache's
    // component

    dfi.setSizeThreshold(1024 * 1024); // 1M, and default is 10k, and
    // excessed will be saved in the
    // cache file
    String path = getServletContext().getRealPath("/WEB-INF/temp");
    dfi.setRepository(new File(path)); // temporary cached file

    ServletFileUpload sf = new ServletFileUpload(dfi);
    sf.setHeaderEncoding("utf-8");
    // sf.setFileSizeMax(1024*1024*5); //set single file's max size

    List<FileItem> lf = sf.parseRequest(new ServletRequestContext(request));
    Map<String, String> map = new HashMap<String, String>();
    List<String> idf = new LinkedList<String>();
    FileItem fi = null; // for stream file
    String ids[];

    for (FileItem item : lf) {
      if (item.isFormField()) { // ordinary input
        String name = item.getFieldName();
        String value = item.getString("utf-8");
        map.put(name, value);
        if ("category".equals(name)) {
          idf.add(value);
        }
        System.out.println(name + " " + value);
      } else { // stream
        String name = item.getName();
        if (name == null || "".equals(name)) {
          continue;
        }
        String ext = name.substring(name.lastIndexOf("."));
        System.out.println(ext);
        if (!types.contains(ext.toLowerCase())) {
          request.setAttribute("message", "sorryia..  i don't support type " + ext);
          return;
        }

        fi = item;
      }
    }

    Book book = new Book();
    ConvertUtils.register(new DateLocaleConverter(), Date.class);
    BeanUtils.populate(book, map);
    BookDaoImpl bdi = new BookDaoImpl();
    ids = new String[idf.size()];
    idf.toArray(ids);
    int id;

    if (method.equals(BOOK_ADD)) {
      id = bdi.add(book, ids);
      request.setAttribute("message", "add book successfully...");
    } else {
      id = Integer.parseInt(map.get("id"));
      book.setId(id);
      bdi.update(book);
      bdi.updateCategory(id, ids);
      request.setAttribute("message", "update book successfully...");
    }

    if (fi != null) { // upload file
      InputStream is = fi.getInputStream();
      String Spath = getServletContext().getRealPath("/img/bookcover/");
      System.out.println(Spath);
      FileOutputStream fos = new FileOutputStream(new File(Spath + id + ".jpg"));
      byte[] b = new byte[1024];
      int len = 0;
      while ((len = is.read(b)) > 0) {
        fos.write(b, 0, len);
      }
      fos.close();
      is.close();
      fi.delete(); // the temporary file would be deleted.
    }
  }
Exemplo n.º 4
0
  public void doGet(HttpServletRequest request, HttpServletResponse response)
      throws ServletException, IOException {

    request.setCharacterEncoding("UTF-8");
    response.setCharacterEncoding("UTF-8");

    PrintWriter out = response.getWriter();

    String idx = request.getParameter("idx");

    // 获得磁盘文件条目工厂
    DiskFileItemFactory factory = new DiskFileItemFactory();
    // 获取文件需要上传到的路径
    String rootPath = getServletContext().getRealPath("/upload");

    // 如果没以下两行设置的话,上传大的 文件 会占用 很多内存,
    // 设置暂时存放的 存储室 , 这个存储室,可以和 最终存储文件 的目录不同
    /** 原理 它是先存到 暂时存储室,然后在真正写到 对应目录的硬盘上, 按理来说 当上传一个文件时,其实是上传了两份,第一个是以 .tem 格式的 然后再将其真正写到 对应目录的硬盘上 */
    File uploadDir = new File(rootPath);
    if (!uploadDir.exists()) {
      uploadDir.mkdirs();
    }
    factory.setRepository(uploadDir);
    // 设置 缓存的大小,当上传文件的容量超过该缓存时,直接放到 暂时存储室
    factory.setSizeThreshold(1024 * 1024);

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

    try {
      // 可以上传多个文件
      List<FileItem> list = (List<FileItem>) upload.parseRequest(request);

      for (FileItem item : list) {
        // 获取表单的属性名字
        String fileName = item.getName();

        // 如果获取的 表单信息是普通的 文本 信息
        if (item.isFormField()) {
        }
        // 对传入的非 简单的字符串进行处理 ,比如说二进制的 图片,电影这些
        else {

          String md5Name = MD5Code.getInstance().getCode(fileName + System.currentTimeMillis());
          String fiffux = fileName.substring(fileName.lastIndexOf("."));

          // 真正写到磁盘上
          // 它抛出的异常 用exception 捕捉
          File f = new File(rootPath + File.separatorChar, md5Name + fiffux);

          if (!f.exists()) {
            f.createNewFile();
          }
          item.write(f); // 第三方提供的

          JSONObject result = new JSONObject();
          JSONObject payload = new JSONObject();

          responseJavaScript(
              out, "parent.uploadImageComplete('" + idx + "','" + (md5Name + fiffux) + "')");

          out.close();
        }
      }

    } catch (FileUploadException e) {
      e.printStackTrace();
    } catch (Exception e) {
      e.printStackTrace();
    }
  }