summaryrefslogtreecommitdiff
path: root/src/gtm/client/test/test_seq.c
blob: 8d45d49297342a58608f7ef465d4248b02088d2e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/*
 * Copyright (c) 2010-2012 Postgres-XC Development Group
 */
#include <sys/types.h>
#include <unistd.h>

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

#define client_log(x)	printf x

int
main(int argc, char *argv[])
{
	int ii;
	pid_t parent_pid;
	GTM_Conn *conn = NULL;
	char connect_string[100];

	//FIXME This statement is wrong
	sprintf(connect_string, "host=%s port=%d node_name=one remote_type=%d", GTM_NODE_COORDINATOR);

	conn = PQconnectGTM(connect_string);
	if (conn == NULL)
	{
		client_log(("Error in connection"));
		exit(1);
	}

	parent_pid = getpid();

	/*
	 * Create sequences
	 */
	for (ii = 0; ii < 20; ii++)
	{
		char buf[100];
		GTM_SequenceKeyData seqkey;
		sprintf(buf, "%d:%d", ii, ii);
		seqkey.gsk_keylen = strlen(buf);
		seqkey.gsk_key = buf;
		if (open_sequence(conn, &seqkey, 10, 1, 10000, 100, false))
		   client_log(("Open seq failed\n"));
		else
			client_log(("Opened Sequence %s\n", seqkey.gsk_key));
	}

	/*
	 * Close the GTM connection
	 */
	GTMPQfinish(conn);

	/*
	 * Start few process which would independently use the sequences
	 */
	for (ii = 0; ii < 3; ii++)
		fork();

	/*
	 * Each process now opens a new connection with the GTM
	 */
	conn = PQconnectGTM(connect_string);

	/*
	 * Try to read/increment the sequence
	 */
	for (ii = 0; ii < 20; ii++)
	{
		char buf[100];
		GTM_SequenceKeyData seqkey;
		GTM_Sequence seqval;
		int jj;

		sprintf(buf, "%d:%d", ii, ii);
		seqkey.gsk_keylen = strlen(buf);
		seqkey.gsk_key = buf;
		if ((seqval = get_current(conn, &seqkey)) == InvalidSequenceValue)
			client_log(("get_current seq failed for sequene %s\n", seqkey.gsk_key));
		else
			client_log(("CURRENT SEQVAL(%s): %lld\n", seqkey.gsk_key, seqval));
		
		for (jj = 0; jj < 5; jj++)
		{
			if ((seqval = get_next(conn, &seqkey)) == InvalidSequenceValue)
				client_log(("get_current seq failed for sequence %s\n", seqkey.gsk_key));
			else
				client_log(("NEXT SEQVAL(%s): %lld ", seqkey.gsk_key, seqval));
		}
		client_log(("\n"));
	}

	/*
	 * The main process now closes the sequences. We want to call close only
	 * once, hence this approach
	 */
	if (getpid() == parent_pid)
	{
		/*
		 * Wait long enough so that all other processes are done
		 */
		sleep(20);
		for (ii = 0; ii < 20; ii++)
		{
			char buf[100];
			GTM_SequenceKeyData seqkey;
			sprintf(buf, "%d:%d", ii, ii);
			seqkey.gsk_keylen = strlen(buf);
			seqkey.gsk_key = buf;
			if (close_sequence(conn, &seqkey))
				client_log(("Close seq failed for sequence %s\n", seqkey.gsk_key));
			else
				client_log(("Sequene closed %s\n", seqkey.gsk_key));
		}
	}
	GTMPQfinish(conn);
	return 0;
}