mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-03 08:00:21 +08:00
402 lines
8.4 KiB
C
402 lines
8.4 KiB
C
/*
|
|
* array_iterator.c --
|
|
*
|
|
* This file defines a new class of operators which take an
|
|
* array and a scalar value, iterate a scalar operator over the
|
|
* elements of the array and the value and compute a result as
|
|
* the logical OR or AND of the iteration results.
|
|
*
|
|
* Copyright (C) 1999, Massimo Dal Zotto <dz@cs.unitn.it>
|
|
* ported to postgreSQL 6.3.2,added oid_functions, 18.1.1999,
|
|
* Tobias Gabele <gabele@wiz.uni-kassel.de>
|
|
*
|
|
* This software is distributed under the GNU General Public License
|
|
* either version 2, or (at your option) any later version.
|
|
*/
|
|
|
|
#include <ctype.h>
|
|
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <string.h>
|
|
|
|
#include "postgres.h"
|
|
#include "miscadmin.h"
|
|
#include "access/xact.h"
|
|
#include "fmgr.h"
|
|
#include "catalog/pg_type.h"
|
|
#include "utils/array.h"
|
|
#include "utils/builtins.h"
|
|
#include "utils/memutils.h"
|
|
#include "utils/syscache.h"
|
|
|
|
#include "array_iterator.h"
|
|
|
|
static int32
|
|
array_iterator(Oid elemtype, Oid proc, int and, ArrayType *array, Datum value)
|
|
{
|
|
HeapTuple typ_tuple;
|
|
Form_pg_type typ_struct;
|
|
bool typbyval;
|
|
int typlen;
|
|
int nitems,
|
|
i;
|
|
Datum result;
|
|
int ndim,
|
|
*dim;
|
|
char *p;
|
|
FmgrInfo finfo;
|
|
|
|
/* Sanity checks */
|
|
if ((array == (ArrayType *) NULL)
|
|
|| (ARR_IS_LO(array) == true))
|
|
{
|
|
/* elog(NOTICE, "array_iterator: array is null"); */
|
|
return (0);
|
|
}
|
|
ndim = ARR_NDIM(array);
|
|
dim = ARR_DIMS(array);
|
|
nitems = getNitems(ndim, dim);
|
|
if (nitems == 0)
|
|
{
|
|
/* elog(NOTICE, "array_iterator: nitems = 0"); */
|
|
return (0);
|
|
}
|
|
|
|
/* Lookup element type information */
|
|
typ_tuple = SearchSysCacheTuple(TYPEOID, ObjectIdGetDatum(elemtype),
|
|
0, 0, 0);
|
|
if (!HeapTupleIsValid(typ_tuple))
|
|
{
|
|
elog(ERROR, "array_iterator: cache lookup failed for type %u", elemtype);
|
|
return 0;
|
|
}
|
|
typ_struct = (Form_pg_type) GETSTRUCT(typ_tuple);
|
|
typlen = typ_struct->typlen;
|
|
typbyval = typ_struct->typbyval;
|
|
|
|
/* Lookup the function entry point */
|
|
fmgr_info(proc, &finfo);
|
|
if (finfo.fn_nargs != 2)
|
|
{
|
|
elog(ERROR, "array_iterator: proc %u does not take 2 args", proc);
|
|
return (0);
|
|
}
|
|
|
|
/* Scan the array and apply the operator to each element */
|
|
result = BoolGetDatum(false);
|
|
p = ARR_DATA_PTR(array);
|
|
for (i = 0; i < nitems; i++)
|
|
{
|
|
if (typbyval)
|
|
{
|
|
switch (typlen)
|
|
{
|
|
case 1:
|
|
result = FunctionCall2(&finfo,
|
|
CharGetDatum(*p),
|
|
value);
|
|
break;
|
|
case 2:
|
|
result = FunctionCall2(&finfo,
|
|
Int16GetDatum(*(int16 *) p),
|
|
value);
|
|
break;
|
|
case 3:
|
|
case 4:
|
|
result = FunctionCall2(&finfo,
|
|
Int32GetDatum(*(int32 *) p),
|
|
value);
|
|
break;
|
|
}
|
|
p += typlen;
|
|
}
|
|
else
|
|
{
|
|
result = FunctionCall2(&finfo, PointerGetDatum(p), value);
|
|
if (typlen > 0)
|
|
p += typlen;
|
|
else
|
|
p += INTALIGN(*(int32 *) p);
|
|
}
|
|
if (DatumGetBool(result))
|
|
{
|
|
if (!and)
|
|
return (1);
|
|
}
|
|
else
|
|
{
|
|
if (and)
|
|
return (0);
|
|
}
|
|
}
|
|
|
|
if (and && DatumGetBool(result))
|
|
return (1);
|
|
else
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* Iterator functions for type _text
|
|
*/
|
|
|
|
int32
|
|
array_texteq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 25, /* text */
|
|
(Oid) 67, /* texteq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_texteq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 25, /* text */
|
|
(Oid) 67, /* texteq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_textregexeq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 25, /* text */
|
|
(Oid) 1254, /* textregexeq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_textregexeq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 25, /* text */
|
|
(Oid) 1254, /* textregexeq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
/*
|
|
* Iterator functions for type _varchar. Note that the regexp
|
|
* operators take the second argument of type text.
|
|
*/
|
|
|
|
int32
|
|
array_varchareq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1043, /* varchar */
|
|
(Oid) 1070, /* varchareq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_varchareq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1043, /* varchar */
|
|
(Oid) 1070, /* varchareq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_varcharregexeq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1043, /* varchar */
|
|
(Oid) 1254, /* textregexeq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_varcharregexeq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1043, /* varchar */
|
|
(Oid) 1254, /* textregexeq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
/*
|
|
* Iterator functions for type _bpchar. Note that the regexp
|
|
* operators take the second argument of type text.
|
|
*/
|
|
|
|
int32
|
|
array_bpchareq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1042, /* bpchar */
|
|
(Oid) 1048, /* bpchareq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_bpchareq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1042, /* bpchar */
|
|
(Oid) 1048, /* bpchareq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_bpcharregexeq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1042, /* bpchar */
|
|
(Oid) 1254, /* textregexeq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_bpcharregexeq(ArrayType *array, char *value)
|
|
{
|
|
return array_iterator((Oid) 1042, /* bpchar */
|
|
(Oid) 1254, /* textregexeq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
/*
|
|
* Iterator functions for type _int4
|
|
*/
|
|
|
|
int32
|
|
array_int4eq(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 65, /* int4eq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_int4eq(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 65, /* int4eq */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_int4ne(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 144, /* int4ne */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_int4ne(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 144, /* int4ne */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_int4gt(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 147, /* int4gt */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_int4gt(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 147, /* int4gt */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_int4ge(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 150, /* int4ge */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_int4ge(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 150, /* int4ge */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_int4lt(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 66, /* int4lt */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_int4lt(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 66, /* int4lt */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_int4le(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 149, /* int4le */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_int4le(ArrayType *array, int4 value)
|
|
{
|
|
return array_iterator((Oid) 23, /* int4 */
|
|
(Oid) 149, /* int4le */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
/* new tobias gabele 1999 */
|
|
|
|
int32
|
|
array_oideq(ArrayType *array, Oid value)
|
|
{
|
|
return array_iterator((Oid) 26, /* oid */
|
|
(Oid) 184, /* oideq */
|
|
0, /* logical or */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
int32
|
|
array_all_oidne(ArrayType *array, Oid value)
|
|
{
|
|
return array_iterator((Oid) 26, /* int4 */
|
|
(Oid) 185, /* oidne */
|
|
1, /* logical and */
|
|
array, (Datum) value);
|
|
}
|
|
|
|
/* end of file */
|
|
|
|
/*
|
|
* Local Variables:
|
|
* tab-width: 4
|
|
* c-indent-level: 4
|
|
* c-basic-offset: 4
|
|
* End:
|
|
*/
|