update.java

Eclips3.2 + mysql5.5 用Struts2编写的图书管理系统

package action;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Date;

import org.apache.struts2.ServletActionContext;

import com.bean.Tbook;
import com.dao.BookDAO;
import com.opensymphony.xwork2.ActionSupport;
/**
 * 
 * 书籍修改action
 *
 */
public class Update extends ActionSupport {
	private static final long serialVersionUID = 572146812454l;

	private static final int BUFFER_SIZE = 16 * 1024;

	private Tbook tbook;

	private String msg;

	private File myFile;

	private String fileName;

	private String contentType;

	private String imageFileName;

	public String getMsg() {
		return msg;
	}

	public void setMsg(String msg) {
		this.msg = msg;
	}

	public Tbook getTbook() {
		return tbook;
	}

	public void setTbook(Tbook tbook) {
		this.tbook = tbook;
	}

	/**
	 * 将文件复制到项目的imgs文件夹下
	 */

	public static void copy(File upFile, File downFile) {
		try {
			InputStream in = null;
			OutputStream out = null;
			try {
				in = new BufferedInputStream(new FileInputStream(upFile),
						BUFFER_SIZE);
				out = new BufferedOutputStream(new FileOutputStream(downFile),
						BUFFER_SIZE);
				byte[] buffer = new byte[BUFFER_SIZE];
				while (in.read(buffer) > 0) {
					out.write(buffer);
				}
			} finally {
				if (null != in) {
					in.close();
				}
				if (null != out) {
					out.close();
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}

	}

	private static String getExtention(String fileName) {
		int pos = fileName.lastIndexOf(".");
		return fileName.substring(pos);
	}

	@Override
	public String execute() throws Exception {
		// TODO Auto-generated method stub
		try {
			if (!fileName.equals("")) {
				imageFileName = new Date().getTime() + getExtention(fileName);
				File imageFile = new File(ServletActionContext
						.getServletContext().getRealPath("/imgs")
						+ "/" + imageFileName);
				copy(myFile, imageFile);
				System.out.println("文件的名称:" + imageFileName);
			}
		} catch (RuntimeException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}

		if (null != imageFileName) {
			tbook.setBookimg(imageFileName);
		}
		if (!new BookDAO().setBook(tbook)) {
			msg = "修改书籍<<" + tbook.getBookname() + ">>失败!请检查输入的信息是否错误!";
			return ERROR;
		}
		msg = "修改书籍<<" + tbook.getBookname() + ">>成功!";
		return SUCCESS;
	}

	public String getMyFileFileName() {
		return fileName;
	}

	public String getMyFileContentType() {
		return contentType;
	}

	public void setMyFileContentType(String contentType) {
		this.contentType = contentType;
	}

	public void setMyFileFileName(String fileName) {
		this.fileName = fileName;
	}

	public String getImageFileName() {
		return imageFileName;
	}

	public String getContentType() {
		return contentType;
	}

	public String getFileName() {
		return fileName;
	}

	public void setFileName(String fileName) {
		this.fileName = fileName;
	}

	public File getMyFile() {
		return myFile;
	}

	public void setMyFile(File myFile) {
		this.myFile = myFile;
	}
}