81,091
社区成员
发帖
与我相关
我的任务
分享
import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.URL;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.io.IOUtils;
import org.apache.log4j.LogManager;
import org.apache.log4j.Logger;
/**
* 文件类
*
*/
public class FileUtil {
public static Map<String,String> contentTypeMap = new HashMap<String,String>();
static{
contentTypeMap.put("pptx", "application/vnd.openxmlformats-officedocument.presentationml.presentation");
contentTypeMap.put("xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
contentTypeMap.put("xls", "application/vnd.ms-excel");
contentTypeMap.put("doc", "application/msword");
contentTypeMap.put("docx", "application/msword");
}
private static Logger logger = LogManager.getLogger(FileUtil.class);
/**
* 上传
* @param request
* @param response
* @throws Exception
*/
public static void upload(HttpServletRequest request,
HttpServletResponse response) throws Exception {
DataInputStream in = null;
FileOutputStream fileOut = null;
try {
response.setCharacterEncoding("utf-8");
PrintWriter out = response.getWriter();
// 定义上载文件的最大字节
int MAX_SIZE = 102400 * 102400;
// 创建根路径的保存变量
String rootPath;
// 声明文件读入类
// 取得客户端的网络地址
String remoteAddr = request.getRemoteAddr();
// 获得服务器的名字
String serverName = request.getServerName();
// 取得互联网程序的绝对地址
String realPath = request.getRealPath(serverName);
realPath = realPath.substring(0, realPath.lastIndexOf("\\"));
// 创建文件的保存目录
rootPath = realPath + "\\upload\\";
// 取得客户端上传的数据类型
String contentType = request.getContentType();
if (contentType.indexOf("multipart/form-data") >= 0) {
// 读入上传的数据
in = new DataInputStream(request.getInputStream());
int formDataLength = request.getContentLength();
if (formDataLength > MAX_SIZE) {
out.println("<P>上传的文件字节数不可以超过" + MAX_SIZE + "</p>");
return;
}
// 保存上传文件的数据
byte[] dataBytes = new byte[formDataLength];
int byteRead = 0;
int totalBytesRead = 0;
// 上传的数据保存在byte数组
while (totalBytesRead < formDataLength) {
byteRead = in.read(dataBytes, totalBytesRead,
formDataLength);
totalBytesRead += byteRead;
}
// 根据byte数组创建字符串
String file = new String(dataBytes);
// out.println(file);
// 取得上传的数据的文件名
String saveFile = file
.substring(file.indexOf("filename=\"") + 10);
saveFile = saveFile.substring(0, saveFile.indexOf("\n"));
saveFile = saveFile.substring(saveFile.lastIndexOf("\\") + 1,
saveFile.indexOf("\""));
int lastIndex = contentType.lastIndexOf("=");
// 取得数据的分隔字符串
String boundary = contentType.substring(lastIndex + 1,
contentType.length());
// 创建保存路径的文件名
String fileName = rootPath + saveFile;
// out.print(fileName);
int pos;
pos = file.indexOf("filename=\"");
pos = file.indexOf("\n", pos) + 1;
pos = file.indexOf("\n", pos) + 1;
pos = file.indexOf("\n", pos) + 1;
int boundaryLocation = file.indexOf(boundary, pos) - 4;
// out.println(boundaryLocation);
// 取得文件数据的开始的位置
int startPos = ((file.substring(0, pos)).getBytes()).length;
// out.println(startPos);
// 取得文件数据的结束的位置
int endPos = ((file.substring(0, boundaryLocation)).getBytes()).length;
// out.println(endPos);
// 检查上载文件是否存在
File checkFile = new File(fileName);
if (checkFile.exists()) {
out.println("<p>" + saveFile + "文件已经存在.</p>");
}
// 检查上载文件的目录是否存在
File fileDir = new File(rootPath);
if (!fileDir.exists()) {
fileDir.mkdirs();
}
in.close();
// 创建文件的写出类
fileOut = new FileOutputStream(fileName);
// 保存文件的数据
fileOut.write(dataBytes, startPos, (endPos - startPos));
fileOut.close();
out.println(saveFile + "文件成功上载.</p>");
} else {
String content = request.getContentType();
out.println("<p>上传的数据类型不是multipart/form-data</p>");
}
} catch (Exception message) {
throw message;
} finally {
if (in != null) {
in.close();
}
if (fileOut != null) {
fileOut.close();
}
}
}
/**
* 获取文件名称
* @param path
* @return
*/
public static String getFileName(String path) {
String fileName = "";
if(path.indexOf("/")!=-1){
fileName = path.substring(path.lastIndexOf("/") + 1,
path.length());
}else{
fileName = path.substring(path.lastIndexOf("/") + 1,
path.length());
}
return fileName;
}
/**
*
* @param path
* @return
*/
public static String getSuffix(String path) {
String suffix = path
.substring(path.lastIndexOf('.') + 1, path.length());
return suffix;
}
/**
* 下载
* @param filePath
* @param request
* @param response
*/
public static void downFile(String filePath, HttpServletRequest request,
HttpServletResponse response) {
InputStream is = null;
OutputStream os = null;
try {
is = new FileInputStream(filePath);
String fileName = getFileName(filePath);
String filename = "";
if (request.getHeader("User-Agent").toUpperCase().indexOf("MSIE") !=-1
||request.getHeader("User-Agent").toUpperCase().indexOf("RV:11")!=-1) {
filename = URLEncoder.encode(fileName, "UTF-8");
}else{
filename = new String(fileName.getBytes("UTF-8"),"ISO8859-1");
}
String suffix = getSuffix(filePath);
String contentType = contentTypeMap.get(suffix);
if(contentType == null){
contentType = "application/octet-stream";
}
response.setHeader("Location",filename);
response.setHeader("Content-Disposition", "attachment;filename="+filename);
response.setContentType(contentType);
os = response.getOutputStream();
IOUtils.copy(is, os);
response.flushBuffer();
os.close();
is.close();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (is != null) {
is.close();
}
if (os != null) {
os.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* 获取文件路径
* @param filePath
* @return
*/
public static String getFilePath(String filePath) {
URL url = Thread.currentThread().getContextClassLoader().getResource(filePath);
return url.getFile();
}
public static boolean delete(String fileName) {
File file = new File(fileName);
if (!file.exists()) {
return false;
} else {
if (file.isFile())
return deleteFile(fileName);
else
return deleteDirectory(fileName);
}
}
/**
* 删除单个文件
*
* @param fileName
* 要删除的文件的文件名
* @return 单个文件删除成功返回true,否则返回false
*/
public static boolean deleteFile(String fileName) {
File file = new File(fileName);
// 如果文件路径所对应的文件存在,并且是一个文件,则直接删除
if (file.exists() && file.isFile()) {
if (file.delete()) {
return true;
} else {
return false;
}
} else {
return false;
}
}
}