scripts/deploy/performance/calculate_result.py

import sys


def read_tps(file):
    tps = []
    lat = []
    with open(file) as f:
        for l in f.readlines():
            s = l.split()
            for r in s:
                if(r.split(':')[0] == 'txn'):
                    tps.append(int(r.split(':')[1]))
            if l.find("client latency") > 0:
                lat.append(float(s[-1].split(':')[-1]))
    return tps, lat

def cal_tps(tps):
    tps_sum = []
    tps_max = 0

    for v in tps:
        if v == 0:
            continue
        tps_max = max(tps_max, v)
        tps_sum.append(v) 

    print("max throughput:",tps_max)
    print("average throughput:",sum(tps_sum)/len(tps_sum))

def cal_lat(lat):
    lat_sum = []
    lat_max = 0
    for v in lat:
        if v == 0:
            continue
        lat_max = max(lat_max, v)
        lat_sum.append(v) 

    print("max latency:",lat_max)
    print("average latency:",sum(lat_sum)/len(lat_sum))

if __name__ == '__main__':
    files = sys.argv[1:]
    print("calculate results, number of nodes:",len(files))


    tps = []
    lat = []
    for f in files:
        t, l=read_tps(f)
        tps += t
        lat += l

    cal_tps(tps)
    cal_lat(lat)