f2a731e0a7
Data_Make_Struct always allocates memory for us, so we
need to explicitly free the pointer in the function we
pass to the GC, not just objects internal to us.
Without this change, a slow but constant growth can be
seen with the trivial code below:
x = Mysql2::Client.new
loop { x.query "select 1" }
121 lines
3.5 KiB
C
121 lines
3.5 KiB
C
#include <ruby.h>
|
|
|
|
#ifdef HAVE_MYSQL_H
|
|
#include <mysql.h>
|
|
#include <mysql_com.h>
|
|
#include <errmsg.h>
|
|
#include <mysqld_error.h>
|
|
#else
|
|
#include <mysql/mysql.h>
|
|
#include <mysql/mysql_com.h>
|
|
#include <mysql/errmsg.h>
|
|
#include <mysql/mysqld_error.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_RUBY_ENCODING_H
|
|
#include <ruby/encoding.h>
|
|
static int utf8Encoding, binaryEncoding;
|
|
#endif
|
|
|
|
#if defined(__GNUC__) && (__GNUC__ >= 3)
|
|
#define RB_MYSQL_UNUSED __attribute__ ((unused))
|
|
#else
|
|
#define RB_MYSQL_UNUSED
|
|
#endif
|
|
|
|
static VALUE cBigDecimal, cDate, cDateTime;
|
|
static ID intern_new, intern_local;
|
|
|
|
/* Mysql2::Error */
|
|
static VALUE cMysql2Error;
|
|
|
|
/* Mysql2::Client */
|
|
typedef struct {
|
|
MYSQL * client;
|
|
} mysql2_client_wrapper;
|
|
#define GetMysql2Client(obj, sval) (sval = ((mysql2_client_wrapper*)(DATA_PTR(obj)))->client);
|
|
static ID sym_socket, sym_host, sym_port, sym_username, sym_password,
|
|
sym_database, sym_reconnect, sym_connect_timeout, sym_id, sym_version,
|
|
sym_sslkey, sym_sslcert, sym_sslca, sym_sslcapath, sym_sslcipher,
|
|
sym_symbolize_keys, sym_async;
|
|
static VALUE rb_mysql_client_new(int argc, VALUE * argv, VALUE klass);
|
|
static VALUE rb_mysql_client_init(int argc, VALUE * argv, VALUE self);
|
|
static VALUE rb_mysql_client_query(int argc, VALUE * argv, VALUE self);
|
|
static VALUE rb_mysql_client_escape(VALUE self, VALUE str);
|
|
static VALUE rb_mysql_client_info(VALUE self);
|
|
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);
|
|
static void rb_mysql_client_free(void * client);
|
|
|
|
/* Mysql2::Result */
|
|
typedef struct {
|
|
VALUE fields;
|
|
VALUE rows;
|
|
unsigned long numberOfFields;
|
|
unsigned long numberOfRows;
|
|
unsigned long lastRowProcessed;
|
|
int resultFreed;
|
|
MYSQL_RES *result;
|
|
} mysql2_result_wrapper;
|
|
#define GetMysql2Result(obj, sval) (sval = (mysql2_result_wrapper*)DATA_PTR(obj));
|
|
static VALUE cMysql2Result;
|
|
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_each(int argc, VALUE * argv, VALUE self);
|
|
static void rb_mysql_result_free(void * wrapper);
|
|
static void rb_mysql_result_mark(void * wrapper);
|
|
static void rb_mysql_result_free_result(mysql2_result_wrapper * wrapper);
|
|
|
|
/*
|
|
* used to pass all arguments to mysql_real_connect while inside
|
|
* rb_thread_blocking_region
|
|
*/
|
|
struct nogvl_connect_args {
|
|
MYSQL *mysql;
|
|
const char *host;
|
|
const char *user;
|
|
const char *passwd;
|
|
const char *db;
|
|
unsigned int port;
|
|
const char *unix_socket;
|
|
unsigned long client_flag;
|
|
};
|
|
|
|
/*
|
|
* used to pass all arguments to mysql_send_query while inside
|
|
* rb_thread_blocking_region
|
|
*/
|
|
struct nogvl_send_query_args {
|
|
MYSQL *mysql;
|
|
VALUE sql;
|
|
};
|
|
|
|
/*
|
|
* partial emulation of the 1.9 rb_thread_blocking_region under 1.8,
|
|
* this is enough for dealing with blocking I/O functions in the
|
|
* presence of threads.
|
|
*/
|
|
#ifndef HAVE_RB_THREAD_BLOCKING_REGION
|
|
# include <rubysig.h>
|
|
# define RUBY_UBF_IO ((rb_unblock_function_t *)-1)
|
|
typedef void rb_unblock_function_t(void *);
|
|
typedef VALUE rb_blocking_function_t(void *);
|
|
static VALUE
|
|
rb_thread_blocking_region(
|
|
rb_blocking_function_t *func, void *data1,
|
|
RB_MYSQL_UNUSED rb_unblock_function_t *ubf,
|
|
RB_MYSQL_UNUSED void *data2)
|
|
{
|
|
VALUE rv;
|
|
|
|
TRAP_BEG;
|
|
rv = func(data1);
|
|
TRAP_END;
|
|
|
|
return rv;
|
|
}
|
|
#endif /* ! HAVE_RB_THREAD_BLOCKING_REGION */
|