prestogres 0.1.0

data/pgpool2/pool_session_context.h

@@ -0,0 +1,203 @@
+/* -*-pgsql-c-*- */
+/*
+ *
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL 
+ * written by Tatsuo Ishii
+ *
+ * Copyright (c) 2003-2011	PgPool Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ *
+ * pool_process_context.h.: process context information
+ *
+ */
+
+#ifndef POOL_SESSION_CONTEXT_H
+#define POOL_SESSION_CONTEXT_H
+#define INIT_LIST_SIZE 8
+
+#include "pool.h"
+#include "pool_process_context.h"
+#include "pool_session_context.h"
+#include "pool_query_context.h"
+#include "parser/pool_memory.h"
+#include "pool_memqcache.h"
+
+/*
+ * Transaction isolation mode
+ */
+typedef enum {
+	POOL_UNKNOWN,				/* Unknown. Need to ask backend */
+	POOL_READ_UNCOMMITTED,		/* Read uncommitted */
+	POOL_READ_COMMITTED,		/* Read committed */
+	POOL_REPEATABLE_READ,		/* Rpeatable read */
+	POOL_SERIALIZABLE			/* Serializable */
+} POOL_TRANSACTION_ISOLATION;
+
+/*
+ * Message content of extended query
+ */
+typedef struct {
+	char kind;	/* one of 'P':Parse, 'B':Bind or 'Q':Query(PREPARE) */
+	int len;	/* in host byte order */
+	char *contents;
+	int num_tsparams;
+	char *name;		/* object name of prepared statement or portal */
+	POOL_QUERY_CONTEXT *query_context;
+	/*
+	 * Following members are only used when memcache is enabled.
+	 */
+	bool is_cache_safe;	/* true if the query can be cached */
+	int param_offset;		/* Offset from contents where actual bind
+							 * parameters are stored.
+							 * This is meaningful only when is_cache_safe is true.
+							 */
+} POOL_SENT_MESSAGE;
+
+/*
+ * List of POOL_SENT_MESSAGE
+ */
+typedef struct {
+	int capacity;	/* capacity of list */
+	int size;		/* number of elements */
+	POOL_SENT_MESSAGE **sent_messages;
+} POOL_SENT_MESSAGE_LIST;
+
+/*
+ * Per session context:
+ */
+typedef struct {
+	POOL_PROCESS_CONTEXT *process_context;		/* belonging process */
+	POOL_CONNECTION *frontend;	/* connection to frontend */
+	POOL_CONNECTION_POOL *backend;		/* connection to backends */
+
+	/* If true, we are waiting for backend response.  For SELECT this
+	 * flags should be kept until all responses are returned from
+	 * backend. i.e. until "Read for Query" packet.
+	 */
+	bool in_progress;
+
+	/* If true, we are doing extended query message */
+	bool doing_extended_query_message;
+
+	/* If true, the command in progress has finished successfully. */
+	bool command_success;
+
+	/* If true, write query has been appeared in this transaction */
+	bool writing_transaction;
+
+	/* If true, error occurred in this transaction */
+	bool failed_transaction;
+
+	/* If true, we skip reading from backends */
+	bool skip_reading_from_backends;
+
+	/* ignore any command until Sync message */
+	bool ignore_till_sync;
+
+	/*
+	 * Transaction isolation mode.
+	 */
+	POOL_TRANSACTION_ISOLATION transaction_isolation;
+
+	/*
+	 * Associated query context, only used for non-extended
+	 * protocol. In extended protocol, the query context resides in
+	 * "PreparedStatementList *pstmt_list" (see below).
+	 */
+	POOL_QUERY_CONTEXT *query_context;
+#ifdef NOT_USED
+	/* where to send map for PREPARE/EXECUTE/DEALLOCATE */
+	POOL_PREPARED_SEND_MAP prep_where;
+#endif /* NOT_USED */
+	POOL_MEMORY_POOL *memory_context;	/* memory context for session */
+
+	/* message which doesn't receive complete message */
+	POOL_SENT_MESSAGE *uncompleted_message;
+
+	POOL_SENT_MESSAGE_LIST message_list;
+
+	int load_balance_node_id;	/* selected load balance node id */
+
+	/*
+	 * If true, UPDATE/DELETE caused difference in number of affected
+	 * tuples in backends.
+	*/
+	bool mismatch_ntuples;
+
+	/*
+	 * If mismatch_ntuples true, this array holds the number of
+	 * affected tuples of each node.
+	 * -1 for down nodes.
+	 */
+	int ntuples[MAX_NUM_BACKENDS];
+
+	/*
+	 * If true, we are executing reset query list.
+	 */
+	bool reset_context;
+
+	/*
+	 * Query cache management area
+	 */
+	POOL_QUERY_CACHE_ARRAY *query_cache_array;	/* pending SELECT results */
+	long long int num_selects;	/* number of successful SELECTs in this transaction */
+} POOL_SESSION_CONTEXT;
+
+extern void pool_init_session_context(POOL_CONNECTION *frontend, POOL_CONNECTION_POOL *backend);
+extern void pool_session_context_destroy(void);
+extern POOL_SESSION_CONTEXT *pool_get_session_context(void);
+extern int pool_get_local_session_id(void);
+extern bool pool_is_query_in_progress(void);
+extern void pool_set_query_in_progress(void);
+extern void pool_unset_query_in_progress(void);
+extern bool pool_is_skip_reading_from_backends(void);
+extern void pool_set_skip_reading_from_backends(void);
+extern void pool_unset_skip_reading_from_backends(void);
+extern bool pool_is_doing_extended_query_message(void);
+extern void pool_set_doing_extended_query_message(void);
+extern void pool_unset_doing_extended_query_message(void);
+extern bool pool_is_ignore_till_sync(void);
+extern void pool_set_ignore_till_sync(void);
+extern void pool_unset_ignore_till_sync(void);
+extern POOL_SENT_MESSAGE *pool_create_sent_message(char kind, int len, char *contents,
+												   int num_tsparams, const char *name,
+												   POOL_QUERY_CONTEXT *query_context);
+extern void pool_add_sent_message(POOL_SENT_MESSAGE *message);
+extern bool pool_remove_sent_message(char kind, const char *name);
+extern void pool_remove_sent_messages(char kind);
+extern void pool_clear_sent_message_list(void);
+extern void pool_sent_message_destroy(POOL_SENT_MESSAGE *message);
+extern POOL_SENT_MESSAGE *pool_get_sent_message(char kind, const char *name);
+extern void pool_unset_writing_transaction(void);
+extern void pool_set_writing_transaction(void);
+extern bool pool_is_writing_transaction(void);
+extern void pool_unset_failed_transaction(void);
+extern void pool_set_failed_transaction(void);
+extern bool pool_is_failed_transaction(void);
+extern void pool_unset_transaction_isolation(void);
+extern void pool_set_transaction_isolation(POOL_TRANSACTION_ISOLATION isolation_level);
+extern POOL_TRANSACTION_ISOLATION pool_get_transaction_isolation(void);
+extern void pool_unset_command_success(void);
+extern void pool_set_command_success(void);
+extern bool pool_is_command_success(void);
+extern void pool_copy_prep_where(bool *src, bool *dest);
+extern bool can_query_context_destroy(POOL_QUERY_CONTEXT *qc);
+
+#ifdef NOT_USED
+extern void pool_add_prep_where(char *name, bool *map);
+extern bool *pool_get_prep_where(char *name);
+extern void pool_delete_prep_where(char *name);
+#endif /* NOT_USED */
+#endif /* POOL_SESSION_CONTEXT_H */

