benchmark.cpp

#include<bits/stdc++.h>
#include <unistd.h> 
#include <pthread.h>
#include <time.h>
#include "kvStore.cpp"

using namespace std;

string sliceToStr(Slice& a) {
    string ret = "";

    for (int i = 0; i < a.size; i++)
        ret += a.data[i];

    return ret;
}

void strToSlice(string l, Slice& a) {
    a.size = l.length();
    a.data = (char*)malloc(a.size);
    strncpy(a.data, l.c_str(), a.size);
}

string random_key(int stringLength){
	string k = "";
	string letters = "";
	for(char i = 'a';i<='z';i++)letters+=i;
	for(char i = 'A';i<='Z';i++)letters+=i;
	for(int i=0;i<stringLength-1;i++)
		k = k + letters[(rand()%52+rand()%10)%52];

	k = k + '\n';

	return k;
}

string random_value(int stringLength){
	string v = "";
	string letters = "";
	for(int i = 32;i<127;i++)letters+=char(i);

	for(int i=0;i<stringLength-1;i++)
		v = v + letters[rand()%95];
	v = v + '\n';

	return v;
}

kvStore kv(10000000);
map<string,string> db;
long long db_size = 0;

void *myThreadFun(void *vargp) 
{ 
	long long int transactions=0;
	time_t start,end;
	int limit = 10;
	time(&start);
	time(&end);
	while((end-start) < limit)
	{
		transactions+=1000;
		for(int i=0;i<10000;i++)
		{
			int x = rand()%5;
			if(x==0)
			{
				string key = random_key(rand()%64 + 1);
				Slice s_key,s_value;
				strToSlice(key,s_key);
				bool ans = kv.get(s_key,s_value);
			}
			else if(x==1)
			{
				string key = random_key(rand()%64 + 1);
				string value = random_value(rand()%255 + 1);
				Slice s_key,s_value,temp;
				strToSlice(key,s_key);
				strToSlice(value,s_value);

				bool check = kv.get(s_key,temp);
				bool ans = kv.put(s_key,s_value);

				if(check == false)
					db_size++;
			}
			else if(x==2)
			{
				int temp=db_size;
				if (temp == 0)
					continue;		
				int rem = rand()%temp;
				Slice s_key,s_value;
				bool check = kv.get(rem,s_key,s_value);
				check = kv.del(s_key);
				db_size--;
			}
			else if(x==3)
			{
				int temp=db_size;
				if (temp == 0)
					continue;
				int rem = rand()%temp;
				Slice s_key,s_value;
				bool check = kv.get(rem,s_key,s_value);
			}
			else if(x==4)
			{
				int temp=db_size;
				if (temp == 0)
					continue;
				int rem = rand()%temp;
				bool check = kv.del(rem);
				db_size--;
			}
		}
		time(&end);
	}
	cout<<"TRANSACTIONS PER SEC = "<<transactions/limit<<endl;
	return NULL;  
} 

int main()
{
	int put_enteries = 100000;
	time_t start_p,end_p;
	float put_time = 0;
	vector<Slice> keys;
	vector<Slice> values;
	for(int i=0;i<put_enteries;i++)
	{
		string key = random_key(rand()%63 + 1);
		string value = random_value(rand()%254 + 1);
		db[key] = value;
		Slice k,v;
		strToSlice(key,k);
		strToSlice(value,v);
		keys.push_back(k);
		values.push_back(v);
		db_size = db.size();
	}

	time(&start_p);
	for(int i=0;i<put_enteries;i++)
		kv.put(keys[i],values[i]);
	time(&end_p);

	cout<<"TIME FOR PUTTING "<<put_enteries<<" ENTERIES = "<<setprecision(9)<<(end_p-start_p)<<" SEC "<<endl;

	bool incorrect = false;

	for(int i=0;i<10000;i++)
	{
		int x = rand()%5;
		if(x==0)
		{
			string key = random_key(rand()%64 + 1);
			Slice s_key,s_value;
			strToSlice(key,s_key);
			bool ans = kv.get(s_key,s_value);
			map<string,string>:: iterator itr = db.find(key);
			if((ans==false && itr != db.end()) || (ans==true && itr->second != sliceToStr(s_value) ))
				incorrect = true;
		}
		else if(x==1)
		{
			int k = rand()%64 + 1;
			int v = rand()%255 + 1;
			string key = random_key(k);
			string value = random_value(v);
			db[key] = value;
			Slice s_key,s_value;
			strToSlice(key,s_key);
			strToSlice(value,s_value);
			bool ans = kv.put(s_key,s_value);

			Slice check;
			bool check2 = kv.get(s_key,check);
			db_size = db.size();
			if(check2 == false || value != sliceToStr(check))
				incorrect = true;
		}
		else if(x==2)
		{
			int rem = rand()%db_size;
			map<string,string>:: iterator itr = db.begin();
			advance(itr,rem);
			string key = itr->first;
			Slice s_key,s_value;
			strToSlice(key,s_key);
			bool check = kv.del(s_key);
			db.erase(itr);
			db_size--;

			bool check2 = kv.get(s_key,s_value);
			if(check2 == true)
				incorrect = true;
		}
		else if(x==3)
		{
			int rem = rand()%db_size;
			Slice s_key,s_value;
			bool check = kv.get(rem,s_key,s_value);
			map<string,string>:: iterator itr = db.begin();
			for(int i=0;i<rem;i++)itr++;
			if( itr->first != sliceToStr(s_key) || itr->second != sliceToStr(s_value))
				incorrect = true;
		}
		else if(x==4)
		{
			int rem = rand()%db_size;
			map<string,string>:: iterator itr = db.begin();
			for(int i=0;i<rem;i++)itr++;
			string key = itr->first;
			bool check = kv.del(rem);
			db.erase(itr);
			db_size--;
			Slice s_key,s_value;
			strToSlice(key,s_key);
			bool check2 = kv.get(s_key,s_value);
			if(check2 == true)
				incorrect = true;
		}
	}
	if(incorrect == true)
	{
		cout<<"INCORRECT OUTPUT"<<endl;
		return 0;
	}
	else
		cout<<"CORRECT OUTPUT"<<endl;
	int threads = 4;

	pthread_t tid[threads];
	for (int i = 0; i < threads; i++) 
	{
		tid[i] = i;
        pthread_create(&tid[i], NULL, myThreadFun, (void *)&tid[i]); 
	}
	for(int i=0;i<threads;i++)
		pthread_join(tid[i],NULL);
	return 0;
}