def set_gateway_parser(parser=None):
"""Set the parser for the gateway arguments
:param parser: an optional existing parser to build upon
:return: the parser
"""
if not parser:
from .base import set_base_parser
parser = set_base_parser()
from .peapods.base import mixin_base_ppr_parser
from .peapods.runtimes.zmq import mixin_zmq_runtime_parser
from .peapods.runtimes.zed import mixin_zed_runtime_parser
from .peapods.runtimes.remote import (
mixin_gateway_parser,
mixin_prefetch_parser,
mixin_http_gateway_parser,
mixin_compressor_parser,
)
from .peapods.pod import mixin_base_pod_parser, mixin_k8s_pod_parser
from .peapods.pea import mixin_pea_parser
mixin_base_ppr_parser(parser)
mixin_zmq_runtime_parser(parser)
mixin_zed_runtime_parser(parser)
mixin_prefetch_parser(parser)
mixin_http_gateway_parser(parser)
mixin_compressor_parser(parser)
mixin_comm_protocol_parser(parser)
mixin_gateway_parser(parser)
mixin_pea_parser(parser)
mixin_k8s_pod_parser(parser)
from ..enums import SocketType, PodRoleType
parser.set_defaults(
name='gateway',
socket_in=SocketType.
PULL_CONNECT, # otherwise there can be only one client at a time
socket_out=SocketType.PUSH_CONNECT,
ctrl_with_ipc=True, # otherwise ctrl port would be conflicted
runtime_cls='GRPCRuntime',
pod_role=PodRoleType.GATEWAY,
)
parser.add_argument(
'--dynamic-routing',
action='store_true',
default=True,
help=
'The Pod will setup the socket types of the HeadPea and TailPea depending on this argument.'
if _SHOW_ALL_ARGS else argparse.SUPPRESS,
)
return parser
def set_gateway_parser(parser=None):
"""Set the parser for the gateway arguments
:param parser: an optional existing parser to build upon
:return: the parser
"""
if not parser:
from .base import set_base_parser
parser = set_base_parser()
from .peapods.base import mixin_base_ppr_parser
from .peapods.runtimes.zmq import mixin_zmq_runtime_parser
from .peapods.runtimes.zed import mixin_zed_runtime_parser
from .peapods.runtimes.remote import (
mixin_gateway_parser,
mixin_prefetch_parser,
mixin_http_gateway_parser,
mixin_compressor_parser,
)
from .peapods.pod import mixin_base_pod_parser, mixin_k8s_pod_parser
from .peapods.pea import mixin_pea_parser
mixin_base_ppr_parser(parser)
mixin_zmq_runtime_parser(parser)
mixin_zed_runtime_parser(parser)
mixin_prefetch_parser(parser)
mixin_http_gateway_parser(parser)
mixin_compressor_parser(parser)
mixin_comm_protocol_parser(parser)
mixin_gateway_parser(parser)
mixin_pea_parser(parser)
mixin_k8s_pod_parser(parser)
from ..enums import SocketType, PodRoleType
parser.set_defaults(
name='gateway',
socket_in=SocketType.
PULL_CONNECT, # otherwise there can be only one client at a time
socket_out=SocketType.PUSH_CONNECT,
ctrl_with_ipc=True, # otherwise ctrl port would be conflicted
runtime_cls='GRPCRuntime',
pod_role=PodRoleType.GATEWAY,
)
return parser
def set_client_cli_parser(parser=None):
"""Set the parser for the cli client
:param parser: an optional existing parser to build upon
:return: the parser
"""
if not parser:
from .base import set_base_parser
parser = set_base_parser()
from .peapods.runtimes.remote import mixin_remote_parser
from .client import mixin_client_features_parser, mixin_comm_protocol_parser
mixin_remote_parser(parser)
mixin_client_features_parser(parser)
mixin_comm_protocol_parser(parser)
return parser
def set_gateway_parser(parser=None):
"""Set the parser for the gateway arguments
:param parser: an optional existing parser to build upon
:return: the parser
"""
if not parser:
from jina.parsers.base import set_base_parser
parser = set_base_parser()
from jina.parsers.orchestrate.base import mixin_base_ppr_parser
from jina.parsers.orchestrate.runtimes.worker import mixin_worker_runtime_parser
from jina.parsers.orchestrate.runtimes.remote import (
mixin_gateway_parser,
mixin_prefetch_parser,
mixin_http_gateway_parser,
mixin_compressor_parser,
)
from jina.parsers.orchestrate.deployment import mixin_base_deployment_parser
from jina.parsers.orchestrate.pod import mixin_pod_parser
mixin_base_ppr_parser(parser)
mixin_worker_runtime_parser(parser)
mixin_prefetch_parser(parser)
mixin_http_gateway_parser(parser)
mixin_compressor_parser(parser)
mixin_comm_protocol_parser(parser)
mixin_gateway_parser(parser)
mixin_pod_parser(parser)
mixin_head_parser(parser)
from jina.enums import DeploymentRoleType
parser.set_defaults(
name='gateway',
runtime_cls='GRPCGatewayRuntime',
deployment_role=DeploymentRoleType.GATEWAY,
)
return parser
def set_client_cli_parser(parser=None):
"""Set the parser for the cli client
:param parser: an optional existing parser to build upon
:return: the parser
"""
if not parser:
from jina.parsers.base import set_base_parser
parser = set_base_parser()
from jina.parsers.client import (
mixin_client_features_parser,
mixin_comm_protocol_parser,
)
from jina.parsers.orchestrate.runtimes.remote import mixin_client_gateway_parser
mixin_client_gateway_parser(parser)
mixin_client_features_parser(parser)
mixin_comm_protocol_parser(parser)
return parser
