add affected_rows method (mysql_affected_rows)

This commit is contained in:
Brian Lopez 2010-04-07 09:47:45 -07:00
parent e20df5c140
commit 8f93f830c1
3 changed files with 39 additions and 15 deletions

View File

@ -254,11 +254,16 @@ static VALUE rb_mysql_client_async_result(VALUE self) {
static VALUE rb_mysql_client_last_id(VALUE self) {
MYSQL * client;
my_ulonglong id;
GetMysql2Client(self, client);
id = mysql_insert_id(client);
return ULL2NUM(id);
return ULL2NUM(mysql_insert_id(client));
}
static VALUE rb_mysql_client_affected_rows(VALUE self) {
MYSQL * client;
GetMysql2Client(self, client);
return ULL2NUM(mysql_affected_rows(client));
}
/* Mysql2::Result */
@ -459,6 +464,7 @@ void Init_mysql2_ext() {
rb_define_method(cMysql2Client, "socket", rb_mysql_client_socket, 0);
rb_define_method(cMysql2Client, "async_result", rb_mysql_client_async_result, 0);
rb_define_method(cMysql2Client, "last_id", rb_mysql_client_last_id, 0);
rb_define_method(cMysql2Client, "affected_rows", rb_mysql_client_affected_rows, 0);
cMysql2Error = rb_define_class_under(mMysql2, "Error", rb_eStandardError);

View File

@ -39,6 +39,7 @@ static VALUE rb_mysql_client_server_info(VALUE self);
static VALUE rb_mysql_client_socket(VALUE self);
static VALUE rb_mysql_client_async_result(VALUE self);
static VALUE rb_mysql_client_last_id(VALUE self);
static VALUE rb_mysql_client_affected_rows(VALUE self);
void rb_mysql_client_free(void * client);
/* Mysql2::Result */

View File

@ -117,18 +117,35 @@ describe Mysql2::Client do
result.class.should eql(Mysql2::Result)
end
context 'write operations api' do
before(:each) do
@client.query "USE test"
@client.query "CREATE TABLE lastIdTest (`id` int(11) NOT NULL AUTO_INCREMENT, blah INT(11), PRIMARY KEY (`id`))"
end
after(:each) do
@client.query "DROP TABLE lastIdTest"
end
it "should respond to #last_id" do
@client.should respond_to(:last_id)
end
it "#last_id should return a Fixnum, the from the last INSERT/UPDATE" do
@client.last_id.should eql(0)
@client.query "USE test"
@client.query "CREATE TABLE lastIdTest (`id` int(11) NOT NULL AUTO_INCREMENT, blah INT(11), PRIMARY KEY (`id`))"
@client.last_id.should eql(0)
@client.query "INSERT INTO lastIdTest (blah) VALUES (1234)"
@client.last_id.should eql(1)
@client.query "DROP TABLE lastIdTest"
@client.last_id.should eql(0)
end
it "should respond to #last_id" do
@client.should respond_to(:last_id)
end
it "#last_id should return a Fixnum, the from the last INSERT/UPDATE" do
@client.query "INSERT INTO lastIdTest (blah) VALUES (1234)"
@client.affected_rows.should eql(1)
@client.query "UPDATE lastIdTest SET blah=4321 WHERE id=1"
@client.affected_rows.should eql(1)
end
end
end