Class: MTBL::Sorter
- Inherits:
-
Object
- Object
- MTBL::Sorter
- Defined in:
- ext/mtbl/ruby-mtbl.c
Instance Method Summary collapse
- #add(key, val) ⇒ Object
- #close ⇒ Object
- #initialize(*args) ⇒ Object constructor
- #write(c_writer) ⇒ Object
Constructor Details
#initialize(*args) ⇒ Object
374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 |
# File 'ext/mtbl/ruby-mtbl.c', line 374
static VALUE rbmtbl_sorter_initialize(int argc, VALUE *argv, VALUE self) {
rbmtbl_sorter_t *sorter;
struct stat ss;
int scanc;
VALUE mergef, tempd, maxm;
Data_Get_Struct(self, rbmtbl_sorter_t, sorter);
scanc = rb_scan_args(argc, argv, "03", &mergef, &tempd, &maxm);
mtbl_sorter_options_set_merge_func(sorter->o, (mtbl_merge_func) rbmbtl_default_merge_func, (void *) sorter);
if (scanc > 1 && mergef != Qnil) {
// TODO: Implement the merge callback
}
if (scanc > 2 && tempd != Qnil) {
if (TYPE(tempd) != T_STRING) {
rb_raise(rb_eArgError, "Temporary directory should be a string");
return Qnil;
}
if (stat(StringValueCStr(tempd), &ss)) {
rb_raise(rb_eArgError, "Temporary directory does not exist: %s", StringValueCStr(tempd));
return Qnil;
}
if (! S_ISDIR(ss.st_mode)) {
rb_raise(rb_eArgError, "Path %s is not a directory", StringValueCStr(tempd));
return Qnil;
}
mtbl_sorter_options_set_temp_dir(sorter->o, StringValueCStr(tempd));
}
if (scanc > 3 && maxm != Qnil) {
mtbl_sorter_options_set_max_memory(sorter->o, NUM2ULONG(maxm));
}
// Verify that c_writer is a MTBL::Writer
sorter->s = mtbl_sorter_init(sorter->o);
if (sorter->s == NULL) {
rb_raise(rb_eRuntimeError, "Failed to create sorter");
return Qnil;
}
return self;
}
|
Instance Method Details
#add(key, val) ⇒ Object
420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 |
# File 'ext/mtbl/ruby-mtbl.c', line 420
static VALUE rbmtbl_sorter_add(VALUE self, VALUE key, VALUE val) {
rbmtbl_sorter_t *sorter;
Data_Get_Struct(self, rbmtbl_sorter_t, sorter);
if (! sorter->s) {
rb_raise(rb_eRuntimeError, "Failed to write key %s: sorter closed", StringValueCStr(key));
return Qnil;
}
if (TYPE(key) != T_STRING) {
rb_raise(rb_eArgError, "Key must be a string");
return Qnil;
}
if (TYPE(val) != T_STRING) {
rb_raise(rb_eArgError, "Value must be a string");
return Qnil;
}
if( mtbl_res_success != mtbl_sorter_add(sorter->s,
(const uint8_t *) RSTRING_PTR(key), RSTRING_LEN(key),
(const uint8_t *) RSTRING_PTR(val), RSTRING_LEN(val))) {
rb_raise(rb_eRuntimeError, "Failed to write key %s, input must be presorted", StringValueCStr(key));
return Qnil;
}
return Qtrue;
}
|
#close ⇒ Object
447 448 449 450 451 452 453 454 455 456 457 |
# File 'ext/mtbl/ruby-mtbl.c', line 447
static VALUE rbmtbl_sorter_close(VALUE self) {
rbmtbl_sorter_t *sorter;
Data_Get_Struct(self, rbmtbl_sorter_t, sorter);
if (sorter->s) {
rbmtbl_sorter_close_handles(sorter);
return Qtrue;
} else {
rb_raise(rb_eRuntimeError, "Sorter is already closed");
return Qfalse;
}
}
|
#write(c_writer) ⇒ Object
459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 |
# File 'ext/mtbl/ruby-mtbl.c', line 459
static VALUE rbmtbl_sorter_write(VALUE self, VALUE c_writer) {
rbmtbl_sorter_t *sorter;
rbmtbl_writer_t *writer;
Data_Get_Struct(self, rbmtbl_sorter_t, sorter);
// Verify MTBL::Writer is c_writer
Data_Get_Struct(c_writer, rbmtbl_writer_t, writer);
if (mtbl_res_success != mtbl_sorter_write(sorter->s, writer->w)) {
rb_raise(rb_eRuntimeError, "Failed to write");
return Qnil;
}
return Qtrue;
}
|