void createInsertCloseReopenTwiceTest() {
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 < 5; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
reopen();
for (int i = 5; i < 10; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
reopen();
for (int i = 10; i < 15; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
}
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);
}
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));
}
}
int
main(int argc, char **argv)
{
ups_env_t* env;
ups_env_create(&env, "test.db", UPS_ENABLE_TRANSACTIONS, 0664, 0);
ups_parameter_t params[] = {
{UPS_PARAM_KEY_TYPE, UPS_TYPE_UINT32},
{0, }
};
ups_txn_t* txn;
ups_txn_begin(&txn, env, 0, 0, 0);
ups_db_t* db;
ups_env_create_db(env, &db, 1, 0, ¶ms[0]);
for (int i = 0; i < 4; i++)
{
ups_key_t key = ups_make_key(&i, sizeof(i));
ups_record_t record = {0};
ups_db_insert(db, txn, &key, &record, 0);
}
uint64_t size;
ups_db_count(db,0,0,&size);
std::cout << size << std::endl;
ups_txn_commit(txn, 0);
return 0;
}
/** Inserts a key/record pair. */
void insert(txn *t, key *k, record *r, uint32_t flags = 0) {
ups_status_t st = ups_db_insert(m_db,
t ? t->get_handle() : 0,
k ? k->get_handle() : 0,
r ? r->get_handle() : 0, flags);
if (st)
throw error(st);
}
void insertBadKeyTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno;
key.flags = 0;
key.data = &recno;
key.size = sizeof(recno);
REQUIRE(UPS_INV_PARAMETER == ups_db_insert(db, 0, &key, &rec, 0));
key.data = 0;
key.size = 8;
REQUIRE(UPS_INV_PARAMETER == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(UPS_INV_PARAMETER == ups_db_insert(db, 0, 0, &rec, 0));
key.data = 0;
key.size = 0;
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE((RecnoType)1ull == *(RecnoType *)key.data);
}
void overflowTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno = std::numeric_limits<RecnoType>::max();
ldb()->_current_record_number = recno;
recno = 0;
key.flags = UPS_KEY_USER_ALLOC;
key.data = &recno;
key.size = sizeof(recno);
REQUIRE(UPS_LIMITS_REACHED == ups_db_insert(db, 0, &key, &rec, 0));
}
void splitTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno;
key.flags = UPS_KEY_USER_ALLOC;
key.data = &recno;
key.size = sizeof(recno);
for (int i = 0; i < 4096; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
}
void eraseLastReopenTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno;
key.data = &recno;
key.flags = UPS_KEY_USER_ALLOC;
key.size = sizeof(recno);
for (int i = 0; i < 5; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
REQUIRE(0 == ups_db_erase(db, 0, &key, 0));
reopen();
for (int i = 5; i < 10; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE((RecnoType)i == recno);
}
}
void envTest() {
ups_key_t key = {};
ups_record_t rec = {};
RecnoType recno;
key.data = &recno;
key.size = sizeof(recno);
key.flags = UPS_KEY_USER_ALLOC;
teardown();
if (sizeof(RecnoType) == 4)
require_create(m_flags, nullptr, UPS_RECORD_NUMBER32, nullptr);
else
require_create(m_flags, nullptr, UPS_RECORD_NUMBER64, nullptr);
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE((RecnoType)1ull == *(RecnoType *)key.data);
if (!is_in_memory()) {
reopen();
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE((RecnoType)2ull == *(RecnoType *)key.data);
}
}
void createInsertCloseReopenTest() {
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 < 5; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
key.flags = UPS_KEY_USER_ALLOC;
key.data = 0;
REQUIRE(UPS_INV_PARAMETER == ups_db_insert(db, 0, &key, &rec, 0));
key.data = &recno;
key.size = sizeof(RecnoType) == 4 ? 8 : 4;
REQUIRE(UPS_INV_KEY_SIZE == ups_db_insert(db, 0, &key, &rec, 0));
key.size = sizeof(recno);
key.flags = 0;
key.size = 0;
REQUIRE(UPS_INV_PARAMETER == ups_db_insert(db, 0, &key, &rec, 0));
key.size = 8;
key.data = 0;
REQUIRE(UPS_INV_PARAMETER == ups_db_insert(db, 0, &key, &rec, 0));
key.data = &recno;
key.size = sizeof(RecnoType);
key.flags = UPS_KEY_USER_ALLOC;
reopen();
for (int i = 5; i < 10; i++) {
REQUIRE(0 == ups_db_insert(db, 0, &key, &rec, 0));
REQUIRE(recno == (RecnoType)i + 1);
}
}
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);
}
