/** Sets the comparison function. */
void set_compare_func(ham_compare_func_t foo) {
ham_status_t st = ham_db_set_compare_func(m_db, foo);
if (st)
throw error(st);
}
int
main(int argc, char **argv) {
ham_status_t st; /* status variable */
ham_env_t *env; /* hamsterdb environment object */
ham_db_t *db; /* hamsterdb database object */
ham_cursor_t *cursor; /* a database cursor */
char line[1024 * 4]; /* a buffer for reading lines */
ham_key_t key;
ham_record_t record;
memset(&key, 0, sizeof(key));
memset(&record, 0, sizeof(record));
printf("This sample uses hamsterdb to sort data.\n");
printf("Reading from stdin...\n");
/*
* Create a new hamsterdb Environment.
* We could create an In-Memory-Environment to speed up the sorting.
*/
st = ham_env_create(&env, "test.db", 0, 0664, 0);
if (st != HAM_SUCCESS) {
printf("ham_env_create() failed with error %d\n", st);
return (-1);
}
st = ham_env_create_db(env, &db, DATABASE_NAME,
HAM_ENABLE_EXTENDED_KEYS | HAM_ENABLE_DUPLICATES, 0);
if (st != HAM_SUCCESS) {
printf("ham_env_create_db() failed with error %d\n", st);
return (-1);
}
/*
* Since we use strings as our database keys we use our own comparison
* function based on strcmp instead of the default memcmp function.
*/
st = ham_db_set_compare_func(db, my_string_compare);
if (st) {
printf("ham_set_compare_func() failed with error %d\n", st);
return (-1);
}
/*
* Now read each line from stdin and split it in words; then each
* word is inserted into the database
*/
while (fgets(line, sizeof(line), stdin)) {
char *start = line, *p;
/*
* strtok is not the best function because it's not threadsafe
* and not flexible, but it's good enough for this example.
*/
while ((p = strtok(start, " \t\r\n"))) {
key.data = p;
key.size = (ham_size_t)strlen(p) + 1; /* also store the terminating
* 0-byte */
st = ham_db_insert(db, 0, &key, &record, 0);
if (st != HAM_SUCCESS && st!=HAM_DUPLICATE_KEY) {
printf("ham_db_insert() failed with error %d\n", st);
return (-1);
}
printf(".");
start = 0;
}
}
/* create a cursor */
st = ham_cursor_create(&cursor, db, 0, 0);
if (st != HAM_SUCCESS) {
printf("ham_cursor_create() failed with error %d\n", st);
return (-1);
}
/* iterate over all items with HAM_CURSOR_NEXT, and print the words */
while (1) {
st = ham_cursor_move(cursor, &key, &record, HAM_CURSOR_NEXT);
if (st != HAM_SUCCESS) {
/* reached end of the database? */
if (st == HAM_KEY_NOT_FOUND)
break;
else {
printf("ham_cursor_next() failed with error %d\n", st);
return (-1);
}
}
/* print the word */
printf("%s\n", (const char *)key.data);
}
/*
* Then close the handles; the flag HAM_AUTO_CLEANUP will automatically
* close all database and cursors, and we do not need to call
* ham_cursor_close and ham_db_close
*/
st = ham_env_close(env, HAM_AUTO_CLEANUP);
if (st != HAM_SUCCESS) {
printf("ham_env_close() failed with error %d\n", st);
return (-1);
}
/* success! */
return (0);
}
