mirror of https://github.com/yandex/odyssey.git
107 lines
2.0 KiB
C
107 lines
2.0 KiB
C
|
|
||
|
|
/*
|
||
|
|
* odissey.
|
||
|
|
*
|
||
|
|
* PostgreSQL connection pooler and request router.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdarg.h>
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <assert.h>
|
||
|
|
|
||
|
|
#include <machinarium.h>
|
||
|
|
#include <soprano.h>
|
||
|
|
|
||
|
|
#include "od_macro.h"
|
||
|
|
#include "od_list.h"
|
||
|
|
#include "od_pid.h"
|
||
|
|
#include "od_syslog.h"
|
||
|
|
#include "od_log.h"
|
||
|
|
#include "od_scheme.h"
|
||
|
|
#include "od_lex.h"
|
||
|
|
#include "od_config.h"
|
||
|
|
#include "od_server.h"
|
||
|
|
#include "od_server_pool.h"
|
||
|
|
#include "od_client.h"
|
||
|
|
#include "od_client_pool.h"
|
||
|
|
|
||
|
|
void od_clientpool_init(od_clientpool_t *pool)
|
||
|
|
{
|
||
|
|
pool->count_active = 0;
|
||
|
|
pool->count_queue = 0;
|
||
|
|
pool->count_pending = 0;
|
||
|
|
od_list_init(&pool->active);
|
||
|
|
od_list_init(&pool->queue);
|
||
|
|
od_list_init(&pool->pending);
|
||
|
|
}
|
||
|
|
|
||
|
|
void od_clientpool_set(od_clientpool_t *pool, od_client_t *client,
|
||
|
|
od_clientstate_t state)
|
||
|
|
{
|
||
|
|
if (client->state == state)
|
||
|
|
return;
|
||
|
|
switch (client->state) {
|
||
|
|
case OD_CUNDEF:
|
||
|
|
break;
|
||
|
|
case OD_CACTIVE:
|
||
|
|
pool->count_active--;
|
||
|
|
break;
|
||
|
|
case OD_CQUEUE:
|
||
|
|
pool->count_queue--;
|
||
|
|
break;
|
||
|
|
case OD_CPENDING:
|
||
|
|
pool->count_pending--;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
od_list_t *target = NULL;
|
||
|
|
switch (state) {
|
||
|
|
case OD_CUNDEF:
|
||
|
|
break;
|
||
|
|
case OD_CACTIVE:
|
||
|
|
target = &pool->active;
|
||
|
|
pool->count_active++;
|
||
|
|
break;
|
||
|
|
case OD_CQUEUE:
|
||
|
|
target = &pool->queue;
|
||
|
|
pool->count_queue++;
|
||
|
|
break;
|
||
|
|
case OD_CPENDING:
|
||
|
|
target = &pool->pending;
|
||
|
|
pool->count_pending++;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
od_list_unlink(&client->link_pool);
|
||
|
|
od_list_init(&client->link_pool);
|
||
|
|
if (target)
|
||
|
|
od_list_push(target, &client->link_pool);
|
||
|
|
client->state = state;
|
||
|
|
}
|
||
|
|
|
||
|
|
od_client_t*
|
||
|
|
od_clientpool_next(od_clientpool_t *pool, od_clientstate_t state)
|
||
|
|
{
|
||
|
|
od_list_t *target = NULL;
|
||
|
|
switch (state) {
|
||
|
|
case OD_CACTIVE:
|
||
|
|
target = &pool->active;
|
||
|
|
break;
|
||
|
|
case OD_CQUEUE:
|
||
|
|
target = &pool->queue;
|
||
|
|
break;
|
||
|
|
case OD_CPENDING:
|
||
|
|
target = &pool->pending;
|
||
|
|
break;
|
||
|
|
case OD_CUNDEF:
|
||
|
|
assert(0);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
if (od_list_empty(target))
|
||
|
|
return NULL;
|
||
|
|
od_client_t *client;
|
||
|
|
client = od_container_of(target->next, od_client_t, link_pool);
|
||
|
|
return client;
|
||
|
|
}
|