/*
 * Copyright (c) 2010-2012 Postgres-XC Development Group
 */

#include <sys/types.h>
#include <unistd.h>

#include "gtm/libpq-fe.h"
#include "gtm/gtm_c.h"
#include "gtm/gtm_client.h"
#include "gtm/register.h"

#include "test_common.h"

pthread_key_t     threadinfo_key;

void
setUp()
{
	sprintf(connect_string, "host=localhost port=6666 node_name=one remote_type=%d",
		GTM_NODE_GTM);
	
	conn = PQconnectGTM(connect_string);

	_ASSERT( conn!=NULL );

	client_log(("PGconnectGTM() ok.\n"));
}

void
tearDown()
{
	GTMPQfinish(conn);
}

void
test01()
{
	GlobalTransactionId gxid;
	int rc;

	SETUP();

	rc = begin_replication_initial_sync(conn);
	if ( rc>=0 )
		client_log(("begin_replication_initial_sync() ok.\n"));
	else
		client_log(("begin_replication_initial_sync() failed.\n"));

	rc = end_replication_initial_sync(conn);
	if ( rc>=0 )
		client_log(("end_replication_initial_sync() ok.\n"));
	else
		client_log(("end_replication_initial_sync() failed.\n"));

	TEARDOWN();
}

void
test04()
{
	GTM_PGXCNodeInfo *data;
	int rc, i;
	GTM_Transactions txn;
	GlobalTransactionId tmp;
	GTM_Timestamp ts;

	//	data = (GTM_PGXCNodeInfo *)malloc( sizeof(GTM_PGXCNodeInfo)*128 );
	//	memset(data, 0, sizeof(GTM_PGXCNodeInfo)*128);

	SETUP();

	tmp = begin_transaction(conn, GTM_ISOLATION_RC, &ts);

	rc = get_txn_gxid_list(conn, &txn);
	if ( rc>=0 )
		client_log(("get_txn_gxid_list() ok.\n"));
	else
		client_log(("get_txn_gxid_list() failed.\n"));

	abort_transaction(conn, tmp);

	TEARDOWN();
}

void
test05()
{
	GlobalTransactionId next_gxid;
	int rc, i;

	SETUP();

	next_gxid = get_next_gxid(conn);
	if ( next_gxid!=InvalidGlobalTransactionId )
		client_log(("get_next_gxid() ok. - %ld\n", next_gxid));
	else
		client_log(("get_txn_gxid_list() failed.\n"));

	TEARDOWN();
}

void
test08()
{
	GlobalTransactionId next_gxid;
	GTM_SequenceKeyData key1;
	GTM_SequenceKeyData key2;
	GTM_SequenceKeyData key3;
	int rc, i;
	GTM_SeqInfo *seq_list[1024];

	SETUP();

	key1.gsk_keylen = strlen("seq1");
	key1.gsk_key    = strdup("seq1");
	key2.gsk_keylen = strlen("seq2");
	key2.gsk_key    = strdup("seq2");
	key3.gsk_keylen = strlen("seq3");
	key3.gsk_key    = strdup("seq3");

	close_sequence(conn, &key1);
	close_sequence(conn, &key2);
	close_sequence(conn, &key3);

	for (i=0 ; i<1024 ; i++)
		seq_list[i] = NULL;

	rc = get_sequence_list(conn, seq_list, 1024);
	if ( rc>=0 )
	{
		client_log(("get_seq_list() ok. - %d\n", rc));

		for (i=0 ; i<rc ; i++)
		{
			client_log(("key = %s\n", seq_list[i]->gs_key->gsk_key));
		}
	}
	else
		client_log(("get_seq_list() failed.\n"));

	TEARDOWN();
}

int
main(int argc, char *argv[])
{
	test01();
	test04();
	/*
	test05();
	test08();
	*/

	return 0;
}