mirror of https://github.com/yandex/odyssey.git
87 lines
1.5 KiB
C
87 lines
1.5 KiB
C
|
|
/*
|
|
* Odyssey.
|
|
*
|
|
* Scalable PostgreSQL connection pooler.
|
|
*/
|
|
|
|
#include <machinarium.h>
|
|
#include <odyssey.h>
|
|
|
|
static int read_attribute_buf_after_key(char **data, size_t *data_size,
|
|
char **out, size_t *out_size)
|
|
{
|
|
char *data_end = *data + *data_size;
|
|
char *begin = *data;
|
|
char *end;
|
|
|
|
if (begin >= data_end || *begin != '=')
|
|
return -1;
|
|
|
|
begin++;
|
|
|
|
end = begin;
|
|
|
|
while (end < data_end && *end && *end != ',')
|
|
end++;
|
|
|
|
if (end < data_end) {
|
|
*data = end + 1;
|
|
} else {
|
|
*data = end;
|
|
}
|
|
*data_size = data_end - *data;
|
|
|
|
if (out)
|
|
*out = begin;
|
|
if (out_size)
|
|
*out_size = end - begin;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int read_attribute_buf(char **data, size_t *data_size, char attr_key,
|
|
char **out, size_t *out_size)
|
|
{
|
|
if (!*data_size || **data != attr_key)
|
|
return -1;
|
|
|
|
char *new_data = *data + 1;
|
|
size_t new_data_size = *data_size - 1;
|
|
|
|
if (read_attribute_buf_after_key(&new_data, &new_data_size, out,
|
|
out_size) == -1)
|
|
return -1;
|
|
|
|
*data = new_data;
|
|
*data_size = new_data_size;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int read_any_attribute_buf(char **data, size_t *data_size, char *attribute_ptr,
|
|
char **out, size_t *out_size)
|
|
{
|
|
if (!*data_size)
|
|
return -1;
|
|
char attribute = **data;
|
|
|
|
if (!isalpha(attribute))
|
|
return -1;
|
|
|
|
if (attribute_ptr != NULL)
|
|
*attribute_ptr = attribute;
|
|
|
|
char *new_data = *data + 1;
|
|
size_t new_data_size = *data_size - 1;
|
|
|
|
if (read_attribute_buf_after_key(&new_data, &new_data_size, out,
|
|
out_size) == -1)
|
|
return -1;
|
|
|
|
*data = new_data;
|
|
*data_size = new_data_size;
|
|
|
|
return 0;
|
|
}
|