1 files changed, 98 insertions, 0 deletions
diff --git a/src/indexer/worker-pool.c b/src/indexer/worker-pool.c
new file mode 100644
index 0000000..e0de136
--- /dev/null
+++ b/src/indexer/worker-pool.c
@@ -0,0 +1,98 @@
+/* Copyright (c) 2011-2018 Dovecot authors, see the included COPYING file */
+
+#include "lib.h"
+#include "ioloop.h"
+#include "llist.h"
+#include "connection.h"
+#include "master-service.h"
+#include "worker-connection.h"
+#include "worker-pool.h"
+
+#define MAX_WORKER_IDLE_SECS (60*5)
+
+struct worker_pool {
+	char *socket_path;
+	indexer_status_callback_t *callback;
+	worker_available_callback_t *avail_callback;
+
+	struct connection_list *connection_list;
+};
+
+struct worker_pool *
+worker_pool_init(const char *socket_path, indexer_status_callback_t *callback,
+		 worker_available_callback_t *avail_callback)
+{
+	struct worker_pool *pool;
+
+	pool = i_new(struct worker_pool, 1);
+	pool->socket_path = i_strdup(socket_path);
+	pool->callback = callback;
+	pool->avail_callback = avail_callback;
+	pool->connection_list = worker_connection_list_create();
+	return pool;
+}
+
+void worker_pool_deinit(struct worker_pool **_pool)
+{
+	struct worker_pool *pool = *_pool;
+
+	*_pool = NULL;
+
+	if (pool->connection_list != NULL)
+		connection_list_deinit(&pool->connection_list);
+
+	i_free(pool->connection_list);
+	i_free(pool->socket_path);
+	i_free(pool);
+}
+
+bool worker_pool_have_connections(struct worker_pool *pool)
+{
+	return pool->connection_list->connections != NULL;
+}
+
+static int worker_pool_add_connection(struct worker_pool *pool,
+				      struct connection **conn_r)
+{
+	struct connection *conn;
+
+	conn = worker_connection_create(pool->socket_path, pool->callback,
+					pool->avail_callback,
+					pool->connection_list);
+	if (connection_client_connect(conn) < 0) {
+		worker_connection_destroy(conn);
+		return -1;
+	}
+
+	*conn_r = conn;
+	return 0;
+}
+
+bool worker_pool_get_connection(struct worker_pool *pool,
+				struct connection **conn_r)
+{
+	unsigned int max_connections;
+
+	max_connections = I_MAX(1, worker_connections_get_process_limit());
+	if (pool->connection_list->connections_count >= max_connections)
+		return FALSE;
+	if (worker_pool_add_connection(pool, conn_r) < 0)
+		return FALSE;
+
+	return TRUE;
+}
+
+struct connection *
+worker_pool_find_username_connection(struct worker_pool *pool,
+				     const char *username)
+{
+	struct connection *list;
+	const char *worker_user;
+
+	for (list = pool->connection_list->connections; list != NULL; list = list->next) {
+		worker_user = worker_connection_get_username(list);
+		if (worker_user != NULL && strcmp(worker_user, username) == 0)
+			return list;
+	}
+	return NULL;
+}