void
test_array_set_data(void)
{
grn_obj *table;
grn_id record_id;
gchar value[] = "sample value";
grn_obj *record_value;
grn_obj *retrieved_record_value;
const gchar *dupped_retrieved_record_value;
const gchar *value_type_name = "value_type";
grn_obj *value_type;
value_type = grn_type_create(context,
value_type_name, strlen(value_type_name),
0, sizeof(value));
table = grn_table_create(context, NULL, 0, NULL,
GRN_OBJ_TABLE_NO_KEY,
NULL, value_type);
record_id = grn_table_add(context, table, NULL, 0, NULL);
record_value = grn_obj_open(context, GRN_BULK, 0, GRN_DB_TEXT);
grn_bulk_write(context, record_value, value, sizeof(value));
grn_test_assert(grn_obj_set_value(context, table, record_id,
record_value, GRN_OBJ_SET));
grn_obj_unlink(context, record_value);
retrieved_record_value = grn_obj_get_value(context, table, record_id, NULL);
dupped_retrieved_record_value =
cut_take_strdup(GRN_BULK_HEAD(retrieved_record_value));
grn_obj_unlink(context, retrieved_record_value);
cut_assert_equal_string(value, dupped_retrieved_record_value);
}
static gboolean
run_test(const gchar **test_case_names, const grn_test_data *data)
{
const gchar *type_name, *table_name;
gchar *path;
grn_test_assert(grn_ctx_init(context, GRN_CTX_USE_QL));
GRN_CTX_SET_ENCODING(context, GRN_ENC_UTF8);
type_name = "name";
type = grn_type_create(context, type_name, strlen(type_name),
GRN_OBJ_KEY_UINT, sizeof(grn_id));
path = g_build_filename(base_dir, "table", NULL);
g_setenv(GRN_TEST_ENV_TABLE_PATH, path, TRUE);
table_name = cut_take_printf("%s: performance-read-write", data->type_name);
g_setenv(GRN_TEST_ENV_TABLE_TYPE, data->type_name, TRUE);
table = grn_table_create(context,
table_name, strlen(table_name),
path, GRN_OBJ_PERSISTENT | data->flags,
type, NULL);
g_free(path);
cut_assert_not_null(table);
return run(test_case_names, data);
}
/*
* 名前が _name_ の型を作成する。
*
* @overload new(name, options={})
* @param name [String] 作成する型の名前
* @param options [::Hash] The name and value
* pairs. Omitted names are initialized as the default value
* @option options [Symbol] :type (:variable)
* :integer（符号付き整数）、:int（:integerの省略
* 形）、:unsigned_integer（符号なし整
* 数）、:uint（:unsigned_integerの省略形）、:float（浮動小数点
* 数）、:variable（可変長文字列）のいずれかを指定する。省略した場
* 合は:variableを指定したものと扱う。
* :variableを指定した場合は必ず +:size+ を指定しなければいけない。
* @option options [Context] :context
* 型の作成時に利用するGroonga::Contextを指定する。省略すると
* Groonga::Context.defaultを用いる。
* @option options [Integer] :size
* +:option+ が:variableの場合は最大長、それ以外の場合は長さを
* 指定する(単位:byte)。
*/
static VALUE
rb_grn_type_initialize (int argc, VALUE *argv, VALUE self)
{
grn_ctx *context;
grn_obj *type;
const char *name = NULL;
unsigned name_size, size = 0;
grn_obj_flags flags = 0;
VALUE rb_name, options, rb_context, rb_type, rb_size;
rb_scan_args(argc, argv, "11", &rb_name, &options);
rb_grn_scan_options(options,
"context", &rb_context,
"type", &rb_type,
"size", &rb_size,
NULL);
name = StringValuePtr(rb_name);
name_size = RSTRING_LEN(rb_name);
context = rb_grn_context_ensure(&rb_context);
if (NIL_P(rb_type) ||
rb_grn_equal_option(rb_type, "variable")) {
flags = GRN_OBJ_KEY_VAR_SIZE;
} else if (rb_grn_equal_option(rb_type, "integer") ||
rb_grn_equal_option(rb_type, "int")) {
flags = GRN_OBJ_KEY_INT;
size = sizeof(int);
} else if (rb_grn_equal_option(rb_type, "unsigned_integer") ||
rb_grn_equal_option(rb_type, "uint")) {
flags = GRN_OBJ_KEY_UINT;
size = sizeof(unsigned int);
} else if (rb_grn_equal_option(rb_type, "float")) {
flags = GRN_OBJ_KEY_FLOAT;
size = sizeof(double);
} else {
rb_raise(rb_eArgError,
":type should be one of "
"[:integer, :int, :unsigned_integer, :uint, "
":float, :variable]: %s",
rb_grn_inspect(options));
}
if (NIL_P(rb_size)) {
if (size == 0)
rb_raise(rb_eArgError, "size is missing: %s",
rb_grn_inspect(options));
} else {
size = NUM2UINT(rb_size);
}
type = grn_type_create(context, name, name_size, flags, size);
rb_grn_object_assign(Qnil, self, rb_context, context, type);
rb_grn_context_check(context, rb_ary_new4(argc, argv));
return Qnil;
}
void
cut_setup(void)
{
gchar *table_path, *vgram_path;
const gchar *type_name, *table_name;
cut_set_fixture_data_dir(grn_test_get_base_dir(),
"fixtures",
"inverted-index",
NULL);
logger = setup_grn_logger();
expected_messages = NULL;
record_ids = NULL;
remove_tmp_directory();
g_mkdir_with_parents(tmp_directory, 0700);
path = g_build_filename(tmp_directory, "inverted-index", NULL);
context = g_new0(grn_ctx, 1);
grn_test_assert(grn_ctx_init(context, GRN_CTX_USE_QL));
GRN_CTX_SET_ENCODING(context, GRN_ENC_UTF8);
db = grn_db_create(context, NULL, NULL);
grn_ctx_use(context, db);
type_name = "name";
type = grn_type_create(context, type_name, strlen(type_name),
GRN_OBJ_KEY_VAR_SIZE, TYPE_SIZE);
table_name = "lexicon";
table_path = g_build_filename(tmp_directory, "lexicon-table", NULL);
lexicon = grn_table_create(context,
table_name, strlen(table_name),
table_path,
GRN_OBJ_PERSISTENT | GRN_OBJ_TABLE_PAT_KEY,
type, NULL);
grn_obj_set_info(context, lexicon, GRN_INFO_DEFAULT_TOKENIZER,
grn_ctx_at(context, GRN_DB_BIGRAM));
g_free(table_path);
vgram_path = g_build_filename(tmp_directory, "vgram", NULL);
/*
vgram = grn_vgram_create(vgram_path);
*/
g_free(vgram_path);
inverted_index = NULL;
}
static grn_obj *
grn_table_factory_make(grn_table_factory *factory)
{
grn_obj *value_type = grn_ctx_get(factory->context, VALUE_TYPE_NAME, strlen(VALUE_TYPE_NAME));
if (!value_type) {
value_type = grn_type_create(factory->context, VALUE_TYPE_NAME, strlen(VALUE_TYPE_NAME),
0, factory->value_size);
}
GRN_CTX_SET_ENCODING(factory->context, factory->encoding);
return grn_table_create(factory->context,
factory->name, factory->name_size,
factory->path, factory->flags,
factory->key_type, value_type);
}
static void
bench_normal(gpointer user_data)
{
BenchmarkData *data = user_data;
grn_obj *table;
grn_obj *value_type = grn_ctx_get(data->context, VALUE_TYPE_NAME, strlen(VALUE_TYPE_NAME));
if (!value_type) {
value_type = grn_type_create(data->context, VALUE_TYPE_NAME, strlen(VALUE_TYPE_NAME),
0, data->value_size);
}
table = grn_table_create(data->context,
data->name, data->name_size,
data->path, data->flags,
data->key_type, value_type);
grn_obj_close(data->context, table);
}
static void
bench_normal_temporary(gpointer user_data)
{
BenchmarkData *data = user_data;
grn_obj *table;
grn_obj *value_type = grn_ctx_get(data->context, VALUE_TYPE_NAME, strlen(VALUE_TYPE_NAME));
if (!value_type) {
value_type = grn_type_create(data->context, VALUE_TYPE_NAME, strlen(VALUE_TYPE_NAME),
0, data->value_size);
}
GRN_CTX_SET_ENCODING(data->context, data->encoding);
table = grn_table_create(data->context,
data->name, data->name_size,
NULL, data->flags & ~GRN_OBJ_PERSISTENT,
data->key_type, value_type);
grn_obj_close(data->context, table);
}
static void
bench_setup(gpointer user_data)
{
BenchmarkData *data = user_data;
const gchar *type_name;
bench_utils_remove_path_recursive_force(data->base_dir);
g_mkdir_with_parents(data->base_dir, 0755);
grn_ctx_init(data->context, GRN_CTX_USE_QL);
type_name = "name";
data->key_type = grn_type_create(data->context,
type_name, strlen(type_name),
GRN_OBJ_KEY_UINT, sizeof(grn_id));
}
void
test_array_truncate(void)
{
grn_obj *table;
gchar value[] = "sample value";
gchar *value_type_name = "value_type";
grn_obj *value_type;
cut_omit("grn_table_truncate() is still buggy.");
value_type = grn_type_create(context,
value_type_name, strlen(value_type_name),
0, sizeof(value));
table = grn_table_create(context, NULL, 0, NULL,
GRN_OBJ_TABLE_NO_KEY,
NULL, value_type);
grn_test_assert_not_nil(grn_table_add(context, table, NULL, 0, NULL));
cut_assert_equal_uint(1, grn_table_size(context, table));
grn_test_assert(grn_table_truncate(context, table));
cut_assert_equal_uint(0, grn_table_size(context, table));
}
int
main(int argc, char **argv)
{
int r, op = 'p', method = 'c';
const char *path;
if (argc < 2) {
fprintf(stderr, "usage: kv dbpath [put|get] [col|table|hash|pat|ql] [value_size]\n");
return -1;
}
path = *argv[1] ? argv[1] : NULL;
if (argc > 2) { op = *argv[2]; }
if (argc > 3) { method = *argv[3]; }
if (argc > 4) { value_size = atoi(argv[4]); }
if (argc > 5) { nloops = atoi(argv[5]); }
if (grn_init()) {
fprintf(stderr, "grn_init() failed\n");
return -1;
}
if (grn_ctx_init(&ctx, (method == 'q' ? GRN_CTX_USE_QL|GRN_CTX_BATCH_MODE : 0))) {
fprintf(stderr, "grn_ctx_init() failed\n");
return -1;
}
srand(0);
if (method == 'h' || method == 'p') {
switch (method) {
case 'h' :
r = (op == 'p') ? hash_put(path) : hash_get(path);
break;
case 'p' :
r = (op == 'p') ? pat_put(path) : pat_get(path);
break;
default :
r = -1;
fprintf(stderr, "invalid method\n");
break;
}
} else {
if (path) { db = grn_db_open(&ctx, path); }
if (!db) { db = grn_db_create(&ctx, path, NULL); }
if (!db) {
fprintf(stderr, "db initialize failed\n");
return -1;
}
value_type = grn_type_create(&ctx, "<value_type>", 12, 0, value_size);
switch (method) {
case 'q' :
r = (op == 'p') ? ql_put() : ql_get();
break;
case 'c' :
r = (op == 'p') ? column_put() : column_get();
break;
case 't' :
r = (op == 'p') ? table_put() : table_get();
//r = (op == 'p') ? table_put_allocate() : table_get();
//r = (op == 'p') ? table_put2() : table_get();
break;
default :
r = -1;
fprintf(stderr, "invalid method\n");
break;
}
if (grn_obj_close(&ctx, db)) {
fprintf(stderr, "grn_obj_close() failed\n");
return -1;
}
}
if (grn_ctx_fin(&ctx)) {
fprintf(stderr, "grn_ctx_fin() failed\n");
return -1;
}
if (grn_fin()) {
fprintf(stderr, "grn_fin() failed\n");
return -1;
}
return r;
}
