1996-07-09 08:22:35 +02:00
|
|
|
/*-------------------------------------------------------------------------
|
|
|
|
*
|
1999-02-14 00:22:53 +01:00
|
|
|
* pg_aggregate.c
|
1997-09-07 07:04:48 +02:00
|
|
|
* routines to support manipulation of the pg_aggregate relation
|
1996-07-09 08:22:35 +02:00
|
|
|
*
|
2001-01-24 20:43:33 +01:00
|
|
|
* Portions Copyright (c) 1996-2001, PostgreSQL Global Development Group
|
2000-01-26 06:58:53 +01:00
|
|
|
* Portions Copyright (c) 1994, Regents of the University of California
|
1996-07-09 08:22:35 +02:00
|
|
|
*
|
|
|
|
*
|
|
|
|
* IDENTIFICATION
|
2002-04-11 22:00:18 +02:00
|
|
|
* $Header: /cvsroot/pgsql/src/backend/catalog/pg_aggregate.c,v 1.44 2002/04/11 19:59:57 tgl Exp $
|
1996-07-09 08:22:35 +02:00
|
|
|
*
|
|
|
|
*-------------------------------------------------------------------------
|
|
|
|
*/
|
1999-07-16 01:04:24 +02:00
|
|
|
#include "postgres.h"
|
1996-07-09 08:22:35 +02:00
|
|
|
|
1999-07-16 01:04:24 +02:00
|
|
|
#include "access/heapam.h"
|
|
|
|
#include "catalog/catname.h"
|
1999-11-22 18:56:41 +01:00
|
|
|
#include "catalog/indexing.h"
|
2002-04-09 22:35:55 +02:00
|
|
|
#include "catalog/namespace.h"
|
1999-07-16 07:00:38 +02:00
|
|
|
#include "catalog/pg_aggregate.h"
|
2002-04-11 22:00:18 +02:00
|
|
|
#include "catalog/pg_language.h"
|
1999-07-16 01:04:24 +02:00
|
|
|
#include "catalog/pg_proc.h"
|
2002-04-11 22:00:18 +02:00
|
|
|
#include "optimizer/cost.h"
|
2000-07-17 05:05:41 +02:00
|
|
|
#include "parser/parse_coerce.h"
|
|
|
|
#include "parser/parse_func.h"
|
2002-03-29 20:06:29 +01:00
|
|
|
#include "parser/parse_type.h"
|
1999-07-16 07:00:38 +02:00
|
|
|
#include "utils/builtins.h"
|
|
|
|
#include "utils/syscache.h"
|
1996-11-04 00:27:08 +01:00
|
|
|
|
2002-04-11 22:00:18 +02:00
|
|
|
|
2002-03-29 20:06:29 +01:00
|
|
|
/*
|
1996-07-09 08:22:35 +02:00
|
|
|
* AggregateCreate
|
|
|
|
*/
|
|
|
|
void
|
2002-03-29 20:06:29 +01:00
|
|
|
AggregateCreate(const char *aggName,
|
|
|
|
Oid aggNamespace,
|
2002-04-09 22:35:55 +02:00
|
|
|
List *aggtransfnName,
|
|
|
|
List *aggfinalfnName,
|
2002-03-29 20:06:29 +01:00
|
|
|
Oid aggBaseType,
|
|
|
|
Oid aggTransType,
|
|
|
|
const char *agginitval)
|
1996-07-09 08:22:35 +02:00
|
|
|
{
|
1997-09-08 04:41:22 +02:00
|
|
|
Relation aggdesc;
|
|
|
|
HeapTuple tup;
|
|
|
|
char nulls[Natts_pg_aggregate];
|
|
|
|
Datum values[Natts_pg_aggregate];
|
|
|
|
Form_pg_proc proc;
|
2000-07-17 05:05:41 +02:00
|
|
|
Oid transfn;
|
2001-03-22 05:01:46 +01:00
|
|
|
Oid finalfn = InvalidOid; /* can be omitted */
|
2000-07-17 05:05:41 +02:00
|
|
|
Oid finaltype;
|
2000-01-10 18:14:46 +01:00
|
|
|
Oid fnArgs[FUNC_MAX_ARGS];
|
2000-07-17 05:05:41 +02:00
|
|
|
int nargs;
|
2002-04-11 22:00:18 +02:00
|
|
|
Oid procOid;
|
1997-09-08 04:41:22 +02:00
|
|
|
TupleDesc tupDesc;
|
2000-07-17 05:05:41 +02:00
|
|
|
int i;
|
1997-09-07 07:04:48 +02:00
|
|
|
|
|
|
|
/* sanity checks */
|
|
|
|
if (!aggName)
|
2001-08-10 17:49:39 +02:00
|
|
|
elog(ERROR, "no aggregate name supplied");
|
1997-09-07 07:04:48 +02:00
|
|
|
|
2000-07-17 05:05:41 +02:00
|
|
|
if (!aggtransfnName)
|
2001-08-10 17:49:39 +02:00
|
|
|
elog(ERROR, "aggregate must have a transition function");
|
2000-03-26 21:43:58 +02:00
|
|
|
|
2000-07-17 05:05:41 +02:00
|
|
|
/* handle transfn */
|
2002-03-29 20:06:29 +01:00
|
|
|
MemSet(fnArgs, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
fnArgs[0] = aggTransType;
|
|
|
|
if (OidIsValid(aggBaseType))
|
1997-09-07 07:04:48 +02:00
|
|
|
{
|
2002-03-29 20:06:29 +01:00
|
|
|
fnArgs[1] = aggBaseType;
|
2000-07-17 05:05:41 +02:00
|
|
|
nargs = 2;
|
1997-09-07 07:04:48 +02:00
|
|
|
}
|
2000-07-17 05:05:41 +02:00
|
|
|
else
|
|
|
|
nargs = 1;
|
2002-04-09 22:35:55 +02:00
|
|
|
transfn = LookupFuncName(aggtransfnName, nargs, fnArgs);
|
|
|
|
if (!OidIsValid(transfn))
|
|
|
|
func_error("AggregateCreate", aggtransfnName, nargs, fnArgs, NULL);
|
|
|
|
tup = SearchSysCache(PROCOID,
|
|
|
|
ObjectIdGetDatum(transfn),
|
|
|
|
0, 0, 0);
|
2000-07-17 05:05:41 +02:00
|
|
|
if (!HeapTupleIsValid(tup))
|
|
|
|
func_error("AggregateCreate", aggtransfnName, nargs, fnArgs, NULL);
|
|
|
|
proc = (Form_pg_proc) GETSTRUCT(tup);
|
2002-03-29 20:06:29 +01:00
|
|
|
if (proc->prorettype != aggTransType)
|
2001-08-10 17:49:39 +02:00
|
|
|
elog(ERROR, "return type of transition function %s is not %s",
|
2002-04-09 22:35:55 +02:00
|
|
|
NameListToString(aggtransfnName), typeidTypeName(aggTransType));
|
2001-03-22 05:01:46 +01:00
|
|
|
|
2000-07-17 05:05:41 +02:00
|
|
|
/*
|
2001-03-22 05:01:46 +01:00
|
|
|
* If the transfn is strict and the initval is NULL, make sure input
|
2002-03-20 20:45:13 +01:00
|
|
|
* type and transtype are the same (or at least binary-compatible),
|
2001-03-22 05:01:46 +01:00
|
|
|
* so that it's OK to use the first input value as the initial
|
|
|
|
* transValue.
|
2000-07-17 05:05:41 +02:00
|
|
|
*/
|
2000-11-16 23:30:52 +01:00
|
|
|
if (proc->proisstrict && agginitval == NULL)
|
2000-07-17 05:05:41 +02:00
|
|
|
{
|
2002-03-29 20:06:29 +01:00
|
|
|
if (!IsBinaryCompatible(aggBaseType, aggTransType))
|
2001-08-10 17:49:39 +02:00
|
|
|
elog(ERROR, "must not omit initval when transfn is strict and transtype is not compatible with input type");
|
1997-09-07 07:04:48 +02:00
|
|
|
}
|
2000-11-16 23:30:52 +01:00
|
|
|
ReleaseSysCache(tup);
|
1997-09-07 07:04:48 +02:00
|
|
|
|
2000-07-17 05:05:41 +02:00
|
|
|
/* handle finalfn, if supplied */
|
1997-09-07 07:04:48 +02:00
|
|
|
if (aggfinalfnName)
|
|
|
|
{
|
2002-04-11 22:00:18 +02:00
|
|
|
MemSet(fnArgs, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
2002-03-29 20:06:29 +01:00
|
|
|
fnArgs[0] = aggTransType;
|
2002-04-09 22:35:55 +02:00
|
|
|
finalfn = LookupFuncName(aggfinalfnName, 1, fnArgs);
|
|
|
|
if (!OidIsValid(finalfn))
|
|
|
|
func_error("AggregateCreate", aggfinalfnName, 1, fnArgs, NULL);
|
|
|
|
tup = SearchSysCache(PROCOID,
|
|
|
|
ObjectIdGetDatum(finalfn),
|
|
|
|
0, 0, 0);
|
1997-09-07 07:04:48 +02:00
|
|
|
if (!HeapTupleIsValid(tup))
|
2000-07-17 05:05:41 +02:00
|
|
|
func_error("AggregateCreate", aggfinalfnName, 1, fnArgs, NULL);
|
1997-09-07 07:04:48 +02:00
|
|
|
proc = (Form_pg_proc) GETSTRUCT(tup);
|
2000-07-17 05:05:41 +02:00
|
|
|
finaltype = proc->prorettype;
|
2000-11-16 23:30:52 +01:00
|
|
|
ReleaseSysCache(tup);
|
1997-09-07 07:04:48 +02:00
|
|
|
}
|
2000-03-26 21:43:58 +02:00
|
|
|
else
|
|
|
|
{
|
2000-04-12 19:17:23 +02:00
|
|
|
/*
|
2000-07-17 05:05:41 +02:00
|
|
|
* If no finalfn, aggregate result type is type of the state value
|
2000-03-26 21:43:58 +02:00
|
|
|
*/
|
2002-03-29 20:06:29 +01:00
|
|
|
finaltype = aggTransType;
|
2000-03-26 21:43:58 +02:00
|
|
|
}
|
2000-07-17 05:05:41 +02:00
|
|
|
Assert(OidIsValid(finaltype));
|
1997-09-07 07:04:48 +02:00
|
|
|
|
2002-04-11 22:00:18 +02:00
|
|
|
/*
|
|
|
|
* Everything looks okay. Try to create the pg_proc entry for the
|
|
|
|
* aggregate. (This could fail if there's already a conflicting entry.)
|
|
|
|
*/
|
|
|
|
MemSet(fnArgs, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
fnArgs[0] = aggBaseType;
|
|
|
|
|
|
|
|
procOid = ProcedureCreate(aggName,
|
|
|
|
aggNamespace,
|
|
|
|
false, /* no replacement */
|
|
|
|
false, /* doesn't return a set */
|
|
|
|
finaltype, /* returnType */
|
|
|
|
INTERNALlanguageId, /* languageObjectId */
|
|
|
|
"aggregate_dummy", /* placeholder proc */
|
|
|
|
"-", /* probin */
|
|
|
|
true, /* isAgg */
|
|
|
|
true, /* (obsolete "trusted") */
|
|
|
|
false, /* isImplicit */
|
|
|
|
false, /* isStrict (not needed for agg) */
|
|
|
|
PROVOLATILE_IMMUTABLE, /* volatility (not needed for agg) */
|
|
|
|
BYTE_PCT, /* default cost values */
|
|
|
|
PERBYTE_CPU,
|
|
|
|
PERCALL_CPU,
|
|
|
|
OUTIN_RATIO,
|
|
|
|
1, /* parameterCount */
|
|
|
|
fnArgs); /* parameterTypes */
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Okay to create the pg_aggregate entry.
|
|
|
|
*/
|
|
|
|
|
1997-09-07 07:04:48 +02:00
|
|
|
/* initialize nulls and values */
|
|
|
|
for (i = 0; i < Natts_pg_aggregate; i++)
|
|
|
|
{
|
|
|
|
nulls[i] = ' ';
|
|
|
|
values[i] = (Datum) NULL;
|
|
|
|
}
|
2002-04-11 22:00:18 +02:00
|
|
|
values[Anum_pg_aggregate_aggfnoid - 1] = ObjectIdGetDatum(procOid);
|
2000-07-17 05:05:41 +02:00
|
|
|
values[Anum_pg_aggregate_aggtransfn - 1] = ObjectIdGetDatum(transfn);
|
|
|
|
values[Anum_pg_aggregate_aggfinalfn - 1] = ObjectIdGetDatum(finalfn);
|
2002-03-29 20:06:29 +01:00
|
|
|
values[Anum_pg_aggregate_aggtranstype - 1] = ObjectIdGetDatum(aggTransType);
|
2000-07-17 05:05:41 +02:00
|
|
|
if (agginitval)
|
|
|
|
values[Anum_pg_aggregate_agginitval - 1] =
|
|
|
|
DirectFunctionCall1(textin, CStringGetDatum(agginitval));
|
1997-09-07 07:04:48 +02:00
|
|
|
else
|
2000-07-17 05:05:41 +02:00
|
|
|
nulls[Anum_pg_aggregate_agginitval - 1] = 'n';
|
1997-09-07 07:04:48 +02:00
|
|
|
|
1999-09-18 21:08:25 +02:00
|
|
|
aggdesc = heap_openr(AggregateRelationName, RowExclusiveLock);
|
1997-09-07 07:04:48 +02:00
|
|
|
tupDesc = aggdesc->rd_att;
|
2002-04-11 22:00:18 +02:00
|
|
|
|
|
|
|
tup = heap_formtuple(tupDesc, values, nulls);
|
|
|
|
heap_insert(aggdesc, tup);
|
1999-11-22 18:56:41 +01:00
|
|
|
|
|
|
|
if (RelationGetForm(aggdesc)->relhasindex)
|
|
|
|
{
|
|
|
|
Relation idescs[Num_pg_aggregate_indices];
|
|
|
|
|
|
|
|
CatalogOpenIndices(Num_pg_aggregate_indices, Name_pg_aggregate_indices, idescs);
|
|
|
|
CatalogIndexInsert(idescs, Num_pg_aggregate_indices, aggdesc, tup);
|
|
|
|
CatalogCloseIndices(Num_pg_aggregate_indices, idescs);
|
|
|
|
}
|
|
|
|
|
1999-09-18 21:08:25 +02:00
|
|
|
heap_close(aggdesc, RowExclusiveLock);
|
1996-07-09 08:22:35 +02:00
|
|
|
}
|