dynamicinvoker.java

webservice压力测试模拟器,可以并发的同时测试多台服务器的性能.

/*根据wsdl定位动态调用其提供的服务 */
package com.gpdi.www.ws.client;

import org.apache.axis.Constants;
import org.apache.axis.encoding.ser.SimpleDeserializer;
import org.apache.axis.wsdl.gen.Parser;
import org.apache.axis.wsdl.symbolTable.*;

import javax.swing.JOptionPane;
import javax.wsdl.*;
import javax.wsdl.extensions.soap.SOAPAddress;
import javax.xml.namespace.QName;
import javax.xml.rpc.Call;
import javax.xml.rpc.encoding.Deserializer;
import javax.xml.rpc.encoding.DeserializerFactory;

import java.util.*;

// Web Service Dynamic Invoker
public class DynamicInvoker
{
	private Parser wsdlParser = null;
	private Map services = null;
    //构造函数
	public DynamicInvoker(String wsdlURL) throws Exception
	{
		//构造一个wsdl分析器
		wsdlParser = new Parser();
		//分析wsdl
		wsdlParser.run(wsdlURL);
		//根据wsdl得到此wsdl提供的服务集合.
		services = enumSymTabEntry(ServiceEntry.class);
	}

	public Map invoke(String serviceName,
					  String portName,
					  String operationName,
					  Vector parameterValues) throws Exception
	{
		//参数的集合.
		Vector inputs = new Vector();
		String returnName = null;
        //根据服务的键取得值.
		ServiceEntry serviceEntry = (ServiceEntry) services.get(serviceName);
		//得到wsdl提供的服务对象.
		Service service = serviceEntry.getService();
		//a Service will be created with a WSDL document and along with a serviceName 
		//you can then ask for a Call object that will allow you to invoke a Web Service. 
		org.apache.axis.client.Service clientService = new
			org.apache.axis.client.Service(wsdlParser, service.getQName());

		Call call = clientService.createCall(QName.valueOf(portName),
											 QName.valueOf(operationName));
		//设置超时时间为一分钟
		( (org.apache.axis.client.Call) call).setTimeout(new Integer(60 * 1000));

		BindingEntry bindingEntry = getBindingEntry(serviceName, portName);
		//根据方法名得到某一方法.
		Operation o = getOperation(bindingEntry, operationName);
		//得到某一方法的参数
		Parameters parameters = bindingEntry.getParameters(o);

		if (parameters.returnParam != null)
		{
			QName returnType = org.apache.axis.wsdl.toJava.Utils.
				getXSIType(parameters.returnParam);
			QName returnQName = parameters.returnParam.getQName();
			returnName = returnQName.getLocalPart();
		}

		int size = parameters.list.size();
		//遍历参数集合.再根据参数的类型(输入、输出、输入输出)分别构造参数对象再加到集合里.
		for (int i = 0; i < size; i++)
		{
			Parameter p = (Parameter) parameters.list.get(i);
			switch (p.getMode())
			{
			case Parameter.IN:
				inputs.add(getParamData( (org.apache.axis.client.Call) call,
										p,
										(String) parameterValues.elementAt(i)));
				break;
			case Parameter.OUT:
				break;
			case Parameter.INOUT:
				inputs.add(getParamData( (org.apache.axis.client.Call) call,
										p,
										(String) parameterValues.elementAt(i)));
				break;
			}
		}

		Object ret = call.invoke(inputs.toArray());
		Map outputs = call.getOutputParams();
		HashMap map = new HashMap();
		if (ret != null && returnName != null)
		{
			map.put(returnName, ret);
		}
		//当输出参数不为空时，则遍历输出参数的集合.
		if (outputs != null)
		{
			for (Iterator i = outputs.keySet().iterator(); i.hasNext(); )
			{
				Object obj = i.next();
				String name;
				Object value;
				//如果输出参数为String类型则直接转换。
				if (obj.getClass().getName().equals("java.lang.String"))
				{
					name = (String) obj;
				}
				else
				{
					name = ( (QName) obj).getLocalPart();
				}
				value = outputs.get(obj);
				map.put(name, value);
			}
		}
		return map;
	}

	public Vector enumServiceNames()
	{
		Vector v = new Vector();
		Iterator i = services.keySet().iterator();
		while (i.hasNext())
		{
			String s = (String) i.next();
			v.addElement(s);
		}
		return v;
	}

