dubbo-remoting/src/test/java/com/alibaba/dubbo/remoting/Main.java

/*
 * Copyright 1999-2011 Alibaba Group.
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *  
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.alibaba.dubbo.remoting;

import java.io.Serializable;
import java.util.Random;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;

import com.alibaba.dubbo.common.URL;
import com.alibaba.dubbo.remoting.exchange.ExchangeChannel;
import com.alibaba.dubbo.remoting.exchange.Exchangers;
import com.alibaba.dubbo.remoting.exchange.ResponseFuture;
import com.alibaba.dubbo.remoting.exchange.support.Replier;
import com.alibaba.dubbo.remoting.exchange.support.ReplierDispatcher;

/**
 * Main
 */

public class Main
{
	public static void main(String[] args) throws Exception
	{
		startServer(9010);
		mutliThreadTest(10,9010);
		dataPackageTest(9010);
	}

	private static void startServer(int port) throws Exception
	{
	    ReplierDispatcher dispatcher = new ReplierDispatcher();
	    dispatcher.addReplier(RpcMessage.class, new RpcMessageHandler());
	    dispatcher.addReplier(Object.class, new Replier<Object>() {
			public Object reply(ExchangeChannel channel, Object msg)
			{
				for(int i=0;i<10000;i++)
					System.currentTimeMillis();
				System.out.println("handle:"+msg+";thread:"+Thread.currentThread().getName());
				return new StringMessage("hello world");
			}
		});
		Exchangers.bind(URL.valueOf("dubbo://localhost:" + port), dispatcher);
	}

	static void dataPackageTest(int port) throws Exception
	{
		ExchangeChannel client = Exchangers.connect(URL.valueOf("dubbo://localhost:" + port));
		Random random = new Random();
		for(int i=5;i<100;i++)
		{
			StringBuilder sb = new StringBuilder();
			for(int j=0;j<i*100;j++)
				sb.append("("+random.nextLong()+")");
			Main.Data d = new Main.Data();
			d.setData(sb.toString());
			client.request(d).get();
		}
		System.out.println("send finished.");
	}

	static void mutliThreadTest(int tc,final int port) throws Exception
	{
		Executor exec = Executors.newFixedThreadPool(tc);
		for(int i=0;i<tc;i++)
			exec.execute(new Runnable(){
				public void run() {
					try{ test(port); }catch(Exception e){ e.printStackTrace(); }
				}
			});
	}

	private static void test(int port) throws Exception
	{
	    ExchangeChannel client = Exchangers.connect(URL.valueOf("dubbo://localhost:" + port));
		MockResult result = (MockResult)client.request(new RpcMessage(DemoService.class.getName(),"plus",new Class<?>[]{int.class, int.class},new Object[]{55,25})).get();
		System.out.println("55+25="+result.getResult());

		for(int i=0;i<100;i++)
			client.request(new RpcMessage(DemoService.class.getName(),"sayHello", new Class<?>[]{String.class},new Object[]{"qianlei"+i}));

		for(int i=0;i<100;i++)
			client.request(new Main.Data());

		System.out.println("=====test invoke=====");
		for(int i=0;i<100;i++){
			ResponseFuture future = client.request(new Main.Data());
			System.out.println("invoke and get");
			System.out.println("invoke result:" + future.get());
		}
		System.out.println("=====the end=====");
	}

	static class Data implements Serializable
	{
		private static final long serialVersionUID = -4666580993978548778L;

		private String mData = "";

		public Data(){}

		public String getData()
		{
			return mData;
		}

		public void setData(String data)
		{
			mData = data;
		}
	}

	static class StringMessage implements Serializable
	{
		private static final long serialVersionUID = 7193122183120113947L;

		private String mText;

		StringMessage(String msg)
		{
			mText = msg;
		}

		public String toString()
		{
			return mText;
		}
	}
}