add tokyotyrant-test

Makefile

@@ -2,7 +2,7 @@ CC = gcc
 CFLAGS = -g -O2 -Wall -I $(HOME)/include -L $(HOME)/lib
 TARGETS = bigmalloc nullcached getsockipmtu echoline cat memcached-benchmark \
 		chunkd-benchmark multimap-memcachedb-test tokyocabinet-test \
-		berkeleydb-test
+		berkeleydb-test tokyotyrant-test
 
 all: $(TARGETS)
 
@@ -38,5 +38,8 @@ tokyocabinet-test: tokyocabinet-test.c
 berkeleydb-test: berkeleydb-test.c
 	$(CC) $(CFLAGS) -o $@ $< -ldb
 
+tokyotyrant-test: tokyotyrant-test.c
+	$(CC) $(CFLAGS) -o $@ $< -ltokyotyrant -ltokyocabinet
+
 clean:
 	-rm -f $(TARGETS)

tokyocabinet-test.c

@@ -171,7 +171,7 @@ static void parse_options(int argc, char **argv)
 {
 	int c;
 
-	while ((c = getopt(argc, argv, "n:l:b:pgcr")) != -1) {
+	while ((c = getopt(argc, argv, "n:l:pgcr")) != -1) {
 		switch (c) {
 		case 'n':
 			count = atol(optarg);

tokyotyrant-test.c

@@ -0,0 +1,166 @@
+#include <tcutil.h>
+#include <tcrdb.h>
+#include <stdlib.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <string.h>
+#include <stdarg.h>
+#include <getopt.h>
+#include <err.h>
+
+static void die(const char *err, ...)
+{
+	va_list params;
+
+	va_start(params, err);
+	verrx(EXIT_FAILURE, err, params);
+	va_end(params);
+}
+
+static TCRDB *open_db(char *host)
+{
+	TCRDB *rdb;
+
+	rdb = tcrdbnew();
+
+	if (!tcrdbopen2(rdb, host)){
+		int ecode = tcrdbecode(rdb);
+		fprintf(stderr, "open error: %s\n", tcrdberrmsg(ecode));
+	}
+	return rdb;
+}
+
+static void close_db(TCRDB *rdb)
+{
+	if (!tcrdbclose(rdb)){
+		int ecode = tcrdbecode(rdb);
+		fprintf(stderr, "close error: %s\n", tcrdberrmsg(ecode));
+	}
+	tcrdbdel(rdb);
+}
+
+static void read_value(char *value)
+{
+}
+
+static void get_db(TCRDB *rdb, unsigned long count)
+{
+	unsigned long i;
+
+	for (i = 0; i < count; i++) {
+		char key[20];
+		char *value;
+
+		sprintf(key, "0x%016lu", i);
+
+		value = tcrdbget2(rdb, key);
+		if (!value) {
+			int ecode = tcrdbecode(rdb);
+			fprintf(stderr, "get error: %s\n", tcrdberrmsg(ecode));
+		}
+		read_value(value);
+		free(value);
+	}
+}
+
+static void cursor_db(TCRDB *rdb, unsigned long count)
+{
+	char *key, *value;
+
+	if (!tcrdbiterinit(rdb)) {
+		int ecode = tcrdbecode(rdb);
+		fprintf(stderr, "iter init error: %s\n", tcrdberrmsg(ecode));
+		return;
+	}
+
+	while ((key = tcrdbiternext2(rdb)) != NULL) {
+		if (strncmp(key, "0x", 2) != 0)
+			continue;
+
+		value = tcrdbget2(rdb, key);
+		read_value(value);
+
+		free(key);
+		free(value);
+	}
+}
+
+void range_db(TCRDB *rdb, unsigned long count)
+{
+	TCLIST *list;
+	char *key, *value;
+	const int batch_count = -1;
+
+	list = tcrdbfwmkeys2(rdb, "0x", batch_count);
+
+	while ((key = tclistshift2(list)) != NULL) {
+		value = tcrdbget2(rdb, key);
+		read_value(value);
+
+		free(key);
+		free(value);
+	}
+	tclistdel(list);
+}
+
+static unsigned long count = 10000000;
+static int value_length = 100;
+static char *host = "localhost:1978";
+
+/*
+ * 'g': get from database
+ * 'c': cursor get from database
+ * 'r': range get from database
+ */
+static int command = 'g';
+
+static void parse_options(int argc, char **argv)
+{
+	int c;
+
+	while ((c = getopt(argc, argv, "n:l:s:gcr")) != -1) {
+		switch (c) {
+		case 'n':
+			count = atol(optarg);
+			break;
+		case 'l':
+			value_length = atoi(optarg);
+			if (value_length <= 0)
+				die("invalid value length");
+			break;
+		case 's':
+			host = optarg;
+			break;
+		case 'g':
+		case 'c':
+		case 'r':
+			command = c;
+			break;
+		default:
+			die("invalid option: %c", c);
+		}
+	}
+}
+
+int main(int argc, char **argv){
+	TCRDB *rdb;
+
+	parse_options(argc, argv);
+
+	rdb = open_db(host);
+
+	switch (command) {
+	case 'g':
+		get_db(rdb, count);
+		break;
+	case 'c':
+		cursor_db(rdb, count);
+		break;
+	case 'r':
+		range_db(rdb, count);
+		break;
+	}
+	close_db(rdb);
+
+	return 0;
+}