课堂上讲课时写的一个简单的仿struts2框架

课堂上讲课时写的一个简单的仿struts2框架

package intecerptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import struts.Action;

/**
*
* @author Administrator
*/
public interface  ActionInvocation {
  public Action getAction() ;
  public String invoke() throws Exception;;
     public HttpServletRequest getRequest() ;
    public HttpServletResponse getResponse();
    public boolean isValidate();
     public void setValidation(boolean v);
}



/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package intecerptor;

import java.util.ArrayList;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import struts.Action;

/**
*
* @author Administrator
*/
public class DefaultActionInvocation implements ActionInvocation {
    private Action action;

   private  HttpServletRequest request;
   private HttpServletResponse response;

    public DefaultActionInvocation(HttpServletRequest request, HttpServletResponse response) {
       this.request=request;
       this.response=response;
    }
    public Action getAction() {
       return action;
    }

    public void setAction(Action action){
        this.action=action;
    }

    List<Interceptor> interceptors=new ArrayList<Interceptor>();

    public void addInterceptor(Interceptor interceptor){
        interceptors.add(interceptor);
    }
    private int index=-1;
    private  String result="";
    public String invoke() throws Exception{
      if(index==interceptors.size()-1){
          if(this.isValidate())
              result=action.execute();
          else
             result=Action.INPUT;
      }else{
          index++;
          ((Interceptor)interceptors.get(index)).interceptor(this);
      }

      return result;

    }

    /**
     * @return the request
     */
    public HttpServletRequest getRequest() {
        return request;
    }

    /**
     * @return the response
     */
    public HttpServletResponse getResponse() {
        return response;
    }

    public boolean isValidate() {
        return v;
    }

    private boolean v;
    public void setValidation(boolean v){
        this.v=v;
    }
   
}




/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package intecerptor;

/**
*
* @author Administrator
*/
public abstract class DefaultInterceptor  implements Interceptor{

    public void destory() throws Exception {
    }

    public void init() throws Exception {      
    }

}



/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package intecerptor;

/**
*
* @author Administrator
*/
public interface  Interceptor {
      public void init() throws Exception;
      public void destory() throws Exception;
      public String interceptor(ActionInvocation  actionInvocation) throws Exception;

}



/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/
package intecerptor;

import java.util.Iterator;
import java.util.Map;
import org.apache.commons.beanutils.BeanUtils;

/**
*
* @author Administrator
*/
public class ParamInterceptor extends DefaultInterceptor {

    public String interceptor(ActionInvocation actionInvocation) throws Exception {
        BeanUtils.populate(actionInvocation.getAction(), actionInvocation.getRequest().getParameterMap());
        String result = actionInvocation.invoke();
        Map param = BeanUtils.describe(actionInvocation.getAction());
        param.remove("class");
        for (Iterator it = param.keySet().iterator(); it.hasNext();) {
            String key = (String) it.next();
            actionInvocation.getRequest().setAttribute(key, param.get(key));
        }
        return result;
    }
}



package struts;

/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

/**
*
* @author Administrator
*/
public interface Action {
    String SUCCESS="success";
    String INPUT ="input";
    String FAILED="failed";
   
   String execute() throws Exception;
}



package struts;


import java.io.InputStream;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import java.util.logging.Level;
import java.util.logging.Logger;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import org.apache.commons.beanutils.BeanUtils;
import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.io.SAXReader;

/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

/**
*
* @author Administrator
*/
public class ActionFactory {

    public static Map<String ,XmlMapping>  actions;
 
    public static void initActions(InputStream input){
        SAXReader reader = new SAXReader();
        try {
            Document document = reader.read(input);
            Element root=document.getRootElement();
            actions=new HashMap<String ,XmlMapping>();
            treeWalk(root,null);

        } catch (DocumentException ex) {
           ex.printStackTrace();
        }
        //
    }


    public static void treeWalk(Element element,XmlMapping xm) {
       for (int i = 0, size = element.nodeCount(); i < size; i++)     {
           Node node = element.node(i);
           if (node instanceof Element) {
                Element el=(Element)node;
               if("action".equals(node.getName())){
                   if(xm==null)
                       xm=new XmlMapping();

               
                 for(Iterator iter = el.attributeIterator(); iter.hasNext();) {
                       Attribute attribute = (Attribute) iter.next();
                       String attr=attribute.getName();
                       if("name".equals(attr)){
                           xm.setActionName(attribute.getValue());
                       }
                       if("type".equals(attr)){
                           xm.setActionType(attribute.getValue());
                       }

                 }
                    actions.put(xm.getActionName(),xm);
               }
               if("result".equals(node.getName())){
                    for(Iterator iter = el.attributeIterator(); iter.hasNext();) {
                       Attribute attribute = (Attribute) iter.next();
                       String attr=attribute.getName();
                       if("name".equals(attr)){
                            xm.getResults().put(attribute.getValue(),el.getText());
                       }


                 }
               }

              treeWalk((Element) node,xm);
           } else { // do something....
           }
       }
}


