SingleTest.java

package com.snowcattle.game.tps.rpc.single;

import com.snowcattle.game.TestStartUp;
import com.snowcattle.game.service.rpc.client.RpcProxyService;
import com.snowcattle.game.common.util.BeanUtil;
import com.snowcattle.game.tps.rpc.RpcTpsRunable;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicLong;

/**
 * Created by jiangwenping on 17/4/19.
 */
public class SingleTest {

    private RpcProxyService rpcProxyService;

    public static void main(String[] args) throws InterruptedException {
        SingleTest singleTest = new SingleTest();
        singleTest.init();
        singleTest.tps();
        singleTest.setTear();
    }

    public void init(){
        TestStartUp.startUpWithSpring();
        rpcProxyService = (RpcProxyService) BeanUtil.getBean("rpcProxyService");
    }

    public void tps() throws InterruptedException {
        AtomicLong atomicLong = new AtomicLong();
        int size = 10000;
        CountDownLatch countDownLatch = new CountDownLatch(1);
        Thread thread = new Thread(new RpcTpsRunable(rpcProxyService, atomicLong, size, countDownLatch));
        long startTime = System.currentTimeMillis();
        thread.start();
        countDownLatch.await();
        long endTime = System.currentTimeMillis();
        long useTime = endTime - startTime;
        System.out.println("rpc 数量" + atomicLong.get() + "时间" + useTime);
    }

    public void setTear(){
        if (rpcProxyService != null) {
            try {
                rpcProxyService.shutdown();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

}