__author__ = "Reinaldo Penno"
__copyright__ = "Copyright(c) 2014, Cisco Systems, Inc."
__license__ = "New-style BSD"
__version__ = "0.1"
__email__ = "repenno@cisco.com"
__status__ = "alpha"


import requests
import time

class Timer(object):
    def __init__(self, verbose=False):
        self.verbose = verbose

    def __enter__(self):
        self.start = time.time()
        return self

    def __exit__(self, *args):
        self.end = time.time()
        self.secs = self.end - self.start
        self.msecs = self.secs * 1000  # millisecs
        if self.verbose:
            print ("elapsed time: %f ms" % self.msecs)
            
# Parametrized single PUT of list + one element.            
JSONPUT = """
{
  "service-functions": {
    "service-function": [
      {
        "ip-mgmt-address": "20.0.0.10",
        "type": "dpi",
        "name": "%d"
      }
    ]
  }
}"""

putheaders = {'content-type': 'application/json'}
getheaders = {'Accept': 'application/json'}
ODLIP   = "127.0.0.1:8080"
DELURL  = "http://" + ODLIP + "/restconf/config/service-function:service-functions/"
GETURL  = "http://" + ODLIP + "/restconf/config/service-function:service-functions/"
PUTURL  = "http://" + ODLIP + "/restconf/config/service-function:service-functions/"

# You probably need to adjust this number based on your OS constraints.
# Maximum number of incremental PUT list elements
numputreq = 1
# Maximum number of GET requests
numgetreq = 100000
# We will present PUT reports every 10000 PUTs
numputstep = 1
# We will present GET reports every 10000 PUTs
numgetstep = 1000


def getperftest():
    s = requests.Session()
    print ("Starting GET Performance. Total of %d requests \n" % numgetreq)
    for numreq in range(0, numgetreq, numgetstep): 
        success = 0      
        with Timer() as t:
            for i in range(numreq, numreq + numgetstep):
                r = s.get(GETURL, stream=False )
                if (r.status_code == 200):
                    success+=1
        print ("=> %d elapsed requests" % (numreq + numgetstep))
        print ("=> %d requests/s in the last %d reqs" % ((numgetstep)/t.secs, numgetstep))
        print ("=> %d successful GET requests in the last %d reqs " % (success, numgetstep))
        print ("\n")

# Based on default parameters performs a single PUT. Always overwrite existing elements
def putperftest():
    s = requests.Session()
    print ("Starting PUT Performance. Total of %d requests\n" % numputreq)
    for numreq in range(0, numputreq, numputstep): 
        success = 0      
        with Timer() as t:
            for i in range(numreq, numreq + numputstep):
                r = s.put(PUTURL, data = (JSONPUT % i), headers=putheaders, stream=False )
                if (r.status_code == 200):
                    success+=1
        print ("=> %d elapsed requests" % (numreq + numputstep))
        print ("=> %d requests/s in the last %d reqs" % ((numputstep)/t.secs, numputstep))
        print ("=> %d successful PUT requests in the last %d reqs " % (success, numputstep))
        print ("\n")

# Delete all service functions
def delallsf():
    print ("Deleting all Service Functions")
    r = requests.delete(DELURL, headers=getheaders)   
    if (r.status_code == 200) or (r.status_code == 500):
        print ("Deleted all Service Functions \n")
        return 0
    else:
        print ("Delete Failed \n")
        exit()
        return -1

if __name__ == "__main__":
    delallsf()
    putperftest()
    getperftest()