// Copyright (c) 2000-2001  David Muse
// See the file COPYING for more information

#include "../../config.h"
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <time.h>

#define FETCH_AT_ONCE		10
#define MAX_ITEM_BUFFER_SIZE	2048
#define MAX_SELECT_LIST_SIZE	256

extern "C" {
	#include <oci.h>
}

struct describe {
	OCIParam	*paramd;
	sb4	dbsize;
	sb2	dbtype;
	text	*buf;
	sb4	buflen;
};
	
/*struct define {
	OCIDefine	*def;
	ub1	buf[MAX_ITEM_BUFFER_SIZE];
	sb2	indp;
	ub2	col_retlen;
	ub2	col_retcode;
};*/

sword		ncols;

describe	desc[MAX_SELECT_LIST_SIZE];
//define		def[MAX_SELECT_LIST_SIZE];

OCIDefine	*def[MAX_SELECT_LIST_SIZE];
ub1		def_buf[MAX_SELECT_LIST_SIZE]
			[FETCH_AT_ONCE][MAX_ITEM_BUFFER_SIZE];
sb2		def_indp[MAX_SELECT_LIST_SIZE][FETCH_AT_ONCE];
ub2		def_col_retlen[MAX_SELECT_LIST_SIZE][FETCH_AT_ONCE];
ub2		def_col_retcode[MAX_SELECT_LIST_SIZE][FETCH_AT_ONCE];

