python多線程性能測試之快速mock數據

Posted on by

背景

在我們測試工作中，性能測試也是避免不瞭的，因此在性能測試前期準備工作中，需要 mock 足夠批量的數據進行壓測。那麼怎麼能在短時間內快速 mock 出想要的格式數據和足夠量的數據進行壓測？那麼往下看。

安裝相關類包

  • pip install kafka
  • pip install appmetrics
  • pip install faker
  • pip install pykafka

快速 mock kafka 批量測試數據

# -* coding:utf8 *-
from pykafka import KafkaClient
import uuid
import time
import threading
from appmetrics import metrics
from faker import Faker
import os
fake = Faker("zh-cn")
PATH = lambda p: os.path.abspath(
    os.path.join(os.path.dirname(__file__), p)
)
meter = metrics.new_meter("meter_test")
host_producer = 'host地址'
def data_info():
    uid = str(uuid.uuid4())
    suid = ''.join(uid.split('-'))
    return suid
def data_result():
    #數據格式可自行定義
    data = f"{data_info()},{fake.phone_number()},111111111111,LOL-UZI"
    return data
def mock_request():
    client_producer = KafkaClient(hosts=host_producer)
    topicdocu = client_producer.topics['XXXXXXX-TOPIC']
    producer = topicdocu.get_producer(sync=False) # sync=False  關閉同步，使用異步
    while True:
        data_uni = data_result()
        producer.produce(bytes(data_uni, encoding='utf-8'))
        meter.notify(1) # 請求一次 記錄器打點一次
        # i = i - 1
    producer.stop()
def print_meter():
    while True:
        print(meter.get())
        time.sleep(1)
def thread_request(nums):
    t1 = []
    for i in range(nums):
        if i == 0:
            #該線程是為瞭記錄每秒打點作用
            t = threading.Thread(target=print_meter, name="T" + str(i))
        else:
            t = threading.Thread(target=mock_request, name="T" + str(i))
        t.setDaemon(True)
        t1.append(t)
    for t in t1:
        t.start()
    for t in t1:
        t.join()
#
#
if __name__ == '__main__':
    thread_request(101)

appmetrics 使用方法

Meters

Meters，度量一系列事件發生的速率 (rate)，例如 TPS。Meters 會統計最近 1 分鐘，5 分鐘，15 分鐘，還有全部時間的速率。

meter = metrics.new_meter(“meter_test”)
meter.notify(1)
meter.notify(1)
meter.notify(3)
meter.get()

返回結果:

{'count': 5, 'kind': 'meter', 'five': 0.0066114184713530035, 'mean': 0.27743058841197027, 'f

以上就是python批量測試多線程之快速mock數據的詳細內容，更多關於python測試多線程mock數據的資料請關註WalkonNet其它相關文章！

推薦閱讀：