def test_status_conflicts(self): zmq_ctx = zmq.Context() bind_socket = socket.Socket(zmq_ctx, zmq.REP) bind_socket.bind("inproc://status_test") with self.assertRaises(Exception): bind_socket.connect("inproc://status_test") connect_socket = socket.Socket(zmq_ctx, zmq.REP) connect_socket.connect("inproc://status_test") with self.assertRaises(Exception): connect_socket.bind("inproc://status_test")
def test_pub_sub(self): zmq_ctx = zmq.Context() pub_socket = socket.Socket(zmq_ctx, zmq.PUB) pub_socket.bind("inproc://req_rep_test") sub_socket = socket.Socket(zmq_ctx, zmq.SUB) sub_socket.connect("inproc://req_rep_test") sub_socket.set_sock_opt(zmq.SUBSCRIBE, b"") thrift_obj = lsdb_types.PrefixDatabase() thrift_obj.thisNodeName = 'some node' pub_socket.send_thrift_obj(thrift_obj) recv_obj = sub_socket.recv_thrift_obj(lsdb_types.PrefixDatabase) self.assertEqual(thrift_obj, recv_obj)
def test_req_rep(self): zmq_ctx = zmq.Context() rep_socket = socket.Socket(zmq_ctx, zmq.REP) rep_socket.bind("inproc://req_rep_test") req_socket = socket.Socket(zmq_ctx, zmq.REQ) req_socket.connect("inproc://req_rep_test") thrift_obj = lsdb_types.PrefixDatabase() thrift_obj.thisNodeName = 'some node' req_socket.send_thrift_obj(thrift_obj) recv_obj = rep_socket.recv_thrift_obj(lsdb_types.PrefixDatabase) self.assertEqual(thrift_obj, recv_obj) rep_socket.send_thrift_obj(recv_obj) recv_obj = req_socket.recv_thrift_obj(lsdb_types.PrefixDatabase) self.assertEqual(thrift_obj, recv_obj)
def test_dealer_dealer(self): zmq_ctx = zmq.Context() d_socket_1 = socket.Socket(zmq_ctx, zmq.DEALER) d_socket_1.bind("inproc://dealer_test") d_socket_2 = socket.Socket(zmq_ctx, zmq.DEALER) d_socket_2.connect("inproc://dealer_test") thrift_obj = lsdb_types.PrefixDatabase() thrift_obj.thisNodeName = 'some node' d_socket_1.send_thrift_obj(thrift_obj) recv_obj = d_socket_2.recv_thrift_obj(lsdb_types.PrefixDatabase) self.assertEqual(thrift_obj, recv_obj) d_socket_2.send_thrift_obj(recv_obj) recv_obj = d_socket_1.recv_thrift_obj(lsdb_types.PrefixDatabase) self.assertEqual(thrift_obj, recv_obj)
def __init__(self, zmq_ctx, decision_cmd_url, timeout=consts.Consts.TIMEOUT_MS, proto_factory=consts.Consts.PROTO_FACTORY): self._decision_cmd_socket = socket.Socket(zmq_ctx, zmq.REQ, timeout, proto_factory) self._decision_cmd_socket.connect(decision_cmd_url)
def __init__(self, zmq_ctx, fib_rep_port, timeout=consts.Consts.TIMEOUT_MS, proto_factory=consts.Consts.PROTO_FACTORY): self._fib_cmd_socket = socket.Socket(zmq_ctx, zmq.REQ, timeout, proto_factory) self._fib_cmd_socket.connect(fib_rep_port)
def _send_recv(): req_socket = socket.Socket(zmq.Context(), zmq.REQ) req_socket.connect("tcp://localhost:5000") req_socket.send_thrift_obj(thrift_obj) print("request sent") recv_obj = req_socket.recv_thrift_obj(lsdb_types.PrefixDatabase) print("reply received") self.assertEqual(thrift_obj, recv_obj)
def __init__(self, zmq_ctx, url): self._prefix_mgr_server_socket = socket.Socket(zmq_ctx, zmq.REP) self._prefix_mgr_server_socket.bind(url) self._prefix_map = { sprint_prefix(prefix_entry1.prefix): prefix_entry1, sprint_prefix(prefix_entry2.prefix): prefix_entry2, sprint_prefix(prefix_entry3.prefix): prefix_entry3 }
def _recv_send(): rep_socket = socket.Socket(zmq.Context(), zmq.REP) rep_socket.bind("tcp://*:5000") recv_obj = rep_socket.recv_thrift_obj(lsdb_types.PrefixDatabase) print("request received") self.assertEqual(thrift_obj, recv_obj) rep_socket.send_thrift_obj(recv_obj) print("reply sent")
def __init__(self, zmq_ctx, cs_cmd_url, timeout=consts.Consts.TIMEOUT_MS, proto_factory=TCompactProtocolFactory): self._cs_cmd_socket = socket.Socket(zmq_ctx, zmq.REQ, timeout, proto_factory) self._cs_cmd_socket.connect(cs_cmd_url)
def __init__(self, zmq_ctx, health_checker_cmd_port, timeout=consts.Consts.TIMEOUT_MS, proto_factory=consts.Consts.PROTO_FACTORY): self._health_checker_cmd_socket = socket.Socket( zmq_ctx, zmq.REQ, timeout, proto_factory) self._health_checker_cmd_socket.connect(health_checker_cmd_port)
def __init__(self, zmq_ctx, lm_cmd_url, timeout=consts.Consts.TIMEOUT_MS, proto_factory=consts.Consts.PROTO_FACTORY): self._lm_cmd_socket = socket.Socket(zmq_ctx, zmq.DEALER, timeout, proto_factory) self._lm_cmd_socket.connect(lm_cmd_url)
def __init__(self, zmq_ctx, kv_store_pub_url, proto_factory=TCompactProtocolFactory): # timeout set as -1 for indefinite blocking self._kv_store_sub_socket = socket.Socket(zmq_ctx, zmq.SUB, -1, proto_factory) self._kv_store_sub_socket.connect(kv_store_pub_url) self._kv_store_sub_socket.set_sock_opt(zmq.SUBSCRIBE, b"")
def __init__( self, zmq_ctx, prefix_mgr_cmd_url, timeout=consts.Consts.TIMEOUT_MS, proto_factory=consts.Consts.PROTO_FACTORY, ): self._prefix_mgr_cmd_socket = socket.Socket(zmq_ctx, zmq.REQ, timeout, proto_factory) self._prefix_mgr_cmd_socket.connect(prefix_mgr_cmd_url)
def __init__(self, zmq_ctx, kv_store_pub_url, proto_factory=consts.Consts.PROTO_FACTORY): # timeout set as -1 for indefinite blocking self._kv_store_sub_socket = socket.Socket(zmq_ctx, zmq.SUB, -1, proto_factory) self._kv_store_sub_socket.connect(kv_store_pub_url) self._kv_store_sub_socket.set_sock_opt(zmq.SUBSCRIBE, b"")
def __init__( self, zmq_ctx, monitor_pub_url, timeout=-1, proto_factory=consts.Consts.PROTO_FACTORY, ): # timeout set as -1 for indefinite blocking self._monitor_sub_socket = socket.Socket(zmq_ctx, zmq.SUB, timeout, proto_factory) self._monitor_sub_socket.connect(monitor_pub_url) self._monitor_sub_socket.set_sock_opt(zmq.SUBSCRIBE, b"")
def run(self, yes): if not yes: yes = utils.yesno("Are you sure to trigger Open/R crash") if not yes: print("Not triggering force crash") return print("Triggering force crash") sock = socket.Socket(self.cli_opts.zmq_ctx, zmq.REQ, timeout=200) sock.set_sock_opt(zmq.LINGER, 1000) sock.connect(consts.Consts.FORCE_CRASH_SERVER_URL) sock.send("User {} issuing crash command".format(os.environ["USER"])) sock.close()
def __init__(self, zmq_ctx, monitor_cmd_url, timeout=consts.Consts.TIMEOUT_MS, proto_factory=TCompactProtocolFactory): self._monitor_cmd_socket = socket.Socket(zmq_ctx, zmq.DEALER, timeout, proto_factory) self._monitor_cmd_socket.connect(monitor_cmd_url)
def __init__(self, zmq_ctx, url): self._lm_server_socket = socket.Socket(zmq_ctx, zmq.DEALER) self._lm_server_socket.bind(url) self._dump_links_cache = dump_links_cache
def __init__(self, zmq_ctx, url): self._monitor_server_socket = socket.Socket(zmq_ctx, zmq.DEALER) self._monitor_server_socket.bind(url) self._monitor_cache = monitor_cache
def __init__(self, zmq_ctx, url): self._kv_store_server_socket = socket.Socket(zmq_ctx, zmq.REP) self._kv_store_server_socket.bind(url) self._kv_store = kv_store_cache
def __init__(self, zmq_ctx, url): self._kv_store_publisher_socket = socket.Socket(zmq_ctx, zmq.PUB) self._kv_store_publisher_socket.bind(url)
def __init__(self, zmq_ctx, url): self._decision_server_socket = socket.Socket(zmq_ctx, zmq.REP) self._decision_server_socket.bind(url) self._route_db_cache = route_db_cache self._adj_db_cachee = adj_dbs_cache self._prefix_db_cache = prefix_dbs_cache
def __init__(self, zmq_ctx, url): self._fib_server_socket = socket.Socket(zmq_ctx, zmq.REP) self._fib_server_socket.bind(url) self._route_db_cache = route_db_cache
def __init__(self, zmq_ctx, url): self._cs_server_socket = socket.Socket(zmq_ctx, zmq.REP) self._cs_server_socket.bind(url) self._store_db = store_db