mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-08-31 14:57:22 +02:00
We used to strategically place newlines after some function call left parentheses to make pgindent move the argument list a few chars to the left, so that the whole line would fit under 80 chars. However, pgindent no longer does that, so the newlines just made the code vertically longer for no reason. Remove those newlines, and reflow some of those lines for some extra naturality. Reviewed-by: Michael Paquier, Tom Lane Discussion: https://postgr.es/m/20200129200401.GA6303@alvherre.pgsql
335 lines
8.3 KiB
C
335 lines
8.3 KiB
C
/*
|
|
* contrib/btree_gist/btree_time.c
|
|
*/
|
|
#include "postgres.h"
|
|
|
|
#include "btree_gist.h"
|
|
#include "btree_utils_num.h"
|
|
#include "utils/builtins.h"
|
|
#include "utils/date.h"
|
|
#include "utils/timestamp.h"
|
|
|
|
typedef struct
|
|
{
|
|
TimeADT lower;
|
|
TimeADT upper;
|
|
} timeKEY;
|
|
|
|
/*
|
|
** time ops
|
|
*/
|
|
PG_FUNCTION_INFO_V1(gbt_time_compress);
|
|
PG_FUNCTION_INFO_V1(gbt_timetz_compress);
|
|
PG_FUNCTION_INFO_V1(gbt_time_fetch);
|
|
PG_FUNCTION_INFO_V1(gbt_time_union);
|
|
PG_FUNCTION_INFO_V1(gbt_time_picksplit);
|
|
PG_FUNCTION_INFO_V1(gbt_time_consistent);
|
|
PG_FUNCTION_INFO_V1(gbt_time_distance);
|
|
PG_FUNCTION_INFO_V1(gbt_timetz_consistent);
|
|
PG_FUNCTION_INFO_V1(gbt_time_penalty);
|
|
PG_FUNCTION_INFO_V1(gbt_time_same);
|
|
|
|
|
|
#ifdef USE_FLOAT8_BYVAL
|
|
#define TimeADTGetDatumFast(X) TimeADTGetDatum(X)
|
|
#else
|
|
#define TimeADTGetDatumFast(X) PointerGetDatum(&(X))
|
|
#endif
|
|
|
|
|
|
static bool
|
|
gbt_timegt(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
const TimeADT *aa = (const TimeADT *) a;
|
|
const TimeADT *bb = (const TimeADT *) b;
|
|
|
|
return DatumGetBool(DirectFunctionCall2(time_gt,
|
|
TimeADTGetDatumFast(*aa),
|
|
TimeADTGetDatumFast(*bb)));
|
|
}
|
|
|
|
static bool
|
|
gbt_timege(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
const TimeADT *aa = (const TimeADT *) a;
|
|
const TimeADT *bb = (const TimeADT *) b;
|
|
|
|
return DatumGetBool(DirectFunctionCall2(time_ge,
|
|
TimeADTGetDatumFast(*aa),
|
|
TimeADTGetDatumFast(*bb)));
|
|
}
|
|
|
|
static bool
|
|
gbt_timeeq(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
const TimeADT *aa = (const TimeADT *) a;
|
|
const TimeADT *bb = (const TimeADT *) b;
|
|
|
|
return DatumGetBool(DirectFunctionCall2(time_eq,
|
|
TimeADTGetDatumFast(*aa),
|
|
TimeADTGetDatumFast(*bb)));
|
|
}
|
|
|
|
static bool
|
|
gbt_timele(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
const TimeADT *aa = (const TimeADT *) a;
|
|
const TimeADT *bb = (const TimeADT *) b;
|
|
|
|
return DatumGetBool(DirectFunctionCall2(time_le,
|
|
TimeADTGetDatumFast(*aa),
|
|
TimeADTGetDatumFast(*bb)));
|
|
}
|
|
|
|
static bool
|
|
gbt_timelt(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
const TimeADT *aa = (const TimeADT *) a;
|
|
const TimeADT *bb = (const TimeADT *) b;
|
|
|
|
return DatumGetBool(DirectFunctionCall2(time_lt,
|
|
TimeADTGetDatumFast(*aa),
|
|
TimeADTGetDatumFast(*bb)));
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
gbt_timekey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
timeKEY *ia = (timeKEY *) (((const Nsrt *) a)->t);
|
|
timeKEY *ib = (timeKEY *) (((const Nsrt *) b)->t);
|
|
int res;
|
|
|
|
res = DatumGetInt32(DirectFunctionCall2(time_cmp, TimeADTGetDatumFast(ia->lower), TimeADTGetDatumFast(ib->lower)));
|
|
if (res == 0)
|
|
return DatumGetInt32(DirectFunctionCall2(time_cmp, TimeADTGetDatumFast(ia->upper), TimeADTGetDatumFast(ib->upper)));
|
|
|
|
return res;
|
|
}
|
|
|
|
static float8
|
|
gbt_time_dist(const void *a, const void *b, FmgrInfo *flinfo)
|
|
{
|
|
const TimeADT *aa = (const TimeADT *) a;
|
|
const TimeADT *bb = (const TimeADT *) b;
|
|
Interval *i;
|
|
|
|
i = DatumGetIntervalP(DirectFunctionCall2(time_mi_time,
|
|
TimeADTGetDatumFast(*aa),
|
|
TimeADTGetDatumFast(*bb)));
|
|
return (float8) Abs(INTERVAL_TO_SEC(i));
|
|
}
|
|
|
|
|
|
static const gbtree_ninfo tinfo =
|
|
{
|
|
gbt_t_time,
|
|
sizeof(TimeADT),
|
|
16, /* sizeof(gbtreekey16) */
|
|
gbt_timegt,
|
|
gbt_timege,
|
|
gbt_timeeq,
|
|
gbt_timele,
|
|
gbt_timelt,
|
|
gbt_timekey_cmp,
|
|
gbt_time_dist
|
|
};
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(time_dist);
|
|
Datum
|
|
time_dist(PG_FUNCTION_ARGS)
|
|
{
|
|
Datum diff = DirectFunctionCall2(time_mi_time,
|
|
PG_GETARG_DATUM(0),
|
|
PG_GETARG_DATUM(1));
|
|
|
|
PG_RETURN_INTERVAL_P(abs_interval(DatumGetIntervalP(diff)));
|
|
}
|
|
|
|
|
|
/**************************************************
|
|
* time ops
|
|
**************************************************/
|
|
|
|
|
|
|
|
Datum
|
|
gbt_time_compress(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
|
|
PG_RETURN_POINTER(gbt_num_compress(entry, &tinfo));
|
|
}
|
|
|
|
|
|
Datum
|
|
gbt_timetz_compress(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
GISTENTRY *retval;
|
|
|
|
if (entry->leafkey)
|
|
{
|
|
timeKEY *r = (timeKEY *) palloc(sizeof(timeKEY));
|
|
TimeTzADT *tz = DatumGetTimeTzADTP(entry->key);
|
|
TimeADT tmp;
|
|
|
|
retval = palloc(sizeof(GISTENTRY));
|
|
|
|
/* We are using the time + zone only to compress */
|
|
tmp = tz->time + (tz->zone * INT64CONST(1000000));
|
|
r->lower = r->upper = tmp;
|
|
gistentryinit(*retval, PointerGetDatum(r),
|
|
entry->rel, entry->page,
|
|
entry->offset, false);
|
|
}
|
|
else
|
|
retval = entry;
|
|
PG_RETURN_POINTER(retval);
|
|
}
|
|
|
|
Datum
|
|
gbt_time_fetch(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
|
|
PG_RETURN_POINTER(gbt_num_fetch(entry, &tinfo));
|
|
}
|
|
|
|
Datum
|
|
gbt_time_consistent(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
TimeADT query = PG_GETARG_TIMEADT(1);
|
|
StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
|
|
|
|
/* Oid subtype = PG_GETARG_OID(3); */
|
|
bool *recheck = (bool *) PG_GETARG_POINTER(4);
|
|
timeKEY *kkk = (timeKEY *) DatumGetPointer(entry->key);
|
|
GBT_NUMKEY_R key;
|
|
|
|
/* All cases served by this function are exact */
|
|
*recheck = false;
|
|
|
|
key.lower = (GBT_NUMKEY *) &kkk->lower;
|
|
key.upper = (GBT_NUMKEY *) &kkk->upper;
|
|
|
|
PG_RETURN_BOOL(gbt_num_consistent(&key, (void *) &query, &strategy,
|
|
GIST_LEAF(entry), &tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
Datum
|
|
gbt_time_distance(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
TimeADT query = PG_GETARG_TIMEADT(1);
|
|
|
|
/* Oid subtype = PG_GETARG_OID(3); */
|
|
timeKEY *kkk = (timeKEY *) DatumGetPointer(entry->key);
|
|
GBT_NUMKEY_R key;
|
|
|
|
key.lower = (GBT_NUMKEY *) &kkk->lower;
|
|
key.upper = (GBT_NUMKEY *) &kkk->upper;
|
|
|
|
PG_RETURN_FLOAT8(gbt_num_distance(&key, (void *) &query, GIST_LEAF(entry),
|
|
&tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
Datum
|
|
gbt_timetz_consistent(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
TimeTzADT *query = PG_GETARG_TIMETZADT_P(1);
|
|
StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
|
|
|
|
/* Oid subtype = PG_GETARG_OID(3); */
|
|
bool *recheck = (bool *) PG_GETARG_POINTER(4);
|
|
timeKEY *kkk = (timeKEY *) DatumGetPointer(entry->key);
|
|
TimeADT qqq;
|
|
GBT_NUMKEY_R key;
|
|
|
|
/* All cases served by this function are inexact */
|
|
*recheck = true;
|
|
|
|
qqq = query->time + (query->zone * INT64CONST(1000000));
|
|
|
|
key.lower = (GBT_NUMKEY *) &kkk->lower;
|
|
key.upper = (GBT_NUMKEY *) &kkk->upper;
|
|
|
|
PG_RETURN_BOOL(gbt_num_consistent(&key, (void *) &qqq, &strategy,
|
|
GIST_LEAF(entry), &tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
|
|
Datum
|
|
gbt_time_union(PG_FUNCTION_ARGS)
|
|
{
|
|
GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
|
|
void *out = palloc(sizeof(timeKEY));
|
|
|
|
*(int *) PG_GETARG_POINTER(1) = sizeof(timeKEY);
|
|
PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
|
|
Datum
|
|
gbt_time_penalty(PG_FUNCTION_ARGS)
|
|
{
|
|
timeKEY *origentry = (timeKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
|
|
timeKEY *newentry = (timeKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
|
|
float *result = (float *) PG_GETARG_POINTER(2);
|
|
Interval *intr;
|
|
double res;
|
|
double res2;
|
|
|
|
intr = DatumGetIntervalP(DirectFunctionCall2(time_mi_time,
|
|
TimeADTGetDatumFast(newentry->upper),
|
|
TimeADTGetDatumFast(origentry->upper)));
|
|
res = INTERVAL_TO_SEC(intr);
|
|
res = Max(res, 0);
|
|
|
|
intr = DatumGetIntervalP(DirectFunctionCall2(time_mi_time,
|
|
TimeADTGetDatumFast(origentry->lower),
|
|
TimeADTGetDatumFast(newentry->lower)));
|
|
res2 = INTERVAL_TO_SEC(intr);
|
|
res2 = Max(res2, 0);
|
|
|
|
res += res2;
|
|
|
|
*result = 0.0;
|
|
|
|
if (res > 0)
|
|
{
|
|
intr = DatumGetIntervalP(DirectFunctionCall2(time_mi_time,
|
|
TimeADTGetDatumFast(origentry->upper),
|
|
TimeADTGetDatumFast(origentry->lower)));
|
|
*result += FLT_MIN;
|
|
*result += (float) (res / (res + INTERVAL_TO_SEC(intr)));
|
|
*result *= (FLT_MAX / (((GISTENTRY *) PG_GETARG_POINTER(0))->rel->rd_att->natts + 1));
|
|
}
|
|
|
|
PG_RETURN_POINTER(result);
|
|
}
|
|
|
|
|
|
Datum
|
|
gbt_time_picksplit(PG_FUNCTION_ARGS)
|
|
{
|
|
PG_RETURN_POINTER(gbt_num_picksplit((GistEntryVector *) PG_GETARG_POINTER(0),
|
|
(GIST_SPLITVEC *) PG_GETARG_POINTER(1),
|
|
&tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
Datum
|
|
gbt_time_same(PG_FUNCTION_ARGS)
|
|
{
|
|
timeKEY *b1 = (timeKEY *) PG_GETARG_POINTER(0);
|
|
timeKEY *b2 = (timeKEY *) PG_GETARG_POINTER(1);
|
|
bool *result = (bool *) PG_GETARG_POINTER(2);
|
|
|
|
*result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
|
|
PG_RETURN_POINTER(result);
|
|
}
|