#include <my_config.h>
#include <mysql.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

int main(int argc, char** argv)
{

	MYSQL* server;
	char *host;
	unsigned int port;
	int rval, iterations,i;
	clock_t begin,end;
	double baseline,test, ratio, result, minimum;

	if(argc < 7){
		fprintf(stderr,"Usage: %s <iterations> <baseline host> <baseline port> <test host> <test port> <max result ratio>\n",argv[0]);
		fprintf(stderr,"The ratio is measured as:\ntest CPU time / baseline CPU time\n");
		fprintf(stderr,"The test fails if this ratio is exceeded.\n");
		return 1;
	}

	iterations = atoi(argv[1]);
	host = strdup(argv[2]);
	port = atoi(argv[3]);
	ratio = atof(argv[6]);
	rval = 0;

	if(ratio <= 0.0){
		return 1;
	}
	
	/**Testing direct connection to master*/

	printf("Connecting to MySQL server through %s:%d.\n",host,port);
	begin = clock();

	for(i = 0;i<iterations;i++)
		{
			if((server = mysql_init(NULL)) == NULL){
				return 1;
			}
		    if(mysql_real_connect(server,host,"maxuser","maxpwd",NULL,port,NULL,0) == NULL){
				fprintf(stderr, "Failed to connect to database: Error: %s\n",
						mysql_error(server));
				rval = 1;
				break;
			}
			mysql_close(server);			
		}

	end = clock();
	baseline = (double)(end - begin)/CLOCKS_PER_SEC;

	free(host);
	host = strdup(argv[4]);
	port = atoi(argv[5]);

	/**Testing connection to master through MaxScale*/

	printf("Connecting to MySQL server through %s:%d.\n",host,port);
	begin = clock();

	for(i = 0;i<iterations;i++)
		{
		    if((server = mysql_init(NULL)) == NULL){
				return 1;
			}
		    if(mysql_real_connect(server,host,"maxuser","maxpwd",NULL,port,NULL,0) == NULL){
				rval = 1;
				fprintf(stderr, "Failed to connect to database: Error: %s\n",
						mysql_error(server));
				break;
			}
			mysql_close(server);			
		}

	end = clock();
	test = (double)(end - begin)/CLOCKS_PER_SEC;

	printf("CPU time used in seconds:\nDirect connection: %f\nThrough MaxScale: %f\n",baseline,test);

	result = test / baseline;
	if(rval){	
		printf("Test failed: Errors during test run.");
	}else if(result  > ratio){
		printf("Test failed: CPU time ratio was %f which exceeded the limit of %f.\n", result, ratio);
		rval = 1;
	}else{
		printf("Test passed: CPU time ratio was %f.\n",result);
	}

	free(host);
	return rval;
}