87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
|
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)
|