nbtcompare.c

Go to the documentation of this file.

/*-------------------------------------------------------------------------
 *
 * nbtcompare.c
 *    Comparison functions for btree access method.
 *
 * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
 * Portions Copyright (c) 1994, Regents of the University of California
 *
 *
 * IDENTIFICATION
 *    src/backend/access/nbtree/nbtcompare.c
 *
 * NOTES
 *
 *  These functions are stored in pg_amproc.  For each operator class
 *  defined on btrees, they compute
 *
 *              compare(a, b):
 *                      < 0 if a < b,
 *                      = 0 if a == b,
 *                      > 0 if a > b.
 *
 *  The result is always an int32 regardless of the input datatype.
 *
 *  Although any negative int32 is acceptable for reporting "<",
 *  and any positive int32 is acceptable for reporting ">", routines
 *  that work on 32-bit or wider datatypes can't just return "a - b".
 *  That could overflow and give the wrong answer.
 *
 *  NOTE: it is critical that the comparison function impose a total order
 *  on all non-NULL values of the data type, and that the datatype's
 *  boolean comparison operators (= < >= etc) yield results consistent
 *  with the comparison routine.  Otherwise bad behavior may ensue.
 *  (For example, the comparison operators must NOT punt when faced with
 *  NAN or other funny values; you must devise some collation sequence for
 *  all such values.)  If the datatype is not trivial, this is most
 *  reliably done by having the boolean operators invoke the same
 *  three-way comparison code that the btree function does.  Therefore,
 *  this file contains only btree support for "trivial" datatypes ---
 *  all others are in the /utils/adt/ files that implement their datatypes.
 *
 *  NOTE: these routines must not leak memory, since memory allocated
 *  during an index access won't be recovered till end of query.  This
 *  primarily affects comparison routines for toastable datatypes;
 *  they have to be careful to free any detoasted copy of an input datum.
 *
 *  NOTE: we used to forbid comparison functions from returning INT_MIN,
 *  but that proves to be too error-prone because some platforms' versions
 *  of memcmp() etc can return INT_MIN.  As a means of stress-testing
 *  callers, this file can be compiled with STRESS_SORT_INT_MIN defined
 *  to cause many of these functions to return INT_MIN or INT_MAX instead of
 *  their customary -1/+1.  For production, though, that's not a good idea
 *  since users or third-party code might expect the traditional results.
 *-------------------------------------------------------------------------
 */
#include "postgres.h"
 
#include <limits.h>
 
#include "utils/fmgrprotos.h"
#include "utils/sortsupport.h"
 
#ifdef STRESS_SORT_INT_MIN
#define A_LESS_THAN_B       INT_MIN
#define A_GREATER_THAN_B    INT_MAX
#else
#define A_LESS_THAN_B       (-1)
#define A_GREATER_THAN_B    1
#endif
 
 
Datum
btboolcmp(PG_FUNCTION_ARGS)
{
    bool        a = PG_GETARG_BOOL(0);
    bool        b = PG_GETARG_BOOL(1);
 
    PG_RETURN_INT32((int32) a - (int32) b);
}
 
Datum
btint2cmp(PG_FUNCTION_ARGS)
{
    int16       a = PG_GETARG_INT16(0);
    int16       b = PG_GETARG_INT16(1);
 
    PG_RETURN_INT32((int32) a - (int32) b);
}
 
static int
btint2fastcmp(Datum x, Datum y, SortSupport ssup)
{
    int16       a = DatumGetInt16(x);
    int16       b = DatumGetInt16(y);
 
    return (int) a - (int) b;
}
 
Datum
btint2sortsupport(PG_FUNCTION_ARGS)
{
    SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
 
    ssup->comparator = btint2fastcmp;
    PG_RETURN_VOID();
}
 
