refactored NodeData and NodeOptions to use server instead of emulation_server

This commit is contained in:
bharnden 2019-10-22 20:50:01 -07:00
parent ab0abd65aa
commit cb81095b64
12 changed files with 23 additions and 25 deletions

View file

@ -775,7 +775,7 @@ class CoreGrpcServer(core_pb2_grpc.CoreApiServicer):
options.image = node_proto.image
options.services = node_proto.services
if node_proto.server:
options.emulation_server = node_proto.server
options.server = node_proto.server
position = node_proto.position
options.set_position(position.x, position.y)

View file

@ -722,7 +722,7 @@ class CoreHandler(socketserver.BaseRequestHandler):
options.icon = message.get_tlv(NodeTlvs.ICON.value)
options.canvas = message.get_tlv(NodeTlvs.CANVAS.value)
options.opaque = message.get_tlv(NodeTlvs.OPAQUE.value)
options.emulation_server = message.get_tlv(NodeTlvs.EMULATION_SERVER.value)
options.server = message.get_tlv(NodeTlvs.EMULATION_SERVER.value)
services = message.get_tlv(NodeTlvs.SERVICES.value)
if services:

View file

@ -24,7 +24,7 @@ def convert_node(node_data):
(NodeTlvs.IP6_ADDRESS, node_data.ip6_address),
(NodeTlvs.MODEL, node_data.model),
(NodeTlvs.EMULATION_ID, node_data.emulation_id),
(NodeTlvs.EMULATION_SERVER, node_data.emulation_server),
(NodeTlvs.EMULATION_SERVER, node_data.server),
(NodeTlvs.SESSION, node_data.session),
(NodeTlvs.X_POSITION, node_data.x_position),
(NodeTlvs.Y_POSITION, node_data.y_position),

View file

@ -64,7 +64,7 @@ NodeData = collections.namedtuple(
"ip6_address",
"model",
"emulation_id",
"emulation_server",
"server",
"session",
"x_position",
"y_position",

View file

@ -87,7 +87,7 @@ class NodeOptions(object):
self.lon = None
self.alt = None
self.emulation_id = None
self.emulation_server = None
self.server = None
self.image = image
def set_position(self, x, y):

View file

@ -673,9 +673,9 @@ class Session(object):
name = f"{node_class.__name__}{_id}"
# verify distributed server
server = self.distributed.servers.get(options.emulation_server)
if options.emulation_server is not None and server is None:
raise CoreError(f"invalid distributed server: {options.emulation_server}")
server = self.distributed.servers.get(options.server)
if options.server is not None and server is None:
raise CoreError(f"invalid distributed server: {options.server}")
# create node
logging.info(
@ -713,7 +713,7 @@ class Session(object):
logging.debug("set node type: %s", node.type)
self.services.add_services(node, node.type, options.services)
# boot nodes if created after runtime, CoreNodes, Physical, and RJ45 are all nodes
# boot nodes after runtime, CoreNodes, Physical, and RJ45 are all nodes
is_boot_node = isinstance(node, CoreNodeBase) and not isinstance(node, Rj45Node)
if self.state == EventTypes.RUNTIME_STATE.value and is_boot_node:
self.write_nodes()

View file

@ -194,9 +194,9 @@ class NodeBase(object):
x, y, _ = self.getposition()
model = self.type
emulation_server = None
server = None
if self.server is not None:
emulation_server = self.server.name
server = self.server.name
services = self.services
if services is not None:
@ -217,7 +217,7 @@ class NodeBase(object):
longitude=lon,
altitude=alt,
model=model,
emulation_server=emulation_server,
server=server,
services=services,
)