/*
   http://bugs.mysql.com/bug.php?id=70214
   
   gcc bug70214.c -o bug70214 -g `mysql_config --include` `mysql_config --libs_r`
   
   ./bug70214.c

*/

#define TESTTIME (90000)
#define NUMTHREADS (10)
char host[]="127.0.0.1";
int port=3306;
char username[]="root";
char password[]="";
char database[]="test";


#include <unistd.h>
#include <pthread.h>

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

pthread_t pthreads[NUMTHREADS];
unsigned long client_version=0;
unsigned long server_version=0;
unsigned long num_queries=0;
int threaddone=0;

int db_query(MYSQL *dbc,char *sql,int showresults);

void *worker_thread(void *arg)
{
	MYSQL *dbc=NULL;
	my_bool auto_reconnect=1;
	int cancelstate=0;
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE,&cancelstate);

	dbc = mysql_init(NULL);
	if(NULL == dbc)
	{
		printf("mysql_init failed\n");
		goto threadexit;
	}
	else
	{
		if(0!=mysql_options(dbc,MYSQL_OPT_RECONNECT,(char*)&auto_reconnect))
		{
			printf("mysql_options() failed to set MYSQL_OPT_RECONNECT");
		}
		if (!mysql_real_connect(dbc,host,username,password,database,port, NULL, CLIENT_FOUND_ROWS|CLIENT_MULTI_STATEMENTS|CLIENT_MULTI_RESULTS))
		{
			printf("mysql_real_connect failed: %s (%d) (%s)", mysql_error(dbc),mysql_errno(dbc),mysql_sqlstate(dbc));
			mysql_close(dbc);
			dbc=NULL;
		}
	}

		
	char shortquery[1024];
	memset(shortquery,0,1024);
	char *c;
	c=NULL;
	while(0==threaddone && NULL!=dbc)
	{

		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","commit");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","rollback");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","start transaction");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","set @a:=floor(cast(rand()*3 as unsigned));");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","set @b:=floor(cast(rand()*3 as unsigned));");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","set @c:=floor(cast(rand()*3 as unsigned));");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","set @d:=floor(cast(rand()*3 as unsigned));");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","update ignore t1 set a=1,b=1,c=1,d=1 where a > 0 limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","update ignore t2 set a=1,b=1,c=1,d=1 where a > 0 limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","update ignore t3 set a=1,b=1,c=1,d=1 where a > 0 limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","update ignore t4 set a=1,b=1,c=1,d=1 where a > 0 limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","insert into t1(a,b,c,d) values(@a,0,0,0) on duplicate key update a=@a");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","insert into t2(a,b,c,d) values(@a,0,0,0) on duplicate key update a=@a");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","insert into t3(a,b,c,d) values(@a,0,0,0) on duplicate key update a=@a");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","insert into t4(a,b,c,d) values(@a,0,0,0) on duplicate key update a=@a");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","delete from t1 where a <> @a limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","delete from t2 where a <> @a limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%5==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","delete from t3 where a <> @a limit 1");
				db_query(dbc,shortquery,0);

		}


		if(lrand48()%33==0)
		{
				c=shortquery;
				c+=sprintf(c,"%s","delete from t4 where a <> @a limit 1");
				db_query(dbc,shortquery,0);

		}


	}
threadexit:
	mysql_close(dbc);
	mysql_thread_end();
	pthread_exit(0);
}


