class TestConsumer(object): """TestConsumer""" def __init__(self): """__init__""" self.logObj = SignalLogHandle("/data/logs/test/") self.baseKafkaConsumer = BaseKafkaConsumer( ['devops-dev1:9090', 'devops-dev1:9091'], ['JOB_TEST'], self.logObj) def testOneMsgConsumer(self): """testOneMsgConsumer""" startTime = time.time() while True: msgs = self.baseKafkaConsumer.get_message() if not msgs: continue break endTime = time.time() print 'Send a single message for {0} second'.format( endTime - startTime) def testMultiConsumer(self, num): """testMultiConsumer :param num: """ index = 1 startTime = 0 while True: msgInterator = self.baseKafkaConsumer.get_message() if not msgInterator: # time.sleep(0.5) continue for msg in msgInterator: if index == 1: startTime = time.time() index += 1 if index % 200 == 0: print 'Current index:', index if index > num: break endTime = time.time() print 'Send {0} messages for {1} second'.format( num, endTime - startTime)
def __init__(self): """__init__""" self.logObj = SignalLogHandle("/data/logs/test/") self.baseKafkaConsumer = BaseKafkaConsumer( ['devops-dev1:9090', 'devops-dev1:9091'], ['JOB_TEST'], self.logObj)