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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
|
/*-------------------------------------------------------------------------
*
* pgxcnode.h
*
* Utility functions to communicate to Datanodes and Coordinators
*
*
* Portions Copyright (c) 2012-2014, TransLattice, Inc.
* Portions Copyright (c) 1996-2011, PostgreSQL Global Development Group ?
* Portions Copyright (c) 2010-2012 Postgres-XC Development Group
*
* src/include/pgxc/pgxcnode.h
*
*-------------------------------------------------------------------------
*/
#ifndef PGXCNODE_H
#define PGXCNODE_H
#include "postgres.h"
#include "gtm/gtm_c.h"
#include "utils/timestamp.h"
#include "nodes/pg_list.h"
#include "utils/snapshot.h"
#include <unistd.h>
#define NO_SOCKET -1
/* Helper structure to access Datanode from Session */
typedef enum
{
DN_CONNECTION_STATE_IDLE, /* idle, ready for query */
DN_CONNECTION_STATE_QUERY, /* query is sent, response expected */
DN_CONNECTION_STATE_CLOSE, /* close is sent, confirmation expected */
DN_CONNECTION_STATE_ERROR_FATAL, /* fatal error */
DN_CONNECTION_STATE_COPY_IN,
DN_CONNECTION_STATE_COPY_OUT
} DNConnectionState;
typedef enum
{
HANDLE_IDLE,
HANDLE_ERROR,
HANDLE_DEFAULT
} PGXCNode_HandleRequested;
#define DN_CONNECTION_STATE_ERROR(dnconn) \
((dnconn)->state == DN_CONNECTION_STATE_ERROR_FATAL \
|| (dnconn)->transaction_status == 'E')
#define HAS_MESSAGE_BUFFERED(conn) \
((conn)->inCursor + 4 < (conn)->inEnd \
&& (conn)->inCursor + ntohl(*((uint32_t *) ((conn)->inBuffer + (conn)->inCursor + 1))) < (conn)->inEnd)
struct pgxc_node_handle
{
Oid nodeoid;
int nodeid;
char nodename[NAMEDATALEN];
char nodehost[NAMEDATALEN];
int nodeport;
/* fd of the connection */
int sock;
/* pid of the remote backend process */
int backend_pid;
/* Connection state */
char transaction_status;
DNConnectionState state;
bool read_only;
struct ResponseCombiner *combiner;
#ifdef DN_CONNECTION_DEBUG
bool have_row_desc;
#endif
char *error;
/* Output buffer */
char *outBuffer;
size_t outSize;
size_t outEnd;
/* Input buffer */
char *inBuffer;
size_t inSize;
size_t inStart;
size_t inEnd;
size_t inCursor;
/*
* Have a variable to enable/disable response checking and
* if enable then read the result of response checking
*
* For details see comments of RESP_ROLLBACK
*/
bool ck_resp_rollback;
bool in_extended_query;
bool needSync;
};
typedef struct pgxc_node_handle PGXCNodeHandle;
/* Structure used to get all the handles involved in a transaction */
typedef struct
{
PGXCNodeHandle *primary_handle; /* Primary connection to PGXC node */
int dn_conn_count; /* number of Datanode Handles including primary handle */
PGXCNodeHandle **datanode_handles; /* an array of Datanode handles */
int co_conn_count; /* number of Coordinator handles */
PGXCNodeHandle **coord_handles; /* an array of Coordinator handles */
} PGXCNodeAllHandles;
extern void InitMultinodeExecutor(bool is_force);
/* Open/close connection routines (invoked from Pool Manager) */
extern void PGXCNodeCleanAndRelease(int code, Datum arg);
extern PGXCNodeHandle *get_any_handle(List *datanodelist);
/* Look at information cached in node handles */
extern int PGXCNodeGetNodeId(Oid nodeoid, char *node_type);
extern int PGXCNodeGetNodeIdFromName(char *node_name, char *node_type);
extern Oid PGXCNodeGetNodeOid(int nodeid, char node_type);
extern PGXCNodeAllHandles *get_handles(List *datanodelist, List *coordlist, bool is_query_coord_only, bool is_global_session);
extern PGXCNodeAllHandles *get_current_handles(void);
extern void pfree_pgxc_all_handles(PGXCNodeAllHandles *handles);
extern void release_handles(void);
extern int get_transaction_nodes(PGXCNodeHandle ** connections,
char client_conn_type,
PGXCNode_HandleRequested type_requested);
extern char* collect_pgxcnode_names(char *nodestring, int conn_count, PGXCNodeHandle ** connections, char client_conn_type);
extern char* collect_localnode_name(char *nodestring);
extern int get_active_nodes(PGXCNodeHandle ** connections);
extern int ensure_in_buffer_capacity(size_t bytes_needed, PGXCNodeHandle * handle);
extern int ensure_out_buffer_capacity(size_t bytes_needed, PGXCNodeHandle * handle);
extern int pgxc_node_send_query(PGXCNodeHandle * handle, const char *query);
extern int pgxc_node_send_rollback(PGXCNodeHandle * handle, const char *query);
extern int pgxc_node_send_describe(PGXCNodeHandle * handle, bool is_statement,
const char *name);
extern int pgxc_node_send_execute(PGXCNodeHandle * handle, const char *portal, int fetch);
extern int pgxc_node_send_close(PGXCNodeHandle * handle, bool is_statement,
const char *name);
extern int pgxc_node_send_sync(PGXCNodeHandle * handle);
extern int pgxc_node_send_bind(PGXCNodeHandle * handle, const char *portal,
const char *statement, int paramlen, char *params);
extern int pgxc_node_send_parse(PGXCNodeHandle * handle, const char* statement,
const char *query, short num_params, Oid *param_types);
extern int pgxc_node_send_flush(PGXCNodeHandle * handle);
extern int pgxc_node_send_query_extended(PGXCNodeHandle *handle, const char *query,
const char *statement, const char *portal,
int num_params, Oid *param_types,
int paramlen, char *params,
bool send_describe, int fetch_size);
extern int pgxc_node_send_plan(PGXCNodeHandle * handle, const char *statement,
const char *query, const char *planstr,
short num_params, Oid *param_types);
extern int pgxc_node_send_gxid(PGXCNodeHandle * handle, GlobalTransactionId gxid);
extern int pgxc_node_send_cmd_id(PGXCNodeHandle *handle, CommandId cid);
extern int pgxc_node_send_snapshot(PGXCNodeHandle * handle, Snapshot snapshot);
extern int pgxc_node_send_timestamp(PGXCNodeHandle * handle, TimestampTz timestamp);
extern bool pgxc_node_receive(const int conn_count,
PGXCNodeHandle ** connections, struct timeval * timeout);
extern int pgxc_node_read_data(PGXCNodeHandle * conn, bool close_if_error);
extern int send_some(PGXCNodeHandle * handle, int len);
extern int pgxc_node_flush(PGXCNodeHandle *handle);
extern char get_message(PGXCNodeHandle *conn, int *len, char **msg);
extern void add_error_message(PGXCNodeHandle * handle, const char *message);
extern Datum pgxc_execute_on_nodes(int numnodes, Oid *nodelist, char *query);
extern void PGXCNodeSetParam(bool local, const char *name, const char *value,
int flags);
extern void PGXCNodeResetParams(bool only_local);
extern char *PGXCNodeGetSessionParamStr(void);
extern char *PGXCNodeGetTransactionParamStr(void);
extern void pgxc_node_set_query(PGXCNodeHandle *handle, const char *set_query);
extern void RequestInvalidateRemoteHandles(void);
extern void RequestRefreshRemoteHandles(void);
extern bool PoolerMessagesPending(void);
extern void PGXCNodeSetConnectionState(PGXCNodeHandle *handle,
DNConnectionState new_state);
extern bool PgxcNodeDiffBackendHandles(List **nodes_alter,
List **nodes_delete, List **nodes_add);
extern void PgxcNodeRefreshBackendHandlesShmem(List *nodes_alter);
extern void HandlePoolerMessages(void);
/* Check health of nodes in the connection pool. */
extern void PoolPingNodeRecheck(Oid nodeoid);
#endif /* PGXCNODE_H */
|
