1 files changed, 76 insertions, 0 deletions
diff --git a/contrib/tsearch2/query_support.c b/contrib/tsearch2/query_support.c
new file mode 100644
index 00000000000..c973def7d4d
--- /dev/null
+++ b/contrib/tsearch2/query_support.c
@@ -0,0 +1,76 @@
+#include "postgres.h"
+#include "fmgr.h"
+
+#include "query_util.h"
+
+PG_FUNCTION_INFO_V1(tsquery_numnode);
+Datum           tsquery_numnode(PG_FUNCTION_ARGS);
+
+Datum
+tsquery_numnode(PG_FUNCTION_ARGS) {
+	QUERYTYPE  *query = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(0)));
+	int nnode = query->size;
+	PG_FREE_IF_COPY(query,0);
+	PG_RETURN_INT32(nnode);
+}
+
+static int
+CompareTSQ( QUERYTYPE *a, QUERYTYPE *b ) {
+	if ( a->size != b->size ) {
+		return ( a->size < b->size ) ? -1 : 1;
+	} else if ( a->len != b->len ) {
+		return ( a->len < b->len ) ? -1 : 1;
+	} else {
+		QTNode	*an = QT2QTN( GETQUERY(a), GETOPERAND(a) ); 
+		QTNode	*bn = QT2QTN( GETQUERY(b), GETOPERAND(b) );
+		int res = QTNodeCompare(an, bn);
+
+		QTNFree(an);
+		QTNFree(bn);
+
+		return res; 
+	}
+
+	return 0;
+}
+
+PG_FUNCTION_INFO_V1(tsquery_cmp);                                                                              \
+Datum   tsquery_cmp(PG_FUNCTION_ARGS);         
+
+Datum
+tsquery_cmp(PG_FUNCTION_ARGS) {
+	QUERYTYPE  *a = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(0)));
+	QUERYTYPE  *b = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(1)));
+	int res = CompareTSQ(a,b);
+
+	PG_FREE_IF_COPY(a,0);
+	PG_FREE_IF_COPY(b,1);
+
+	PG_RETURN_INT32(res);
+}
+
+#define CMPFUNC( NAME, ACTION ) 									\
+PG_FUNCTION_INFO_V1(NAME);										\
+Datum	NAME(PG_FUNCTION_ARGS);										\
+													\
+Datum													\
+NAME(PG_FUNCTION_ARGS) { 										\
+	QUERYTYPE  *a = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(0)));	\
+	QUERYTYPE  *b = (QUERYTYPE *) DatumGetPointer(PG_DETOAST_DATUM_COPY(PG_GETARG_DATUM(1)));	\
+	int res = CompareTSQ(a,b);									\
+													\
+	PG_FREE_IF_COPY(a,0);										\
+	PG_FREE_IF_COPY(b,1);										\
+													\
+	PG_RETURN_BOOL( ACTION );									\
+}
+
+CMPFUNC( tsquery_lt, res <0 ); 
+CMPFUNC( tsquery_le, res<=0 ); 
+CMPFUNC( tsquery_eq, res==0 ); 
+CMPFUNC( tsquery_ge, res>=0 ); 
+CMPFUNC( tsquery_gt, res >0 );
+CMPFUNC( tsquery_ne, res!=0 );
+
+
+