Skip to content

Commit 9d9bbd1

Browse files
authoredJul 30, 2022
Reverse proxy plugin constructor now accepts client connection object (#1235)
1 parent dc06ea4 commit 9d9bbd1

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed
 

‎proxy/http/server/plugin.py

+14
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,20 @@ def on_access_log(self, context: Dict[str, Any]) -> Optional[Dict[str, Any]]:
128128
class ReverseProxyBasePlugin(ABC):
129129
"""ReverseProxy base plugin class."""
130130

131+
def __init__(
132+
self,
133+
uid: str,
134+
flags: argparse.Namespace,
135+
client: HttpClientConnection,
136+
event_queue: EventQueue,
137+
upstream_conn_pool: Optional['UpstreamConnectionPool'] = None,
138+
):
139+
self.uid = uid
140+
self.flags = flags
141+
self.client = client
142+
self.event_queue = event_queue
143+
self.upstream_conn_pool = upstream_conn_pool
144+
131145
@abstractmethod
132146
def routes(self) -> List[Union[str, Tuple[str, List[bytes]]]]:
133147
"""List of routes registered by plugin.

‎proxy/http/server/reverse.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,9 @@ def __init__(self, *args: Any, **kwargs: Any):
4040
self.choice: Optional[Url] = None
4141
self.plugins: List['ReverseProxyBasePlugin'] = []
4242
for klass in self.flags.plugins[b'ReverseProxyBasePlugin']:
43-
plugin: 'ReverseProxyBasePlugin' = klass()
43+
plugin: 'ReverseProxyBasePlugin' = klass(
44+
self.uid, self.flags, self.client, self.event_queue, self.upstream_conn_pool,
45+
)
4446
self.plugins.append(plugin)
4547

4648
def handle_upstream_data(self, raw: memoryview) -> None:

0 commit comments

Comments
 (0)
Please sign in to comment.