From d9b0536b128e10447a986f7b6ea2b5c93252c80f Mon Sep 17 00:00:00 2001
From: "Steinar H. Gunderson" <sgunderson@bigfoot.com>
Date: Thu, 11 Apr 2013 00:24:33 +0200
Subject: [PATCH] Move stopping into ServerPool.

---
 main.cpp       | 4 +---
 serverpool.cpp | 7 +++++++
 serverpool.h   | 3 +++
 3 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/main.cpp b/main.cpp
index 1979bb3..1f7313a 100644
--- a/main.cpp
+++ b/main.cpp
@@ -396,9 +396,7 @@ int main(int argc, char **argv)
 	for (size_t i = 0; i < inputs.size(); ++i) {
 		inputs[i]->stop();
 	}
-	for (int i = 0; i < num_servers; ++i) {	
-		servers->get_server(i)->stop();
-	}
+	servers->stop();
 
 	fprintf(stderr, "Serializing state and re-execing...\n");
 	int state_fd = make_tempfile(collect_state(
diff --git a/serverpool.cpp b/serverpool.cpp
index 09fe435..cf6933c 100644
--- a/serverpool.cpp
+++ b/serverpool.cpp
@@ -59,6 +59,13 @@ void ServerPool::run()
 	}
 }
 	
+void ServerPool::stop()
+{
+	for (int i = 0; i < num_servers; ++i) {
+		servers[i].stop();
+	}
+}
+	
 vector<ClientStats> ServerPool::get_client_stats() const
 {
 	vector<ClientStats> ret;
diff --git a/serverpool.h b/serverpool.h
index c22fb67..5dee5f7 100644
--- a/serverpool.h
+++ b/serverpool.h
@@ -35,6 +35,9 @@ public:
 	// Starts all the servers.
 	void run();
 
+	// Stops all the servers.
+	void stop();
+
 	std::vector<ClientStats> get_client_stats() const;
 
 private:
-- 
2.39.2