下载处理Servlet工具类

原文：http://javatar.javaeye.com/blog/72265

 

2. package com.xxx.servlet;   
3.   
4. import java.io.File;   
5. import java.io.FileInputStream;   
6. import java.io.IOException;   
7. import java.io.InputStream;   
8. import java.io.OutputStream;   
9.   
10. import javax.servlet.ServletConfig;   
11. import javax.servlet.ServletException;   
12. import javax.servlet.http.HttpServlet;   
13. import javax.servlet.http.HttpServletRequest;   
14. import javax.servlet.http.HttpServletResponse;   
15.   
16. /**  
17.  * 下载处理Servlet  
18.  * @author liangfei0201@163.com  
19.  *  
20.  */  
21. public class DownloadServlet extends HttpServlet {   
22.   
23.     private static final long serialVersionUID = 1L;   
24.        
25.     private static final String DEFAULT_ENCODING = "UTF-8";   
26.        
27.     private static final int DEFAULT_BUFFER_SIZE = 4096;   
28.        
29.     private String encoding = DEFAULT_ENCODING;   
30.        
31.     private int bufferSize = DEFAULT_BUFFER_SIZE;   
32.        
33.     public void init(ServletConfig sc) throws ServletException {   
34.         super.init(sc);   
35.   
36.         String paramEncoding = sc.getInitParameter("encoding");   
37.         if (paramEncoding != null && paramEncoding.trim().length() > 0) {   
38.             encoding = paramEncoding.trim();   
39.         }   
40.            
41.         String paramSize = sc.getInitParameter("bufferSize");   
42.         if (paramSize != null && paramSize.matches("^[0-9]+$")) {   
43.             bufferSize = Integer.parseInt(paramSize);   
44.         }   
45.     }   
46.   
47.     public void doGet(HttpServletRequest request, HttpServletResponse response)   
48.             throws ServletException, IOException {   
49.         doPost(request, response);   
50.     }   
51.   
52.     public void doPost(HttpServletRequest request, HttpServletResponse response)   
53.             throws ServletException, IOException {   
54.         // 查找文件   
55.         File file = new File(getServletContext().getRealPath("/") + request.getServletPath());   
56.         if (! file.exists()) {   
57.             response.sendError(HttpServletResponse.SC_NOT_FOUND);   
58.             return ;   
59.         }   
60.            
61.         // 设置response   
62.         response.setContentType("application/x-msdownload; charset=" + encoding);   
63.         response.addHeader("content-type", "application/x-msdownload;");   
64.         response.addHeader("content-disposition", "attachment; filename="+ response.encodeURL(file.getName()));   
65.         response.setContentLength((int)file.length());   
66.            
67.         // 输出   
68.         InputStream in = new FileInputStream(file);   
69.         OutputStream out = response.getOutputStream();   
70.         try {   
71.             int length = 0;   
72.             byte[] buffer = new byte[bufferSize];   
73.             while ((length = in.read(buffer)) != -1) {   
74.                 out.write(buffer, 0, length);   
75.             }   
76.             out.flush();   
77.         } finally {   
78.             if (in != null) {   
79.                 in.close();   
80.             }   
81.         }   
82.     }   
83.   
84. }  

 

 

1. <servlet>  
2.     <servlet-name>download</servlet-name>  
3.     <servlet-class>com.xxx.servlet.DownloadServlet</servlet-class>  
4.     <init-param>  
5.         <param-name>encoding</param-name>  
6.         <param-value>GBK</param-value>  
7.     </init-param>  
8.     <init-param>  
9.         <param-name>bufferSize</param-name>  
10.         <param-value>8192</param-value>  
11.         </init-param>  
12.     <load-on-startup>4</load-on-startup>  
13. </servlet>  
14. <servlet-mapping>  
15.     <servlet-name>download</servlet-name>  
16.     <url-pattern>/download/*</url-pattern>  
17. </servlet-mapping>  
18. <servlet-mapping>  
19.     <servlet-name>download</servlet-name>  
20.     <url-pattern>*.csv</url-pattern>  
21. </servlet-mapping>  
22. <servlet-mapping>  
23.     <servlet-name>download</servlet-name>  
24.     <url-pattern>*.txt</url-pattern>  
25. </servlet-mapping>