Meld rb_raise_mysql2_stmt_error2 into rb_raise_mysql2_stmt_error

Author: sodabrew

ext/mysql2/result.c

@@ -357,11 +357,7 @@ static VALUE rb_mysql_result_fetch_row_stmt(VALUE self, MYSQL_FIELD * fields, co
   }
 
   if (mysql_stmt_bind_result(wrapper->stmt_wrapper->stmt, wrapper->result_buffers)) {
-    rb_raise_mysql2_stmt_error2(wrapper->stmt_wrapper->stmt
-#ifdef HAVE_RUBY_ENCODING_H
-      , conn_enc
-#endif
-      );
+    rb_raise_mysql2_stmt_error(wrapper->stmt_wrapper);
   }
 
   {
@@ -372,11 +368,7 @@ static VALUE rb_mysql_result_fetch_row_stmt(VALUE self, MYSQL_FIELD * fields, co
 
       case 1:
         /* error */
-        rb_raise_mysql2_stmt_error2(wrapper->stmt_wrapper->stmt
-#ifdef HAVE_RUBY_ENCODING_H
-          , conn_enc
-#endif
-          );
+        rb_raise_mysql2_stmt_error(wrapper->stmt_wrapper);
 
       case MYSQL_NO_DATA:
         /* no more row */

ext/mysql2/statement.c

@@ -2,7 +2,7 @@
 
 VALUE cMysql2Statement;
 extern VALUE mMysql2, cMysql2Error, cBigDecimal, cDateTime, cDate;
-static VALUE sym_stream, intern_error_number_eql, intern_sql_state_eql, intern_each;
+static VALUE sym_stream, intern_new_with_args, intern_each;
 static VALUE intern_usec, intern_sec, intern_min, intern_hour, intern_day, intern_month, intern_year;
 
 #define GET_STATEMENT(self) \
@@ -31,15 +31,17 @@ void decr_mysql2_stmt(mysql_stmt_wrapper *stmt_wrapper) {
   }
 }
 
-VALUE rb_raise_mysql2_stmt_error2(MYSQL_STMT *stmt
-#ifdef HAVE_RUBY_ENCODING_H
-  , rb_encoding *conn_enc
-#endif
-  ) {
-  VALUE rb_error_msg = rb_str_new2(mysql_stmt_error(stmt));
-  VALUE rb_sql_state = rb_tainted_str_new2(mysql_stmt_sqlstate(stmt));
-  VALUE e = rb_exc_new3(cMysql2Error, rb_error_msg);
+
+void rb_raise_mysql2_stmt_error(mysql_stmt_wrapper *stmt_wrapper) {
+  VALUE e;
+  GET_CLIENT(stmt_wrapper->client);
+  VALUE rb_error_msg = rb_str_new2(mysql_stmt_error(stmt_wrapper->stmt));
+  VALUE rb_sql_state = rb_tainted_str_new2(mysql_stmt_sqlstate(stmt_wrapper->stmt));
+
 #ifdef HAVE_RUBY_ENCODING_H
+  rb_encoding *conn_enc;
+  conn_enc = rb_to_encoding(wrapper->encoding);
+
   rb_encoding *default_internal_enc = rb_default_internal_encoding();
 
   rb_enc_associate(rb_error_msg, conn_enc);
@@ -49,30 +51,13 @@ VALUE rb_raise_mysql2_stmt_error2(MYSQL_STMT *stmt
     rb_sql_state = rb_str_export_to_enc(rb_sql_state, default_internal_enc);
   }
 #endif
-  rb_funcall(e, intern_error_number_eql, 1, UINT2NUM(mysql_stmt_errno(stmt)));
-  rb_funcall(e, intern_sql_state_eql, 1, rb_sql_state);
-  rb_exc_raise(e);
-  return Qnil;
-}
 
-static void rb_raise_mysql2_stmt_error(VALUE self) {
-#ifdef HAVE_RUBY_ENCODING_H
-  rb_encoding *conn_enc;
-#endif
-  GET_STATEMENT(self);
-
-#ifdef HAVE_RUBY_ENCODING_H
-  {
-    GET_CLIENT(stmt_wrapper->client);
-    conn_enc = rb_to_encoding(wrapper->encoding);
-  }
-#endif
-
-  rb_raise_mysql2_stmt_error2(stmt_wrapper->stmt
-#ifdef HAVE_RUBY_ENCODING_H
-  , conn_enc
-#endif
-  );
+  e = rb_funcall(cMysql2Error, intern_new_with_args, 4,
+                 rb_error_msg,
+                 LONG2FIX(wrapper->server_version),
+                 UINT2NUM(mysql_stmt_errno(stmt_wrapper->stmt)),
+                 rb_sql_state);
+  rb_exc_raise(e);
 }
 
 
@@ -146,7 +131,7 @@ VALUE rb_mysql_stmt_new(VALUE rb_client, VALUE sql) {
     args.sql_len = RSTRING_LEN(sql);
 
     if ((VALUE)rb_thread_call_without_gvl(nogvl_prepare_statement, &args, RUBY_UBF_IO, 0) == Qfalse) {
-      rb_raise_mysql2_stmt_error(rb_stmt);
+      rb_raise_mysql2_stmt_error(stmt_wrapper);
     }
   }
 
@@ -335,13 +320,13 @@ static VALUE execute(int argc, VALUE *argv, VALUE self) {
     // copies bind_buffers into internal storage
     if (mysql_stmt_bind_param(stmt, bind_buffers)) {
       FREE_BINDS;
-      rb_raise_mysql2_stmt_error(self);
+      rb_raise_mysql2_stmt_error(stmt_wrapper);
     }
   }
 
   if ((VALUE)rb_thread_call_without_gvl(nogvl_execute, stmt, RUBY_UBF_IO, 0) == Qfalse) {
     FREE_BINDS;
-    rb_raise_mysql2_stmt_error(self);
+    rb_raise_mysql2_stmt_error(stmt_wrapper);
   }
 
   FREE_BINDS;
@@ -352,7 +337,7 @@ static VALUE execute(int argc, VALUE *argv, VALUE self) {
       // either CR_OUT_OF_MEMORY or CR_UNKNOWN_ERROR. both fatal.
 
       MARK_CONN_INACTIVE(stmt_wrapper->client);
-      rb_raise_mysql2_stmt_error(self);
+      rb_raise_mysql2_stmt_error(stmt_wrapper);
     }
     // no data and no error, so query was not a SELECT
     return Qnil;
@@ -367,7 +352,7 @@ static VALUE execute(int argc, VALUE *argv, VALUE self) {
     // recieve the whole result set from the server
     if (rb_thread_call_without_gvl(nogvl_stmt_store_result, stmt, RUBY_UBF_IO, 0) == Qfalse) {
       mysql_free_result(metadata);
-      rb_raise_mysql2_stmt_error(self);
+      rb_raise_mysql2_stmt_error(stmt_wrapper);
     }
     MARK_CONN_INACTIVE(stmt_wrapper->client);
   }
@@ -440,8 +425,7 @@ void init_mysql2_statement() {
 
   sym_stream = ID2SYM(rb_intern("stream"));
 
-  intern_error_number_eql = rb_intern("error_number=");
-  intern_sql_state_eql = rb_intern("sql_state=");
+  intern_new_with_args = rb_intern("new_with_args");
   intern_each = rb_intern("each");
 
   intern_usec = rb_intern("usec");

ext/mysql2/statement.h

@@ -13,10 +13,6 @@ void init_mysql2_statement();
 void decr_mysql2_stmt(mysql_stmt_wrapper *stmt_wrapper);
 
 VALUE rb_mysql_stmt_new(VALUE rb_client, VALUE sql);
-VALUE rb_raise_mysql2_stmt_error2(MYSQL_STMT *stmt
-#ifdef HAVE_RUBY_ENCODING_H
-  , rb_encoding* conn_enc
-#endif
-  );
+void rb_raise_mysql2_stmt_error(mysql_stmt_wrapper *stmt_wrapper);
 
 #endif