webService快速入门入门系列《5》----自定义拦截器

上篇写到用cfx自带的拦截器。这次我们用自定义的拦截器，以后工作中会经常用到

在服务器端自定义拦截器

发布：

package com.lzh.ws;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import javax.jws.WebService;

import org.apache.cxf.binding.soap.SoapMessage;
import org.apache.cxf.endpoint.Server;
import org.apache.cxf.interceptor.Interceptor;
import org.apache.cxf.interceptor.LoggingInInterceptor;
import org.apache.cxf.interceptor.LoggingOutInterceptor;
import org.apache.cxf.jaxws.JaxWsServerFactoryBean;
import org.apache.cxf.phase.AbstractPhaseInterceptor;
@WebService(endpointInterface="com.lzh.ws.WebServiceTest")
public class WebServiceTestImpl implements WebServiceTest {
    private static HashMap<User, ArrayList> map;
    private static User user;
    private static ArrayList<Dog> list;
    static{
        user = new User(1,"刘中华");
        Dog dog1=new Dog(001,"小狗",12);
        Dog dog2=new Dog(002,"大狗",22);
        list=new ArrayList<Dog>();
        list.add(dog1);
        list.add(dog2);
        map=new HashMap<User, ArrayList>();
        map.put(user, list);
    }

    public int add(int i, int j) {
        // TODO Auto-generated method stub
        return i+j;
    }

    public List<Dog> getDogByUser(User u) {
        // TODO Auto-generated method stub

        if(user.equals(u)){
            ArrayList<Dog> list = map.get(u);
            return list;
        }
        return null;
    }


    public static void main(String[] args) {
        String url="http://localhost:9999/web";
//      EndpointImpl endpointImpl = (EndpointImpl) Endpoint.publish(url, new WebServiceTestImpl());
        /**
         * 注意，一定要是JaxWsServerFactoryBean这个类，这个类是cfx的类，不能用Endpoint这个类，这个是java自带的。
         * 给一个url和实体类.class    必须实体类的！！！！！！！
         *       
         * */
        JaxWsServerFactoryBean factory = new JaxWsServerFactoryBean();
        factory.setAddress(url);
        factory.setServiceClass(WebServiceTestImpl.class);
        /**
         * 添加拦截器，getInInterceptors()得到拦截器集合list。然后添加的是日志in   和  日志out
         */
    /*  factory.getInInterceptors().add(new LoggingInInterceptor());
        factory.getOutInterceptors().add(new LoggingOutInterceptor());*/

        /**
         * 添加自定义拦截器  AuthInInterceptor类是自定义的，需要继承AbstractPhaseInterceptor<SoapMessage>类
         */
        factory.getInInterceptors().add(new AuthInInterceptor());
        /**
         * 创建服务，启动
         */
        Server server = factory.create();
        server.start();
        System.out.println("发布webservice成功!");  
    }

    public HashMap<String, Dog> getDogMap() {
        HashMap<String, Dog> map= new HashMap<String, Dog>();
        int i=0;
        for(Dog d:list){
            map.put("第 "+ i++ +"条狗" + "",d);
        }
        return map;
    }

}

AuthInInterceptor拦截器类 名字随便起

package com.lzh.ws;

import org.apache.cxf.binding.soap.SoapMessage;
import org.apache.cxf.interceptor.Fault;
import org.apache.cxf.phase.AbstractPhaseInterceptor;
import org.apache.cxf.phase.Phase;

public class AuthInInterceptor extends AbstractPhaseInterceptor<SoapMessage>{

    public AuthInInterceptor() {
        super(Phase.PRE_INVOKE);  //调用之前拦截  SOAP信息
        // TODO Auto-generated constructor stub
    }
    //SoapMessage 是SOAP消息，就是客户端访问服务端的时候的信息，我们要做的就是在访问服务器之前解析这个消息，看看有没有我们所需要的东西，比如账号，密码之类的。没有就不让访问。他是xml文件。带的信息是在heard里面，不是在body里面
    public void handleMessage(SoapMessage msg) throws Fault {
        // TODO Auto-generated method stub
        //所携带的额外信息实在头部的，所以要得到headers，他是个list集合。如果没有就抛异常
        List<Header> headers = msg.getHeaders();
        if(headers==null||headers.size()<1){
            throw new Fault(new Exception("没有header,调用不了"));
        }
        Header header = headers.get(0);
        Element element = (Element) header.getObject();
        NodeList userId = element.getElementsByTagName("userId");
        NodeList userPass = element.getElementsByTagName("userPass");


        if(userId.getLength() != 1){
            throw new Fault(new Exception("用户格式不对"));
        }
        if(userPass.getLength() != 1){
            throw new Fault(new Exception("密码格式不对"));
        }

        String u = userId.item(0).getTextContent();
        String p = userPass.item(0).getTextContent();

        if(u.equals("lzh") && p.equals("123")){
            System.out.println("成功");
        }else{
            throw new Fault(new Exception("不对!!!!"));
        }


    }

}

