prestogres 0.1.0

data/pgpool2/sql/pgpool-recovery/pgpool-recovery.sql.in

@@ -0,0 +1,19 @@
+CREATE OR REPLACE FUNCTION pgpool_recovery(text, text, text)
+RETURNS bool
+AS 'MODULE_PATHNAME', 'pgpool_recovery'
+LANGUAGE C STRICT;
+
+CREATE OR REPLACE FUNCTION pgpool_remote_start(text, text)
+RETURNS bool
+AS 'MODULE_PATHNAME', 'pgpool_remote_start'
+LANGUAGE C STRICT;
+
+CREATE OR REPLACE FUNCTION pgpool_pgctl(text, text)
+RETURNS bool
+AS '$libdir/pgpool-recovery', 'pgpool_pgctl'
+LANGUAGE C STRICT;
+
+CREATE OR REPLACE FUNCTION pgpool_switch_xlog(text)
+RETURNS text
+AS 'MODULE_PATHNAME', 'pgpool_switch_xlog'
+LANGUAGE C STRICT;

data/pgpool2/sql/pgpool-recovery/pgpool_recovery--1.0.sql

@@ -0,0 +1,24 @@
+-- complain if script is sourced in psql, rather than via CREATE EXTENSION
+\echo Use "CREATE EXTENSION pgpool_recovery" to load this file. \quit
+
+CREATE FUNCTION pgpool_recovery(IN script_name text,
+	   IN remote_host text,
+	   IN remote_data_directory text)
+RETURNS bool
+AS 'MODULE_PATHNAME', 'pgpool_recovery'
+LANGUAGE C STRICT;
+
+CREATE FUNCTION pgpool_remote_start(IN remote_host text, IN remote_data_directory text)
+RETURNS bool
+AS 'MODULE_PATHNAME', 'pgpool_remote_start'
+LANGUAGE C STRICT;
+
+CREATE OR REPLACE FUNCTION pgpool_pgctl(IN action text, IN stop_mode text)
+RETURNS bool
+AS '$libdir/pgpool-recovery', 'pgpool_pgctl'
+LANGUAGE C STRICT;
+
+CREATE OR REPLACE FUNCTION pgpool_switch_xlog(IN arcive_dir text)
+RETURNS text
+AS 'MODULE_PATHNAME', 'pgpool_switch_xlog'
+LANGUAGE C STRICT;

data/pgpool2/sql/pgpool-recovery/pgpool_recovery.control

@@ -0,0 +1,5 @@
+# pgpool-recovery extension
+comment = 'recovery functions for pgpool-II'
+default_version = '1.0'
+module_pathname = '$libdir/pgpool-recovery'
+relocatable = true

data/pgpool2/sql/pgpool-recovery/uninstall_pgpool-recovery.sql

@@ -0,0 +1,3 @@
+DROP FUNCTION pgpool_switch_xlog(text);
+DROP FUNCTION pgpool_remote_start(text, text);
+DROP FUNCTION pgpool_recovery(text, text, text);

data/pgpool2/sql/pgpool-regclass/pgpool-regclass.c

