void createInsertManyCloseTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno, value = 1;
key.flags = UPS_KEY_USER_ALLOC;
key.data = &recno;
key.size = sizeof(recno);
rec.data = &value;
rec.size = sizeof(value);
for (int i = 0; i < 500; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
key.size = sizeof(RecnoType) == 4 ? 8 : 4;
REQUIRE(UPS_INV_KEY_SIZE == ups_db_find(db, 0, &key, &rec, 0));
key.size = 0;
key.data = &key;
REQUIRE(UPS_INV_KEY_SIZE == ups_db_find(db, 0, &key, &rec, 0));
for (int i = 0; i < 500; i++) {
recno = i + 1;
memset(&key, 0, sizeof(key));
memset(&rec, 0, sizeof(rec));
key.data = &recno;
key.size = sizeof(recno);
REQUIRE(0 == ups_db_find(db, 0, &key, &rec, 0));
}
}
void overwriteCursorTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno, value;
ups_cursor_t *cursor;
REQUIRE(0 == ups_cursor_create(&cursor, db, 0, 0));
key.data = &recno;
key.flags = UPS_KEY_USER_ALLOC;
key.size = sizeof(recno);
REQUIRE(0 == ups_cursor_insert(cursor, &key, &rec, 0));
value = 0x13ull;
memset(&rec, 0, sizeof(rec));
rec.data = &value;
rec.size = sizeof(value);
REQUIRE(0 == ups_cursor_insert(cursor, &key, &rec, UPS_OVERWRITE));
memset(&rec, 0, sizeof(rec));
REQUIRE(0 == ups_db_find(db, 0, &key, &rec, 0));
REQUIRE(value == *(RecnoType *)rec.data);
REQUIRE(0 == ups_cursor_close(cursor));
}
void overwriteTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno, value;
key.data = &recno;
key.flags = UPS_KEY_USER_ALLOC;
key.size = sizeof(recno);
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
value = 0x13ull;
memset(&rec, 0, sizeof(rec));
rec.data = &value;
rec.size = sizeof(value);
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, UPS_OVERWRITE));
key.size = sizeof(RecnoType) == 4 ? 8 : 4;
REQUIRE(UPS_INV_KEY_SIZE ==
ups_db_insert(db, 0, &key, &rec, UPS_OVERWRITE));
key.size = 8;
key.data = 0;
REQUIRE(UPS_INV_PARAMETER ==
ups_db_insert(db, 0, &key, &rec, UPS_OVERWRITE));
key.data = &recno;
key.size = sizeof(RecnoType);
memset(&rec, 0, sizeof(rec));
REQUIRE(0 == ups_db_find(db, 0, &key, &rec, 0));
REQUIRE(value == *(RecnoType *)rec.data);
}
/** Finds a record by looking up the key. */
record &find(txn *t, key *k, record *r, uint32_t flags = 0) {
ups_status_t st = ups_db_find(m_db,
t ? t->get_handle() : 0,
k ? k->get_handle() : 0,
r->get_handle(), flags);
if (st)
throw error(st);
return (*r);
}
/** Finds a record by looking up the key. */
record find(txn *t, key *k, uint32_t flags = 0) {
record r;
ups_status_t st = ups_db_find(m_db,
t ? t->get_handle() : 0,
k ? k->get_handle() : 0,
r.get_handle(), flags);
if (st)
throw error(st);
return (r);
}
int
main(int argc, char **argv) {
int i;
ups_status_t st; /* status variable */
ups_env_t *env; /* upscaledb Environment object */
ups_db_t *db; /* upscaledb Database object */
ups_key_t key = {0}; /* the structure for a key */
ups_record_t record = {0}; /* the structure for a record */
/*
* Connect to the server which should listen at 8080. The server is
* implemented in server1.c.
*/
st = ups_env_create(&env, "ups://localhost:8080/env1.db", 0, 0, 0);
if (st != UPS_SUCCESS)
error("ups_env_create", st);
/* now open a Database in this Environment */
st = ups_env_open_db(env, &db, 13, 0, 0);
if (st != UPS_SUCCESS)
error("ups_env_open_db", st);
/* now we can insert, delete or lookup values in the database */
for (i = 0; i < LOOP; i++) {
key.data = &i;
key.size = sizeof(i);
record.size = key.size;
record.data = key.data;
st = ups_db_insert(db, 0, &key, &record, 0);
if (st != UPS_SUCCESS)
error("ups_db_insert", st);
}
/* now lookup all values */
for (i = 0; i < LOOP; i++) {
key.data = &i;
key.size = sizeof(i);
st = ups_db_find(db, 0, &key, &record, 0);
if (st != UPS_SUCCESS)
error("ups_db_find", st);
/* check if the value is ok */
if (*(int *)record.data != i) {
printf("ups_db_find() ok, but returned bad value\n");
return (-1);
}
}
/* erase everything */
for (i = 0; i < LOOP; i++) {
key.data = &i;
key.size = sizeof(i);
st = ups_db_erase(db, 0, &key, 0);
if (st != UPS_SUCCESS)
error("ups_db_erase", st);
}
/* and make sure that the database is empty */
for (i = 0; i < LOOP; i++) {
key.data = &i;
key.size = sizeof(i);
st = ups_db_find(db, 0, &key, &record, 0);
if (st != UPS_KEY_NOT_FOUND)
error("ups_db_find", st);
}
/* close the database handle */
st = ups_db_close(db, 0);
if (st != UPS_SUCCESS)
error("ups_db_close", st);
/* close the environment handle */
st = ups_env_close(env, 0);
if (st != UPS_SUCCESS)
error("ups_env_close", st);
printf("success!\n");
return (0);
}
