struts2 上传 下载

package com.dynastarter.sc.file.action;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
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.util.ArrayList;
import java.util.Date;
import java.util.List;

import org.apache.struts2.ServletActionContext;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Controller;
import org.springside.modules.orm.Page;
import org.springside.modules.security.springsecurity.SpringSecurityUtils;

import com.dynastarter.sc.file.baseaction.BaseAction;
import com.dynastarter.sc.file.po.FileManager;
import com.dynastarter.sc.file.po.HistoryManager;
import com.dynastarter.sc.file.po.User;
import com.dynastarter.sc.file.service.FileManagerService;
import com.dynastarter.sc.file.service.HistoryManagerService;
import com.dynastarter.sc.file.service.UserService;


@Controller
@Scope("prototype")
public class DocumentManagerAction extends BaseAction{

public DocumentManagerAction(){
System.out.println("DocumentManagerAction======>");
}

private final int BUFFER_SIZE = 16 * 1024 ;

private int id;
private File file; //上传文件
private String fileFileName; //上传文件�
private String fileContentType; //文件类型
private int parentId;

private String downFileName;

private Page<FileManager> page=new Page<FileManager>();

@Autowired
FileManagerService fileManagerService;

@Autowired
UserService userSerivce;

@Autowired
HistoryManagerService historyService;

public File getFile() {
return file;
}

public void setFile(File file) {
this.file = file;
}



public String getFileContentType() {
return fileContentType;
}

public void setFileContentType(String fileContentType) {
this.fileContentType = fileContentType;
}

public String getFileFileName() {
return fileFileName;
}

public void setFileFileName(String fileFileName) {
this.fileFileName = fileFileName;
}


/**
* 创建目录
* @return
*/
public String createDirectory() throws Exception{
String filePath=getRequest().getRealPath("/");
String parentPath=fileManagerService.getFileManager(new Long(parentId)).getFilename();
//创建目录
fileFileName=java.net.URLDecoder.decode(fileFileName, "UTF-8"); //将js代码中的编码的过的参数解码

File myFilePath = new File(filePath+"/"+parentPath,fileFileName);
if(!myFilePath.exists())
{

//创建该目录
myFilePath.mkdirs();
//保存数据库
FileManager fileManager=new FileManager();
fileManager.setFilename(fileFileName);
fileManager.setFileContentType("directory");
fileManager.setFileSize(0);
fileManager.setParentDirectory(parentId);
fileManager.setUploadTime(new Date());
fileManager.setIsDirectory(true);
fileManager.setIsFile(false);
//
String loginName=SpringSecurityUtils.getCurrentUserName();
User user=userSerivce.login(loginName,SpringSecurityUtils.getCurrentUser().getPassword());
if(user!=null){
fileManager.setUser(user);
}
else{
user=userSerivce.getUser(new Long(1));
fileManager.setUser(user);
}
fileManagerService.save(fileManager);


HistoryManager history=new HistoryManager();
history.setFileName(fileFileName);
history.setFileLength(0);
history.setFileType("目录");
history.setOpetionType("创建目录");
history.setUploadUser(user);
history.setUploadTime(new Date());
historyService.save(history);

getResponse().setContentType("json;charset=UTF-8");
StringBuffer sb=new StringBuffer("{");
sb.append("'id':"+fileManager.getId()+",'status':'success'");
sb.append("}");
getResponse().getWriter().write(sb.toString());
}
else{
getResponse().setContentType("json;charset=UTF-8");
getResponse().getWriter().write("{'status':'allReady'}");
}
return null;
}

public void copy(File src, File dst) {
try {
InputStream in = null ;
OutputStream out = null ;
try {
in = new BufferedInputStream( new FileInputStream(src), BUFFER_SIZE);
out = new BufferedOutputStream( new FileOutputStream(dst), BUFFER_SIZE);
byte [] buffer = new byte [BUFFER_SIZE];
while (in.read(buffer) > 0 ) {
out.write(buffer);
}
out.flush();

} finally {
if ( null != in) {
in.close();
}
if ( null != out) {
out.close();
}
}
} catch (Exception e) {
e.printStackTrace();
}
}

/**
* 批量删除
* @return
*/
public String modelDelete(){
return null;
}


/**
* 上传文件
* @return
*/
public String upLoadFile() throws Exception{
try {
List<String> list=new ArrayList<String>();
list.add("txt");
list.add("jpg");
list.add("png");
list.add("gif");
list.add("doc");
list.add("rar");
list.add("xls");
list.add("zip");

String tagDirectory = getRequest().getRealPath("/");
FileManager parentFileManager=fileManagerService.getFileManager(new Long(parentId));
FileManager rootFileManager=fileManagerService.getFileManager(new Long(1));

File target = new File(tagDirectory+"/"+rootFileManager.getFilename()+"/"+parentFileManager.getFilename(), fileFileName);
if(!target.exists()){
String houZhuiMing=fileFileName.substring(fileFileName.lastIndexOf(".")+1);
if(list.contains(houZhuiMing.toLowerCase())){
copy(file,target);

//保存到数据库
FileManager fileManager=new FileManager();
fileManager.setFilename(fileFileName);
fileManager.setFileSize((int)file.length());
fileManager.setFileContentType(houZhuiMing);
fileManager.setParentDirectory(parentId);
fileManager.setIsDirectory(false);
fileManager.setIsFile(true);
fileManager.setUploadTime(new Date());
String loginName=SpringSecurityUtils.getCurrentUserName();
User user=userSerivce.login(loginName,SpringSecurityUtils.getCurrentUser().getPassword());
if(user!=null){
fileManager.setUser(user);
}
else{
user=userSerivce.getUser(new Long(1));
fileManager.setUser(user);
}
fileManagerService.save(fileManager);


HistoryManager history=new HistoryManager();
history.setFileName(fileFileName);
history.setFileLength(fileManager.getFileSize()/1024);
history.setFileType("目录");
history.setOpetionType("上传文件");
history.setUploadUser(user);
history.setUploadTime(new Date());
historyService.save(history);

getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("上传成功");
}
else{
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("只能上传图片,文本,word,zip,rar,xls文件");
}
}
else{
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("文件已经存在");
}
} catch (Exception e) {
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("上传失败");
throw e;
}
return null;
}


/**
* 修改目录
* @return
*/
public String updateDirectory() throws Exception{
String filePath=getRequest().getRealPath("/");
if(parentId==1){
FileManager fileManager=fileManagerService.getFileManager(new Long(parentId));
File file=new File(filePath+"/"+fileManager.getFilename());
String newFileName=filePath.substring(0,filePath.lastIndexOf("\\"))+"\\"+fileFileName;
File file1=new File(newFileName);
StringBuffer sb=new StringBuffer("{");
if(file1.exists()){ //先判断该目录是否存在
sb.append("'status':'allReady'");
}
else{
file.renameTo(new File(newFileName));
fileManager.setFilename(fileFileName);
fileManagerService.updateFileManager(fileManager);
sb.append("'status':'success'");
}
sb.append("}");


HistoryManager history=new HistoryManager();
history.setFileName(fileFileName);
history.setFileLength(0);
history.setFileType("目录");
history.setOpetionType("修改目录");
String loginName=SpringSecurityUtils.getCurrentUserName();
User user=userSerivce.login(loginName,SpringSecurityUtils.getCurrentUser().getPassword());
history.setUploadUser(user);
history.setUploadTime(new Date());
historyService.save(history);

getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write(sb.toString());
}
else{
FileManager fileManager=fileManagerService.getFileManager(new Long(parentId));
FileManager parentFile=fileManagerService.getFileManager(new Long(fileManager.getParentDirectory()));
filePath=filePath+"\\"+parentFile.getFilename()+"\\"+fileManager.getFilename();
fileManager.setFilename(fileFileName);
File file=new File(filePath);
String newFileName=filePath.substring(0,filePath.lastIndexOf("\\"))+"\\"+fileFileName;
File file1=new File(newFileName);
StringBuffer sb=new StringBuffer("{");
if(file1.exists()){ //先判断该目录是否存在
sb.append("'status':'allReady'");
}
else{
file.renameTo(new File(newFileName));
fileManagerService.updateFileManager(fileManager);
sb.append("'status':'success'");
}
sb.append("}");

HistoryManager history=new HistoryManager();
history.setFileName(fileFileName);
history.setFileLength(0);
history.setFileType("目录");
history.setOpetionType("修改目录");
String loginName=SpringSecurityUtils.getCurrentUserName();
User user=userSerivce.login(loginName,SpringSecurityUtils.getCurrentUser().getPassword());
history.setUploadUser(user);
history.setUploadTime(new Date());
historyService.save(history);

getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write(sb.toString());


}
return null;
}


/**
* 删除文件
* @return
* @throws Exception
*/
public String deleteDirectoryOrFile()throws Exception{
try {
FileManager fileManager=fileManagerService.getFileManager(new Long(parentId));
String filePath=getRequest().getRealPath("/");
FileManager parentFileManager=fileManagerService.getFileManager(new Long(fileManager.getParentDirectory()));
FileManager rootFileManager=fileManagerService.getFileManager(new Long(1));
File dFile=null;
if(parentFileManager.getFilename()==rootFileManager.getFilename()){ //删除目录
dFile=new File(filePath+"/"+rootFileManager.getFilename()+"/"+fileManager.getFilename());
}
else{ //删除文件
dFile = new File(filePath+"/"+rootFileManager.getFilename()+"/"+parentFileManager.getFilename()+"/"+fileManager.getFilename());
}
if(dFile.exists())
{
dFile.delete();

HistoryManager history=new HistoryManager();
history.setFileName(fileManager.getFilename());
history.setFileLength(0);
history.setFileType("目录");
history.setOpetionType("删除文件,目录");
String loginName=SpringSecurityUtils.getCurrentUserName();
User user=userSerivce.login(loginName,SpringSecurityUtils.getCurrentUser().getPassword());
history.setUploadUser(user);
history.setUploadTime(new Date());
historyService.save(history);

fileManagerService.deleteFileManager(new Long(parentId));
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("删除成功!~");
}
else{
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("该文件已经删除或者不存在!~");
}
} catch (IOException e) {
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("删除失败!~");
}
return null;
}

/**
* @return
* @throws Exception
*/
public String loadDocument() throws Exception{
StringBuilder sb=new StringBuilder("[");
String root=getRequest().getParameter("root");
if(root!=null&&"true".equals(root)){
FileManager fileManager= fileManagerService.getFileManager(new Long(1));
sb.append("{");
sb.append("'id':").append(""+fileManager.getId()+",");
sb.append("'fileName':").append("'"+fileManager.getFilename()+"',");
sb.append("'fileSize':").append(fileManager.getFileSize()+",");
sb.append("'fileContentType':").append("'"+fileManager.getFileContentType()+"',");
sb.append("'uploadUser':").append("'"+fileManager.getUser().getLoginName()+"',");
sb.append("'uploadTime':").append("'"+fileManager.getUploadTime()+"',");
sb.append("'parentDirectory':").append("'"+fileManager.getParentDirectory()+"',");
sb.append("'isDirectory':").append("'"+fileManager.getIsDirectory()+"',");
sb.append("'isFile':").append("'"+fileManager.getIsFile()+"',");
sb.append("'childFiles':");
sb.append("''");
sb.append("}");
}
else{
List<FileManager> list=fileManagerService.getFileManagerByParentId(id);
for(FileManager fileManager1:list){
sb.append("{");
sb.append("'id':").append(""+fileManager1.getId()+",");
sb.append("'fileName':").append("'"+fileManager1.getFilename()+"',");
sb.append("'fileSize':").append(fileManager1.getFileSize()+",");
sb.append("'fileContentType':").append("'"+fileManager1.getFileContentType()+"',");
sb.append("'uploadUser':").append("'"+fileManager1.getUser().getLoginName()+"',");
sb.append("'parentDirectory':").append("'"+fileManager1.getParentDirectory()+"',");
sb.append("'uploadTime':").append("'"+fileManager1.getUploadTime()+"',");
sb.append("'isDirectory':").append("'"+fileManager1.getIsDirectory()+"',");
sb.append("'isFile':").append("'"+fileManager1.getIsFile()+"'");
sb.append("},");
}
if(sb.lastIndexOf(",")>0){
sb=new StringBuilder(sb.substring(0,sb.lastIndexOf(",")));
}
}

sb.append("]");
System.out.println(sb.toString());
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write(sb.toString());
return null;
}

@Override
public String execute() throws Exception {
return SUCCESS;
}

/**
* 下载文件
* @return
*/
public InputStream getDownLoadFile() throws Exception{
FileManager fileManager=fileManagerService.getFileManager(new Long(parentId));
fileFileName=fileManager.getFilename();
FileManager parentFileManager=fileManagerService.getFileManager(new Long(fileManager.getParentDirectory()));
String filePath=getRequest().getRealPath("/");
FileManager rootFileManager=fileManagerService.getFileManager(new Long(1));
filePath=filePath+"/"+rootFileManager.getFilename()+"/"+parentFileManager.getFilename()+"/"+fileManager.getFilename();
File file=new File(filePath);
if(file.exists()){ //判断文件是否存在
HistoryManager history=new HistoryManager();
history.setFileName(fileManager.getFilename() );
history.setFileLength(fileManager.getFileSize()/1024);
history.setFileType("文件");
history.setOpetionType("下载文件");
String loginName=SpringSecurityUtils.getCurrentUserName();
User user=userSerivce.login(loginName,SpringSecurityUtils.getCurrentUser().getPassword());
history.setUploadUser(user);
history.setUploadTime(new Date());
historyService.save(history);

return ServletActionContext.getServletContext().getResourceAsStream(rootFileManager.getFilename()+"/"+parentFileManager.getFilename()+"/"+fileManager.getFilename());
}
else{ //
getResponse().setContentType("text/html;charset=UTF-8");
getResponse().getWriter().write("对不起该文件不存在");
}
return null;
}



public int getId() {
return id;
}

public void setId(int id) {
this.id = id;
}

public int getParentId() {
return parentId;
}

public void setParentId(int parentId) {
this.parentId = parentId;
}

public String getDownFileName() {
try {
return new String(fileFileName.getBytes(), "ISO8859-1");
} catch (Exception e) {
e.printStackTrace();
}
return null;
}

public void setDownFileName(String downFileName) {
this.downFileName = downFileName;
}


}


----
xml配置:
<package name="document" extends="struts-default" namespace="/document">
<action name="documentManagerAction" class="documentManagerAction">
<result name="success" type="stream">
<param name="contentType">application/octet-stream</param>
<param name="contentDisposition">attachment;filename=${downFileName}</param>
<param name="inputName">downLoadFile</param>
<param name="bufferSize">4096</param>
</result>
</action>

</package>