test/tools/odl-mdsal-clustering-tests/clustering-performance-test/inventory_perf.py

   1 __author__ = "Jan Medved"
   2 __copyright__ = "Copyright(c) 2014, Cisco Systems, Inc."
   3 __license__ = "New-style BSD"
   4 __email__ = "jmedved@cisco.com"
   5
   6 import argparse
   7 import requests
   8 import time
   9 import threading
  10
  11 class Counter(object):
  12     def __init__(self, start=0):
  13         self.lock = threading.Lock()
  14         self.value = start
  15     def increment(self, value=1):
  16         self.lock.acquire()
  17         try:
  18             self.value = self.value + value
  19         finally:
  20             self.lock.release()
  21
  22
  23 class Timer(object):
  24     def __init__(self, verbose=False):
  25         self.verbose = verbose
  26
  27     def __enter__(self):
  28         self.start = time.time()
  29         return self
  30
  31     def __exit__(self, *args):
  32         self.end = time.time()
  33         self.secs = self.end - self.start
  34         self.msecs = self.secs * 1000  # millisecs
  35         if self.verbose:
  36             print ("elapsed time: %f ms" % self.msecs)
  37
  38 # Initialize the totals over all threads
  39 total_requests = Counter(0)
  40 total_req_rate = Counter(0.0)
  41
  42 total_mbytes = Counter(0.0)
  43 total_mb_rate = Counter(0.0)
  44
  45 putheaders = {'content-type': 'application/json'}
  46 getheaders = {'Accept': 'application/json'}
  47
  48 INVENTORY_URL = 'http://localhost:8080/restconf/operational/opendaylight-inventory:nodes'
  49 N1T0_URL = 'http://localhost:8080/restconf/operational/opendaylight-inventory:nodes/node/openflow:1/table/0'
  50
  51 num_threads = 1
  52
  53 print_lock = threading.Lock()
  54
  55
  56 def get_inventory(tnum, url, hdrs, rnum, cond):
  57     total_len = float(0)
  58     results = {}
  59
  60     with print_lock:
  61         print 'Thread %d: Getting %s' % (tnum, url)
  62
  63     s = requests.Session()
  64     with Timer() as t:
  65         for i in range(rnum):
  66             r = s.get(url, headers=hdrs, stream=False )
  67             total_len += len(r.content)
  68
  69             try:
  70                 results[r.status_code] += 1
  71             except(KeyError):
  72                 results[r.status_code] = 1
  73
  74     total = sum(results.values())
  75     rate = total/t.secs
  76     total_requests.increment(total)
  77     total_req_rate.increment(rate)
  78
  79     mbytes = total_len / (1024*1024)
  80     mrate = mbytes/t.secs
  81     total_mbytes.increment(mbytes)
  82     total_mb_rate.increment(mrate)
  83
  84     with print_lock:
  85         print '\nThread %d: ' % tnum
  86         print '    Elapsed time: %.2f,' % t.secs
  87         print '    Requests: %d, Requests/sec: %.2f' % (total, rate)
  88         print '    Volume: %.2f MB, Rate: %.2f MByte/s' % (mbytes, mrate)
  89         print '    Results: ',
  90         print results
  91
  92     with cond:
  93         cond.notifyAll()
  94
  95
  96 if __name__ == "__main__":
  97
  98     parser = argparse.ArgumentParser(description='Restconf test program')
  99     parser.add_argument('--odlhost', default='127.0.0.1', help='host where '
 100                         'odl controller is running (default is 127.0.0.1)')
 101     parser.add_argument('--odlport', default='8080', help='port on '
 102                         'which odl\'s RESTCONF is listening (default is 8080)')
 103     parser.add_argument('--requests', type=int, default=10, help='number of '
 104                         'requests to send')
 105     parser.add_argument('--url', default='restconf/operational/opendaylight-inventory:nodes',
 106                         help='Url to send.')
 107     parser.add_argument('--nthreads', type=int, default=1,
 108                         help='Number of request worker threads, default=1')
 109     in_args = parser.parse_args()
 110
 111     url = 'http://' + in_args.odlhost + ":" + in_args.odlport + '/' + in_args.url
 112
 113     threads = []
 114     nthreads = int(in_args.nthreads)
 115     cond = threading.Condition()
 116
 117     for i in range(nthreads):
 118         t = threading.Thread(target=get_inventory,
 119                              args=(i,url, getheaders, int(in_args.requests), cond))
 120         threads.append(t)
 121         t.start()
 122
 123     finished = 0
 124     while finished < nthreads:
 125         with cond:
 126             cond.wait()
 127             finished = finished + 1
 128
 129     print '\nAggregate requests: %d, Aggregate requests/sec: %.2f' % (total_requests.value,
 130                                                                     total_req_rate.value)
 131     print 'Aggregate Volume: %.2f MB, Aggregate Rate: %.2f MByte/s' % (total_mbytes.value,
 132                                                                        total_mb_rate.value)
 133
 134 #    get_inventory(url, getheaders, int(in_args.requests))
 135
 136 #    get_inventory(N1T0_URL, getheaders, 100)