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

   1 #!/usr/bin/python
   2 import time
   3
   4 import flow_config_blaster
   5 import json
   6
   7
   8 class FlowConfigBulkBlaster(flow_config_blaster.FlowConfigBlaster):
   9     """
  10     Reusing FlowConfigBlaster class providing flow operations based on bulk processing
  11     """
  12
  13     FLW_ADD_RPC_URL = "restconf/operations/sal-bulk-flow:add-flows-rpc"
  14     FLW_REMOVE_RPC_URL = "restconf/operations/sal-bulk-flow:remove-flows-rpc"
  15     FLW_ADD_DS_URL = "restconf/operations/sal-bulk-flow:add-flows-ds"
  16     FLW_REMOVE_DS_URL = "restconf/operations/sal-bulk-flow:remove-flows-ds"
  17
  18     def __init__(self, *args, **kwargs):
  19         super(FlowConfigBulkBlaster, self).__init__(*args, **kwargs)
  20         self.bulk_type = 'RPC'
  21
  22     def update_post_url_template(self, action):
  23         """
  24         Update url templates (defined in parent class) in order to point to bulk API rpcs.
  25         :param action: user intention (currently only 'ADD' is supported)
  26         """
  27         if self.bulk_type == 'RPC':
  28             self.post_url_template = 'http://%s:' + self.port + '/'
  29             if action == 'ADD':
  30                 self.post_url_template += self.FLW_ADD_RPC_URL
  31             elif action == 'REMOVE':
  32                 self.post_url_template += self.FLW_REMOVE_RPC_URL
  33         elif self.bulk_type == 'DS':
  34             self.post_url_template = 'http://%s:' + self.port + '/'
  35             if action == 'ADD':
  36                 self.post_url_template += self.FLW_ADD_DS_URL
  37             elif action == 'REMOVE':
  38                 self.post_url_template += self.FLW_REMOVE_DS_URL
  39
  40     def assemble_post_url(self, host, node):
  41         """
  42         Format url to final form using substitutions. Here the node is ignored.
  43         :param host: controller address
  44         :param node: node identifier
  45         :return: finalized url
  46         """
  47         return self.post_url_template % host
  48
  49     def create_flow_from_template(self, flow_id, ipaddr, node_id):
  50         """
  51         Create one flow beased on template and given values
  52         :param flow_id: flow identifier
  53         :param ipaddr: part of flow match
  54         :param node_id: node identifier
  55         :return: flow structure ready to use
  56         """
  57         # python 2.7 specific syntax (super)
  58         flow = super(FlowConfigBulkBlaster, self).create_flow_from_template(flow_id, ipaddr, node_id)
  59         flow_id = flow['id']
  60         del(flow['id'])
  61         if self.bulk_type == 'DS':
  62             flow['flow-id'] = flow_id
  63         flow['node'] = '/opendaylight-inventory:nodes/opendaylight-inventory' \
  64                        ':node[opendaylight-inventory:id="openflow:{}"]'.format(node_id)
  65         return flow
  66
  67     def convert_to_json(self, flow_list, node_id=None):
  68         """
  69         Converts given list of flows into string of json form.
  70         :param flow_list: list of flows to convert
  71         :param node_id: identifier of corresponding node
  72         :return: json string
  73         """
  74         json_input = None
  75         if self.bulk_type == 'RPC':
  76             json_input = {'input': {'bulk-flow-item': flow_list}}
  77         elif self.bulk_type == 'DS':
  78             json_input = {'input': {'bulk-flow-ds-item': flow_list}}
  79
  80         flow_data = json.dumps(json_input)
  81         # print flow_data
  82         return flow_data
  83
  84
  85 if __name__ == "__main__":
  86     ############################################################################
  87     # This program executes the base performance test. The test adds flows into
  88     # the controller's config space. This function is basically the CLI frontend
  89     # to the FlowConfigBlaster class and drives its main functions: adding and
  90     # deleting flows from the controller's config data store
  91     ############################################################################
  92     parser = flow_config_blaster.create_arguments_parser()
  93     parser.add_argument('--bulk-type', default='RPC', dest='bulk_type',
  94                         choices=['RPC', 'DS'],
  95                         help='Bulk type to use: RPC, DS (default is RPC)')
  96
  97     in_args = parser.parse_args()
  98
  99     if in_args.file != '':
 100         flow_template = flow_config_blaster.get_json_from_file(in_args.file)
 101     else:
 102         flow_template = None
 103
 104     fcbb = FlowConfigBulkBlaster(in_args.host, in_args.port, in_args.cycles,
 105                                  in_args.threads, in_args.fpr, in_args.nodes,
 106                                  in_args.flows, in_args.startflow, in_args.auth)
 107     fcbb.bulk_type = in_args.bulk_type
 108     fcbb.update_post_url_template('ADD')
 109
 110     # Run through <cycles>, where <threads> are started in each cycle and
 111     # <flows> are added from each thread
 112     fcbb.add_blaster()
 113
 114     print '\n*** Total flows added: %s' % fcbb.get_ok_flows()
 115     print '    HTTP[OK] results:  %d\n' % fcbb.get_ok_rqsts()
 116
 117     if in_args.delay > 0:
 118         print '*** Waiting for %d seconds before the delete cycle ***\n' % in_args.delay
 119         time.sleep(in_args.delay)
 120
 121     # Run through <cycles>, where <threads> are started in each cycle and
 122     # <flows> previously added in an add cycle are deleted in each thread
 123     if in_args.delete:
 124         fcbb.delete_blaster()
 125         print '\n*** Total flows deleted: %s' % fcbb.get_ok_flows()
 126         print '    HTTP[OK] results:    %d\n' % fcbb.get_ok_rqsts()