Example #1
0
    def __init__(self, fileName = None, controller = None):

        docontroller = False
        self.controller = None
        if docontroller:
            if not controller:

                self.controller = ODLClient(topology=self)
            else:
                self.controller = controller
        # Build topology
        self.builder = TopoBuilder(fileName = fileName, controller = self.controller)
        self.buildNodes()
        if docontroller and not controller:
            # now that self.builder is ready
            self.controller.init()
Example #2
0
class TestbedTopology (GenericTopologyProvider):

    def displayDot(self):
        sys.stdout.write('.')
        sys.stdout.flush()

    def buildSwitch(self, switch):
        enosSwitch = TestbedNode(switch.name, props=dict(switch.props, controller=self.controller))
        self.addNode(enosSwitch)
        switch.props['enosNode'] = enosSwitch

    def buildHost(self,host):
        enosHost = TestbedHost(host.name,props=host.props)
        self.addNode(enosHost)
        host.props['enosNode'] = enosHost

    def buildLink(self,link):
        p1 = link.props['endpoints'][0]
        p2 = link.props['endpoints'][1]
        n1 = p1.props['node']
        n2 = p2.props['node']
        node1 = n1.props['enosNode']
        node2 = n2.props['enosNode']
        port1 = TestbedPort(port=p1)
        port1.props['enosNode'] = node1
        port1.setNode(node1)
        p1.props['enosPort'] = port1
        port2 = TestbedPort(port=p2)
        port2.props['enosNode'] = node2
        port2.setNode(node2)
        p2.props['enosPort'] = port2
        self.addPort (node1,port1)
        self.addPort (node2,port2)
        p1.props['switch'] = node1
        p2.props['switch'] = node2
        l = TestbedLink(node1,port1,node2,port2,props=link.props)
        link.props['enosLink'] = l
        self.addLink(l)
        # Links are assumed to be uni-directional. Create reverse link
        r = TestbedLink(node1=node2,port1=port2,node2=node1,port2=port1)
        self.addLink(r)

    def buildNodes(self):
        for switch in self.builder.switchIndex.values():
            self.buildSwitch(switch)
        for host in self.builder.hostIndex.values():
            self.buildHost(host)
        for link in self.builder.linkIndex.values():
            self.buildLink(link)


    def toGraph(self):
        graph = self.getGraph(TopologyProvider.WeightType.TrafficEngineering)
        # Add a non null arbitrary weight
        for e in graph.edgeSet():
            graph.setEdgeWeight(e, 1)

    def __init__(self, fileName = None, controller = None):

        docontroller = False
        self.controller = None
        if docontroller:
            if not controller:

                self.controller = ODLClient(topology=self)
            else:
                self.controller = controller
        # Build topology
        self.builder = TopoBuilder(fileName = fileName, controller = self.controller)
        self.buildNodes()
        if docontroller and not controller:
            # now that self.builder is ready
            self.controller.init()