2
* Drizzle Client & Protocol Library
4
* Copyright (C) 2008 Eric Day (eday@oddments.org)
7
* Redistribution and use in source and binary forms, with or without
8
* modification, are permitted provided that the following conditions are
11
* * Redistributions of source code must retain the above copyright
12
* notice, this list of conditions and the following disclaimer.
14
* * Redistributions in binary form must reproduce the above
15
* copyright notice, this list of conditions and the following disclaimer
16
* in the documentation and/or other materials provided with the
19
* * The names of its contributors may not be used to endorse or
20
* promote products derived from this software without specific prior
23
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
39
* @brief Row definitions
42
#include <libdrizzle-2.0/common.h>
48
uint64_t drizzle_row_read(drizzle_result_st *result, drizzle_return_t *ret_ptr)
50
if ((result->column_current != result->column_count) && (!(result->options & DRIZZLE_RESULT_BUFFER_COLUMN)))
52
drizzle_set_error(result->con->drizzle, "drizzle_row_read", "cannot retrieve rows until all columns are retrieved");
53
*ret_ptr= DRIZZLE_RETURN_NOT_READY;
57
if (drizzle_state_none(result->con))
59
drizzle_state_push(result->con, drizzle_state_row_read);
60
drizzle_state_push(result->con, drizzle_state_packet_read);
63
*ret_ptr= drizzle_state_loop(result->con);
65
return result->row_current;
68
drizzle_row_t drizzle_row_buffer(drizzle_result_st *result,
69
drizzle_return_t *ret_ptr)
76
return drizzle_row_t();
79
drizzle_return_t unused;
85
if (result->row == NULL)
87
if (drizzle_row_read(result, ret_ptr) == 0 || *ret_ptr != DRIZZLE_RETURN_OK)
92
result->row= new (std::nothrow) drizzle_row_t_type[result->column_count *2];
94
if (result->row == NULL)
96
*ret_ptr= DRIZZLE_RETURN_MEMORY;
97
return drizzle_row_t();
99
result->field_sizes= reinterpret_cast<size_t *>(result->row + result->column_count);
104
drizzle_field_t field= drizzle_field_buffer(result, &total, ret_ptr);
105
if (*ret_ptr == DRIZZLE_RETURN_ROW_END)
110
if (*ret_ptr != DRIZZLE_RETURN_OK)
112
if (*ret_ptr != DRIZZLE_RETURN_IO_WAIT)
114
delete[] result->row;
115
result->field_sizes= NULL;
121
result->row[result->field_current - 1]= field;
122
result->field_sizes[result->field_current - 1]= total;
125
*ret_ptr= DRIZZLE_RETURN_OK;
132
void drizzle_row_free(drizzle_result_st *result, drizzle_row_t row)
136
for (x= 0; x < result->column_count; x++)
137
drizzle_field_free(row[x]);
142
size_t *drizzle_row_field_sizes(drizzle_result_st *result)
144
return result->field_sizes;
147
drizzle_row_t drizzle_row_next(drizzle_result_st *result)
149
if (result->row_current == result->row_count)
152
result->field_sizes= result->field_sizes_list->at(static_cast<size_t>(result->row_current));
153
result->row_current++;
154
return result->row_list->at(static_cast<size_t>(result->row_current) - 1);
157
drizzle_row_t drizzle_row_prev(drizzle_result_st *result)
159
if (result->row_current == 0)
162
result->row_current--;
163
result->field_sizes= result->field_sizes_list->at(static_cast<size_t>(result->row_current));
164
return result->row_list->at(static_cast<size_t>(result->row_current));
167
void drizzle_row_seek(drizzle_result_st *result, uint64_t row)
169
if (row <= result->row_count)
170
result->row_current= row;
173
drizzle_row_t drizzle_row_index(drizzle_result_st *result, uint64_t row)
175
if (row >= result->row_count)
178
return (*result->row_list)[static_cast<size_t>(row)];
181
uint64_t drizzle_row_current(drizzle_result_st *result)
183
return result->row_current;
190
drizzle_return_t drizzle_row_write(drizzle_result_st *result)
192
if (drizzle_state_none(result->con))
193
drizzle_state_push(result->con, drizzle_state_row_write);
195
return drizzle_state_loop(result->con);
199
* Internal state functions.
202
drizzle_return_t drizzle_state_row_read(drizzle_con_st *con)
204
drizzle_log_debug(con->drizzle, "drizzle_state_row_read");
206
if (con->packet_size != 0 && con->buffer_size < con->packet_size &&
207
con->buffer_size < 5)
209
drizzle_state_push(con, drizzle_state_read);
210
return DRIZZLE_RETURN_OK;
213
if (con->packet_size == 5 && con->buffer_ptr[0] == 254)
215
/* Got EOF packet, no more rows. */
216
con->result->row_current= 0;
217
con->result->warning_count= drizzle_get_byte2(con->buffer_ptr + 1);
218
con->status= (drizzle_con_status_t)drizzle_get_byte2(con->buffer_ptr + 3);
220
con->buffer_size-= 5;
222
else if (con->buffer_ptr[0] == 255)
224
drizzle_state_pop(con);
225
drizzle_state_push(con, drizzle_state_result_read);
226
return DRIZZLE_RETURN_OK;
228
else if (con->result->options & DRIZZLE_RESULT_ROW_BREAK)
230
con->result->options&= ~DRIZZLE_RESULT_ROW_BREAK;
234
con->result->row_count++;
235
con->result->row_current++;
236
con->result->field_current= 0;
239
drizzle_state_pop(con);
241
return DRIZZLE_RETURN_OK;
244
drizzle_return_t drizzle_state_row_write(drizzle_con_st *con)
246
uint8_t *start= con->buffer_ptr + con->buffer_size;
248
drizzle_log_debug(con->drizzle, "drizzle_state_row_write");
250
/* Flush buffer if there is not enough room. */
251
if (((size_t)DRIZZLE_MAX_BUFFER_SIZE - (size_t)(start - con->buffer)) < 4)
253
drizzle_state_push(con, drizzle_state_write);
254
return DRIZZLE_RETURN_OK;
257
drizzle_set_byte3(start, con->packet_size);
258
start[3]= con->packet_number;
259
con->packet_number++;
261
con->buffer_size+= 4;
263
drizzle_state_pop(con);
265
return DRIZZLE_RETURN_OK;