#include <u.h>
#include <libc.h>
#include <thread.h>

#include <map.h>
#include <bench.h>

void
bench_mapset(B *b)
{
	int i;
	char buf[32];
	Map *m;

	m = mapnew(nil);
	assert(m != nil);

	benchreset(b);

	for(i = 0; i < b->N; i++){
		snprint(buf, sizeof(buf), "%d", i);
		assert(mapset(m, buf, (void*)i) == 0);
	}

	mapfree(m);
}

void
bench_mapget(B *b)
{
	int i, n;
	char buf[32];
	Map *m;

	m = mapnew(nil);
	assert(m != nil);

	for(i = 1; i < 100000; i++){
		snprint(buf, sizeof(buf), "%d", i);
		assert(mapset(m, buf, (void*)i) == 0);
	}

	benchreset(b);
	
	for(i = 0; i < b->N; i++){
		n = i;
		if(n >= 100000)
			n = n % 100000;
		if(n == 0)
			n = 1;
		snprint(buf, sizeof(buf), "%d", n);
		if(mapget(m, buf) == nil){
			fprint(2, "get %d/%d == nil\n", n, b->N);
			mapdump(m, 2);
		}
	}

	mapfree(m);
}

void
bench_mapdelete(B *b)
{
	int i;
	char buf[32];
	Map *m;

	m = mapnew(nil);
	assert(m != nil);

	for(i = 1; i < b->N; i++){
		snprint(buf, sizeof(buf), "%d", i);
		assert(mapset(m, buf, (void*)i) == 0);
	}

	benchreset(b);

	for(i = 1; i < b->N; i++){
		snprint(buf, sizeof(buf), "%d", i);
		mapdelete(m, buf);
	}

	mapfree(m);
}

void
bench_snprintn(B *b)
{
	int i;
	char buf[32];

	for(i = 0; i < b->N; i++){
		snprint(buf, sizeof(buf), "%d", i);
	}
}

void
test_mapset(void)
{
	int i;
	char buf[32];
	void *v;
	vlong v1, v2;
	Map *m;

	m = mapnew(nil);
	assert(m != nil);

	v1 = nsec();

	for(i = 0; i < 10000000; i++){
		snprint(buf, sizeof(buf), "%d", i);
		assert(mapset(m, buf, (void*)(i+1)) == 0);
	}

	v2 = nsec();
	fprint(2, "delta = %llud ms\n", (v2-v1)/1000000LL);

	v1 = nsec();
	for(i = 0; i < 10000000; i++){
		snprint(buf, sizeof(buf), "%d", i);
		v = mapget(m, buf);
		//fprint(2, "mapget('%s') == %#p\n", buf, v);
		assert((void*)(i+1) == v);
	}

	v2 = nsec();
	fprint(2, "delta = %llud ms\n", (v2-v1)/1000000LL);

	//mapdump(m, 2);
}

static void
tfree(void *v)
{
	fprint(2, "free %#p\n", v);
	free(v);
}

static void
test_mapupdate(void)
{
	int rv;
	Map *m;

	m = mapnew(tfree);
	assert(m != nil);

	rv = mapset(m, "foo", malloc(0));
	assert(rv == 0);

	rv = mapset(m, "foo", malloc(0));
	assert(rv == 0);

	mapdump(m, 2);
	mapfree(m);
}

int
atnote(void *, char*)
{
	threadexitsall(nil);
	return NDFLT;
}

void
usage(void)
{
	fprint(2, "usage: %s\n", argv0);
	exits("usage");
}

void
threadmain(int argc, char *argv[])
{
	ARGBEGIN{
	default:
		usage();
	}ARGEND

//#include <pool.h>
//mainmem->flags = POOL_PARANOIA;

	threadnotify(atnote, 1);

	benchinit();
	//BENCHTIME = 10000000000LL;

	//bench("snprint", bench_snprintn);

	//test_mapset();
	test_mapupdate();
	//bench("mapset", bench_mapset);
	//bench("mapget", bench_mapget);
	//bench("mapdelete", bench_mapdelete);

	sleep(90000);
	threadexitsall(nil);
}