    public static String getResult(String actionName, String code){
        XmlMapping mp=(XmlMapping)actions.get(actionName);
        String result=mp.getResults().get(code);
        return result;
    }
    public static  Action createAction(String actionName,Map parameter){
         Action action=null;
         try {
            String type=((XmlMapping)actions.get(actionName)).getActionType();
            action=(Action) Class.forName(type).newInstance();
           BeanUtils.populate(action,parameter);
        } catch (Exception ex) {
           ex.printStackTrace();
        }
          return action;
    }

    public static void main(String[] args ){
        try {
            try {
                Action action = (Action) Class.forName("actions.LoginAction").newInstance();
                System.out.println(action);
            } catch (InstantiationException ex) {
                Logger.getLogger(ActionFactory.class.getName()).log(Level.SEVERE, null, ex);
            } catch (IllegalAccessException ex) {
                Logger.getLogger(ActionFactory.class.getName()).log(Level.SEVERE, null, ex);
            }
        } catch (ClassNotFoundException ex) {
            Logger.getLogger(ActionFactory.class.getName()).log(Level.SEVERE, null, ex);
        }
    }
}



/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

import java.util.HashMap;
import java.util.Map;

/**
*
* @author Administrator
*/
public abstract class ActionSupport implements Action  ,Validator{
    protected String message;
    private Map<String,String> fieldErrors =new HashMap<String,String>();

    public abstract String execute() throws Exception ;

    public void validatoin() throws Exception {
    
    }

    /**
     * @return the message
     */
    public String getMessage() {
        return message;
    }

    /**
     * @param message the message to set
     */
    public void setMessage(String message) {
        this.message = message;
    }
    public boolean isFieldErrors(){
        return !fieldErrors.isEmpty();
    }

    /**
     * @return the fieldErrors
     */
    public Map<String, String> getFieldErrors() {
        return fieldErrors;
    }

    /**
     * @param fieldErrors the fieldErrors to set
     */
    public void setFieldErrors(Map<String, String> fieldErrors) {
        this.fieldErrors = fieldErrors;
    }

    public void addFiledErrors(String key,String value){
        fieldErrors.put(key, value);
    }
}


/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

import java.util.Map;

/**
*
* @author Administrator
*/
public interface  ContextAware {
      void setContext(Map context);
      Map getContext();
}


/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

import java.util.Map;

/**
*
* @author Administrator
*/
public interface  RequestAware {
      void setRequest(Map request);
      Map getRequest();
}


/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

import java.util.Map;

/**
*
* @author Administrator
*/
public interface  SessionAware {
      void setSession(Map session);
      Map getSession();
}


/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

/**
*
* @author Administrator
*/
public interface  Validator {

    void  validatoin() throws Exception;
}

/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

import java.util.HashMap;
import java.util.Map;

/**
*
* @author Administrator
*/
public class XmlMapping {
   private String actionName;
   private String actionType;
   private Map<String ,String>  results= new HashMap<String ,String>();;

    /**
     * @return the actionName
     */
    public String getActionName() {
        return actionName;
    }

    /**
     * @param actionName the actionName to set
     */
    public void setActionName(String actionName) {
        this.actionName = actionName;
    }

    /**
     * @return the actionType
     */
    public String getActionType() {
        return actionType;
    }

    /**
     * @param actionType the actionType to set
     */
    public void setActionType(String actionType) {
        this.actionType = actionType;
    }

    /**
     * @return the results
     */
    public Map<String, String> getResults() {
        return results;
    }

    /**
     * @param results the results to set
     */
    public void setResults(Map<String, String> results) {
        this.results = results;
    }
  

}


/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/

package struts;

import intecerptor.DefaultActionInvocation;
import intecerptor.LogInterceptor;
import intecerptor.ParamInterceptor;
import intecerptor.TimeInterceptor;
import intecerptor.ValidatorInterceptor;
import java.io.IOException;
import java.io.InputStream;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.beanutils.BeanUtils;

/**
*
* @author Administrator
*/
public class StrutsDispatcher extends HttpServlet {
  
  

  
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
         doPost( request,  response);
    }

    /**
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        String next;
        String action=request.getRequestURI().replaceAll(request.getContextPath(), "");//login.action
        action=action.substring(1,action.lastIndexOf("."));//login
        if(action!=null&&!"".equals(action)){
            Action actionInstance =ActionFactory.createAction(action,request.getParameterMap());
            try {
                DefaultActionInvocation da=new DefaultActionInvocation(request,response);
                da.setAction(actionInstance);
                ParamInterceptor paramInterceptor=new ParamInterceptor();
                LogInterceptor log=new LogInterceptor();
                TimeInterceptor time=new TimeInterceptor();
                ValidatorInterceptor validator=new ValidatorInterceptor();
                da.addInterceptor(paramInterceptor);
               // da.addInterceptor(validator);
                da.addInterceptor(log);
                da.addInterceptor(time);

                 String  result=da.invoke();
                 result=ActionFactory.getResult(action, result);
                 request.getRequestDispatcher(result).forward(request, response);
            } catch (Exception ex) {
                Logger.getLogger(StrutsDispatcher.class.getName()).log(Level.SEVERE, null, ex);
            }

        }
    }

    @Override
    public void init(ServletConfig config) throws ServletException {
        String file=config.getInitParameter("struts");
        System.out.println(config.getServletContext());
        InputStream input=config.getServletContext().getResourceAsStream(file);
        System.out.println(input);
        ActionFactory.initActions(input);
    }

  

}