Datum
btint4cmp(PG_FUNCTION_ARGS)
{
    int32       a = PG_GETARG_INT32(0);
    int32       b = PG_GETARG_INT32(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btint4sortsupport(PG_FUNCTION_ARGS)
{
    SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
 
    ssup->comparator = ssup_datum_int32_cmp;
    PG_RETURN_VOID();
}
 
Datum
btint8cmp(PG_FUNCTION_ARGS)
{
    int64       a = PG_GETARG_INT64(0);
    int64       b = PG_GETARG_INT64(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
#if SIZEOF_DATUM < 8
static int
btint8fastcmp(Datum x, Datum y, SortSupport ssup)
{
    int64       a = DatumGetInt64(x);
    int64       b = DatumGetInt64(y);
 
    if (a > b)
        return A_GREATER_THAN_B;
    else if (a == b)
        return 0;
    else
        return A_LESS_THAN_B;
}
#endif
 
Datum
btint8sortsupport(PG_FUNCTION_ARGS)
{
    SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
 
#if SIZEOF_DATUM >= 8
    ssup->comparator = ssup_datum_signed_cmp;
#else
    ssup->comparator = btint8fastcmp;
#endif
    PG_RETURN_VOID();
}
 
Datum
btint48cmp(PG_FUNCTION_ARGS)
{
    int32       a = PG_GETARG_INT32(0);
    int64       b = PG_GETARG_INT64(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btint84cmp(PG_FUNCTION_ARGS)
{
    int64       a = PG_GETARG_INT64(0);
    int32       b = PG_GETARG_INT32(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btint24cmp(PG_FUNCTION_ARGS)
{
    int16       a = PG_GETARG_INT16(0);
    int32       b = PG_GETARG_INT32(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btint42cmp(PG_FUNCTION_ARGS)
{
    int32       a = PG_GETARG_INT32(0);
    int16       b = PG_GETARG_INT16(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btint28cmp(PG_FUNCTION_ARGS)
{
    int16       a = PG_GETARG_INT16(0);
    int64       b = PG_GETARG_INT64(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btint82cmp(PG_FUNCTION_ARGS)
{
    int64       a = PG_GETARG_INT64(0);
    int16       b = PG_GETARG_INT16(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
Datum
btoidcmp(PG_FUNCTION_ARGS)
{
    Oid         a = PG_GETARG_OID(0);
    Oid         b = PG_GETARG_OID(1);
 
    if (a > b)
        PG_RETURN_INT32(A_GREATER_THAN_B);
    else if (a == b)
        PG_RETURN_INT32(0);
    else
        PG_RETURN_INT32(A_LESS_THAN_B);
}
 
static int
btoidfastcmp(Datum x, Datum y, SortSupport ssup)
{
    Oid         a = DatumGetObjectId(x);
    Oid         b = DatumGetObjectId(y);
 
    if (a > b)
        return A_GREATER_THAN_B;
    else if (a == b)
        return 0;
    else
        return A_LESS_THAN_B;
}
 
Datum
btoidsortsupport(PG_FUNCTION_ARGS)
{
    SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
 
    ssup->comparator = btoidfastcmp;
    PG_RETURN_VOID();
}
 
Datum
btoidvectorcmp(PG_FUNCTION_ARGS)
{
    oidvector  *a = (oidvector *) PG_GETARG_POINTER(0);
    oidvector  *b = (oidvector *) PG_GETARG_POINTER(1);
    int         i;
 
    /* We arbitrarily choose to sort first by vector length */
    if (a->dim1 != b->dim1)
        PG_RETURN_INT32(a->dim1 - b->dim1);
 
    for (i = 0; i < a->dim1; i++)
    {
        if (a->values[i] != b->values[i])
        {
            if (a->values[i] > b->values[i])
                PG_RETURN_INT32(A_GREATER_THAN_B);
            else
                PG_RETURN_INT32(A_LESS_THAN_B);
        }
    }
    PG_RETURN_INT32(0);
}
 
Datum
btcharcmp(PG_FUNCTION_ARGS)
{
    char        a = PG_GETARG_CHAR(0);
    char        b = PG_GETARG_CHAR(1);
 
    /* Be careful to compare chars as unsigned */
    PG_RETURN_INT32((int32) ((uint8) a) - (int32) ((uint8) b));
}