update grpc to allow for configuring and created distributed nodes

daemon/examples/grpc/distributed_switch.py

@@ -0,0 +1,86 @@
+import argparse
+import logging
+
+from core.api.grpc import client, core_pb2
+
+
+def log_event(event):
+    logging.info("event: %s", event)
+
+
+def main(args):
+    core = client.CoreGrpcClient()
+
+    with core.context_connect():
+        # create session
+        response = core.create_session()
+        session_id = response.session_id
+        logging.info("created session: %s", response)
+
+        # add distributed server
+        server_name = "core2"
+        response = core.add_session_server(session_id, server_name, args.server)
+        logging.info("added session server: %s", response)
+
+        # handle events session may broadcast
+        core.events(session_id, log_event)
+
+        # change session state
+        response = core.set_session_state(
+            session_id, core_pb2.SessionState.CONFIGURATION
+        )
+        logging.info("set session state: %s", response)
+
+        # create switch node
+        switch = core_pb2.Node(type=core_pb2.NodeType.SWITCH)
+        response = core.add_node(session_id, switch)
+        logging.info("created switch: %s", response)
+        switch_id = response.node_id
+
+        # helper to create interfaces
+        interface_helper = client.InterfaceHelper(ip4_prefix="10.83.0.0/16")
+
+        # create node one
+        position = core_pb2.Position(x=100, y=50)
+        node = core_pb2.Node(position=position)
+        response = core.add_node(session_id, node)
+        logging.info("created node one: %s", response)
+        node_one_id = response.node_id
+
+        # create link
+        interface_one = interface_helper.create_interface(node_one_id, 0)
+        response = core.add_link(session_id, node_one_id, switch_id, interface_one)
+        logging.info("created link from node one to switch: %s", response)
+
+        # create node two
+        position = core_pb2.Position(x=200, y=50)
+        node = core_pb2.Node(position=position, server=server_name)
+        response = core.add_node(session_id, node)
+        logging.info("created node two: %s", response)
+        node_two_id = response.node_id
+
+        # create link
+        interface_one = interface_helper.create_interface(node_two_id, 0)
+        response = core.add_link(session_id, node_two_id, switch_id, interface_one)
+        logging.info("created link from node two to switch: %s", response)
+
+        # change session state
+        response = core.set_session_state(
+            session_id, core_pb2.SessionState.INSTANTIATION
+        )
+        logging.info("set session state: %s", response)
+
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.DEBUG)
+    parser = argparse.ArgumentParser(description="Run distributed_switch example")
+    parser.add_argument(
+        "-a",
+        "--address",
+        help="local address that distributed servers will use for gre tunneling",
+    )
+    parser.add_argument(
+        "-s", "--server", help="distributed server to use for creating nodes"
+    )
+    args = parser.parse_args()
+    main(args)