/*
* Document-method: Bignum#to_msgpack
*
* call-seq:
* bignum.to_msgpack(out = '') -> String
*
* Serializes the Bignum into raw bytes.
*/
static VALUE MessagePack_Bignum_to_msgpack(int argc, VALUE *argv, VALUE self)
{
ARG_BUFFER(out, argc, argv);
if(RBIGNUM_POSITIVE_P(self)) {
msgpack_pack_uint64(out, rb_big2ull(self));
} else {
msgpack_pack_int64(out, rb_big2ll(self));
}
return out;
}
/*
* Document-method: Bignum#to_msgpack
*
* call-seq:
* bignum.to_msgpack(out = '') -> String
*
* Serializes the Bignum into raw bytes.
*/
static VALUE MessagePack_Bignum_to_msgpack(int argc, VALUE *argv, VALUE self)
{
ARG_BUFFER(out, argc, argv);
// FIXME bignum
if(RBIGNUM_SIGN(self)) { // positive
msgpack_pack_uint64(out, rb_big2ull(self));
} else { // negative
msgpack_pack_int64(out, rb_big2ll(self));
}
return out;
}
void mochilo_pack_bignum(mochilo_buf *buf, VALUE rb_bignum)
{
if (RBIGNUM_POSITIVE_P(rb_bignum)) {
uint64_t bignum = rb_big2ull(rb_bignum);
mochilo_buf_putc(buf, MSGPACK_T_UINT64);
mochilo_buf_put64be(buf, &bignum);
} else {
int64_t bignum = rb_big2ll(rb_bignum);
mochilo_buf_putc(buf, MSGPACK_T_INT64);
mochilo_buf_put64be(buf, &bignum);
}
}
static VALUE bignum_spec_rb_big2ll(VALUE self, VALUE num) {
return rb_ll2inum(rb_big2ll(num));
}
/* call-seq: stmt.execute
*
* Executes the current prepared statement, returns +stmt+.
*/
static VALUE execute(int argc, VALUE *argv, VALUE self) {
MYSQL_STMT *stmt;
MYSQL_BIND *bind_buffers;
unsigned long bind_count;
long i;
Data_Get_Struct(self, MYSQL_STMT, stmt);
bind_count = mysql_stmt_param_count(stmt);
if (argc != (long)bind_count) {
rb_raise(cMysql2Error, "Bind parameter count (%ld) doesn't match number of arguments (%d)", bind_count, argc);
}
// setup any bind variables in the query
if (bind_count > 0) {
bind_buffers = xcalloc(bind_count, sizeof(MYSQL_BIND));
for (i = 0; i < argc; i++) {
bind_buffers[i].buffer = NULL;
switch (TYPE(argv[i])) {
case T_NIL:
bind_buffers[i].buffer_type = MYSQL_TYPE_NULL;
break;
case T_FIXNUM:
#if SIZEOF_INT < SIZEOF_LONG
bind_buffers[i].buffer_type = MYSQL_TYPE_LONGLONG;
bind_buffers[i].buffer = malloc(sizeof(long long int));
*(long*)(bind_buffers[i].buffer) = FIX2LONG(argv[i]);
#else
bind_buffers[i].buffer_type = MYSQL_TYPE_LONG;
bind_buffers[i].buffer = malloc(sizeof(int));
*(long*)(bind_buffers[i].buffer) = FIX2INT(argv[i]);
#endif
break;
case T_BIGNUM:
bind_buffers[i].buffer_type = MYSQL_TYPE_LONGLONG;
bind_buffers[i].buffer = malloc(sizeof(long long int));
*(LONG_LONG*)(bind_buffers[i].buffer) = rb_big2ll(argv[i]);
break;
case T_FLOAT:
bind_buffers[i].buffer_type = MYSQL_TYPE_DOUBLE;
bind_buffers[i].buffer = malloc(sizeof(double));
*(double*)(bind_buffers[i].buffer) = NUM2DBL(argv[i]);
break;
case T_STRING:
bind_buffers[i].buffer_type = MYSQL_TYPE_STRING;
bind_buffers[i].buffer = RSTRING_PTR(argv[i]);
bind_buffers[i].buffer_length = RSTRING_LEN(argv[i]);
unsigned long *len = malloc(sizeof(long));
(*len) = RSTRING_LEN(argv[i]);
bind_buffers[i].length = len;
break;
default:
// TODO: what Ruby type should support MYSQL_TYPE_TIME
if (CLASS_OF(argv[i]) == rb_cTime || CLASS_OF(argv[i]) == cDateTime) {
bind_buffers[i].buffer_type = MYSQL_TYPE_DATETIME;
bind_buffers[i].buffer = malloc(sizeof(MYSQL_TIME));
MYSQL_TIME t;
VALUE rb_time = argv[i];
memset(&t, 0, sizeof(MYSQL_TIME));
t.second_part = 0;
t.neg = 0;
t.second = FIX2INT(rb_funcall(rb_time, rb_intern("sec"), 0));
t.minute = FIX2INT(rb_funcall(rb_time, rb_intern("min"), 0));
t.hour = FIX2INT(rb_funcall(rb_time, rb_intern("hour"), 0));
t.day = FIX2INT(rb_funcall(rb_time, rb_intern("day"), 0));
t.month = FIX2INT(rb_funcall(rb_time, rb_intern("month"), 0));
t.year = FIX2INT(rb_funcall(rb_time, rb_intern("year"), 0));
*(MYSQL_TIME*)(bind_buffers[i].buffer) = t;
} else if (CLASS_OF(argv[i]) == cDate) {
bind_buffers[i].buffer_type = MYSQL_TYPE_DATE;
bind_buffers[i].buffer = malloc(sizeof(MYSQL_TIME));
MYSQL_TIME t;
VALUE rb_time = argv[i];
memset(&t, 0, sizeof(MYSQL_TIME));
t.second_part = 0;
t.neg = 0;
t.day = FIX2INT(rb_funcall(rb_time, rb_intern("day"), 0));
t.month = FIX2INT(rb_funcall(rb_time, rb_intern("month"), 0));
t.year = FIX2INT(rb_funcall(rb_time, rb_intern("year"), 0));
*(MYSQL_TIME*)(bind_buffers[i].buffer) = t;
} else if (CLASS_OF(argv[i]) == cBigDecimal) {
bind_buffers[i].buffer_type = MYSQL_TYPE_NEWDECIMAL;
}
break;
}
}
// copies bind_buffers into internal storage
if (mysql_stmt_bind_param(stmt, bind_buffers)) {
FREE_BINDS;
rb_raise(cMysql2Error, "%s", mysql_stmt_error(stmt));
}
}
if (rb_thread_blocking_region(nogvl_execute, stmt, RUBY_UBF_IO, 0) == Qfalse) {
FREE_BINDS;
rb_raise(cMysql2Error, "%s", mysql_stmt_error(stmt));
}
if (bind_count > 0) {
FREE_BINDS;
}
return self;
}
/**
* call-seq:
* 23874.localize(:decimal, 'es_ES') => 23.874
*/
VALUE rb_numeric_localize(int argc, VALUE *argv, VALUE self)
{
VALUE style, options;
UNumberFormatStyle formatStyle;
char *locale = NULL;
UNumberFormat *format;
UErrorCode status;
UChar result[256];
/* arguments */
rb_scan_args(argc, argv, "02", &style, &options);
if (style == Qnil) {
formatStyle = UNUM_DECIMAL;
} else {
ID style_ID;
Check_Type(style, T_SYMBOL);
style_ID = SYM2ID(style);
if (style_ID == rb_intern("decimal")) {
formatStyle = UNUM_DECIMAL;
} else if (style_ID == rb_intern("currency")) {
formatStyle = UNUM_CURRENCY;
} else if (style_ID == rb_intern("percent")) {
formatStyle = UNUM_PERCENT;
} else if (style_ID == rb_intern("scientific")) {
formatStyle = UNUM_SCIENTIFIC;
} else if (style_ID == rb_intern("spellout")) {
formatStyle = UNUM_SPELLOUT;
} else {
rb_raise(rb_eArgError, "unsupported format style %s", rb_id2name(style_ID));
}
}
if (options != Qnil) {
VALUE rb_locale;
Check_Type(options, T_HASH);
rb_locale = rb_hash_aref(options, ID2SYM(rb_intern("locale")));
if (rb_locale != Qnil) {
locale = StringValuePtr(rb_locale);
}
}
/* formatter */
status = U_ZERO_ERROR;
format = unum_open(formatStyle, NULL, 0, locale, NULL, &status);
RAISE_ON_ERROR(status);
/* set format attributes */
if (options != Qnil) {
VALUE currency, precision, round_mode, round_increment;
switch (formatStyle) {
case UNUM_CURRENCY:
currency = rb_hash_aref(options, ID2SYM(rb_intern("currency")));
if (currency != Qnil) {
UChar *uStr;
int32_t uStrLen;
uStr = u_strFromRString(currency, &uStrLen);
status = U_ZERO_ERROR;
unum_setTextAttribute(format, UNUM_CURRENCY_CODE, uStr, uStrLen, &status);
RAISE_ON_ERROR(status);
}
case UNUM_DECIMAL:
/* precision */
precision = rb_hash_aref(options, ID2SYM(rb_intern("precision")));
if (precision != Qnil) {
Check_Type(precision, T_FIXNUM);
status = U_ZERO_ERROR;
unum_setAttribute(format, UNUM_FRACTION_DIGITS, NUM2INT(precision));
RAISE_ON_ERROR(status);
}
round_mode = rb_hash_aref(options, ID2SYM(rb_intern("round_mode")));
if (round_mode != Qnil) {
ID round_mode_ID;
UNumberFormatRoundingMode rounding_mode;
Check_Type(round_mode, T_SYMBOL);
round_mode_ID = SYM2ID(round_mode);
if (round_mode_ID == rb_intern("ceil")) {
rounding_mode = UNUM_ROUND_CEILING;
} else if (round_mode_ID == rb_intern("floor")) {
rounding_mode = UNUM_ROUND_FLOOR;
} else if (round_mode_ID == rb_intern("down")) {
rounding_mode = UNUM_ROUND_DOWN;
} else if (round_mode_ID == rb_intern("up")) {
rounding_mode = UNUM_ROUND_UP;
} else if (round_mode_ID == rb_intern("halfeven")) {
rounding_mode = UNUM_FOUND_HALFEVEN;
} else if (round_mode_ID == rb_intern("halfdown")) {
rounding_mode = UNUM_ROUND_HALFDOWN;
} else if (round_mode_ID == rb_intern("halfup")) {
rounding_mode = UNUM_ROUND_HALFUP;
} else {
rb_raise(rb_eArgError, "unsupported rounding mode '%s'", rb_id2name(round_mode_ID));
}
status = U_ZERO_ERROR;
unum_setAttribute(format, UNUM_ROUNDING_MODE, rounding_mode);
RAISE_ON_ERROR(status);
}
round_increment = rb_hash_aref(options, ID2SYM(rb_intern("round_increment")));
if (round_increment != Qnil) {
Check_Type(round_increment, T_FLOAT);
status = U_ZERO_ERROR;
unum_setDoubleAttribute(format, UNUM_ROUNDING_INCREMENT, NUM2DBL(round_increment));
RAISE_ON_ERROR(status);
}
}
}
/* format */
status = U_ZERO_ERROR;
switch (TYPE(self)) {
case T_FIXNUM:
unum_format(format, NUM2INT(self), result, 256, NULL, &status);
break;
case T_FLOAT:
unum_formatDouble(format, NUM2DBL(self), result, 256, NULL, &status);
break;
case T_BIGNUM:
unum_formatInt64(format, rb_big2ll(self), result, 256, NULL, &status);
break;
}
RAISE_ON_ERROR(status);
/* free resources */
unum_close(format);
return u_strToRString(result, -1);
}
/* call-seq: stmt.execute
*
* Executes the current prepared statement, returns +result+.
*/
static VALUE execute(int argc, VALUE *argv, VALUE self) {
MYSQL_BIND *bind_buffers = NULL;
unsigned long *length_buffers = NULL;
unsigned long bind_count;
long i;
MYSQL_STMT *stmt;
MYSQL_RES *metadata;
VALUE current;
VALUE resultObj;
VALUE *params_enc;
int is_streaming;
#ifdef HAVE_RUBY_ENCODING_H
rb_encoding *conn_enc;
#endif
GET_STATEMENT(self);
GET_CLIENT(stmt_wrapper->client);
#ifdef HAVE_RUBY_ENCODING_H
conn_enc = rb_to_encoding(wrapper->encoding);
#endif
/* Scratch space for string encoding exports, allocate on the stack. */
params_enc = alloca(sizeof(VALUE) * argc);
stmt = stmt_wrapper->stmt;
bind_count = mysql_stmt_param_count(stmt);
if (argc != (long)bind_count) {
rb_raise(cMysql2Error, "Bind parameter count (%ld) doesn't match number of arguments (%d)", bind_count, argc);
}
// setup any bind variables in the query
if (bind_count > 0) {
bind_buffers = xcalloc(bind_count, sizeof(MYSQL_BIND));
length_buffers = xcalloc(bind_count, sizeof(unsigned long));
for (i = 0; i < argc; i++) {
bind_buffers[i].buffer = NULL;
params_enc[i] = Qnil;
switch (TYPE(argv[i])) {
case T_NIL:
bind_buffers[i].buffer_type = MYSQL_TYPE_NULL;
break;
case T_FIXNUM:
#if SIZEOF_INT < SIZEOF_LONG
bind_buffers[i].buffer_type = MYSQL_TYPE_LONGLONG;
bind_buffers[i].buffer = xmalloc(sizeof(long long int));
*(long*)(bind_buffers[i].buffer) = FIX2LONG(argv[i]);
#else
bind_buffers[i].buffer_type = MYSQL_TYPE_LONG;
bind_buffers[i].buffer = xmalloc(sizeof(int));
*(long*)(bind_buffers[i].buffer) = FIX2INT(argv[i]);
#endif
break;
case T_BIGNUM:
bind_buffers[i].buffer_type = MYSQL_TYPE_LONGLONG;
bind_buffers[i].buffer = xmalloc(sizeof(long long int));
*(LONG_LONG*)(bind_buffers[i].buffer) = rb_big2ll(argv[i]);
break;
case T_FLOAT:
bind_buffers[i].buffer_type = MYSQL_TYPE_DOUBLE;
bind_buffers[i].buffer = xmalloc(sizeof(double));
*(double*)(bind_buffers[i].buffer) = NUM2DBL(argv[i]);
break;
case T_STRING:
{
params_enc[i] = argv[i];
#ifdef HAVE_RUBY_ENCODING_H
params_enc[i] = rb_str_export_to_enc(params_enc[i], conn_enc);
#endif
set_buffer_for_string(&bind_buffers[i], &length_buffers[i], params_enc[i]);
}
break;
default:
// TODO: what Ruby type should support MYSQL_TYPE_TIME
if (CLASS_OF(argv[i]) == rb_cTime || CLASS_OF(argv[i]) == cDateTime) {
MYSQL_TIME t;
VALUE rb_time = argv[i];
bind_buffers[i].buffer_type = MYSQL_TYPE_DATETIME;
bind_buffers[i].buffer = xmalloc(sizeof(MYSQL_TIME));
memset(&t, 0, sizeof(MYSQL_TIME));
t.neg = 0;
t.second_part = FIX2INT(rb_funcall(rb_time, intern_usec, 0));
t.second = FIX2INT(rb_funcall(rb_time, intern_sec, 0));
t.minute = FIX2INT(rb_funcall(rb_time, intern_min, 0));
t.hour = FIX2INT(rb_funcall(rb_time, intern_hour, 0));
t.day = FIX2INT(rb_funcall(rb_time, intern_day, 0));
t.month = FIX2INT(rb_funcall(rb_time, intern_month, 0));
t.year = FIX2INT(rb_funcall(rb_time, intern_year, 0));
*(MYSQL_TIME*)(bind_buffers[i].buffer) = t;
} else if (CLASS_OF(argv[i]) == cDate) {
MYSQL_TIME t;
VALUE rb_time = argv[i];
bind_buffers[i].buffer_type = MYSQL_TYPE_DATE;
bind_buffers[i].buffer = xmalloc(sizeof(MYSQL_TIME));
memset(&t, 0, sizeof(MYSQL_TIME));
t.second_part = 0;
t.neg = 0;
t.day = FIX2INT(rb_funcall(rb_time, intern_day, 0));
t.month = FIX2INT(rb_funcall(rb_time, intern_month, 0));
t.year = FIX2INT(rb_funcall(rb_time, intern_year, 0));
*(MYSQL_TIME*)(bind_buffers[i].buffer) = t;
} else if (CLASS_OF(argv[i]) == cBigDecimal) {
bind_buffers[i].buffer_type = MYSQL_TYPE_NEWDECIMAL;
// DECIMAL are represented with the "string representation of the
// original server-side value", see
// https://dev.mysql.com/doc/refman/5.7/en/c-api-prepared-statement-type-conversions.html
// This should be independent of the locale used both on the server
// and the client side.
VALUE rb_val_as_string = rb_funcall(argv[i], intern_to_s, 0);
params_enc[i] = rb_val_as_string;
#ifdef HAVE_RUBY_ENCODING_H
params_enc[i] = rb_str_export_to_enc(params_enc[i], conn_enc);
#endif
set_buffer_for_string(&bind_buffers[i], &length_buffers[i], params_enc[i]);
}
break;
}
}
// copies bind_buffers into internal storage
if (mysql_stmt_bind_param(stmt, bind_buffers)) {
FREE_BINDS;
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(stmt_wrapper);
}
FREE_BINDS;
metadata = mysql_stmt_result_metadata(stmt);
if (metadata == NULL) {
if (mysql_stmt_errno(stmt) != 0) {
// either CR_OUT_OF_MEMORY or CR_UNKNOWN_ERROR. both fatal.
wrapper->active_thread = Qnil;
rb_raise_mysql2_stmt_error(stmt_wrapper);
}
// no data and no error, so query was not a SELECT
return Qnil;
}
current = rb_hash_dup(rb_iv_get(stmt_wrapper->client, "@query_options"));
(void)RB_GC_GUARD(current);
Check_Type(current, T_HASH);
is_streaming = (Qtrue == rb_hash_aref(current, sym_stream));
if (!is_streaming) {
// recieve the whole result set from the server
if (mysql_stmt_store_result(stmt)) {
mysql_free_result(metadata);
rb_raise_mysql2_stmt_error(stmt_wrapper);
}
wrapper->active_thread = Qnil;
}
resultObj = rb_mysql_result_to_obj(stmt_wrapper->client, wrapper->encoding, current, metadata, self);
if (!is_streaming) {
// cache all result
rb_funcall(resultObj, intern_each, 0);
}
return resultObj;
}
