404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
|
# File 'ext/mysql2_ext.c', line 404
static VALUE rb_mysql_result_each(int argc, VALUE * argv, VALUE self) {
VALUE dataset, opts, block;
MYSQL_RES * result;
unsigned long numRows, i;
GetMysql2Result(self, result);
rb_scan_args(argc, argv, "01&", &opts, &block);
// force-start at the beginning of the result set for proper
// behavior of #each
mysql_data_seek(result, 0);
numRows = mysql_num_rows(result);
if (numRows == 0) {
return Qnil;
}
// TODO: allow yielding datasets of configurable size
// like find_in_batches from AR...
if (block != Qnil) {
for (i = 0; i < numRows; i++) {
VALUE row = rb_mysql_result_fetch_row(argc, argv, self);
if (row == Qnil) {
return Qnil;
}
rb_yield(row);
}
} else {
dataset = rb_ary_new2(numRows);
for (i = 0; i < numRows; i++) {
VALUE row = rb_mysql_result_fetch_row(argc, argv, self);
if (row == Qnil) {
return Qnil;
}
rb_ary_store(dataset, i, row);
}
return dataset;
}
return Qnil;
}
|