OCIEnv		*env;
OCIServer	*srv;
OCIError	*err;
OCISvcCtx	*svc;
OCISession	*session;
OCITrans	*trans;
OCIStmt		*stmt;
int		fetchatonce=FETCH_AT_ONCE;

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

	if (argc<6) {
		printf("usage: ora8test user password sid query iterations\n");
		exit(0);
	}

	char	*user=argv[1];
	char	*password=argv[2];
	char	*sid=argv[3];
	char	*query=argv[4];
	int	iterations=atoi(argv[5]);

	#if defined(HAVE_PUTENV)
		char	*sidstr=new char[strlen(sid)+12];
		sprintf(sidstr,"ORACLE_SID=%s",sid);
		putenv(sidstr);
		delete[] sidstr;
		sidstr=new char[strlen(sid)+10];
		sprintf(sidstr,"TWO_TASK=%s",sid);
		putenv(sidstr);
		delete[] sidstr;
	#elif defined(HAVE_SETENV)
		setenv("ORACLE_SID",sid,1);
		setenv("TWO_TATK",sid,1);
	#endif

	// init the timer
	time_t	starttime=time(NULL);
	printf("oratest running, please wait...\n");
	clock();

	for (int count=0; count<iterations; count++) {

		// initialize OCI
		OCIInitialize(OCI_DEFAULT,(dvoid *)0, 
			(dvoid*(*)(dvoid *,size_t))0,
			(dvoid*(*)(dvoid *,dvoid *,size_t))0, 
			(void(*)(dvoid *,dvoid *))0);

		// init the environment
		OCIEnvInit((OCIEnv **)&env,OCI_DEFAULT,(size_t)0,(dvoid **)0); 

		// allocate an error handle
		OCIHandleAlloc((dvoid *)env,(dvoid **)&err,OCI_HTYPE_ERROR,
				(size_t)0,(dvoid **)0);
		// allocate a server handle
		OCIHandleAlloc((dvoid *)env,(dvoid **)&srv,OCI_HTYPE_SERVER,
				(size_t)0,(dvoid **)0);
		// allocate a service context handle
		OCIHandleAlloc((dvoid *)env,(dvoid **)&svc,OCI_HTYPE_SVCCTX,
				(size_t)0,(dvoid **)0);



		// attach to the server
		OCIServerAttach(srv,err,(text *)sid,strlen(sid),0);

		// attach the server to the service
		OCIAttrSet((dvoid *)svc,OCI_HTYPE_SVCCTX,
				(dvoid *)srv,(ub4)0,
				OCI_ATTR_SERVER,(OCIError *)err);

		// allocate a session handle
		OCIHandleAlloc((dvoid *)env,
				(dvoid **)&session,(ub4)OCI_HTYPE_SESSION,
				(size_t)0,(dvoid **)0);

		// set username and password
		OCIAttrSet((dvoid *)session,(ub4)OCI_HTYPE_SESSION,
				(dvoid *)user,(ub4)strlen(user),
				(ub4)OCI_ATTR_USERNAME,err);
		OCIAttrSet((dvoid *)session,(ub4)OCI_HTYPE_SESSION,
				(dvoid *)password,(ub4)strlen(password),
				(ub4)OCI_ATTR_PASSWORD,err);

		// start a session
		OCISessionBegin(svc,err,session,
				OCI_CRED_RDBMS,(ub4)OCI_DEFAULT);

		// attach the session to the service
		OCIAttrSet((dvoid *)svc,(ub4)OCI_HTYPE_SVCCTX,
					(dvoid *)session,(ub4)0,
					(ub4)OCI_ATTR_SESSION,err);
	
		// allocate a transaction handle
		OCIHandleAlloc((dvoid *)env,
				(dvoid **)&trans,OCI_HTYPE_TRANS,0,0);

		// attach the transaction to the service
		OCIAttrSet((dvoid *)svc,OCI_HTYPE_SVCCTX,
				(dvoid *)trans,(ub4)0,
				(ub4)OCI_ATTR_TRANS,err);

		// initialize the column count
		ncols=0;

		// allocate a statement handle
		OCIHandleAlloc((dvoid *)env,(dvoid **)&stmt,OCI_HTYPE_STMT,
					(size_t)0,(dvoid **)0);

		// set the number of rows to prefetch
		OCIAttrSet((dvoid *)stmt,OCI_HTYPE_STMT,
				(dvoid *)&fetchatonce,(ub4)0,
				OCI_ATTR_PREFETCH_ROWS,(OCIError *)err);

		// prepare the query
		OCIStmtPrepare(stmt,err,(text *)query,(ub4)strlen(query),
					(ub4)OCI_NTV_SYNTAX,(ub4)OCI_DEFAULT);

		// execute the query
		OCIStmtExecute(svc,stmt,err,0,(ub4)0,NULL,NULL,
						OCI_DEFAULT);


		OCIAttrGet((dvoid *)stmt,OCI_HTYPE_STMT,
				(dvoid *)&ncols,(ub4 *)0,
				OCI_ATTR_PARAM_COUNT,err);

		// run through the columns...
		for (int i=0; i<ncols; i++) {

			// get the entire column definition
			OCIParamGet(stmt,OCI_HTYPE_STMT,err,
				(dvoid **)&desc[i].paramd,i+1);

			// get the column name
			OCIAttrGet((dvoid *)desc[i].paramd,OCI_DTYPE_PARAM,
				(dvoid **)&desc[i].buf,
				(ub4 *)&desc[i].buflen,
				(ub4)OCI_ATTR_NAME,err);

			// get the column type
			OCIAttrGet((dvoid *)desc[i].paramd,OCI_DTYPE_PARAM,
				(dvoid *)&desc[i].dbtype,(ub4 *)0,
				(ub4)OCI_ATTR_DATA_TYPE,err);

			// get the column size
			OCIAttrGet((dvoid *)desc[i].paramd,
				OCI_DTYPE_PARAM,
				(dvoid *)&desc[i].dbsize,(ub4 *)0,
				(ub4)OCI_ATTR_DATA_SIZE,err);

				// translate to a NULL terminated string
			/*OCIDefineByPos(stmt,&def[i].def,err,
				i+1,
				(dvoid *)&def[i].buf,
				(sb4)MAX_ITEM_BUFFER_SIZE,
				SQLT_STR,
				(dvoid *)&def[i].indp,
				&def[i].col_retlen,
				&def[i].col_retcode,
				OCI_DEFAULT);*/
			OCIDefineByPos(stmt,&def[i],err,
				i+1,
				(dvoid *)def_buf[i],
				(sb4)MAX_ITEM_BUFFER_SIZE,
				SQLT_STR,
				(dvoid *)def_indp[i],
				(ub2 *)def_col_retlen[i],
				def_col_retcode[i],
				OCI_DEFAULT);

		}

		// go fetch all rows and columns
		/*while (OCIStmtFetch(stmt,err,1,
				OCI_FETCH_NEXT,OCI_DEFAULT)==OCI_SUCCESS) {
			for (int i=0; i<ncols; i++) {
				printf("\"%s\n,",def_buf[i][j]);
			}
			printf("\n");
		}*/
		ub4	newpos;
		int	oldpos=0;
		for (;;) {
			OCIStmtFetch(stmt,err,fetchatonce,
				OCI_FETCH_NEXT,OCI_DEFAULT);
			OCIAttrGet(stmt,OCI_HTYPE_STMT,
					(dvoid *)&newpos,(ub4 *)0,
					OCI_ATTR_ROW_COUNT,err);
			if (newpos==oldpos) {
				break;
			}
			for (int j=0; j<newpos-oldpos; j++) {
				for (int i=0; i<ncols; i++) {
					printf("\"%s\",",def_buf[i][j]);
				}
				printf("\n");
			}
			oldpos=newpos;
		}

		// free resources
		OCIHandleFree(stmt,OCI_HTYPE_STMT);
		for (int i=0; i<ncols; i++) {
			/*if (def[i].def) {
				OCIHandleFree(def[i].def,OCI_HTYPE_DEFINE);
			}*/
			if (def[i]) {
				OCIHandleFree(def[i],OCI_HTYPE_DEFINE);
			}
		}
	
		// log off
		OCIHandleFree(trans,OCI_HTYPE_TRANS);
		OCISessionEnd(svc,err,session,OCI_DEFAULT);
		OCIHandleFree(session,OCI_HTYPE_SESSION);
		OCIServerDetach(srv,err,OCI_DEFAULT);

		// clean up
		OCIHandleFree(svc,OCI_HTYPE_SVCCTX);
		OCIHandleFree(srv,OCI_HTYPE_SERVER);
		OCIHandleFree(err,OCI_HTYPE_ERROR);
	}

	printf("total system time used: %d\n",clock());
	printf("total real time: %d\n",time(NULL)-starttime);
}