客户端要也要添加自定义拦截器，添加头部信息

package com.lzh.ws.client;

import java.util.List;

import org.apache.cxf.frontend.ClientProxy;
import org.apache.cxf.interceptor.LoggingInInterceptor;
import org.apache.cxf.interceptor.LoggingOutInterceptor;

import com.lzh.ws.Dog;
import com.lzh.ws.Entry;
import com.lzh.ws.StringDog;
import com.lzh.ws.User;
import com.lzh.ws.WebServiceTest;
import com.lzh.ws.WebServiceTestImplService;

public class Client {
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        WebServiceTestImplService impl = new WebServiceTestImplService();
        WebServiceTest implPort = impl.getWebServiceTestImplPort();

        /**
         * 添加日志拦截器     先创建一个代理类，是cfx的类，参数是上面的implPort（这个类是webService的远程代理），
         */
        org.apache.cxf.endpoint.Client proxy = ClientProxy.getClient(implPort);
        proxy.getOutInterceptors().add(new YanZhengInInterceptor("lzh","123"));
        proxy.getOutInterceptors().add(new LoggingOutInterceptor());


        int add = implPort.add(1, 2);
        System.out.println(add);
        User user = new User();
        user.setId(1);
        user.setName("刘中华");
        List<Dog> list = implPort.getDogByUser(user);
        for(Dog d:list){
            System.out.println(d.getName());
        }
        System.out.println("-------------");
        StringDog dogMap = implPort.getDogMap();
        for(Entry e:dogMap.getList()){
            System.out.println(e.getValue().getName());
        }
    }
}

客户端拦截器YanZhengInInterceptor

package com.lzh.ws.client;

import java.util.List;

import javax.xml.namespace.QName;

import org.apache.cxf.binding.soap.SoapMessage;
import org.apache.cxf.headers.Header;
import org.apache.cxf.helpers.DOMUtils;
import org.apache.cxf.interceptor.Fault;
import org.apache.cxf.phase.AbstractPhaseInterceptor;
import org.apache.cxf.phase.Phase;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

public class YanZhengInInterceptor extends AbstractPhaseInterceptor<SoapMessage> {
    private String username;
    private String pass;
    public YanZhengInInterceptor(String username,String pass) {
        super(Phase.PREPARE_SEND);   //在消息发送之前调用
        this.username=username;
        this.pass=pass;
    }

    public void handleMessage(SoapMessage msg) throws Fault {
        // TODO Auto-generated method stub
        List<Header> headers = msg.getHeaders();
        /**
         * 得到了soap对象就要把验证信息添加到头部。他是个xml片段。
         */
        Document doc = DOMUtils.createDocument();
        //创建头部
        Element authHeader = doc.createElement("authHeader");
        //创建username和userpass 并赋值   创建元素的名字要和服务器那边解析的要一致
        Element userId = doc.createElement("userId");
        userId.setTextContent(username);
        Element userPass = doc.createElement("userPass");
        userPass.setTextContent(pass);
        //把创建的子元素添加到头部。
        authHeader.appendChild(userPass);
        authHeader.appendChild(userId);
        //把创建的头部添加到头部
        headers.add(new Header(new QName("刘中华"), authHeader));
        /**
         * 所得到的格式比如是：
         * <soap:Envelope>
         * <soap:Header>
         *      <authHeader>
         *          <userPass>1234</userPass>
         *          <userId>lzh</userId>
         *      </authHeader>
         * </soap:Header>
         * <soap:Body>
         *      <ns2:add xmlns:ns2="http://ws.lzh.com/">
         *          <arg0>1</arg0>
         *          <arg1>2</arg1>
         *      </ns2:add>
         * </soap:Body>
         * </soap:Envelope>
         */
    }

}