1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
# -*- test-case-name: twisted.protocols.haproxy.test.test_parser -*-
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
"""
Parser for 'haproxy:' string endpoint.
"""
from typing import Mapping, Tuple
from zope.interface import implementer
from twisted.internet import interfaces
from twisted.internet.endpoints import (
IStreamServerEndpointStringParser,
_WrapperServerEndpoint,
quoteStringArgument,
serverFromString,
)
from twisted.plugin import IPlugin
from . import proxyEndpoint
def unparseEndpoint(args: Tuple[object, ...], kwargs: Mapping[str, object]) -> str:
"""
Un-parse the already-parsed args and kwargs back into endpoint syntax.
@param args: C{:}-separated arguments
@param kwargs: C{:} and then C{=}-separated keyword arguments
@return: a string equivalent to the original format which this was parsed
as.
"""
description = ":".join(
[quoteStringArgument(str(arg)) for arg in args]
+ sorted(
"{}={}".format(
quoteStringArgument(str(key)), quoteStringArgument(str(value))
)
for key, value in kwargs.items()
)
)
return description
@implementer(IPlugin, IStreamServerEndpointStringParser)
class HAProxyServerParser:
"""
Stream server endpoint string parser for the HAProxyServerEndpoint type.
@ivar prefix: See L{IStreamServerEndpointStringParser.prefix}.
"""
prefix = "haproxy"
def parseStreamServer(
self, reactor: interfaces.IReactorCore, *args: object, **kwargs: object
) -> _WrapperServerEndpoint:
"""
Parse a stream server endpoint from a reactor and string-only arguments
and keyword arguments.
@param reactor: The reactor.
@param args: The parsed string arguments.
@param kwargs: The parsed keyword arguments.
@return: a stream server endpoint
@rtype: L{IStreamServerEndpoint}
"""
subdescription = unparseEndpoint(args, kwargs)
wrappedEndpoint = serverFromString(reactor, subdescription)
return proxyEndpoint(wrappedEndpoint)
|
