diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-10-19 12:37:32 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-10-19 12:37:32 +1300 |
commit | 264a09e7789500673282075b2f5b9956268f0cea (patch) | |
tree | 31199e39c194f08982e554e5d1621d035a7affae /mitmproxy/controller.py | |
parent | da8dec982373b5137c5700818505887fe0249e98 (diff) | |
download | mitmproxy-264a09e7789500673282075b2f5b9956268f0cea.tar.gz mitmproxy-264a09e7789500673282075b2f5b9956268f0cea.tar.bz2 mitmproxy-264a09e7789500673282075b2f5b9956268f0cea.zip |
Let's dispense with this fiction that we support multiple servers
It's simple enough to re-add if needed, and until then is just cruft.
Diffstat (limited to 'mitmproxy/controller.py')
-rw-r--r-- | mitmproxy/controller.py | 20 |
1 files changed, 6 insertions, 14 deletions
diff --git a/mitmproxy/controller.py b/mitmproxy/controller.py index a225634a..a88500a8 100644 --- a/mitmproxy/controller.py +++ b/mitmproxy/controller.py @@ -87,14 +87,14 @@ class Master: """ The master handles mitmproxy's main event loop. """ - def __init__(self, opts, *servers): + def __init__(self, opts, server): self.options = opts or options.Options() self.addons = addons.Addons(self) self.event_queue = queue.Queue() self.should_exit = threading.Event() - self.servers = [] - for i in servers: - self.add_server(i) + self.server = server + channel = Channel(self.event_queue, self.should_exit) + server.set_channel(channel) @contextlib.contextmanager def handlecontext(self): @@ -121,16 +121,9 @@ class Master: with self.handlecontext(): self.addons("log", LogEntry(e, level)) - def add_server(self, server): - # We give a Channel to the server which can be used to communicate with the master - channel = Channel(self.event_queue, self.should_exit) - server.set_channel(channel) - self.servers.append(server) - def start(self): self.should_exit.clear() - for server in self.servers: - ServerThread(server).start() + ServerThread(self.server).start() def run(self): self.start() @@ -168,8 +161,7 @@ class Master: return changed def shutdown(self): - for server in self.servers: - server.shutdown() + self.server.shutdown() self.should_exit.set() self.addons.done() |