bast
c5c0d42eac
All checks were successful
MT5 Helm GitOps Deploy / build-and-deploy (push) Has been skipped
45 lines
2 KiB
Python
Executable file
45 lines
2 KiB
Python
Executable file
#!/usr/bin/env python
|
|
"""
|
|
The registry server listens to broadcasts on UDP port 18812, answering to
|
|
discovery queries by clients and registering keepalives from all running
|
|
servers. In order for clients to use discovery, a registry service must
|
|
be running somewhere on their local network.
|
|
"""
|
|
from plumbum import cli
|
|
from rpyc.utils.registry import REGISTRY_PORT, DEFAULT_PRUNING_TIMEOUT
|
|
from rpyc.utils.registry import UDPRegistryServer, TCPRegistryServer
|
|
from rpyc.lib import setup_logger
|
|
|
|
|
|
class RegistryServer(cli.Application):
|
|
mode = cli.SwitchAttr(["-m", "--mode"], cli.Set("UDP", "TCP"), default="UDP",
|
|
help="Serving mode")
|
|
|
|
ipv6 = cli.Flag(["-6", "--ipv6"], help="use ipv6 instead of ipv4")
|
|
|
|
port = cli.SwitchAttr(["-p", "--port"], cli.Range(0, 65535), default=REGISTRY_PORT,
|
|
help="The UDP/TCP listener port")
|
|
|
|
logfile = cli.SwitchAttr(["--logfile"], str, default=None,
|
|
help="The log file to use; the default is stderr")
|
|
|
|
quiet = cli.SwitchAttr(["-q", "--quiet"], help="Quiet mode (only errors are logged)")
|
|
|
|
pruning_timeout = cli.SwitchAttr(["-t", "--timeout"], int,
|
|
default=DEFAULT_PRUNING_TIMEOUT, help="Set a custom pruning timeout (in seconds)")
|
|
|
|
allow_listing = cli.SwitchAttr(["-l", "--listing"], bool, default=False, help="Enable/disable listing on registry")
|
|
|
|
def main(self):
|
|
if self.mode.upper() == "UDP":
|
|
server = UDPRegistryServer(host='::' if self.ipv6 else '0.0.0.0', port=self.port,
|
|
pruning_timeout=self.pruning_timeout, allow_listing=self.allow_listing)
|
|
elif self.mode.upper() == "TCP":
|
|
server = TCPRegistryServer(port=self.port, pruning_timeout=self.pruning_timeout,
|
|
allow_listing=self.allow_listing)
|
|
setup_logger(self.quiet, self.logfile)
|
|
server.start()
|
|
|
|
|
|
def main():
|
|
RegistryServer.run()
|