mirror of
https://github.com/haiwen/ccnet-server.git
synced 2025-04-29 02:53:20 +00:00
63 lines
1.3 KiB
Python
63 lines
1.3 KiB
Python
import sys
|
|
import os
|
|
from ccnet import ClientPool, RpcClientBase
|
|
from pysearpc import searpc_func
|
|
import threading
|
|
import logging
|
|
|
|
RPC_SERVICE_NAME = 'test-rpcserver'
|
|
CCNET_CONF_DIR = os.path.expanduser('~/.ccnet')
|
|
|
|
class TestRpcClient(RpcClientBase):
|
|
def __init__(self, client_pool, *args, **kwargs):
|
|
RpcClientBase.__init__(self, client_pool, RPC_SERVICE_NAME, *args, **kwargs)
|
|
|
|
@searpc_func('string', ['string', 'int'])
|
|
def str_mul(self, s, n):
|
|
pass
|
|
|
|
|
|
class Worker(threading.Thread):
|
|
def __init__(self, rpc):
|
|
threading.Thread.__init__(self)
|
|
self.rpc = rpc
|
|
|
|
def run(self):
|
|
s = 'abcdef'
|
|
n = 100
|
|
assert self.rpc.str_mul(s, n) == s * n
|
|
|
|
def test(n):
|
|
rpcclient = TestRpcClient(ClientPool(CCNET_CONF_DIR, CCNET_CONF_DIR))
|
|
|
|
workers = []
|
|
for i in xrange(n):
|
|
t = Worker(rpcclient)
|
|
t.start()
|
|
workers.append(t)
|
|
|
|
for t in workers:
|
|
t.join()
|
|
|
|
def setup_logging():
|
|
kw = {
|
|
'format': '[%(asctime)s][%(module)s]: %(message)s',
|
|
'datefmt': '%m/%d/%Y %H:%M:%S',
|
|
'level': logging.DEBUG,
|
|
'stream': sys.stdout,
|
|
}
|
|
|
|
logging.basicConfig(**kw)
|
|
|
|
def main():
|
|
setup_logging()
|
|
if len(sys.argv) > 1:
|
|
test(int(sys.argv[1]))
|
|
else:
|
|
test(100)
|
|
|
|
print 'test passed'
|
|
|
|
if __name__ == '__main__':
|
|
main()
|