data/pgpool2/pool_shmem.c

@@ -0,0 +1,185 @@
+/* -*-pgsql-c-*- */
+/*
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL
+ * written by Tatsuo Ishii
+ *
+ * Portions Copyright (c) 2003-2013, PgPool Global Development Group
+ * Portions Copyright (c) 2003-2004, PostgreSQL Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ *
+ */
+#include "pool.h"
+
+#include <errno.h>
+#include <string.h>
+#include <sys/shm.h>
+
+#include "pool_ipc.h"
+
+
+#ifdef SHM_SHARE_MMU			/* use intimate shared memory on Solaris */
+#define PG_SHMAT_FLAGS			SHM_SHARE_MMU
+#else
+#define PG_SHMAT_FLAGS			0
+#endif
+
+
+#define MAX_ON_EXITS 64
+
+static struct ONEXIT
+{
+	void		(*function) (int code, Datum arg);
+	Datum		arg;
+}	on_shmem_exit_list[MAX_ON_EXITS];
+
+static int	on_shmem_exit_index;
+
+
+static void IpcMemoryDetach(int status, Datum shmaddr);
+static void IpcMemoryDelete(int status, Datum shmId);
+
+
+/*
+ * Create a shared memory segment of the given size and initialize.  Also,
+ * register an on_shmem_exit callback to release the storage.
+ */
+void *
+pool_shared_memory_create(size_t size)
+{
+	int			shmid;
+	void	   *memAddress;
+
+	/* Try to create new segment */
+	shmid = shmget(IPC_PRIVATE, size, IPC_CREAT | IPC_EXCL | IPCProtection);
+
+	if (shmid < 0)
+	{
+		pool_error("could not create shared memory segment: %s",
+				   strerror(errno));
+		return NULL;
+	}
+
+	/* Register on-exit routine to delete the new segment */
+	on_shmem_exit(IpcMemoryDelete, shmid);
+
+	/* OK, should be able to attach to the segment */
+	memAddress = shmat(shmid, NULL, PG_SHMAT_FLAGS);
+
+	if (memAddress == (void *) -1)
+	{
+		pool_error("shmat(id=%d) failed: %s", shmid, strerror(errno));
+		return NULL;
+	}
+
+	/* Register on-exit routine to detach new segment before deleting */
+	on_shmem_exit(IpcMemoryDetach, (Datum) memAddress);
+
+	return memAddress;
+}
+
+/*
+ * Removes a shared memory segment from process' address spaceq (called as
+ * an on_shmem_exit callback, hence funny argument list)
+ */
+static void
+IpcMemoryDetach(int status, Datum shmaddr)
+{
+	if (shmdt((void *) shmaddr) < 0)
+		pool_log("shmdt(%p) failed: %s", (void *) shmaddr, strerror(errno));
+}
+
+/*
+ * Deletes a shared memory segment (called as an on_shmem_exit callback,
+ * hence funny argument list)
+ */
+static void
+IpcMemoryDelete(int status, Datum shmId)
+{
+  	struct shmid_ds shmStat;
+
+  	/*
+  	 * Is a previously-existing shmem segment still existing and in use?
+  	 */
+  	if (shmctl(shmId, IPC_STAT, &shmStat) < 0
+  		&& (errno == EINVAL || errno == EACCES))
+  		return;
+  	else if (shmStat.shm_nattch != 0)
+  		return;
+
+	if (shmctl(shmId, IPC_RMID, NULL) < 0)
+		pool_log("shmctl(%lu, %d, 0) failed: %s",
+				 shmId, IPC_RMID, strerror(errno));
+}
+
+void
+pool_shmem_exit(int code)
+{
+	shmem_exit(code);
+	/* Close syslog connection here as this function is always called on exit */
+	closelog();
+}
+
+/*
+ * Run all of the on_shmem_exit routines --- but don't actually exit.  This
+ * is used by the postmaster to re-initialize shared memory and semaphores
+ * after a backend dies horribly.
+ */
+void
+shmem_exit(int code)
+{
+	pool_debug("shmem_exit(%d)", code);
+
+	/*
+	 * Call all the registered callbacks.
+	 *
+	 * As with proc_exit(), we remove each callback from the list before
+	 * calling it, to avoid infinite loop in case of error.
+	 */
+	while (--on_shmem_exit_index >= 0)
+		(*on_shmem_exit_list[on_shmem_exit_index].function) (code,
+								on_shmem_exit_list[on_shmem_exit_index].arg);
+
+	on_shmem_exit_index = 0;
+}
+
+/*
+ * This function adds a callback function to the list of functions invoked
+ * by shmem_exit().
+ */
+void
+on_shmem_exit(void (*function) (int code, Datum arg), Datum arg)
+{
+	if (on_shmem_exit_index >= MAX_ON_EXITS)
+		pool_error("out of on_shmem_exit slots");
+	else
+	{
+		on_shmem_exit_list[on_shmem_exit_index].function = function;
+		on_shmem_exit_list[on_shmem_exit_index].arg = arg;
+
+		++on_shmem_exit_index;
+	}
+}
+
+/*
+ * This function clears all on_proc_exit() and on_shmem_exit() registered
+ * functions.  This is used just after forking a backend, so that the
+ * backend doesn't believe it should call the postmaster's on-exit routines
+ * when it exits...
+ */
+void
+on_exit_reset(void)
+{
+	on_shmem_exit_index = 0;
+}