int main(int argc, const char *argv[])
{
	MYSQL *dbc=NULL;
	int i=0,err=0;

	srand48((unsigned long)1404249423);
	time_t timestart=0,timenow=0;

	char shortquery[1024]={0};
	char *c=NULL;
	my_init();
	if (!(dbc = mysql_init(NULL)))
	{
		printf("mysql_init failed\n");
		dbc=NULL;
		goto threadexit;
	}
	else
	{
		if (!mysql_real_connect(dbc,host,username,password,database,port, NULL, CLIENT_FOUND_ROWS|CLIENT_MULTI_STATEMENTS|CLIENT_MULTI_RESULTS))
		{
			printf("mysql_real_connect failed: %s (%d) (%s)", mysql_error(dbc),mysql_errno(dbc),mysql_sqlstate(dbc));
			mysql_close(dbc);
			dbc=NULL;
			goto threadexit;
		}
	}

	printf("running initializations..\n");
	client_version=mysql_get_client_version();
	server_version=mysql_get_server_version(dbc);
	printf("client version=%lu\n",client_version);
	printf("server version=%lu\n",server_version);
	if((client_version/10000) < (server_version/10000))
	{
		printf("different client and server version!  please upgrade client library!\n");
		//goto threadexit;
	}

	if (!mysql_thread_safe())
	{
		printf("non-threadsafe client detected!  please rebuild and link with libmysql_r!\n");
	}

				c=shortquery;
				c+=sprintf(c,"%s","set global innodb_file_format=barracuda");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","set global innodb_file_per_table=1");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","set global innodb_flush_log_at_trx_commit=0");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","set global innodb_stats_persistent=0;");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","set global innodb_stats_auto_recalc=0");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","set global innodb_stats_on_metadata=0");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","drop table if exists t1");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","drop table if exists t2");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","drop table if exists t3");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","drop table if exists t4");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","create table t1(a tinyint primary key,b tinyint,c tinyint,d tinyint,key(a),key(b),key(c),key(a,b,c),key(a,b),key(b,c),key(c,a),key(a,c),key(c,b),key(c,a,b),key(d,a,b),key(b,a,d),key(d,a))engine=innodb default charset latin1 row_format=dynamic partition by key(a) partitions 3");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","create table t2(a tinyint primary key,b tinyint,c tinyint,d tinyint,key(a),key(b),key(c),key(a,b,c),key(a,b),key(b,c),key(c,a),key(a,c),key(c,b),key(c,a,b),key(d,a,b),key(b,a,d),key(d,a))engine=innodb default charset latin1 row_format=compact  partition by key(a) partitions 3");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","create table t3(a tinyint primary key,b tinyint,c tinyint,d tinyint,key(a),key(b),key(c),key(a,b,c),key(a,b),key(b,c),key(c,a),key(a,c),key(c,b),key(c,a,b),key(d,a,b),key(b,a,d),key(d,a))engine=innodb default charset latin1 row_format=compressed  partition by key(a) partitions 3");
				db_query(dbc,shortquery,0);

				c=shortquery;
				c+=sprintf(c,"%s","create table t4(a tinyint primary key,b tinyint,c tinyint,d tinyint,key(a),key(b),key(c),key(a,b,c),key(a,b),key(b,c),key(c,a),key(a,c),key(c,b),key(c,a,b),key(d,a,b),key(b,a,d),key(d,a))engine=innodb default charset latin1 row_format=compressed  partition by hash(a) partitions 3");
				db_query(dbc,shortquery,0);

	mysql_close(dbc);

	printf("about to spawn %d threads\n",NUMTHREADS);
	for (i=0;i<NUMTHREADS;i++)
	{
		err=pthread_create(&pthreads[i], NULL, worker_thread, (void *)i);
		if(err!=0)
		{
			printf("error spawning thread %lu, pthread_create returned %lu\n",(unsigned long)i,(unsigned long)err);
		}
		printf(".");
	}
	printf("\n");
	printf("completed spawning new database worker threads\n");

	printf("testcase is now running, so watch for error output\n");

	timestart=time(NULL);
	timenow=time(NULL);
	for(i=0;(timenow-timestart) < TESTTIME;timenow=time(NULL))
	{
		usleep(1000*1000);
		printf("queries: %09lu\n",num_queries);
	}
	threaddone=1;

	printf("waiting for worker threads to finish...\n");

	for (i=0;i<NUMTHREADS;i++)
	{
		pthread_join(pthreads[i], NULL);
	}

	exit(0);
threadexit:
	exit(-1);
}


int db_query(MYSQL *dbc,char *sql,int showresults)
{
	int res=0;
	MYSQL_RES *r=NULL;
	MYSQL_ROW w;
	MYSQL_FIELD *field=NULL;
	int moreresult=0;
	unsigned int i=0;
	unsigned int myerrno=0;
	if(NULL == dbc) return 0;

	res = mysql_query(dbc,sql);
	if(res != 0 && showresults > 0)
	{
		myerrno=mysql_errno(dbc);
		printf("query failed '%s' : %d (%s) (%s)\n",sql,myerrno,mysql_error(dbc),mysql_sqlstate(dbc));
		return 0;
	}

	num_queries++;
	do
	{
		r = mysql_use_result(dbc);
		if(r)
		{
			unsigned int numfields = mysql_num_fields(r);
			//unsigned int numrows=mysql_num_rows(r);
			while(0!=(field = mysql_fetch_field(r)))
			{
					//print metadata information about each field
					if(showresults > 1)
					{
						printf("%s	",field->name);
					}
			}
			if(showresults > 1)
			{
				printf("\n------------------------------------\n");
			}

			while (0!=(w = mysql_fetch_row(r)))
			{
				for(i = 0; i < numfields; i++)
				{
					//print each field here
					if(showresults > 1)
					{
						printf("%s\t",w[i]);
					}
				}

				if(showresults > 1)
				{
						printf("\n");
				}
			}
			if(showresults > 1)
			{
				printf("\n");
			}
			mysql_free_result(r);
		}
		else //no rows returned. was it a select?
		{
			if(mysql_field_count(dbc) > 0 && showresults > 0)
			{
				printf("No results for '%s'.  (%d) - %s (%s)\n",sql,mysql_errno(dbc),mysql_error(dbc),mysql_sqlstate(dbc));
				return 0;
			}
			else //it could have been some insert/update/delete
			{
				//this is successful query
			}
		}
		moreresult=mysql_next_result(dbc);
		if(moreresult > 0 && showresults > 0)
		{
			printf("mysql_next_result returned %d, mysql error %s, (%d) (%s)\n",moreresult,mysql_error(dbc),mysql_errno(dbc),mysql_sqlstate(dbc));
			return 0;
		}
	} while (0==moreresult);
	return 1;
}