#include <windows.h>
#include <sqltypes.h>
#include <sql.h>
#include <sqlext.h>
#include <stdio.h>

#define TEST_ROW_COUNT 100000
#define COMMAND_MAX_SIZE 500

void HandleDiagnosticRecord(SQLHANDLE hHandle, 
							SQLSMALLINT hType,
							RETCODE RetCode)
{
	SQLSMALLINT iRec = 0;
	SQLINTEGER  iError;
	SQLWCHAR       wszMessage[1000];
	SQLWCHAR       wszState[SQL_SQLSTATE_SIZE+1];

	if (RetCode == SQL_INVALID_HANDLE){
		printf("Invalid handle!\n");
		return;
	}

	while ( SQLGetDiagRec(hType,
		hHandle,
		++iRec,
		wszState,
		&iError,
		wszMessage,
		(SQLSMALLINT)(sizeof(wszMessage) / sizeof(SQLWCHAR)),
		(SQLSMALLINT *)NULL ) 
		== SQL_SUCCESS )
	{
		if (wcsncmp(wszState, L"01004", 5))
		{
			wprintf(L"[%5.5s] %s (%d)\n", wszState, wszMessage, iError);
		}
	}
}



int main(int argc, char *argv[])
{
	SQLHSTMT stmt;
	SQLHENV env;
	SQLHDBC dbc;
	RETCODE retcode;
	clock_t time;
	SQLWCHAR command[COMMAND_MAX_SIZE];
	int i;

	// настраиваем всякую ODBC-шную дрянь

	// Allocate an environment handle 
	SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &env);
	// We want ODBC 3 support 
	SQLSetEnvAttr(env, SQL_ATTR_ODBC_VERSION, (void *) SQL_OV_ODBC3, 0);
	// Allocate a connection handle 
	SQLAllocHandle(SQL_HANDLE_DBC, env, &dbc);

	// Connect to the DSN 
	SQLDriverConnect(dbc, GetDesktopWindow(), NULL, SQL_NTS, NULL, 0, NULL, SQL_DRIVER_COMPLETE);

	// Check for success 
	printf("Check sql connection:\n");

	// проверяем подключением

	if(SQL_SUCCESS!=SQLAllocHandle(SQL_HANDLE_STMT, dbc, &stmt))	{
		printf("SQL connection failed!\n");
	}
	else{
		printf("SQL connection Success!\n");

		// дропаем тестовую таблицу

		wsprintf(command, L"DROP TABLE TEST;");
		retcode = SQLExecDirect(stmt,command, SQL_NTS);
		if (retcode != SQL_SUCCESS)
		HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);

		// создаём опять тестовую таблицу

		wsprintf(command, L"CREATE TABLE TEST ( field1 int, field2 int, field3 int );");
		executeDirectCommand(command);
		retcode = SQLExecDirect(stmt,command, SQL_NTS);
		if (retcode != SQL_SUCCESS)
		HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);

		// команда для INSERT-а

		wsprintf(command, L"INSERT INTO dbo.TEST VALUES ( 1, 2, 3 );");
		time = clock();
		for(i=0; i<(uint32_t)TEST_ROW_COUNT; i++){

			// крутим команду 100 000 раз

			retcode = SQLExecDirect(stmt,command, SQL_NTS);
			if (retcode != SQL_SUCCESS)
			HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);
		}
		time = clock() - time;
		printf("Execution time = %4.1f", (float)time/CLOCKS_PER_SEC);
	}
	retcode = SQLFreeStmt(stmt, SQL_DROP);
	if (retcode != SQL_SUCCESS)
		HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);
	retcode = SQLDisconnect(dbc);
	if (retcode != SQL_SUCCESS)
		HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);
	retcode = SQLFreeConnect(dbc);
	if (retcode != SQL_SUCCESS)
		HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);
	retcode = SQLFreeEnv(env);
	if (retcode != SQL_SUCCESS)
		HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, retcode);

	getchar();
}

clock_t test2_BindParameters(void){
	clock_t time;
	SQLWCHAR command[COMMAND_MAX_SIZE];
	uint32_t *field1Rows;
	uint32_t *field2Rows;
	uint32_t *field3Rows;
	SQLUINTEGER paramProcessed;
	SQLUSMALLINT *ret;
	uint32_t i;
	RETCODE rc;

	field1Rows = (uint32_t*)malloc(TEST_ROW_COUNT*sizeof(uint32_t));
	if (field1Rows == NULL){
		printf("Malloc failed!\n");
		return 0;
	}
	field2Rows = (uint32_t*)malloc(TEST_ROW_COUNT*sizeof(uint32_t));
	if (field2Rows == NULL){
		printf("Malloc failed!\n");
		return 0;
	}
	field3Rows = (uint32_t*)malloc(TEST_ROW_COUNT*sizeof(uint32_t));
	if (field3Rows == NULL){
		printf("Malloc failed!\n");
		return 0;
	}
	ret = (SQLUSMALLINT*)malloc(TEST_ROW_COUNT*sizeof(uint32_t));	
	if (ret == NULL){
		printf("Malloc failed!\n");
		return 0;
	}

	for(i=0; i<(uint32_t)TEST_ROW_COUNT; i++){
		field1Rows[i] = i;
		field2Rows[i] = i;
		field3Rows[i] = i;
	}

	time = clock();

	rc = SQLSetStmtAttr(stmt, SQL_ATTR_PARAM_BIND_TYPE, SQL_PARAM_BIND_BY_COLUMN, 0);
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	rc = SQLSetStmtAttr(stmt, SQL_ATTR_PARAMSET_SIZE, (SQLPOINTER)TEST_ROW_COUNT, 0);
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	rc = SQLSetStmtAttr(stmt, SQL_ATTR_PARAMS_PROCESSED_PTR, (SQLPOINTER)&paramProcessed, 0);
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	rc = SQLSetStmtAttr(stmt, SQL_ATTR_PARAM_STATUS_PTR, (SQLPOINTER)ret, 0);
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	rc = SQLBindParameter( stmt,
								   1,
								   SQL_PARAM_INPUT, 
								   SQL_C_SLONG, 
								   SQL_INTEGER,
								   NULL, 
								   0, 
								   field1Rows,
								   NULL, 
								   NULL
								   );
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	rc = SQLBindParameter( stmt,
								   2,
								   SQL_PARAM_INPUT, 
								   SQL_C_SLONG, 
								   SQL_INTEGER,
								   NULL, 
								   0, 
								   field2Rows,
								   NULL, 
								   NULL
								   );
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	rc = SQLBindParameter( stmt,
								   3,
								   SQL_PARAM_INPUT, 
								   SQL_C_SLONG, 
								   SQL_INTEGER,
								   NULL, 
								   0, 
								   field3Rows,
								   NULL, 
								   NULL
								   );
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	
	wsprintf(command, L"INSERT INTO dbo.TEST VALUES (?, ?, ?);");

	//time = clock();

	rc = SQLExecDirect(stmt, command, SQL_NTS);
	if (rc != SQL_SUCCESS) HandleDiagnosticRecord(stmt, SQL_HANDLE_STMT, rc);

	time = clock()-time;

	free(field1Rows);
	free(field2Rows);
	free(field3Rows);
	free(ret);

	return time;
}

wsprintf(command, L"INSERT INTO dbo.TEST VALUES ( 1, 2, 3 );INSERT INTO dbo.TEST VALUES ( 1, 2, 3 );");