int
pat_put(const char *path)
{
int i;
grn_obj buf;
grn_pat *pat = grn_pat_create(&ctx, path, key_size, value_size,
GRN_OBJ_PERSISTENT|GRN_OBJ_KEY_VAR_SIZE);
if (!pat) { return -1; }
GRN_TEXT_INIT(&buf, 0);
for (i = 0; i < nloops; i++) {
int key = GENKEY(i);
GRN_BULK_REWIND(&buf);
grn_text_itoh(&ctx, &buf, key, key_size);
{
void *value;
grn_id rid = grn_pat_add(&ctx, pat,
GRN_BULK_HEAD(&buf), key_size, &value, NULL);
if (!rid) {
fprintf(stderr, "table_lookup failed");
} else {
memcpy(value, GRN_BULK_HEAD(&buf), value_size);
}
}
}
grn_obj_close(&ctx, &buf);
grn_pat_close(&ctx, pat);
return 0;
}
int
pat_get(const char *path)
{
int i;
grn_obj buf;
grn_pat *pat = grn_pat_open(&ctx, path);
if (!pat) { return -1; }
GRN_TEXT_INIT(&buf, 0);
for (i = 0; i < nloops; i++) {
int key = GENKEY(i);
GRN_BULK_REWIND(&buf);
grn_text_itoh(&ctx, &buf, key, key_size);
{
void *value;
grn_id rid = grn_pat_get(&ctx, pat, GRN_BULK_HEAD(&buf), key_size, &value);
if (!rid) {
fprintf(stderr, "table_lookup failed");
} else {
if (memcmp(value, GRN_BULK_HEAD(&buf), value_size)) {
fprintf(stderr, "value unmatch %d: %d != %d\n", i, *((int *)value), key);
}
}
}
}
grn_obj_close(&ctx, &buf);
grn_pat_close(&ctx, pat);
return 0;
}
int
table_get(void)
{
int i;
grn_obj buf;
grn_obj *table = grn_ctx_get(&ctx, "<t1>", 4);
if (!table) { return -1; }
GRN_TEXT_INIT(&buf, 0);
for (i = 0; i < nloops; i++) {
int key = GENKEY(i);
GRN_BULK_REWIND(&buf);
grn_text_itoh(&ctx, &buf, key, key_size);
{
grn_id rid = grn_table_get(&ctx, table, GRN_BULK_HEAD(&buf), key_size);
if (!rid) {
fprintf(stderr, "table_lookup failed");
} else {
grn_obj obj, *p;
GRN_TEXT_INIT(&obj, 0);
p = grn_obj_get_value(&ctx, table, rid, &obj);
if (!p) {
fprintf(stderr, "grn_obj_get_value failed\n");
} else {
if (memcmp(GRN_BULK_HEAD(p), GRN_BULK_HEAD(&buf), value_size)) {
fprintf(stderr, "value unmatch\n");
}
grn_obj_close(&ctx, p);
}
}
}
}
grn_obj_close(&ctx, &buf);
return 0;
}
int
table_put_allocate(void)
{
int i;
grn_obj *buf;
grn_obj *key_type = grn_ctx_at(&ctx, GRN_DB_SHORT_TEXT);
grn_obj *table = grn_table_create(&ctx, "<t1>", 4, NULL,
GRN_OBJ_TABLE_HASH_KEY|GRN_OBJ_PERSISTENT,
key_type, value_type);
if (!table) { return -1; }
for (i = 0; i < nloops; i++) {
int key = GENKEY(i);
buf = grn_obj_open(&ctx, GRN_BULK, 0, 0);
grn_text_itoh(&ctx, buf, key, key_size);
{
grn_id rid = grn_table_add(&ctx, table, GRN_BULK_HEAD(buf), key_size, NULL);
if (!rid) {
fprintf(stderr, "table_lookup failed");
} else {
grn_obj *value_buf;
value_buf = grn_obj_open(&ctx, GRN_BULK, 0, 0);
grn_text_itoh(&ctx, value_buf, key, key_size);
if (grn_obj_set_value(&ctx, table, rid, value_buf, GRN_OBJ_SET)) {
fprintf(stderr, "grn_obj_set_value failed");
}
grn_obj_close(&ctx, value_buf);
}
}
grn_obj_close(&ctx, buf);
}
return 0;
}
int
ql_get(void)
{
int i, key;
grn_obj buf;
GRN_TEXT_INIT(&buf, 0);
for (i = 0; i < nloops; i++) {
key = GENKEY(i);
GRN_BULK_REWIND(&buf);
GRN_TEXT_PUTS(&ctx, &buf, "(<t1> : \"");
grn_text_itoh(&ctx, &buf, key, key_size);
GRN_TEXT_PUTS(&ctx, &buf, "\").c1");
GRN_TEXT_PUTC(&ctx, &buf, '\0');
EVAL(&ctx, GRN_BULK_HEAD(&buf));
}
grn_obj_close(&ctx, &buf);
return 0;
}
int
column_get(void)
{
int i, s = 0;
grn_obj buf;
grn_obj *table = grn_ctx_get(&ctx, "<t1>", 4);
grn_obj *column = grn_ctx_get(&ctx, "<t1>.c1", 7);
if (!table || !column) { return -1; }
GRN_TEXT_INIT(&buf, 0);
for (i = 0; i < nloops; i++) {
int key = GENKEY(i);
GRN_BULK_REWIND(&buf);
grn_text_itoh(&ctx, &buf, key, key_size);
{
grn_id rid = grn_table_get(&ctx, table, GRN_BULK_HEAD(&buf), key_size);
if (!rid) {
fprintf(stderr, "table_lookup failed\n");
} else {
grn_obj obj, *p;
unsigned int v = key % value_size;
GRN_TEXT_INIT(&obj, 0);
p = grn_obj_get_value(&ctx, column, rid, &obj);
if (!p) {
fprintf(stderr, "grn_obj_get_value failed\n");
} else {
if (GRN_BULK_VSIZE(p) != v) {
fprintf(stderr, "value_size unmatch %d (%ld:%u)\n", i, GRN_BULK_VSIZE(p), v);
} else {
if (v && GRN_BULK_HEAD(p)[v-1] != GRN_BULK_HEAD(&buf)[0]) {
fprintf(stderr, "value unmatch\n");
} else {
s++;
}
}
grn_obj_close(&ctx, p);
}
}
}
}
grn_obj_close(&ctx, &buf);
if (i != s) { printf("successed: %d\n", s); }
return 0;
}
static void test_write(LSMDB_env *const env) {
uint8_t k[KEY_SIZE];
uint8_t d[DATA_SIZE] = {};
for(int i = 0; i < WRITES / TXN_SIZE; ++i) {
LSMDB_txn *txn;
chk( lsmdb_txn_begin(env, NULL, MDB_RDWR, &txn) );
for(int j = 0; j < TXN_SIZE; ++j) {
GENKEY(k);
MDB_val key = { sizeof(k), &k };
MDB_val data = { sizeof(d), &d };
chk( lsmdb_put(txn, &key, &data, PUT_FLAGS) );
}
chk( lsmdb_autocompact(txn) );
lsmdb_txn_commit(txn);
}
}
int
ql_put(void)
{
int i, key;
grn_obj buf;
GRN_TEXT_INIT(&buf, 0);
EVAL(&ctx, "(ptable '<t1>)");
EVAL(&ctx, "(<t1> ::def :c1 <text>)");
EVAL(&ctx, "(<t1> ::load :c1)");
for (i = 0; i < nloops; i++) {
key = GENKEY(i);
GRN_BULK_REWIND(&buf);
grn_text_itoh(&ctx, &buf, key, key_size);
GRN_TEXT_PUTC(&ctx, &buf, '\t');
grn_text_itoh(&ctx, &buf, key, value_size);
GRN_TEXT_PUTC(&ctx, &buf, '\0');
EVAL(&ctx, GRN_BULK_HEAD(&buf));
}
grn_obj_close(&ctx, &buf);
return 0;
}
int
column_put(void)
{
int i, s = 0;
grn_obj buf;
grn_obj *key_type = grn_ctx_at(&ctx, GRN_DB_SHORT_TEXT);
grn_obj *table = grn_table_create(&ctx, "<t1>", 4, NULL,
GRN_OBJ_TABLE_HASH_KEY|GRN_OBJ_PERSISTENT,
key_type, NULL);
grn_obj *value_type = grn_ctx_at(&ctx, GRN_DB_TEXT);
grn_obj *column = grn_column_create(&ctx, table, "c1", 2, NULL,
GRN_OBJ_PERSISTENT, value_type);
if (!table || !column) { return -1; }
GRN_TEXT_INIT(&buf, 0);
for (i = 0; i < nloops; i++) {
int key = GENKEY(i);
GRN_BULK_REWIND(&buf);
grn_text_itoh(&ctx, &buf, key, key_size);
{
grn_id rid = grn_table_add(&ctx, table, GRN_BULK_HEAD(&buf), key_size, NULL);
if (!rid) {
fprintf(stderr, "table_lookup failed");
} else {
unsigned int v = key % value_size;
GRN_BULK_REWIND(&buf);
if (v) {
grn_bulk_space(&ctx, &buf, v -1);
GRN_TEXT_PUTC(&ctx, &buf, GRN_BULK_HEAD(&buf)[0]);
s += v;
}
if (grn_obj_set_value(&ctx, column, rid, &buf, GRN_OBJ_SET)) {
fprintf(stderr, "grn_obj_set_value failed");
}
}
}
}
grn_obj_close(&ctx, &buf);
printf("total size: %d\n", s);
return 0;
}
