stats: Wait for clients to disconnect at shutdown

diff --git a/src/stats/main.c b/src/stats/main.c
index 28e068523774938da67bdc44277be550f70e9916..85f6e462c18877fb369601e732260788dbb8dc42 100644 (file)
--- a/src/stats/main.c
+++ b/src/stats/main.c
@@ -40,6 +40,11 @@ static void client_connected(struct master_service_connection *conn)
        master_service_client_connection_accept(conn);
 }
 
+static void stats_die(void)
+{
+       /* just wait for existing stats clients to disconnect from us */
+}
+
 static void main_preinit(void)
 {
        restrict_access_by_env(RESTRICT_ACCESS_FLAG_ALLOW_ROOT, NULL);
@@ -85,6 +90,7 @@ int main(int argc, char *argv[])
                                                &error) < 0)
                i_fatal("Error reading configuration: %s", error);
        master_service_init_log(master_service, "stats: ");
+       master_service_set_die_callback(master_service, stats_die);
 
        main_preinit();