@@ -0,0 +1,206 @@
+/* -*-pgsql-c-*- */
+/*
+ * $Header$
+ *
+ * Copyright (c) 2003-2012	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.
+ *
+ * pgpool-regclass.c is similar to PostgreSQL builtin function
+ * reglcass but does not throw exceptions.
+ * If something goes wrong, it returns InvalidOid.
+ */
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include "postgres.h"
+#include "utils/builtins.h"
+#include "utils/syscache.h"
+#include "utils/elog.h"
+#include "catalog/namespace.h"
+#include "nodes/makefuncs.h"
+#include "commands/dbcommands.h"
+#include "fmgr.h"
+#include "funcapi.h"
+
+#include <stdlib.h>
+
+#ifdef PG_MODULE_MAGIC
+PG_MODULE_MAGIC;
+#endif
+
+PG_FUNCTION_INFO_V1(pgpool_regclass);
+
+extern Datum pgpool_regclass(PG_FUNCTION_ARGS);
+
+static List *
+MystringToQualifiedNameList(const char *string);
+
+static RangeVar *
+MymakeRangeVarFromNameList(List *names);
+
+extern Oid			MyDatabaseId;
+
+#if !defined(PG_VERSION_NUM) || (PG_VERSION_NUM < 90100)
+static Oid
+get_namespace_oid(const char *nspname, bool missing_ok);
+#endif
+
+Datum
+pgpool_regclass(PG_FUNCTION_ARGS)
+{
+	char		*pro_name_or_oid = PG_GETARG_CSTRING(0);
+	Oid	result;
+	List *names;
+	RangeVar   *rel;
+
+	names = MystringToQualifiedNameList(pro_name_or_oid);
+	if (names == NIL)
+		PG_RETURN_OID(InvalidOid);
+
+	rel = MymakeRangeVarFromNameList(names);
+	if (rel == NULL)
+		PG_RETURN_OID(InvalidOid);
+
+	/*
+	 * Check to see if cross-database reference is used.
+	 * This is to be done in RangeVarGetRelid(). The reason
+	 * we do this here is, to avoid raising an error
+	 * in RangeVarGetRelid().
+	 */
+	if (rel->catalogname)
+	{
+		if (strcmp(rel->catalogname, get_database_name(MyDatabaseId)) != 0)
+			PG_RETURN_OID(InvalidOid);
+	}
+
+	/*
+	 * Check to see if schema exists.
+	 * This is to be done in RangeVarGetRelid(). The reason
+	 * we do this here is, to avoid raising an error
+	 * in RangeVarGetRelid().
+	 */
+	if (rel->schemaname)
+	{
+		if (get_namespace_oid(rel->schemaname, true) == InvalidOid)
+			PG_RETURN_OID(InvalidOid);
+	}
+#if !defined(PG_VERSION_NUM) || (PG_VERSION_NUM < 90200)
+	result = RangeVarGetRelid(rel, true);
+#else
+	/* RangeVarGetRelid() of PostgreSQL 9.2 or later, has third
+	 * argument "missing_ok" which suppresses ERROR exception, but
+	 * returns invlaid_oid. See include/catalog/namespace.h */
+	result = RangeVarGetRelid(rel, true, true);
+#endif
+	PG_RETURN_OID(result);
+}
+
+/*
+ * Given a C string, parse it into a qualified-name list.
+ * Stolen from PostgreSQL, removing elog calls.
+ */
+static List *
+MystringToQualifiedNameList(const char *string)
+{
+	char	   *rawname;
+	List	   *result = NIL;
+	List	   *namelist;
+	ListCell   *l;
+
+	/* We need a modifiable copy of the input string. */
+	rawname = pstrdup(string);
+
+	if (!SplitIdentifierString(rawname, '.', &namelist))
+		return NIL;
+
+	if (namelist == NIL)
+		return NIL;
+
+	foreach(l, namelist)
+	{
+		char	   *curname = (char *) lfirst(l);
+
+		result = lappend(result, makeString(pstrdup(curname)));
+	}
+
+	pfree(rawname);
+	list_free(namelist);
+
+	return result;
+}
+
+/*
+ * makeRangeVarFromNameList
+ *		Utility routine to convert a qualified-name list into RangeVar form.
+ *		Stolen from PostgreSQL, removing ereport calls.
+ */
+static RangeVar *
+MymakeRangeVarFromNameList(List *names)
+{
+/*
+ * Number of arguments of makeRangeVar() has been increased in 8.4 or
+ * later.
+ */
+#if PG_VERSION_NUM >= 80400
+	RangeVar   *rel = makeRangeVar(NULL, NULL, -1);
+#else
+	RangeVar   *rel = makeRangeVar(NULL, NULL);
+#endif
+
+	switch (list_length(names))
+	{
+		case 1:
+			rel->relname = strVal(linitial(names));
+			break;
+		case 2:
+			rel->schemaname = strVal(linitial(names));
+			rel->relname = strVal(lsecond(names));
+			break;
+		case 3:
+			rel->catalogname = strVal(linitial(names));
+			rel->schemaname = strVal(lsecond(names));
+			rel->relname = strVal(lthird(names));
+			break;
+		default:
+			rel = NULL;
+			break;
+	}
+
+	return rel;
+}
+
+#if !defined(PG_VERSION_NUM) || (PG_VERSION_NUM < 90100)
+/*
+ * get_namespace_oid - given a namespace name, look up the OID
+ *
+ * If missing_ok is false, throw an error if namespace name not found.  If
+ * true, just return InvalidOid.
+ *
+ * This function was stolen from PostgreSQL 9.0 and modified to not
+ * use GetSysCacheOid1. Since it is new in 9.0.
+ */
+static Oid
+get_namespace_oid(const char *nspname, bool missing_ok)
+{
+	Oid			oid;
+
+	oid = GetSysCacheOid(NAMESPACENAME, CStringGetDatum(nspname), 0, 0, 0);
+	if (!OidIsValid(oid) && !missing_ok)
+        ereport(ERROR,
+                (errcode(ERRCODE_UNDEFINED_SCHEMA),
+                 errmsg("schema \"%s\" does not exist", nspname)));
+
+	return oid;
+}
+#endif