	public Vector enumPortNames(String serviceName)
	{
		Vector v = new Vector();
		ServiceEntry serviceEntry = (ServiceEntry) services.get(serviceName);
		Map ports = serviceEntry.getService().getPorts();
		Iterator i = ports.keySet().iterator();
		while (i.hasNext())
		{
			String s = (String) i.next();
			v.addElement(s);
		}
		return v;
	}

	public Vector enumOperationNames(String serviceName, String portName)
	{
		Vector v = new Vector();
		BindingEntry entry = getBindingEntry(serviceName, portName);
		Set operations = entry.getOperations();
		Iterator i = operations.iterator();
		while (i.hasNext())
		{
			Operation o = (Operation) i.next();
			String s = o.getName();
			v.addElement(s);
		}
		return v;
	}

	public Parameters enumParameters(String serviceName, String portName,
									 String operationName)
	{
		BindingEntry entry = getBindingEntry(serviceName, portName);
		Operation o = getOperation(entry, operationName);
		Parameters parameters = entry.getParameters(o);
		return parameters;
	}

	public String getParameterModeString(Parameter p)
	{
		String ret = null;
		switch (p.getMode())
		{
		case Parameter.IN:
			ret = "[IN]";
			break;
		case Parameter.INOUT:
			ret = "[IN, OUT]";
			break;
		case Parameter.OUT:
			ret = "[OUT]";
			break;
		}
		return ret;
	}

	private BindingEntry getBindingEntry(String serviceName, String portName)
	{
		//得到指定的web服务.从Map里根据主键找值.
		ServiceEntry serviceEntry = (ServiceEntry) services.get(serviceName);
		//根据web服务得到端口.
		Port port = serviceEntry.getService().getPort(portName);
		//绑定端口.
		Binding binding = port.getBinding();
		SymbolTable table = wsdlParser.getSymbolTable();
		return table.getBindingEntry(binding.getQName());
	}
    //根据方法名从端口绑定的服务的集合里得到某一个方法对象.
	private Operation getOperation(BindingEntry entry, String operationName)
	{
		Set operations = entry.getOperations();
		Iterator i = operations.iterator();
		while (i.hasNext())
		{
			Operation o = (Operation) i.next();
			if (operationName.equals(o.getName()))
			{
				return o;
			}
		}
		return null;
	}

	// return Map of <QName.getLocalPart, SymTabEntry>
	private Map enumSymTabEntry(Class cls)
	{
		//用来返回的Hashmap
		HashMap ret = new HashMap();
		//根据wsdl分析器分析出来的
		HashMap map = wsdlParser.getSymbolTable().getHashMap();
		//HashMap的方法entrySet()将HashMap里一行记录(键值)变成一个对象(Map.Entry)放到Set集合里去.
		Iterator iterator = map.entrySet().iterator();
		while (iterator.hasNext())
		{
			//HashMap根据值取主键
			Map.Entry entry = (Map.Entry) iterator.next();
			//取键
			QName key = (QName) entry.getKey();
			//取值
			Vector v = (Vector) entry.getValue();
			//得到SymTabEntry的个数.
			int size = v.size();
			for (int i = 0; i < size; ++i)
			{
				SymTabEntry symTabEntry = (SymTabEntry) v.elementAt(i);
				//判断传进来的对象同wsdl分析器分析出来的对象是否为同一类型对象.如是则将其加入到返回的HashMap里去.
				if (cls.isInstance(symTabEntry))
				{
					ret.put(key.getLocalPart(), symTabEntry);
				}
			}
		}
		return ret;
	}
    //打包参数对象
	private Object getParamData(org.apache.axis.client.Call c, Parameter p,
								String arg) throws Exception
	{
		// Get the QName representing the parameter type
		QName paramType = org.apache.axis.wsdl.toJava.Utils.getXSIType(p);

		TypeEntry type = p.getType();
		//如果参数的类型为基本的数据类型就序列化它。否则抛出不能转换的巽常!
		if (type instanceof BaseType && ( (BaseType) type).isBaseType())
		{
			DeserializerFactory factory = c.getTypeMapping().getDeserializer(
				paramType);
			Deserializer deserializer = factory.getDeserializerAs(Constants.
				AXIS_SAX);
			if (deserializer instanceof SimpleDeserializer)
			{
				return ( (SimpleDeserializer) deserializer).makeValue(arg);
			}
		}
		throw new RuntimeException("not know how to convert '" + arg
								   + "' into " + c);
	}
}