学习webservice之cxf（7）：cxf自定义拦截器

package com.rg2.webservice.impl;

import org.apache.cxf.jaxws.JaxWsServerFactoryBean;

import com.rg2.intercepter.MyIntercepter;
import com.rg2.webservice.HelloWorld;

public class Server {

    public static void main(String[] args) {
        System.out.println("web service start");
        HelloWorld implementor = new HelloWorldImpl();
        String address = "http://localhost/helloWorld";
//        Endpoint.publish(address, implementor);//jdk实现暴露webservice接口
        JaxWsServerFactoryBean factoryBean = new JaxWsServerFactoryBean();
        factoryBean.setAddress(address);//设置暴露地址
        factoryBean.setServiceClass(HelloWorld.class);//接口类
        factoryBean.setServiceBean(implementor);//设置实现类
        factoryBean.getInInterceptors().add(new LoggingInInterceptor());//添加in拦截器 日志拦截器
        factoryBean.getOutInterceptors().add(new LoggingInInterceptor());//添加out拦截器 日志拦截器
        
        factoryBean.getInInterceptors().add(new MyIntercepter());
        factoryBean.create();//创建webservice接口
        System.out.println("web service started");
    }

}

package com.rg2.intercepter;

import java.util.List;

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

public class MyIntercepter extends AbstractPhaseInterceptor<SoapMessage> {

    public MyIntercepter() {
        super(Phase.PRE_INVOKE);//在调用方法之前调用自定义拦截器
    }

    @Override
    public void handleMessage(SoapMessage message) throws Fault {
        List<Header> headers = message.getHeaders();
        if(null == headers && headers.size() == 0) {
            throw new Fault(new IllegalArgumentException("没有Header，拦截器实施拦截"));
        }
        Header firstHeader = headers.get(0);
        Element ele = (Element) firstHeader.getObject();
        NodeList uList = ele.getElementsByTagName("userName");
        NodeList pList = ele.getElementsByTagName("password");
        if(uList.getLength()!=1){
            throw new Fault(new IllegalArgumentException("用户名格式不对"));
        }
        if(pList.getLength()!=1){
            throw new Fault(new IllegalArgumentException("密码格式不对"));
        }
        String userName = uList.item(0).getTextContent();
        String password = pList.item(0).getTextContent();
        if(!userName.equals("rg2")||!password.equals("123456")){
            throw new Fault(new IllegalArgumentException("用户名或者密码错误！"));
        }
    }

}

package com.rg2.webservice;

import java.util.List;

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

import com.rg2.interceptor.AddHeaderIntercepter;

public class Client {

    public static void main(String[] args) {
        HelloWorldService service = new HelloWorldService();
        HelloWorld helloWorldPort = service.getHelloWorldPort();
        org.apache.cxf.endpoint.Client client = ClientProxy.getClient(helloWorldPort);
        
        client.getOutInterceptors().add(new AddHeaderIntercepter("rg2","123456"));//添加自定义拦截器
        client.getInInterceptors().add(new LoggingInInterceptor());//添加in拦截器 日志拦截器
        client.getOutInterceptors().add(new LoggingInInterceptor());//添加out拦截器 日志拦截器
        
        MyRoleArray getroles = helloWorldPort.getroles();
        List<MyRole> roleList = getroles.item;
        for (int i = 0; i < roleList.size(); i++) {
            MyRole my = roleList.get(i);
            System.out.print(my.key + ":");
            for (Role role : my.value) {
                System.out.print(role.getId()+","+role.getRoleName());
            }
            System.out.println("===============");
        }
    }

}

package com.rg2.interceptor;

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.interceptor.Interceptor;
import org.apache.cxf.message.Message;
import org.apache.cxf.phase.AbstractPhaseInterceptor;
import org.apache.cxf.phase.Phase;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

public class AddHeaderIntercepter extends AbstractPhaseInterceptor<SoapMessage> {
    
    private String userName;
    private String password;
    
    public AddHeaderIntercepter(String userName, String password) {
        super(Phase.PREPARE_SEND);// 准备发送SOAP消息的时候调用拦截器
        this.userName = userName;
        this.password = password;
    }

    @Override
    public void handleMessage(SoapMessage message) throws Fault {
        List<Header> headerList = message.getHeaders();
        
        Document doc = DOMUtils.createDocument();
        Element ele = doc.createElement("authHeader");
        Element uElement = doc.createElement("userName");
        uElement.setTextContent(userName);
        Element pElement = doc.createElement("password");
        pElement.setTextContent(password);
        
        ele.appendChild(uElement);
        ele.appendChild(pElement);
        
        headerList.add(new Header(new QName("rg2"), ele));
        
    }

}