data/pgpool2/sql/pgpool-regclass/pgpool-regclass.sql.in

@@ -0,0 +1,4 @@
+CREATE OR REPLACE FUNCTION pg_catalog.pgpool_regclass(cstring)
+RETURNS oid
+AS 'MODULE_PATHNAME', 'pgpool_regclass'
+LANGUAGE C STRICT;

data/pgpool2/sql/pgpool-regclass/pgpool_regclass--1.0.sql

@@ -0,0 +1,7 @@
+-- complain if script is sourced in psql, rather than via CREATE EXTENSION
+\echo Use "CREATE EXTENSION pgpool_regclass" to load this file. \quit
+
+CREATE FUNCTION pg_catalog.pgpool_regclass(IN expression cstring)
+RETURNS oid
+AS 'MODULE_PATHNAME', 'pgpool_regclass'
+LANGUAGE C STRICT;

data/pgpool2/sql/pgpool-regclass/pgpool_regclass.control

@@ -0,0 +1,5 @@
+# pgpool-regclass extension
+comment = 'replacement for regclass'
+default_version = '1.0'
+module_pathname = '$libdir/pgpool-regclass'
+relocatable = true

data/pgpool2/sql/pgpool-regclass/uninstall_pgpool-regclass.sql

@@ -0,0 +1 @@
+DROP FUNCTION pgpool_regclass(cstring);

data/pgpool2/sql/system_db.sql

@@ -0,0 +1,38 @@
+-- $Header$
+
+DROP TABLE pgpool_catalog.dist_def;
+DROP TABLE pgpool_catalog.replicate_def;
+DROP TABLE pgpool_catalog.query_cache;
+DROP SCHEMA pgpool_catalog;
+
+
+CREATE SCHEMA pgpool_catalog;
+
+CREATE TABLE pgpool_catalog.dist_def(
+	dbname TEXT,
+	schema_name TEXT,
+	table_name TEXT,
+	col_name TEXT NOT NULL CHECK (col_name = ANY (col_list)),
+	col_list TEXT[] NOT NULL,
+	type_list TEXT[] NOT NULL,
+	dist_def_func TEXT NOT NULL,
+	PRIMARY KEY (dbname,schema_name,table_name)
+);
+
+CREATE TABLE pgpool_catalog.replicate_def(
+	dbname TEXT,
+	schema_name TEXT,
+	table_name TEXT,
+	col_list TEXT[] NOT NULL,
+	type_list TEXT[] NOT NULL,
+	PRIMARY KEY (dbname,schema_name,table_name)
+);
+
+CREATE TABLE pgpool_catalog.query_cache (
+	hash TEXT,
+	query TEXT,
+	value bytea,
+	dbname TEXT,
+	create_time TIMESTAMP WITH TIME ZONE,
+	PRIMARY KEY(hash, dbname)
+);

data/pgpool2/strlcpy.c

