ngx_int_t ngx_http_drizzle_output_row(ngx_http_request_t *r, uint64_t row) { u_char *pos, *last; ngx_http_upstream_t *u = r->upstream; size_t size; ngx_http_upstream_drizzle_peer_data_t *dp = u->peer.data; size = sizeof(uint8_t); pos = ngx_http_drizzle_request_mem(r, dp, size); if (pos == NULL) { return NGX_ERROR; } last = pos; *last++ = (row != 0); if (row == 0) { dp->seen_stream_end = 1; } return ngx_http_drizzle_submit_mem(r, dp, size, row == 0 ? 1 : 0 /* last_buf */); }
ngx_int_t ngx_http_drizzle_output_result_header(ngx_http_request_t *r, drizzle_result_st *res) { u_char *pos, *last; ngx_int_t rc; ngx_http_upstream_t *u = r->upstream; const char *errstr; size_t size; uint16_t errstr_len; uint16_t col_count; unsigned last_buf; ngx_http_upstream_drizzle_peer_data_t *dp = u->peer.data; errstr = drizzle_result_error(res); errstr_len = (uint16_t) strlen(errstr); col_count = drizzle_result_column_count(res); size = sizeof(uint8_t) /* endian type */ + sizeof(uint32_t) /* format version */ + sizeof(uint8_t) /* result type */ + sizeof(uint16_t) /* standard error code */ + sizeof(uint16_t) /* driver-specific error code */ + sizeof(uint16_t) /* driver-specific errstr len */ + errstr_len /* driver-specific errstr data */ + sizeof(uint64_t) /* rows affected */ + sizeof(uint64_t) /* insert id */ + sizeof(uint16_t) /* column count */ ; pos = ngx_http_drizzle_request_mem(r, dp, size); if (pos == NULL) { return NGX_ERROR; } last = pos; #if NGX_HAVE_LITTLE_ENDIAN *last++ = 0; #else /* big endian */ *last++ = 1; #endif /* RDS format version */ *(uint32_t *) last = (uint32_t) resty_dbd_stream_version; last += sizeof(uint32_t); /* result type fixed to 0 */ *last++ = 0; /* standard error code * FIXME: define the standard error code set and map * libdrizzle's to it. */ *(uint16_t *) last = drizzle_result_error_code(res); last += sizeof(uint16_t); /* driver-specific error code */ *(uint16_t *) last = drizzle_result_error_code(res); last += sizeof(uint16_t); /* driver-specific errstr len */ *(uint16_t *) last = errstr_len; last += sizeof(uint16_t); /* driver-specific errstr data */ if (errstr_len) { last = ngx_copy(last, (u_char *) errstr, errstr_len); } /* affected rows */ *(uint64_t *) last = drizzle_result_affected_rows(res); last += sizeof(uint64_t); /* insert id */ *(uint64_t *) last = drizzle_result_insert_id(res); last += sizeof(uint64_t); /* column count */ *(uint16_t *) last = col_count; last += sizeof(uint16_t); if ((size_t) (last - pos) != size) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "drizzle: FATAL: output result header buffer error"); return NGX_ERROR; } if (col_count == 0) { /* we suppress row terminator here when there's no columns */ dp->seen_stream_end = 1; if (r == r->main) { last_buf = 1; } else { last_buf = 0; } rc = ngx_http_drizzle_submit_mem(r, dp, size, last_buf); if (rc != NGX_OK) { return NGX_ERROR; } ngx_http_upstream_drizzle_done(r, u, dp, NGX_OK); return NGX_DONE; } return ngx_http_drizzle_submit_mem(r, dp, size, 0 /* last_buf */); }
ngx_int_t ngx_http_drizzle_output_field(ngx_http_request_t *r, size_t offset, size_t len, size_t total, drizzle_field_t field) { u_char *pos, *last; ngx_http_upstream_t *u = r->upstream; size_t size = 0; ngx_http_upstream_drizzle_peer_data_t *dp = u->peer.data; if (offset == 0) { if (len == 0 && total != 0) { return NGX_DONE; } size = sizeof(uint32_t); /* field total length */ } /* (more) field data */ size += (uint32_t) len; /* request memory */ pos = ngx_http_drizzle_request_mem(r, dp, size); if (pos == NULL) { return NGX_ERROR; } last = pos; /* fill in the buffer */ if (offset == 0) { /* field total length */ if (field == NULL) { *(uint32_t *) last = (uint32_t) -1; } else { *(uint32_t *) last = (uint32_t) total; } last += sizeof(uint32_t); } /* field data */ if (len) { last = ngx_copy(last, field, (uint32_t) len); } if ((size_t) (last - pos) != size) { dd("offset %d, len %d, size %d", (int) offset, (int) len, (int) size); ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "drizzle: FATAL: output field buffer error"); return NGX_ERROR; } return ngx_http_drizzle_submit_mem(r, dp, size, 0 /* last_buf */); }
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 */); }
ngx_int_t ngx_http_drizzle_output_result_header(ngx_http_request_t *r, drizzle_result_st *res) { u_char *pos, *last; ngx_int_t rc; ngx_http_upstream_t *u = r->upstream; const char *errstr; size_t size; uint16_t errstr_len; uint16_t col_count; uint16_t errcode; ngx_http_upstream_drizzle_peer_data_t *dp = u->peer.data; errcode = drizzle_result_error_code(res); if (dp->enable_charset && ! dp->has_set_names) { if (errcode != 0) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "drizzle: FATAL: failed to set names 'utf8' " "(error %d)", (int) errcode); return NGX_ERROR; } if (dp->drizzle_con && dp->drizzle_res.con) { dd("before drizzle result free"); dd("%p vs. %p", dp->drizzle_res.con, dp->drizzle_con); drizzle_result_free(&dp->drizzle_res); dd("after drizzle result free"); } /* ngx_http_upstream_drizzle_done(r, u, dp, NGX_OK); */ dd("returning DONE when set names"); return NGX_DONE; } errstr = drizzle_result_error(res); errstr_len = (uint16_t) ngx_strlen(errstr); col_count = drizzle_result_column_count(res); size = sizeof(uint8_t) /* endian type */ + sizeof(uint32_t) /* format version */ + sizeof(uint8_t) /* result type */ + sizeof(uint16_t) /* standard error code */ + sizeof(uint16_t) /* driver-specific error code */ + sizeof(uint16_t) /* driver-specific errstr len */ + errstr_len /* driver-specific errstr data */ + sizeof(uint64_t) /* rows affected */ + sizeof(uint64_t) /* insert id */ + sizeof(uint16_t); /* column count */ pos = ngx_http_drizzle_request_mem(r, dp, size); if (pos == NULL) { return NGX_ERROR; } last = pos; #if NGX_HAVE_LITTLE_ENDIAN *last++ = 0; #else /* big endian */ *last++ = 1; #endif /* RDS format version */ *(uint32_t *) last = (uint32_t) resty_dbd_stream_version; last += sizeof(uint32_t); /* result type fixed to 0 */ *last++ = 0; /* standard error code * FIXME: define the standard error code set and map * libdrizzle's to it. */ *(uint16_t *) last = errcode; last += sizeof(uint16_t); /* driver-specific error code */ *(uint16_t *) last = drizzle_result_error_code(res); last += sizeof(uint16_t); /* driver-specific errstr len */ *(uint16_t *) last = errstr_len; last += sizeof(uint16_t); /* driver-specific errstr data */ if (errstr_len) { last = ngx_copy(last, (u_char *) errstr, errstr_len); } /* affected rows */ *(uint64_t *) last = drizzle_result_affected_rows(res); last += sizeof(uint64_t); /* insert id */ *(uint64_t *) last = drizzle_result_insert_id(res); last += sizeof(uint64_t); /* column count */ *(uint16_t *) last = col_count; last += sizeof(uint16_t); if ((size_t) (last - pos) != size) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "drizzle: FATAL: output result header buffer error"); return NGX_ERROR; } if (col_count == 0) { dd("Col count is ZERO"); /* we suppress row terminator here when there's no columns */ dp->seen_stream_end = 1; rc = ngx_http_drizzle_submit_mem(r, dp, size); if (rc != NGX_OK) { return NGX_ERROR; } dd("about to be done..."); ngx_http_upstream_drizzle_done(r, u, dp, NGX_DONE); dd("i am returning DONE"); return NGX_DONE; } return ngx_http_drizzle_submit_mem(r, dp, size); }