odyssey/sources/route.h

#ifndef ODYSSEY_ROUTE_H
#define ODYSSEY_ROUTE_H

/*
 * Odyssey.
 *
 * Scalable PostgreSQL connection pooler.
*/

typedef struct od_route od_route_t;

struct od_route
{
	od_rule_t         *rule;
	od_route_id_t      id;
	od_stat_t          stats;
	od_stat_t          stats_prev;
	int                stats_mark;
	od_server_pool_t   server_pool;
	od_client_pool_t   client_pool;
	kiwi_params_lock_t params;
	pthread_mutex_t    lock;
	od_list_t          link;
};

static inline void
od_route_init(od_route_t *route)
{
	route->rule = NULL;
	od_route_id_init(&route->id);
	od_server_pool_init(&route->server_pool);
	od_client_pool_init(&route->client_pool);
	route->stats_mark = 0;
	od_stat_init(&route->stats);
	od_stat_init(&route->stats_prev);
	kiwi_params_lock_init(&route->params);
	od_list_init(&route->link);
	pthread_mutex_init(&route->lock, NULL);
}

static inline od_route_t*
od_route_allocate(void)
{
	od_route_t *route = malloc(sizeof(*route));
	if (route == NULL)
		return NULL;
	od_route_init(route);
	return route;
}

static inline void
od_route_free(od_route_t *route)
{
	od_route_id_free(&route->id);
	od_server_pool_free(&route->server_pool);
	kiwi_params_lock_free(&route->params);
	pthread_mutex_destroy(&route->lock);
	free(route);
}

static inline int
od_route_is_dynamic(od_route_t *route)
{
	return route->rule->db_is_default || route->rule->user_is_default;
}

static inline int
od_route_match_compare_client_cb(od_client_t *client, void **argv)
{
	return od_id_mgr_cmp(&client->id, argv[0]);
}

static inline od_client_t*
od_route_match_client(od_route_t *route, od_id_t *id)
{
	void *argv[] = { id };
	od_client_t *match;
	match = od_client_pool_foreach(&route->client_pool, OD_CLIENT_ACTIVE,
	                               od_route_match_compare_client_cb,
	                               argv);
	if (match)
		return match;
	match = od_client_pool_foreach(&route->client_pool, OD_CLIENT_QUEUE,
	                               od_route_match_compare_client_cb,
	                               argv);
	if (match)
		return match;
	match = od_client_pool_foreach(&route->client_pool, OD_CLIENT_PENDING,
	                               od_route_match_compare_client_cb,
	                               argv);
	if (match)
		return match;

	return NULL;
}

static inline void
od_route_kill_client(od_route_t *route, od_id_t *id)
{
	od_client_t *client;
	client = od_route_match_client(route, id);
	if (client)
		od_client_kill(client);
}

static inline int
od_route_kill_cb(od_client_t *client, void **argv)
{
	(void)argv;
	od_client_kill(client);
	return 0;
}

static inline void
od_route_kill_client_pool(od_route_t *route)
{
	od_client_pool_foreach(&route->client_pool, OD_CLIENT_ACTIVE,
	                       od_route_kill_cb, NULL);
	od_client_pool_foreach(&route->client_pool, OD_CLIENT_PENDING,
	                       od_route_kill_cb, NULL);
	od_client_pool_foreach(&route->client_pool, OD_CLIENT_QUEUE,
	                       od_route_kill_cb, NULL);
}

static inline void
od_route_lock(od_route_t *route)
{
	pthread_mutex_lock(&route->lock);
}

static inline void
od_route_unlock(od_route_t *route)
{
	pthread_mutex_unlock(&route->lock);
}

#endif /* ODYSSEY_ROUTE_H */