tsquery_op.c File Reference

#include "postgres.h"
#include "lib/qunique.h"
#include "tsearch/ts_utils.h"
#include "utils/fmgrprotos.h"
#include "varatt.h"

Include dependency graph for tsquery_op.c:

Go to the source code of this file.

Macros
#define	CMPFUNC(NAME, CONDITION)

Functions
Datum	tsquery_numnode (PG_FUNCTION_ARGS)
static QTNode *	join_tsqueries (TSQuery a, TSQuery b, int8 operator, uint16 distance)
Datum	tsquery_and (PG_FUNCTION_ARGS)
Datum	tsquery_or (PG_FUNCTION_ARGS)
Datum	tsquery_phrase_distance (PG_FUNCTION_ARGS)
Datum	tsquery_phrase (PG_FUNCTION_ARGS)
Datum	tsquery_not (PG_FUNCTION_ARGS)
static int	CompareTSQ (TSQuery a, TSQuery b)
Datum	tsquery_cmp (PG_FUNCTION_ARGS)
	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)
TSQuerySign	makeTSQuerySign (TSQuery a)
static char **	collectTSQueryValues (TSQuery a, int *nvalues_p)
static int	cmp_string (const void *a, const void *b)
Datum	tsq_mcontains (PG_FUNCTION_ARGS)
Datum	tsq_mcontained (PG_FUNCTION_ARGS)

Macro Definition Documentation

CMPFUNC

#define CMPFUNC	(		NAME,
			CONDITION
	)

Value:

Datum                                           \
NAME(PG_FUNCTION_ARGS) {                        \
    TSQuery  a = PG_GETARG_TSQUERY_COPY(0);     \
    TSQuery  b = PG_GETARG_TSQUERY_COPY(1);     \
    int res = CompareTSQ(a,b);                  \
												\
	PG_FREE_IF_COPY(a,0);                      \
    PG_FREE_IF_COPY(b,1);                       \
												\
	PG_RETURN_BOOL( CONDITION );               \
}   \
/* keep compiler quiet - no extra ; */          \
extern int no_such_variable

Definition at line 227 of file tsquery_op.c.

Function Documentation

cmp_string()

static int cmp_string ( const void *  a, const void *  b  )

static

Definition at line 298 of file tsquery_op.c.

{
    const char *sa = *((char *const *) a);
    const char *sb = *((char *const *) b);
 
    return strcmp(sa, sb);
}

References a, and b.

Referenced by tsq_mcontains().

CMPFUNC() [1/6]

CMPFUNC	(	tsquery_eq	,
		res	= =0
	)

CMPFUNC() [2/6]

CMPFUNC	(	tsquery_ge	,
		res >=	0
	)

CMPFUNC() [3/6]

CMPFUNC	(	tsquery_gt	,
		res	,
		0
	)

CMPFUNC() [4/6]

CMPFUNC	(	tsquery_le	,
		res<=	0
	)

CMPFUNC() [5/6]

CMPFUNC

(

tsquery_lt

)

CMPFUNC() [6/6]

CMPFUNC	(	tsquery_ne	,
		res !	= 0
	)

collectTSQueryValues()

static char ** collectTSQueryValues ( TSQuery  a, int *  nvalues_p  )

static

Definition at line 267 of file tsquery_op.c.

{
    QueryItem  *ptr = GETQUERY(a);
    char       *operand = GETOPERAND(a);
    char      **values;
    int         nvalues = 0;
    int         i;
 
    values = (char **) palloc(sizeof(char *) * a->size);
 
    for (i = 0; i < a->size; i++)
    {
        if (ptr->type == QI_VAL)
        {
            int         len = ptr->qoperand.length;
            char       *val;
 
            val = palloc(len + 1);
            memcpy(val, operand + ptr->qoperand.distance, len);
            val[len] = '\0';
 
            values[nvalues++] = val;
        }
        ptr++;
    }
 
    *nvalues_p = nvalues;
    return values;
}

References a, QueryOperand::distance, GETOPERAND, GETQUERY, i, len, QueryOperand::length, palloc(), QI_VAL, QueryItem::qoperand, QueryItem::type, val, and values.