@@ -0,0 +1,85 @@
+/* -*-pgsql-c-*- */
+/*
+ *
+ * $Header$
+ *
+ * This file was imported from PostgreSQL source code.
+ * See below for the copyright and description.
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL
+ * written by Tatsuo Ishii
+ *
+ * Portions Copyright (c) 2003-2008	PgPool Global Development Group
+ *
+ */
+/*-------------------------------------------------------------------------
+ *
+ * strlcpy.c
+ *	  strncpy done right
+ *
+ * Portions Copyright (c) 1996-2006, PostgreSQL Global Development Group
+ *
+ *
+ * IDENTIFICATION
+ *	  $PostgreSQL: pgsql/src/port/strlcpy.c,v 1.3 2006/10/04 00:30:14 momjian Exp $
+ *
+ * This file was taken from OpenBSD and is used on platforms that don't
+ * provide strlcpy().  The OpenBSD copyright terms follow.
+ *-------------------------------------------------------------------------
+ */
+
+/*	$OpenBSD: strlcpy.c,v 1.11 2006/05/05 15:27:38 millert Exp $	*/
+
+/*
+ * Copyright (c) 1998 Todd C. Miller <Todd.Miller@courtesan.com>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <unistd.h>
+#include "pool.h"
+
+/*
+ * Copy src to string dst of size siz.	At most siz-1 characters
+ * will be copied.	Always NUL terminates (unless siz == 0).
+ * Returns strlen(src); if retval >= siz, truncation occurred.
+ * Function creation history:  http://www.gratisoft.us/todd/papers/strlcpy.html
+ */
+size_t
+strlcpy(char *dst, const char *src, size_t siz)
+{
+	char	   *d = dst;
+	const char *s = src;
+	size_t		n = siz;
+
+	/* Copy as many bytes as will fit */
+	if (n != 0)
+	{
+		while (--n != 0)
+		{
+			if ((*d++ = *s++) == '\0')
+				break;
+		}
+	}
+
+	/* Not enough room in dst, add NUL and traverse rest of src */
+	if (n == 0)
+	{
+		if (siz != 0)
+			*d = '\0';			/* NUL-terminate dst */
+		while (*s++)
+			;
+	}
+
+	return (s - src - 1);		/* count does not include NUL */
+}

data/pgpool2/test/C/test_extended.c

@@ -0,0 +1,98 @@
+/* -*-pgsql-c-*- */
+/*
+ * $Header$
+ *
+ * pgpool: a language independent connection pool server for PostgreSQL
+ * written by Tatsuo Ishii
+ *
+ * Copyright (c) 2003-2010	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.
+ *
+ * Extended protocol test program.
+ * Features are:
+ * libq trace on/off
+ * explicit transaction on/off
+ * multiple SQL statements
+ */
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+
+#include "libpq-fe.h"
+#include "libpq/libpq-fs.h"
+
+int main(int argc, char **argv) {
+	char *connect_string = "user=t-ishii dbname=test port=5432";
+	int doTrace = 1;	/* set 1 to start libpq trace */
+	int doTransaction = 1;		/* set 1 to start explicit transaction */
+	int doSleep = 0;	/* set non 0 seconds to sleep after connection and beforfe starting command */
+
+	/* SQL commands to be executed */
+	static char *commands[] = {
+	  "SAVEPOINT S1",
+	  "UPDATE t1 SET k = 1",
+	  "ROLLBACK TO S1",
+	  "SELECT 1",
+	  "RELEASE SAVEPOINT S1"
+	};
+
+	PGconn *conn;
+	PGresult *res;
+	int i;
+
+	conn = PQconnectdb(connect_string);
+	if (PQstatus(conn) == CONNECTION_BAD) {
+		printf("Unable to connect to db\n");
+		PQfinish(conn);
+		return 1;
+	}
+
+	if (doSleep)
+	  sleep(doSleep);
+
+	if(doTrace == 1)
+	  PQtrace(conn, stdout);
+
+	if(doTransaction)
+	  PQexec(conn,"BEGIN;");
+
+	for (i=0;i<sizeof(commands)/sizeof(char *);i++)
+	{
+		char *command = commands[i];
+	
+		res = PQexecParams(conn, command, 0, NULL, NULL, NULL, NULL, 0);
+		switch(PQresultStatus(res)) {
+			case PGRES_COMMAND_OK:
+			case PGRES_TUPLES_OK:
+				fprintf(stderr, "\"%s\" : succeeded\n", command);
+				break;
+			default:
+				fprintf(stderr, "\"%s\" failed: %s\n", command, PQresultErrorMessage(res));
+				break;
+		}
+		PQclear(res);
+
+	}
+
+	if(doTransaction == 1) {
+		PQexec(conn,"COMMIT;");
+	}
+	
+	PQfinish(conn);
+	return 0;
+}