mirror of https://github.com/yandex/odyssey.git
98 lines
2.0 KiB
C
98 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 <signal.h>
|
||
|
|
|
||
|
|
#include <machinarium.h>
|
||
|
|
#include <soprano.h>
|
||
|
|
|
||
|
|
#include "od_macro.h"
|
||
|
|
#include "od_version.h"
|
||
|
|
#include "od_list.h"
|
||
|
|
#include "od_pid.h"
|
||
|
|
#include "od_syslog.h"
|
||
|
|
#include "od_log.h"
|
||
|
|
#include "od_daemon.h"
|
||
|
|
#include "od_scheme.h"
|
||
|
|
#include "od_lex.h"
|
||
|
|
#include "od_config.h"
|
||
|
|
#include "od_msg.h"
|
||
|
|
#include "od_instance.h"
|
||
|
|
|
||
|
|
#include "od_server.h"
|
||
|
|
#include "od_server_pool.h"
|
||
|
|
#include "od_client.h"
|
||
|
|
#include "od_client_pool.h"
|
||
|
|
#include "od_route_id.h"
|
||
|
|
#include "od_route.h"
|
||
|
|
|
||
|
|
#include "od_pooler.h"
|
||
|
|
#include "od_relay.h"
|
||
|
|
#include "od_frontend.h"
|
||
|
|
|
||
|
|
static inline void
|
||
|
|
od_relay(void *arg)
|
||
|
|
{
|
||
|
|
od_relay_t *relay = arg;
|
||
|
|
od_instance_t *instance = relay->pooler->instance;
|
||
|
|
|
||
|
|
od_log(&instance->log, NULL, "relay: started");
|
||
|
|
|
||
|
|
for (;;) {
|
||
|
|
machine_msg_t msg;
|
||
|
|
msg = machine_queue_get(relay->pooler->task_queue, UINT32_MAX);
|
||
|
|
if (msg == NULL)
|
||
|
|
break;
|
||
|
|
|
||
|
|
od_msg_t msg_type;
|
||
|
|
msg_type = machine_msg_get_type(msg);
|
||
|
|
switch (msg_type) {
|
||
|
|
case OD_MCLIENT_NEW:
|
||
|
|
{
|
||
|
|
od_client_t *client;
|
||
|
|
client = *(od_client_t**)machine_msg_get_data(msg);
|
||
|
|
client->relay = relay;
|
||
|
|
int64_t coroutine_id;
|
||
|
|
coroutine_id = machine_coroutine_create(od_frontend_main, client);
|
||
|
|
if (coroutine_id == -1) {
|
||
|
|
od_error(&relay->pooler->instance->log, client->io,
|
||
|
|
"failed to create coroutine");
|
||
|
|
machine_close(client->io);
|
||
|
|
od_client_free(client);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
client->coroutine_id = coroutine_id;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
machine_msg_free(msg);
|
||
|
|
}
|
||
|
|
|
||
|
|
od_log(&instance->log, NULL, "relay: stopped");
|
||
|
|
}
|
||
|
|
|
||
|
|
void od_relay_init(od_relay_t *relay, od_pooler_t *pooler)
|
||
|
|
{
|
||
|
|
relay->machine = -1;
|
||
|
|
relay->pooler = pooler;
|
||
|
|
}
|
||
|
|
|
||
|
|
int od_relay_start(od_relay_t *relay)
|
||
|
|
{
|
||
|
|
relay->machine = machine_create("relay", od_relay, relay);
|
||
|
|
if (relay->machine == -1) {
|
||
|
|
od_error(&relay->pooler->instance->log, NULL, "failed to start relay");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|