TarsCpp/examples/CoroutineDemo/testCoro/main.cpp

/**
 * Tencent is pleased to support the open source community by making Tars available.
 *
 * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 
 * in compliance with the License. You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * 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.
 */

#include "BServant.h"
#include "servant/Communicator.h"
#include "servant/CoroutineScheduler.h"
#include <iostream>

using namespace std;
using namespace Test;
using namespace tars;

//继承框架的协程类
class TestCoroutine : public Coroutine
{
public:
	TestCoroutine(int iNum, const string &sObj);

	~TestCoroutine() {}

	void handle();

private:
	int _num;
	string _sObj;
	Communicator _comm;
	BServantPrx _prx;
};

TestCoroutine::TestCoroutine(int iNum, const string &sObj)
: _num(iNum)
, _sObj(sObj)
{
	_comm.setProperty("locator", "tars.tarsregistry.QueryObj@tcp -h 10.208.139.242 -p 17890 -t 10000");
	_comm.stringToProxy(_sObj, _prx);
}

void TestCoroutine::handle()
{
	string sIn(32,'a');
	string sOut("");
	unsigned long sum = 0;

	for(int i = 0; i < _num; i++) 
	{
		try
		{
			int iRet = _prx->testCoroSerial(sIn, sOut);
			if(iRet == 0)
			{
				++sum;
			}

            sOut = "";
			iRet = _prx->testCoroParallel(sIn, sOut);
			if(iRet == 0)
			{
				++sum;
			}
		}
		catch(TC_Exception &e)
		{
			cout << "i: " << i << "exception: " << e.what() << endl;
		}
		catch(...)
		{
			cout << "i: " << i << "unknown exception." << endl;
		}
	}
	cout << "succ:" << sum <<endl;
}

int main(int argc,char ** argv)
{
	if(argc != 3)
	{
		cout << "usage: " << argv[0] << " CallTimes sObj" << endl;
		return -1;
	}

	tars::Int32 iNum = TC_Common::strto<tars::Int32>(string(argv[1]));

	string sObj = string(argv[2]);

	TestCoroutine testCoro(iNum, sObj);

	testCoro.setCoroInfo(10, 128, 128*1024);

	testCoro.start();
		
	testCoro.getThreadControl().join();
	
    return 0;
}