github-migration/mysql2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename method to each, seek dataset to beginning on each call of #each

Browse Source
This commit is contained in:
Brian Lopez 2010-04-02 15:32:58 -07:00
parent d90e2a3187
commit 872bdb0caf
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ext/mysql2_ext.c
View File

@ -208,7 +208,7 @@ static VALUE rb_mysql_result_fetch_row(int argc, VALUE * argv, VALUE self) {
return rowHash;
}
static VALUE rb_mysql_result_fetch_rows(int argc, VALUE * argv, VALUE self) {
static VALUE rb_mysql_result_each(int argc, VALUE * argv, VALUE self) {
VALUE dataset, opts, block;
MYSQL_RES * result;
unsigned long numRows, i;
@ -217,6 +217,10 @@ static VALUE rb_mysql_result_fetch_rows(int argc, VALUE * argv, VALUE self) {
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;
@ -262,7 +266,7 @@ void Init_mysql2_ext() {
cMysql2Result = rb_define_class_under(mMysql2, "Result", rb_cObject);
// rb_define_method(cMysql2Result, "fetch_row", rb_mysql_result_fetch_row, -1);
// rb_define_method(cMysql2Result, "fetch_rows", rb_mysql_result_fetch_rows, -1);
rb_define_method(cMysql2Result, "each", rb_mysql_result_fetch_rows, -1);
rb_define_method(cMysql2Result, "each", rb_mysql_result_each, -1);
VALUE mEnumerable = rb_const_get(rb_cObject, rb_intern("Enumerable"));
rb_include_module(cMysql2Result, mEnumerable);

2
ext/mysql2_ext.h
View File

@ -31,5 +31,5 @@ VALUE cMysql2Result;
static ID sym_symbolize_keys;
static VALUE rb_mysql_result_to_obj(MYSQL_RES * res);
static VALUE rb_mysql_result_fetch_row(int argc, VALUE * argv, VALUE self);
static VALUE rb_mysql_result_fetch_rows(int argc, VALUE * argv, VALUE self);
static VALUE rb_mysql_result_each(int argc, VALUE * argv, VALUE self);
void rb_mysql_result_free(void * result);