Referenced by tsq_mcontains().

CompareTSQ()

static int CompareTSQ ( TSQuery  a, TSQuery  b  )

static

Definition at line 189 of file tsquery_op.c.

{
    if (a->size != b->size)
    {
        return (a->size < b->size) ? -1 : 1;
    }
    else if (VARSIZE(a) != VARSIZE(b))
    {
        return (VARSIZE(a) < VARSIZE(b)) ? -1 : 1;
    }
    else if (a->size != 0)
    {
        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;
}

References a, b, GETOPERAND, GETQUERY, QT2QTN(), QTNFree(), QTNodeCompare(), and VARSIZE().

Referenced by tsquery_cmp().

join_tsqueries()

static QTNode * join_tsqueries ( TSQuery  a, TSQuery  b, int8  operator, uint16  distance  )

static

Definition at line 33 of file tsquery_op.c.

{
    QTNode     *res = (QTNode *) palloc0(sizeof(QTNode));
 
    res->flags |= QTN_NEEDFREE;
 
    res->valnode = (QueryItem *) palloc0(sizeof(QueryItem));
    res->valnode->type = QI_OPR;
    res->valnode->qoperator.oper = operator;
    if (operator == OP_PHRASE)
        res->valnode->qoperator.distance = distance;
 
    res->child = (QTNode **) palloc0(sizeof(QTNode *) * 2);
    res->child[0] = QT2QTN(GETQUERY(b), GETOPERAND(b));
    res->child[1] = QT2QTN(GETQUERY(a), GETOPERAND(a));
    res->nchild = 2;
 
    return res;
}

References a, b, QTNode::child, QueryOperator::distance, QTNode::flags, GETOPERAND, GETQUERY, QTNode::nchild, OP_PHRASE, QueryOperator::oper, palloc0(), QI_OPR, QueryItem::qoperator, QT2QTN(), QTN_NEEDFREE, QueryItem::type, and QTNode::valnode.

Referenced by tsquery_and(), tsquery_or(), and tsquery_phrase_distance().

makeTSQuerySign()

TSQuerySign makeTSQuerySign

(

TSQuery

a

)

Definition at line 250 of file tsquery_op.c.

{
    int         i;
    QueryItem  *ptr = GETQUERY(a);
    TSQuerySign sign = 0;
 
    for (i = 0; i < a->size; i++)
    {
        if (ptr->type == QI_VAL)
            sign |= ((TSQuerySign) 1) << (((unsigned int) ptr->qoperand.valcrc) % TSQS_SIGLEN);
        ptr++;
    }
 
    return sign;
}

References a, GETQUERY, i, QI_VAL, QueryItem::qoperand, sign, TSQS_SIGLEN, QueryItem::type, and QueryOperand::valcrc.

Referenced by gtsquery_compress(), and gtsquery_consistent().

tsq_mcontained()

Datum tsq_mcontained

(

PG_FUNCTION_ARGS

)

Definition at line 354 of file tsquery_op.c.

{
    PG_RETURN_DATUM(DirectFunctionCall2(tsq_mcontains,
                                        PG_GETARG_DATUM(1),
                                        PG_GETARG_DATUM(0)));
}

References DirectFunctionCall2, PG_GETARG_DATUM, PG_RETURN_DATUM, and tsq_mcontains().

tsq_mcontains()

Datum tsq_mcontains

(

PG_FUNCTION_ARGS

)

Definition at line 307 of file tsquery_op.c.

{
    TSQuery     query = PG_GETARG_TSQUERY(0);
    TSQuery     ex = PG_GETARG_TSQUERY(1);
    char      **query_values;
    int         query_nvalues;
    char      **ex_values;
    int         ex_nvalues;
    bool        result = true;
 
    /* Extract the query terms into arrays */
    query_values = collectTSQueryValues(query, &query_nvalues);
    ex_values = collectTSQueryValues(ex, &ex_nvalues);
 
    /* Sort and remove duplicates from both arrays */
    qsort(query_values, query_nvalues, sizeof(char *), cmp_string);
    query_nvalues = qunique(query_values, query_nvalues, sizeof(char *),
                            cmp_string);
    qsort(ex_values, ex_nvalues, sizeof(char *), cmp_string);
    ex_nvalues = qunique(ex_values, ex_nvalues, sizeof(char *), cmp_string);
 
    if (ex_nvalues > query_nvalues)
        result = false;
    else
    {
        int         i;
        int         j = 0;
 
        for (i = 0; i < ex_nvalues; i++)
        {
            for (; j < query_nvalues; j++)
            {
                if (strcmp(ex_values[i], query_values[j]) == 0)
                    break;
            }
            if (j == query_nvalues)
            {
                result = false;
                break;
            }
        }
    }
 
    PG_RETURN_BOOL(result);
}

References cmp_string(), collectTSQueryValues(), i, j, PG_GETARG_TSQUERY, PG_RETURN_BOOL, qsort, and qunique().

Referenced by tsq_mcontained().

tsquery_and()

Datum tsquery_and

(

PG_FUNCTION_ARGS

)

Definition at line 54 of file tsquery_op.c.

{
    TSQuery     a = PG_GETARG_TSQUERY_COPY(0);
    TSQuery     b = PG_GETARG_TSQUERY_COPY(1);
    QTNode     *res;
    TSQuery     query;
 
    if (a->size == 0)
    {
        PG_FREE_IF_COPY(a, 1);
        PG_RETURN_POINTER(b);
    }
    else if (b->size == 0)
    {
        PG_FREE_IF_COPY(b, 1);
        PG_RETURN_POINTER(a);
    }
 
    res = join_tsqueries(a, b, OP_AND, 0);
 
    query = QTN2QT(res);
 
    QTNFree(res);
    PG_FREE_IF_COPY(a, 0);
    PG_FREE_IF_COPY(b, 1);
 
    PG_RETURN_TSQUERY(query);
}

References a, b, join_tsqueries(), OP_AND, PG_FREE_IF_COPY, PG_GETARG_TSQUERY_COPY, PG_RETURN_POINTER, PG_RETURN_TSQUERY, QTN2QT(), and QTNFree().

tsquery_cmp()

Datum tsquery_cmp

(

PG_FUNCTION_ARGS

)

Definition at line 215 of file tsquery_op.c.

{
    TSQuery     a = PG_GETARG_TSQUERY_COPY(0);
    TSQuery     b = PG_GETARG_TSQUERY_COPY(1);
    int         res = CompareTSQ(a, b);
 
    PG_FREE_IF_COPY(a, 0);
    PG_FREE_IF_COPY(b, 1);
 
    PG_RETURN_INT32(res);
}

References a, b, CompareTSQ(), PG_FREE_IF_COPY, PG_GETARG_TSQUERY_COPY, and PG_RETURN_INT32.

tsquery_not()

Datum tsquery_not

(

PG_FUNCTION_ARGS

)

Definition at line 159 of file tsquery_op.c.

{
    TSQuery     a = PG_GETARG_TSQUERY_COPY(0);
    QTNode     *res;
    TSQuery     query;
 
    if (a->size == 0)
        PG_RETURN_POINTER(a);
 
    res = (QTNode *) palloc0(sizeof(QTNode));
 
    res->flags |= QTN_NEEDFREE;
 
    res->valnode = (QueryItem *) palloc0(sizeof(QueryItem));
    res->valnode->type = QI_OPR;
    res->valnode->qoperator.oper = OP_NOT;
 
    res->child = (QTNode **) palloc0(sizeof(QTNode *));
    res->child[0] = QT2QTN(GETQUERY(a), GETOPERAND(a));
    res->nchild = 1;
 
    query = QTN2QT(res);
 
    QTNFree(res);
    PG_FREE_IF_COPY(a, 0);
 
    PG_RETURN_POINTER(query);
}

References a, QTNode::child, QTNode::flags, GETOPERAND, GETQUERY, QTNode::nchild, OP_NOT, QueryOperator::oper, palloc0(), PG_FREE_IF_COPY, PG_GETARG_TSQUERY_COPY, PG_RETURN_POINTER, QI_OPR, QueryItem::qoperator, QT2QTN(), QTN2QT(), QTN_NEEDFREE, QTNFree(), QueryItem::type, and QTNode::valnode.

tsquery_numnode()

Datum tsquery_numnode

(

PG_FUNCTION_ARGS

)

Definition at line 23 of file tsquery_op.c.

{
    TSQuery     query = PG_GETARG_TSQUERY(0);
    int         nnode = query->size;
 
    PG_FREE_IF_COPY(query, 0);
    PG_RETURN_INT32(nnode);
}

References PG_FREE_IF_COPY, PG_GETARG_TSQUERY, PG_RETURN_INT32, and TSQueryData::size.

tsquery_or()

Datum tsquery_or

(

PG_FUNCTION_ARGS

)

Definition at line 84 of file tsquery_op.c.

{
    TSQuery     a = PG_GETARG_TSQUERY_COPY(0);
    TSQuery     b = PG_GETARG_TSQUERY_COPY(1);
    QTNode     *res;
    TSQuery     query;
 
    if (a->size == 0)
    {
        PG_FREE_IF_COPY(a, 1);
        PG_RETURN_POINTER(b);
    }
    else if (b->size == 0)
    {
        PG_FREE_IF_COPY(b, 1);
        PG_RETURN_POINTER(a);
    }
 
    res = join_tsqueries(a, b, OP_OR, 0);
 
    query = QTN2QT(res);
 
    QTNFree(res);
    PG_FREE_IF_COPY(a, 0);
    PG_FREE_IF_COPY(b, 1);
 
    PG_RETURN_TSQUERY(query);
}

References a, b, join_tsqueries(), OP_OR, PG_FREE_IF_COPY, PG_GETARG_TSQUERY_COPY, PG_RETURN_POINTER, PG_RETURN_TSQUERY, QTN2QT(), and QTNFree().

tsquery_phrase()

Datum tsquery_phrase

(

PG_FUNCTION_ARGS

)

Definition at line 150 of file tsquery_op.c.

{
    PG_RETURN_DATUM(DirectFunctionCall3(tsquery_phrase_distance,
                                        PG_GETARG_DATUM(0),
                                        PG_GETARG_DATUM(1),
                                        Int32GetDatum(1)));
}

References DirectFunctionCall3, Int32GetDatum(), PG_GETARG_DATUM, PG_RETURN_DATUM, and tsquery_phrase_distance().

tsquery_phrase_distance()

Datum tsquery_phrase_distance

(

PG_FUNCTION_ARGS

)

Definition at line 114 of file tsquery_op.c.

{
    TSQuery     a = PG_GETARG_TSQUERY_COPY(0);
    TSQuery     b = PG_GETARG_TSQUERY_COPY(1);
    QTNode     *res;
    TSQuery     query;
    int32       distance = PG_GETARG_INT32(2);
 
    if (distance < 0 || distance > MAXENTRYPOS)
        ereport(ERROR,
                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                 errmsg("distance in phrase operator must be an integer value between zero and %d inclusive",
                        MAXENTRYPOS)));
    if (a->size == 0)
    {
        PG_FREE_IF_COPY(a, 1);
        PG_RETURN_POINTER(b);
    }
    else if (b->size == 0)
    {
        PG_FREE_IF_COPY(b, 1);
        PG_RETURN_POINTER(a);
    }
 
    res = join_tsqueries(a, b, OP_PHRASE, (uint16) distance);
 
    query = QTN2QT(res);
 
    QTNFree(res);
    PG_FREE_IF_COPY(a, 0);
    PG_FREE_IF_COPY(b, 1);
 
    PG_RETURN_TSQUERY(query);
}

References a, b, ereport, errcode(), errmsg(), ERROR, join_tsqueries(), MAXENTRYPOS, OP_PHRASE, PG_FREE_IF_COPY, PG_GETARG_INT32, PG_GETARG_TSQUERY_COPY, PG_RETURN_POINTER, PG_RETURN_TSQUERY, QTN2QT(), and QTNFree().

Referenced by tsquery_phrase().