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
|
|
|
*
|
2002-06-20 22:29:54 +02:00
|
|
|
* Portions Copyright (c) 1996-2002, 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-09-04 22:31:48 +02:00
|
|
|
* $Header: /cvsroot/pgsql/src/backend/catalog/pg_aggregate.c,v 1.55 2002/09/04 20:31:14 momjian 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"
|
2002-07-17 00:12:20 +02:00
|
|
|
#include "catalog/dependency.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"
|
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;
|
2002-09-04 22:31:48 +02:00
|
|
|
ObjectAddress myself,
|
|
|
|
referenced;
|
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;
|
2002-08-22 02:01:51 +02:00
|
|
|
if (aggBaseType == ANYOID)
|
|
|
|
nargs = 1;
|
|
|
|
else
|
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
|
|
|
}
|
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-09-04 22:31:48 +02:00
|
|
|
NameListToString(aggtransfnName), format_type_be(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-09-04 22:31:48 +02:00
|
|
|
* type and transtype are the same (or at least binary-compatible), so
|
|
|
|
* that it's OK to use the first input value as the initial
|
2001-03-22 05:01:46 +01:00
|
|
|
* 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
|
2002-09-04 22:31:48 +02:00
|
|
|
* aggregate. (This could fail if there's already a conflicting
|
|
|
|
* entry.)
|
2002-04-11 22:00:18 +02:00
|
|
|
*/
|
|
|
|
MemSet(fnArgs, 0, FUNC_MAX_ARGS * sizeof(Oid));
|
|
|
|
fnArgs[0] = aggBaseType;
|
|
|
|
|
|
|
|
procOid = ProcedureCreate(aggName,
|
|
|
|
aggNamespace,
|
2002-09-04 22:31:48 +02:00
|
|
|
false, /* no replacement */
|
|
|
|
false, /* doesn't return a set */
|
|
|
|
finaltype, /* returnType */
|
|
|
|
INTERNALlanguageId, /* languageObjectId */
|
2002-05-22 19:21:02 +02:00
|
|
|
0,
|
2002-09-04 22:31:48 +02:00
|
|
|
"aggregate_dummy", /* placeholder proc */
|
|
|
|
"-", /* probin */
|
|
|
|
true, /* isAgg */
|
|
|
|
false, /* security invoker (currently not
|
|
|
|
* definable for agg) */
|
|
|
|
false, /* isStrict (not needed for agg) */
|
|
|
|
PROVOLATILE_IMMUTABLE, /* volatility (not
|
|
|
|
* needed for agg) */
|
|
|
|
1, /* parameterCount */
|
|
|
|
fnArgs); /* parameterTypes */
|
2002-04-11 22:00:18 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* 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);
|
2002-05-22 00:05:55 +02:00
|
|
|
simple_heap_insert(aggdesc, tup);
|
1999-11-22 18:56:41 +01:00
|
|
|
|
2002-08-05 05:29:17 +02:00
|
|
|
CatalogUpdateIndexes(aggdesc, tup);
|
1999-11-22 18:56:41 +01:00
|
|
|
|
1999-09-18 21:08:25 +02:00
|
|
|
heap_close(aggdesc, RowExclusiveLock);
|
2002-07-17 00:12:20 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Create dependencies for the aggregate (above and beyond those
|
|
|
|
* already made by ProcedureCreate). Note: we don't need an explicit
|
|
|
|
* dependency on aggTransType since we depend on it indirectly through
|
|
|
|
* transfn.
|
|
|
|
*/
|
|
|
|
myself.classId = RelOid_pg_proc;
|
|
|
|
myself.objectId = procOid;
|
|
|
|
myself.objectSubId = 0;
|
|
|
|
|
|
|
|
/* Depends on transition function */
|
|
|
|
referenced.classId = RelOid_pg_proc;
|
|
|
|
referenced.objectId = transfn;
|
|
|
|
referenced.objectSubId = 0;
|
|
|
|
recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
|
|
|
|
|
|
|
|
/* Depends on final function, if any */
|
|
|
|
if (OidIsValid(finalfn))
|
|
|
|
{
|
|
|
|
referenced.classId = RelOid_pg_proc;
|
|
|
|
referenced.objectId = finalfn;
|
|
|
|
referenced.objectSubId = 0;
|
|
|
|
recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
|
|
|
|
}
|
1996-07-09 08:22:35 +02:00
|
|
|
}
|