コード例 #1
0
    def receive(self, num_msgs):
        start_time = time.time()
        timeout_stats = 'cherami_client_python.{}.receive.timeout'.format(
            self.tchannel.name)
        duration_stats = 'cherami_client_python.{}.receive.duration'.format(
            self.tchannel.name)

        msgs = []
        end_time = time.time() + self.timeout_seconds
        while len(msgs) < num_msgs:
            seconds_remaining = end_time - time.time()
            if seconds_remaining <= 0:
                stats.count(timeout_stats, 1)
                stats.timing(duration_stats,
                             util.time_diff_in_ms(start_time, time.time()))
                return msgs
            try:
                msgs.append(
                    self.msg_queue.get(block=True, timeout=seconds_remaining))
                self.msg_queue.task_done()

                util.stats_count(self.tchannel.name,
                                 'consumer_msg_queue.dequeue', None, 1)
            except queue.Empty:
                pass
        stats.timing(duration_stats,
                     util.time_diff_in_ms(start_time, time.time()))
        return msgs
コード例 #2
0
ファイル: util.py プロジェクト: mfateev/cherami-client-python
def stats_count(client_name, stats_name, hostport, count):
    overall_stats = 'cherami_client_python.{}.{}'.format(
        client_name, stats_name)
    stats.count(overall_stats, count)

    if hostport:
        hostport_stats = 'cherami_client_python.{}.{}.{}'\
                            .format(client_name, hostport.replace('.', '_').replace(':', '_'), stats_name)
        stats.count(hostport_stats, count)
コード例 #3
0
 def test_count_sample(self):
     stats.count('foo', 1, 0.5)
     line = mockserver.readline()
     self.assertEqual(line, 'foo:1|c|@0.500000')
コード例 #4
0
 def test_count(self):
     stats.count('foo', 1)
     line = mockserver.readline()
     self.assertEqual(line, 'foo:1|c')
コード例 #5
0
ファイル: test_stats.py プロジェクト: JeremyGrosser/clay
 def test_count_sample(self):
     stats.count('foo', 1, 0.5)
     line = mockserver.readline()
     self.assertEqual(line, 'foo:1|c|@0.500000')
コード例 #6
0
ファイル: test_stats.py プロジェクト: JeremyGrosser/clay
 def test_count(self):
     stats.count('foo', 1)
     line = mockserver.readline()
     self.assertEqual(line, 'foo:1|c')