data/pgpool2/pool_signal.c

@@ -0,0 +1,158 @@
+/* -*-pgsql-c-*- */
+/*
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL
+ * written by Tatsuo Ishii
+ *
+ * Portions Copyright (c) 2003-2008, PgPool Global Development Group
+ * Portions Copyright (c) 2003-2004, PostgreSQL Global Development Group
+ *
+ * Permission to use, copy, modify, and distribute this software and
+ * its documentation for any purpose and without fee is hereby
+ * granted, provided that the above copyright notice appear in all
+ * copies and that both that copyright notice and this permission
+ * notice appear in supporting documentation, and that the name of the
+ * author not be used in advertising or publicity pertaining to
+ * distribution of the software without specific, written prior
+ * permission. The author makes no representations about the
+ * suitability of this software for any purpose.  It is provided "as
+ * is" without express or implied warranty.
+ */
+
+/*
+ * Signal stuff. Stolen from PostgreSQL source code.
+ */
+
+#include "config.h"
+#include "pool_signal.h"
+#include <signal.h>
+
+#ifdef HAVE_SIGPROCMASK
+sigset_t	UnBlockSig,
+			BlockSig,
+			AuthBlockSig;
+
+#else
+int			UnBlockSig,
+			BlockSig,
+			AuthBlockSig;
+#endif
+
+/*
+ * Initialize BlockSig, UnBlockSig, and AuthBlockSig.
+ *
+ * BlockSig is the set of signals to block when we are trying to block
+ * signals.  This includes all signals we normally expect to get, but NOT
+ * signals that should never be turned off.
+ *
+ * AuthBlockSig is the set of signals to block during authentication;
+ * it's essentially BlockSig minus SIGTERM, SIGQUIT, SIGALRM.
+ *
+ * UnBlockSig is the set of signals to block when we don't want to block
+ * signals (is this ever nonzero??)
+ */
+void
+poolinitmask(void)
+{
+#ifdef HAVE_SIGPROCMASK
+	sigemptyset(&UnBlockSig);
+	sigfillset(&BlockSig);
+	sigfillset(&AuthBlockSig);
+
+	/*
+	 * Unmark those signals that should never be blocked. Some of these
+	 * signal names don't exist on all platforms.  Most do, but might as
+	 * well ifdef them all for consistency...
+	 */
+#ifdef SIGTRAP
+	sigdelset(&BlockSig, SIGTRAP);
+	sigdelset(&AuthBlockSig, SIGTRAP);
+#endif
+#ifdef SIGABRT
+	sigdelset(&BlockSig, SIGABRT);
+	sigdelset(&AuthBlockSig, SIGABRT);
+#endif
+#ifdef SIGILL
+	sigdelset(&BlockSig, SIGILL);
+	sigdelset(&AuthBlockSig, SIGILL);
+#endif
+#ifdef SIGFPE
+	sigdelset(&BlockSig, SIGFPE);
+	sigdelset(&AuthBlockSig, SIGFPE);
+#endif
+#ifdef SIGSEGV
+	sigdelset(&BlockSig, SIGSEGV);
+	sigdelset(&AuthBlockSig, SIGSEGV);
+#endif
+#ifdef SIGBUS
+	sigdelset(&BlockSig, SIGBUS);
+	sigdelset(&AuthBlockSig, SIGBUS);
+#endif
+#ifdef SIGSYS
+	sigdelset(&BlockSig, SIGSYS);
+	sigdelset(&AuthBlockSig, SIGSYS);
+#endif
+#ifdef SIGCONT
+	sigdelset(&BlockSig, SIGCONT);
+	sigdelset(&AuthBlockSig, SIGCONT);
+#endif
+#ifdef SIGTERM
+	sigdelset(&AuthBlockSig, SIGTERM);
+#endif
+#ifdef SIGQUIT
+	sigdelset(&AuthBlockSig, SIGQUIT);
+#endif
+#ifdef SIGALRM
+	sigdelset(&AuthBlockSig, SIGALRM);
+#endif
+#else
+	UnBlockSig = 0;
+	BlockSig = sigmask(SIGHUP) | sigmask(SIGQUIT) |
+		sigmask(SIGTERM) | sigmask(SIGALRM) |
+		sigmask(SIGINT) | sigmask(SIGUSR1) |
+		sigmask(SIGUSR2) | sigmask(SIGCHLD) |
+		sigmask(SIGWINCH) | sigmask(SIGFPE);
+	AuthBlockSig = sigmask(SIGHUP) |
+		sigmask(SIGINT) | sigmask(SIGUSR1) |
+		sigmask(SIGUSR2) | sigmask(SIGCHLD) |
+		sigmask(SIGWINCH) | sigmask(SIGFPE);
+#endif
+}
+
+
+/* Win32 signal handling is in backend/port/win32/signal.c */
+#ifndef WIN32
+
+/*
+ * We need to check actually the system has the posix signals or not, but...
+ */
+#define HAVE_POSIX_SIGNALS
+/*
+ * Set up a signal handler
+ */
+pool_sighandler_t
+pool_signal(int signo, pool_sighandler_t func)
+{
+#if !defined(HAVE_POSIX_SIGNALS)
+	return signal(signo, func);
+#else
+	struct sigaction act,
+				oact;
+
+	act.sa_handler = func;
+	sigemptyset(&act.sa_mask);
+	act.sa_flags = 0;
+	if (signo != SIGALRM)
+		act.sa_flags |= SA_RESTART;
+#ifdef SA_NOCLDSTOP
+	if (signo == SIGCHLD)
+		act.sa_flags |= SA_NOCLDSTOP;
+#endif
+	if (sigaction(signo, &act, &oact) < 0)
+		return SIG_ERR;
+	return oact.sa_handler;
+#endif   /* !HAVE_POSIX_SIGNALS */
+}
+
+#endif   /* WIN32 */