static PyObject*
get_column_desc(int i, drizzle_column_type_t *types, drizzle_column_st *column)
{
PyObject *o = NULL;
const char *name;
drizzle_column_type_t type;
drizzle_column_flags_t flags;
drizzle_charset_t charset;
uint32_t size;
uint8_t decimals = 0, null_ok = 0;
size_t max_size;
name = drizzle_column_name(column);
DEBUG("name:%s", name);
type = drizzle_column_type(column);
DEBUG("type:%u", type);
size = drizzle_column_size(column);
DEBUG("size:%u", size);
max_size = drizzle_column_size(column);
DEBUG("max_size:%" PRIu64, max_size);
charset = drizzle_column_charset(column);
DEBUG("charset:%u", charset);
decimals = drizzle_column_decimals(column);
DEBUG("decimals:%d", decimals);
flags = drizzle_column_flags(column);
DEBUG("flags:%d", flags);
if (flags & DRIZZLE_COLUMN_FLAGS_NOT_NULL) {
null_ok = 0;
} else {
null_ok = 1;
}
types[i] = type;
o = Py_BuildValue("(siiiiii)",
name, type, 0, size, size, decimals, null_ok);
return o;
}
static void _client(drizzle_con_st *con, client_state_st *state)
{
drizzle_return_t ret;
drizzle_column_st *column;
drizzle_row_t row;
size_t *field_sizes;
switch(state->state)
{
case CLIENT_STATE_START:
drizzle_test("drizzle_query_str");
(void)drizzle_query_str(con, &state->result, "SELECT", &ret);
if (ret == DRIZZLE_RETURN_IO_WAIT)
{
state->state = CLIENT_STATE_START;
return;
}
else if (ret == DRIZZLE_RETURN_COULD_NOT_CONNECT)
{
state->state= CLIENT_STATE_DONE;
return;
}
else if (ret != DRIZZLE_RETURN_OK)
{
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
}
case CLIENT_STATE_RESULT:
drizzle_test("drizzle_result_buffer");
ret = drizzle_result_buffer(&state->result);
if (ret == DRIZZLE_RETURN_IO_WAIT)
{
state->state = CLIENT_STATE_RESULT;
return;
}
else if (ret != DRIZZLE_RETURN_OK)
{
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
}
drizzle_test("drizzle_con_protocol_version");
if (drizzle_con_protocol_version(con) != 10)
{
drizzle_test_error("no match");
}
drizzle_test("drizzle_con_server_version");
if (strcmp(drizzle_con_server_version(con), "test_version"))
drizzle_test_error("no match");
drizzle_test("drizzle_con_thread_id");
if (drizzle_con_thread_id(con) != 1)
drizzle_test_error("no match");
drizzle_test("drizzle_con_scramble");
if (memcmp(drizzle_con_scramble(con), "ABCDEFGHIJKLMNOPQRST",
DRIZZLE_MAX_SCRAMBLE_SIZE))
{
drizzle_test_error("no match");
}
/* We default to MySQL protocol right now, which sets this flag. */
drizzle_test("drizzle_con_capabilities");
if (drizzle_con_capabilities(con) != DRIZZLE_CAPABILITIES_PROTOCOL_41)
drizzle_test_error("no match");
drizzle_test("drizzle_con_charset");
if (drizzle_con_charset(con) != 8)
drizzle_test_error("no match");
drizzle_test("drizzle_con_status");
if (drizzle_con_status(con) != DRIZZLE_CON_STATUS_NONE)
drizzle_test_error("no match");
drizzle_test("drizzle_con_packet_size");
if (drizzle_con_max_packet_size(con) != DRIZZLE_MAX_PACKET_SIZE)
drizzle_test_error("no match");
drizzle_test("drizzle_column_next");
column= drizzle_column_next(&state->result);
if (column == NULL)
drizzle_test_error("column not found");
drizzle_test("drizzle_column_drizzle_result");
if (drizzle_column_drizzle_result(column) != &state->result)
drizzle_test_error("no match");
drizzle_test("drizzle_column_catalog");
if (strcmp(drizzle_column_catalog(column), "test_catalog"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_db");
if (strcmp(drizzle_column_db(column), "test_db"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_table");
if (strcmp(drizzle_column_table(column), "test_table"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_orig_table");
if (strcmp(drizzle_column_orig_table(column), "test_orig_table"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_name");
if (strcmp(drizzle_column_name(column), "test_column_1"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_orig_name");
if (strcmp(drizzle_column_orig_name(column), "test_orig_column_1"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_charset");
if (drizzle_column_charset(column) != 8)
drizzle_test_error("no match");
drizzle_test("drizzle_column_size");
if (drizzle_column_size(column) != 32)
drizzle_test_error("no match");
drizzle_test("drizzle_column_max_size");
if (drizzle_column_max_size(column) != 12)
drizzle_test_error("no match");
drizzle_test("drizzle_column_type_drizzle");
if (drizzle_column_type_drizzle(column) != DRIZZLE_COLUMN_TYPE_DRIZZLE_VARCHAR)
drizzle_test_error("no match");
drizzle_test("drizzle_column_flags");
if (drizzle_column_flags(column) != DRIZZLE_COLUMN_FLAGS_NONE)
drizzle_test_error("no match");
drizzle_test("drizzle_column_next");
column= drizzle_column_next(&state->result);
if (column == NULL)
drizzle_test_error("column not found");
drizzle_test("drizzle_column_name");
if (strcmp(drizzle_column_name(column), "test_column_2"))
drizzle_test_error("no match");
drizzle_test("drizzle_column_next");
column= drizzle_column_next(&state->result);
if (column != NULL)
drizzle_test_error("column found");
drizzle_test("drizzle_column_prev");
column= drizzle_column_prev(&state->result);
if (column == NULL)
drizzle_test_error("column not found");
drizzle_test("drizzle_column_name");
if (strcmp(drizzle_column_name(column), "test_column_2"))
drizzle_test_error("no match");
drizzle_test("drizzle_row_next");
row= drizzle_row_next(&state->result);
if (row == NULL)
drizzle_test_error("row not found");
if (strcmp(row[0], "test_field_1") || strcmp(row[1], "test_field_2"))
drizzle_test_error("no match");
drizzle_test("drizzle_row_field_sizes");
field_sizes= drizzle_row_field_sizes(&state->result);
if (field_sizes[0] != 12 || field_sizes[1] != 12)
drizzle_test_error("no match");
drizzle_test("drizzle_row_prev");
row = drizzle_row_prev(&state->result);
if (row == NULL)
drizzle_test_error("row not found");
if (strcmp(row[0], "test_field_1") || strcmp(row[1], "test_field_2"))
drizzle_test_error("no match");
drizzle_test("drizzle_row_next");
row = drizzle_row_next(&state->result);
drizzle_test("drizzle_row_index");
row = drizzle_row_index(&state->result, 0);
if (row == NULL)
drizzle_test_error("row not found");
if (strcmp(row[0], "test_field_1") || strcmp(row[1], "test_field_2"))
drizzle_test_error("no match");
drizzle_test("drizzle_row_index");
row = drizzle_row_index(&state->result, 1);
if (row != NULL)
drizzle_test_error("row found");
drizzle_test("drizzle_result_free");
drizzle_result_free(&state->result);
state->state = CLIENT_STATE_DONE;
case CLIENT_STATE_DONE:
return;
default:
drizzle_test_error("invalid client state");
}
}
ngx_int_t
ngx_http_drizzle_output_col(ngx_http_request_t *r, drizzle_column_st *col)
{
u_char *pos, *last;
ngx_http_upstream_t *u = r->upstream;
drizzle_column_type_t col_type = 0;
uint16_t std_col_type = 0;
const char *col_name = NULL;
uint16_t col_name_len = 0;
size_t size;
ngx_http_upstream_drizzle_peer_data_t *dp = u->peer.data;
if (col == NULL) {
return NGX_ERROR;
}
col_type = drizzle_column_type(col);
col_name = drizzle_column_name(col);
col_name_len = (uint16_t) strlen(col_name);
size = sizeof(uint16_t) /* std col type */
+ sizeof(uint16_t) /* driver-specific col type */
+ sizeof(uint16_t) /* col name str len */
+ col_name_len /* col name str len */
;
pos = ngx_http_drizzle_request_mem(r, dp, size);
if (pos == NULL) {
return NGX_ERROR;
}
last = pos;
/* std column type */
std_col_type = (uint16_t) ngx_http_drizzle_std_col_type(col_type);
#if 0
dd("std col type for %s: %d, %d (%d, %d, %d)",
col_name, std_col_type, rds_col_type_blob,
rds_rough_col_type_str,
rds_rough_col_type_str << 14,
(uint16_t) (19 | (rds_rough_col_type_str << 14))
);
#endif
*(uint16_t *) last = std_col_type;
last += sizeof(uint16_t);
/* drizzle column type */
*(uint16_t *) last = col_type;
last += sizeof(uint16_t);
/* column name string length */
*(uint16_t *) last = col_name_len;
last += sizeof(uint16_t);
/* column name string data */
last = ngx_copy(last, col_name, col_name_len);
if ((size_t) (last - pos) != size) {
ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
"drizzle: FATAL: output column buffer error");
return NGX_ERROR;
}
return ngx_http_drizzle_submit_mem(r, dp, size, 0 /* last_buf */);
}
