fileaction.java

spring+struts+hibernate做的银行系统

/*
 * Created on 2006-4-14
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package cn.com.tym.blank.action;


import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.sql.DataSource;

import org.apache.struts.action.ActionError;
import org.apache.struts.action.ActionErrors;
import org.apache.struts.action.ActionForm;
import org.apache.struts.action.ActionForward;
import org.apache.struts.action.ActionMapping;

import org.apache.struts.actions.DispatchAction;
import org.apache.struts.upload.FormFile;

import cn.com.tym.blank.form.FileForm;
import cn.com.tym.pub.ConnPool;
import cn.com.tym.pub.DB;
import cn.com.tym.pub.EmployeeLogin;
import cn.com.tym.table.bean.RegisterBean;
import cn.com.tym.util.tablebean.UtilMyphotofile;
import cn.com.tym.util.tablebean.UtilMyphotos;
import cn.com.tym.util.tablebean.UtilRegister;
import cn.com.tym.util.text.DealText;
import cn.com.tym.vip.VipUtil;

/**
 * @author jack_booth
 *
 * TODO To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
public class FileAction extends DispatchAction{

	
	  public ActionForward createNewFile(ActionMapping mapping,
  			ActionForm form,
			HttpServletRequest request,
			HttpServletResponse response){
		 
		 
		if (EmployeeLogin.isLogin(request) == 0) {
			return mapping.findForward("welcome");
		}
		 ActionErrors errors = new ActionErrors();
		String id=((RegisterBean)request.getSession().getAttribute("user")).getRid();
	  	
		FileForm frm=(FileForm)form;
	  	 
	  	String PageForward="errors";
	
	 
	    UtilMyphotofile util=new UtilMyphotofile();
		  UtilRegister util2=new  UtilRegister();  
			 
			  RegisterBean bean=new RegisterBean();  
			  
	  	
	        DB db=null;	
	        try
	        {

	        		ConnPool pool=new ConnPool();
    		        DataSource dataSource=pool.Tomcat_JDBC_DataSource();
    		        db = new DB(dataSource);
    		        
    		        bean=util2.getViparticular(id,db);
    		        request.setAttribute("vipes",bean);
    		        request.setAttribute("mmId",id);
    		      int count=util.fileCount(id,db);
    		      //不包含默认目录
    	  if(count<=5)
	           {
    	  		int one =util.newfile(id,frm.getFilename(),db);
		          
    	  		if(one==1){
 
    	  			errors.add("newfiles",new ActionError("vip.myfile.newfiles.suceess"));
    	  			saveErrors(request,errors);
    	  			PageForward="myfile";
    	  		}
    	  		if(one==2){
    	  			
    	  		 	errors.add("filesome",new ActionError("vip.myfile.some"));
    	  			 saveErrors(request,errors); 
    	  			PageForward="myfile";
	           }
    	  		
	           }
		        
	           else{
	        	errors.add("size",new ActionError("vip.myfile.size"));
	        	saveErrors(request,errors);
	        	PageForward="myfile";
	  
	           }
	        }catch(Exception e)
	         {
	            e.printStackTrace();
	           
	         }
	        finally{
	        	db.close();
	        }
  	  	  	return mapping.findForward(PageForward);
	  }
	        
 
	  	public ActionForward fileUpload(ActionMapping mapping, 
	  			 ActionForm form,
	  			 HttpServletRequest request, 
	  			 HttpServletResponse response){
	  		 
	  		if (EmployeeLogin.isLogin(request) == 0) {
	  			return mapping.findForward("login");
	  		}
	  	 
			String id=((RegisterBean)request.getSession().getAttribute("user")).getRid();
	  		
	  		DB db=null;
	  		String PageForward="errors";
	  		FileForm frm =(FileForm)form;
	  		
	  	  
			 
			  RegisterBean bean=new RegisterBean();  
		      

	 
	  		String dir=servlet.getServletContext().getRealPath("/images/viphoto/");

	  		  ActionErrors errors=new ActionErrors();	
	  		FormFile file=frm.getUpfile();
	   
	  		if(file==null){
	         	errors.add("filesome",new ActionError("vip.upfile.some"));
	        	saveErrors(request,errors);
	  			 PageForward="myPhotoPage";
	  		}	
	  		UtilMyphotos util=new UtilMyphotos();
	  		 

	  	       try{
	  			 	ConnPool pool=new ConnPool();
	  		        DataSource dataSource=pool.Tomcat_JDBC_DataSource();
	  		        db = new DB(dataSource);
	    		    
	  				int  i=file.getFileName().lastIndexOf(".");
	  				String name2=file.getFileName().substring(i);			 
	  				String  name3=DealText.getDateTime();
	  				
	  				String ImageNewName=name3+name2;
	  				String dirDB="images/viphoto/"+ImageNewName;
	  		      request.setAttribute("mmId",id);
	  		        int su=util.insertPhoto(id,frm.getDirname(),frm.getFilename(),dirDB,frm.getContent(),db);
	  		 
	  		    if(su==0){
	  	 
	  	        	errors.add("error",new ActionError("vip.upfile.error"));
	  	     	saveErrors(request,errors);
		  		       ArrayList arraylist=util.getFilenames(id,db);
			  		      
			  		       request.setAttribute("result",arraylist);
	  				 PageForward="myPhotoPage";
	  		    }
	  	        if(su==2){
	  	    
	  	        	//����ͬ��ͼƬ,�����,
	  		        	errors.add("some",new ActionError("vip.myfile.some"));
	  		        	saveErrors(request,errors);
	 	  		       ArrayList arraylist=util.getFilenames(id,db);
	 	  		      
	 	  		       request.setAttribute("result",arraylist);
	  					 PageForward="myPhotoPage";
	  	        }
	  	        if(su==1){//�ϴ��ɹ�
	   
	  				String size=Integer.toString(file.getFileSize())+"bytes";
	  				InputStream streamIn=file.getInputStream();

	  	
	  				OutputStream streamOut=new FileOutputStream(dir+"/"+ImageNewName);		
	  				int bytesRead=0;			
	  				byte[] buffer=new byte[8192];
	  				while((bytesRead=streamIn.read(buffer,0,8192))!=-1){
	  					streamOut.write(buffer,0,bytesRead);
	  				}
	  				streamOut.close();
	  				streamIn.close();
	  	       	errors.add("filesuccess",new ActionError("vip.upfile.success"));
	  	         	saveErrors(request,errors);
	  				PageForward="upPhotoSuccess";		
	  	        }

	  		}catch(Exception e)
	  		{
	  		 
	  			e.printStackTrace();
	  			PageForward="error";
	  		}
	  	
	  	  	return mapping.findForward(PageForward);
	  	}
	/**
		  public ActionForward getFile(ActionMapping mapping,
	  			ActionForm form, 
				HttpServletRequest request,
				HttpServletResponse response){
			 
			 
			if (EmployeeLogin.isLogin(request) == 0) {
				return mapping.findForward("welcome");
			}
		 
			
			String id=((RegisterBean)request.getSession().getAttribute("user")).getRid();; 
		  	String PageForward="myfilePage";
	 
			ArrayList arraylist=new ArrayList();
			VipUtil util=new VipUtil();
	        try
	        {
	          
	         	ConnPool pool=new ConnPool();
		        DataSource dataSource=pool.Tomcat_JDBC_DataSource();
		        DB db = new DB(dataSource);
	           arraylist=util.getFilename(id,db);
	   		request.setAttribute("result",arraylist);
	        }catch(Exception e){
	            PageForward="errors";
	        }
		
		  	return mapping.findForward(PageForward);
	 
		  }
		  
*/
		    public ActionForward deleteFile(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response)
		    {
		        String PageForward;
		        if(EmployeeLogin.isLogin(request) == 0)
		            return mapping.findForward("welcome");
		        PageForward = "error";
		        String filenameId = request.getParameter("filenameId");
		         ActionErrors errors = new ActionErrors();
		        DB db = null;
		        VipUtil util = new VipUtil();
		        try
		        {
		            ConnPool pool = new ConnPool();
		            javax.sql.DataSource dataSource = pool.Tomcat_JDBC_DataSource();
		            db = new DB(dataSource);
		           
		            boolean check = util.isdeleteFilename(filenameId, db);
		            if(check)
		            {
		                 errors.add("vip.newfile.delete", new ActionError("vip.newfile.delete"));
		                  saveErrors(request, errors);
		                PageForward = "myfile";
		            }
		        }
		        catch(Exception e)
		        {
		        	 
		            e.printStackTrace();
		        }
		        finally
		        { 
		            db.close();
		            db=null;
		        }
		        return mapping.findForward(PageForward);
		    }

		    public ActionForward makeFile(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response)
		    {
		        if(EmployeeLogin.isLogin(request) == 0)
		            return mapping.findForward("welcome");
				String id=((RegisterBean)request.getSession().getAttribute("user")).getRid();
		        String PageForward = "errors";
		        FileForm frm = (FileForm)form;
		         ActionErrors errors= new ActionErrors();
		        VipUtil util = new VipUtil();
		        try
		        {
		            ConnPool pool = new ConnPool();
		            javax.sql.DataSource dataSource = pool.Tomcat_JDBC_DataSource();
		            DB db = new DB(dataSource);
		            if(!frm.getFilename().equals(""))
		            {
		                int bb = util.setFilename(id, frm.getFileId(), frm.getFilename(), db);
		                if(bb == 1)
		                {
		                   errors.add("newfile",new ActionError("vip.newfilename"));
		                    saveErrors(request, errors);
		                    PageForward = "myfile";
		                } else
		                {
		              errors.add("errorfile",new ActionError("vip.newfilenamerror"));
		                saveErrors(request, errors);
		                    PageForward = "myfile";
		                }
		            } else
		            {
		                errors.add("errorfile", new ActionError("vip.newfilenamerror"));
		              saveErrors(request, errors);
		                PageForward = "myfile";
		            }
		        }
		        catch(Exception e)
		        {
		            PageForward = "fail";
		            e.printStackTrace();
		        }
		        return mapping.findForward(PageForward);
		    }
 
		    public ActionForward myPhotos(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response)
		    {
		        if(EmployeeLogin.isLogin(request) == 0)
		            return mapping.findForward("admin");
				String id=((RegisterBean)request.getSession().getAttribute("user")).getRid();
		        String PageForward = "errors";
		        FileForm frm = (FileForm)form;
		         ActionErrors errors = new ActionErrors();
		        VipUtil util = new VipUtil();
		        try
		        {
		            ConnPool pool = new ConnPool();
		            javax.sql.DataSource dataSource = pool.Tomcat_JDBC_DataSource();
		            DB db = new DB(dataSource);
		            int bb = util.setFilename(id, frm.getFileId(), frm.getFilename(), db);
		        }
		        catch(Exception e)
		        {
		            e.printStackTrace();
		        }
		        return mapping.findForward(PageForward);
		    } 

	
}