Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Pass collations to functions in FunctionCallInfoData, not FmgrInfo.
authorTom Lane <tgl@sss.pgh.pa.us>
Tue, 12 Apr 2011 23:19:24 +0000 (19:19 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Tue, 12 Apr 2011 23:19:24 +0000 (19:19 -0400)
Since collation is effectively an argument, not a property of the function,
FmgrInfo is really the wrong place for it; and this becomes critical in
cases where a cached FmgrInfo is used for varying purposes that might need
different collation settings.  Fix by passing it in FunctionCallInfoData
instead.  In particular this allows a clean fix for bug #5970 (record_cmp
not working).  This requires touching a bit more code than the original
method, but nobody ever thought that collations would not be an invasive
patch...

49 files changed:
contrib/btree_gin/btree_gin.c
contrib/btree_gist/btree_text.c
src/backend/access/common/reloptions.c
src/backend/access/common/scankey.c
src/backend/access/gin/ginget.c
src/backend/access/gin/ginutil.c
src/backend/access/gist/gistget.c
src/backend/access/gist/gistscan.c
src/backend/access/hash/hashutil.c
src/backend/access/index/indexam.c
src/backend/access/nbtree/nbtinsert.c
src/backend/access/nbtree/nbtsearch.c
src/backend/access/nbtree/nbtsort.c
src/backend/access/nbtree/nbtutils.c
src/backend/commands/analyze.c
src/backend/commands/trigger.c
src/backend/commands/tsearchcmds.c
src/backend/executor/execGrouping.c
src/backend/executor/execQual.c
src/backend/executor/execUtils.c
src/backend/executor/functions.c
src/backend/executor/nodeAgg.c
src/backend/executor/nodeIndexscan.c
src/backend/executor/nodeMergeAppend.c
src/backend/executor/nodeMergejoin.c
src/backend/executor/nodeSubplan.c
src/backend/executor/nodeWindowAgg.c
src/backend/optimizer/path/indxpath.c
src/backend/tcop/fastpath.c
src/backend/tsearch/wparser.c
src/backend/utils/adt/arrayfuncs.c
src/backend/utils/adt/int.c
src/backend/utils/adt/like.c
src/backend/utils/adt/oid.c
src/backend/utils/adt/ri_triggers.c
src/backend/utils/adt/rowtypes.c
src/backend/utils/adt/selfuncs.c
src/backend/utils/cache/catcache.c
src/backend/utils/fmgr/README
src/backend/utils/fmgr/fmgr.c
src/backend/utils/sort/tuplesort.c
src/include/access/gin_private.h
src/include/access/skey.h
src/include/access/valid.h
src/include/fmgr.h
src/include/nodes/execnodes.h
src/include/utils/selfuncs.h
src/include/utils/tuplesort.h
src/pl/plpgsql/src/pl_comp.c

index 144f6db18466ae02c097bf84d2002104c3b666de..7ca153e0928930199154977deb8a29b8cfdc3da3 100644 (file)
@@ -121,7 +121,7 @@ gin_compare_prefix_##type(PG_FUNCTION_ARGS)                                 \
    int32       res,                                                        \
                cmp;                                                        \
                                                                            \
-   cmp = DatumGetInt32(DirectFunctionCall2WithCollation(                   \
+   cmp = DatumGetInt32(DirectFunctionCall2Coll(                            \
                TypeInfo_##type.typecmp,                                    \
                DEFAULT_COLLATION_OID,                                      \
                (data->strategy == BTLessStrategyNumber ||                  \
index 665dfe78b449eb340e85817eff8640db5bbb5a5b..c6b57f82de9875776bd39e731a6a06a296486d03 100644 (file)
@@ -33,37 +33,55 @@ Datum       gbt_text_same(PG_FUNCTION_ARGS);
 static bool
 gbt_textgt(const void *a, const void *b)
 {
-   return (DatumGetBool(DirectFunctionCall2WithCollation(text_gt, DEFAULT_COLLATION_OID, PointerGetDatum(a), PointerGetDatum(b))));
+   return DatumGetBool(DirectFunctionCall2Coll(text_gt,
+                                               DEFAULT_COLLATION_OID,
+                                               PointerGetDatum(a),
+                                               PointerGetDatum(b)));
 }
 
 static bool
 gbt_textge(const void *a, const void *b)
 {
-   return (DatumGetBool(DirectFunctionCall2WithCollation(text_ge, DEFAULT_COLLATION_OID, PointerGetDatum(a), PointerGetDatum(b))));
+   return DatumGetBool(DirectFunctionCall2Coll(text_ge,
+                                               DEFAULT_COLLATION_OID,
+                                               PointerGetDatum(a),
+                                               PointerGetDatum(b)));
 }
 
 static bool
 gbt_texteq(const void *a, const void *b)
 {
-   return (DatumGetBool(DirectFunctionCall2WithCollation(texteq, DEFAULT_COLLATION_OID, PointerGetDatum(a), PointerGetDatum(b))));
+   return DatumGetBool(DirectFunctionCall2Coll(texteq,
+                                               DEFAULT_COLLATION_OID,
+                                               PointerGetDatum(a),
+                                               PointerGetDatum(b)));
 }
 
 static bool
 gbt_textle(const void *a, const void *b)
 {
-   return (DatumGetBool(DirectFunctionCall2WithCollation(text_le, DEFAULT_COLLATION_OID, PointerGetDatum(a), PointerGetDatum(b))));
+   return DatumGetBool(DirectFunctionCall2Coll(text_le,
+                                               DEFAULT_COLLATION_OID,
+                                               PointerGetDatum(a),
+                                               PointerGetDatum(b)));
 }
 
 static bool
 gbt_textlt(const void *a, const void *b)
 {
-   return (DatumGetBool(DirectFunctionCall2WithCollation(text_lt, DEFAULT_COLLATION_OID, PointerGetDatum(a), PointerGetDatum(b))));
+   return DatumGetBool(DirectFunctionCall2Coll(text_lt,
+                                               DEFAULT_COLLATION_OID,
+                                               PointerGetDatum(a),
+                                               PointerGetDatum(b)));
 }
 
 static int32
 gbt_textcmp(const bytea *a, const bytea *b)
 {
-   return DatumGetInt32(DirectFunctionCall2WithCollation(bttextcmp, DEFAULT_COLLATION_OID, PointerGetDatum(a), PointerGetDatum(b)));
+   return DatumGetInt32(DirectFunctionCall2Coll(bttextcmp,
+                                                DEFAULT_COLLATION_OID,
+                                                PointerGetDatum(a),
+                                                PointerGetDatum(b)));
 }
 
 static gbtree_vinfo tinfo =
index d2e0531e3548c5811ea898c6c5e8dee7987f0837..465742556f5b990b9bff095e757802477c01d299 100644 (file)
@@ -1206,7 +1206,7 @@ index_reloptions(RegProcedure amoptions, Datum reloptions, bool validate)
    /* Can't use OidFunctionCallN because we might get a NULL result */
    fmgr_info(amoptions, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 2, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 2, InvalidOid, NULL, NULL);
 
    fcinfo.arg[0] = reloptions;
    fcinfo.arg[1] = BoolGetDatum(validate);
index b632408da4732eaeba661bbd525e898971c44f60..c879b8aaa44f861a37535db085347b46b88fa7ea 100644 (file)
@@ -42,11 +42,11 @@ ScanKeyEntryInitialize(ScanKey entry,
    entry->sk_attno = attributeNumber;
    entry->sk_strategy = strategy;
    entry->sk_subtype = subtype;
+   entry->sk_collation = collation;
    entry->sk_argument = argument;
    if (RegProcedureIsValid(procedure))
    {
        fmgr_info(procedure, &entry->sk_func);
-       entry->sk_func.fn_collation = collation;
    }
    else
    {
@@ -83,9 +83,9 @@ ScanKeyInit(ScanKey entry,
    entry->sk_attno = attributeNumber;
    entry->sk_strategy = strategy;
    entry->sk_subtype = InvalidOid;
+   entry->sk_collation = DEFAULT_COLLATION_OID;
    entry->sk_argument = argument;
    fmgr_info(procedure, &entry->sk_func);
-   entry->sk_func.fn_collation = DEFAULT_COLLATION_OID;
 }
 
 /*
@@ -111,7 +111,7 @@ ScanKeyEntryInitializeWithInfo(ScanKey entry,
    entry->sk_attno = attributeNumber;
    entry->sk_strategy = strategy;
    entry->sk_subtype = subtype;
+   entry->sk_collation = collation;
    entry->sk_argument = argument;
    fmgr_info_copy(&entry->sk_func, finfo, CurrentMemoryContext);
-   entry->sk_func.fn_collation = collation;
 }
index a4771654a6d129439f2bf2a9842de30b7cfbae85..227f84d9881b6cf7ddd02c2d3eb7354adc130bd0 100644 (file)
@@ -55,15 +55,16 @@ callConsistentFn(GinState *ginstate, GinScanKey key)
     */
    key->recheckCurItem = true;
 
-   return DatumGetBool(FunctionCall8(&ginstate->consistentFn[key->attnum - 1],
-                                     PointerGetDatum(key->entryRes),
-                                     UInt16GetDatum(key->strategy),
-                                     key->query,
-                                     UInt32GetDatum(key->nuserentries),
-                                     PointerGetDatum(key->extra_data),
-                                     PointerGetDatum(&key->recheckCurItem),
-                                     PointerGetDatum(key->queryValues),
-                                     PointerGetDatum(key->queryCategories)));
+   return DatumGetBool(FunctionCall8Coll(&ginstate->consistentFn[key->attnum - 1],
+                                         ginstate->compareCollation[key->attnum - 1],
+                                         PointerGetDatum(key->entryRes),
+                                         UInt16GetDatum(key->strategy),
+                                         key->query,
+                                         UInt32GetDatum(key->nuserentries),
+                                         PointerGetDatum(key->extra_data),
+                                         PointerGetDatum(&key->recheckCurItem),
+                                         PointerGetDatum(key->queryValues),
+                                         PointerGetDatum(key->queryCategories)));
 }
 
 /*
@@ -250,9 +251,10 @@ collectMatchBitmap(GinBtreeData *btree, GinBtreeStack *stack,
             * case cmp < 0 => not match and continue scan
             *----------
             */
-           cmp = DatumGetInt32(FunctionCall4(&btree->ginstate->comparePartialFn[attnum - 1],
-                                             scanEntry->queryKey,
-                                             idatum,
+           cmp = DatumGetInt32(FunctionCall4Coll(&btree->ginstate->comparePartialFn[attnum - 1],
+                                                 btree->ginstate->compareCollation[attnum - 1],
+                                                 scanEntry->queryKey,
+                                                 idatum,
                                         UInt16GetDatum(scanEntry->strategy),
                                    PointerGetDatum(scanEntry->extra_data)));
 
@@ -1175,9 +1177,10 @@ matchPartialInPendingList(GinState *ginstate, Page page,
         * case cmp < 0 => not match and continue scan
         *----------
         */
-       cmp = DatumGetInt32(FunctionCall4(&ginstate->comparePartialFn[entry->attnum - 1],
-                                         entry->queryKey,
-                                         datum[off - 1],
+       cmp = DatumGetInt32(FunctionCall4Coll(&ginstate->comparePartialFn[entry->attnum - 1],
+                                             ginstate->compareCollation[entry->attnum - 1],
+                                             entry->queryKey,
+                                             datum[off - 1],
                                          UInt16GetDatum(entry->strategy),
                                        PointerGetDatum(entry->extra_data)));
        if (cmp == 0)
index 716cf3a734865d7ee3502f6377c12a5f50c9c1c3..a712331cf475fbe54f1761ef7a23780aeae9a54c 100644 (file)
@@ -63,23 +63,6 @@ initGinState(GinState *state, Relation index)
        fmgr_info_copy(&(state->compareFn[i]),
                       index_getprocinfo(index, i + 1, GIN_COMPARE_PROC),
                       CurrentMemoryContext);
-
-       /*
-        * If the index column has a specified collation, index_getprocinfo
-        * will have installed it into the fmgr info, and we should honor it.
-        * However, we may have a collatable storage type for a noncollatable
-        * indexed data type (for instance, hstore uses text index entries).
-        * If there's no index collation then specify default collation in
-        * case the comparison function needs one.  This is harmless if the
-        * comparison function doesn't care about collation, so we just do it
-        * unconditionally.  (We could alternatively call get_typcollation,
-        * but that seems like expensive overkill --- there aren't going to be
-        * any cases where a GIN storage type has a nondefault collation.)
-        */
-       if (!OidIsValid(state->compareFn[i].fn_collation))
-           fmgr_info_set_collation(DEFAULT_COLLATION_OID,
-                                   &(state->compareFn[i]));
-
        fmgr_info_copy(&(state->extractValueFn[i]),
                       index_getprocinfo(index, i + 1, GIN_EXTRACTVALUE_PROC),
                       CurrentMemoryContext);
@@ -98,18 +81,29 @@ initGinState(GinState *state, Relation index)
            fmgr_info_copy(&(state->comparePartialFn[i]),
                   index_getprocinfo(index, i + 1, GIN_COMPARE_PARTIAL_PROC),
                           CurrentMemoryContext);
-
-           /* As above, install collation spec in case compare fn needs it */
-           if (!OidIsValid(state->comparePartialFn[i].fn_collation))
-               fmgr_info_set_collation(DEFAULT_COLLATION_OID,
-                                       &(state->comparePartialFn[i]));
-
            state->canPartialMatch[i] = true;
        }
        else
        {
            state->canPartialMatch[i] = false;
        }
+
+       /*
+        * If the index column has a specified collation, we should honor that
+        * while doing comparisons.  However, we may have a collatable storage
+        * type for a noncollatable indexed data type (for instance, hstore
+        * uses text index entries).  If there's no index collation then
+        * specify default collation in case the comparison function needs
+        * collation.  This is harmless if the comparison function doesn't
+        * care about collation, so we just do it unconditionally.  (We could
+        * alternatively call get_typcollation, but that seems like expensive
+        * overkill --- there aren't going to be any cases where a GIN storage
+        * type has a nondefault collation.)
+        */
+       if (OidIsValid(index->rd_indcollation[i]))
+           state->compareCollation[i] = index->rd_indcollation[i];
+       else
+           state->compareCollation[i] = DEFAULT_COLLATION_OID;
    }
 }
 
@@ -298,8 +292,9 @@ ginCompareEntries(GinState *ginstate, OffsetNumber attnum,
        return 0;
 
    /* both not null, so safe to call the compareFn */
-   return DatumGetInt32(FunctionCall2(&ginstate->compareFn[attnum - 1],
-                                      a, b));
+   return DatumGetInt32(FunctionCall2Coll(&ginstate->compareFn[attnum - 1],
+                                          ginstate->compareCollation[attnum - 1],
+                                          a, b));
 }
 
 /*
@@ -334,6 +329,7 @@ typedef struct
 typedef struct
 {
    FmgrInfo   *cmpDatumFunc;
+   Oid         collation;
    bool        haveDups;
 } cmpEntriesArg;
 
@@ -355,8 +351,9 @@ cmpEntries(const void *a, const void *b, void *arg)
    else if (bb->isnull)
        res = -1;               /* not-NULL "<" NULL */
    else
-       res = DatumGetInt32(FunctionCall2(data->cmpDatumFunc,
-                                         aa->datum, bb->datum));
+       res = DatumGetInt32(FunctionCall2Coll(data->cmpDatumFunc,
+                                             data->collation,
+                                             aa->datum, bb->datum));
 
    /*
     * Detect if we have any duplicates.  If there are equal keys, qsort must
@@ -456,6 +453,7 @@ ginExtractEntries(GinState *ginstate, OffsetNumber attnum,
        }
 
        arg.cmpDatumFunc = &ginstate->compareFn[attnum - 1];
+       arg.collation = ginstate->compareCollation[attnum - 1];
        arg.haveDups = false;
        qsort_arg(keydata, *nentries, sizeof(keyEntryData),
                  cmpEntries, (void *) &arg);
index e4488a925de687e153677be332fd217d522c3415..4eb31318ffd3e99d3d7a5e583940862403634e72 100644 (file)
@@ -137,12 +137,13 @@ gistindex_keytest(IndexScanDesc scan,
             */
            recheck = true;
 
-           test = FunctionCall5(&key->sk_func,
-                                PointerGetDatum(&de),
-                                key->sk_argument,
-                                Int32GetDatum(key->sk_strategy),
-                                ObjectIdGetDatum(key->sk_subtype),
-                                PointerGetDatum(&recheck));
+           test = FunctionCall5Coll(&key->sk_func,
+                                    key->sk_collation,
+                                    PointerGetDatum(&de),
+                                    key->sk_argument,
+                                    Int32GetDatum(key->sk_strategy),
+                                    ObjectIdGetDatum(key->sk_subtype),
+                                    PointerGetDatum(&recheck));
 
            if (!DatumGetBool(test))
                return false;
@@ -195,11 +196,12 @@ gistindex_keytest(IndexScanDesc scan,
             * can't tolerate lossy distance calculations on leaf tuples;
             * there is no opportunity to re-sort the tuples afterwards.
             */
-           dist = FunctionCall4(&key->sk_func,
-                                PointerGetDatum(&de),
-                                key->sk_argument,
-                                Int32GetDatum(key->sk_strategy),
-                                ObjectIdGetDatum(key->sk_subtype));
+           dist = FunctionCall4Coll(&key->sk_func,
+                                    key->sk_collation,
+                                    PointerGetDatum(&de),
+                                    key->sk_argument,
+                                    Int32GetDatum(key->sk_strategy),
+                                    ObjectIdGetDatum(key->sk_subtype));
 
            *distance_p = DatumGetFloat8(dist);
        }
index 67308ed37e5cb3370da15a89b001c7e032f35bfb..5662a3a4aab08bbc4841b3acb5961975dc6ef0b6 100644 (file)
@@ -169,8 +169,7 @@ gistrescan(PG_FUNCTION_ARGS)
         * comparisons. The original operator is passed to the Consistent
         * function in the form of its strategy number, which is available
         * from the sk_strategy field, and its subtype from the sk_subtype
-        * field.  Also, preserve sk_func.fn_collation which is the input
-        * collation for the operator.
+        * field.
         *
         * Next, if any of keys is a NULL and that key is not marked with
         * SK_SEARCHNULL/SK_SEARCHNOTNULL then nothing can be found (ie, we
@@ -181,10 +180,8 @@ gistrescan(PG_FUNCTION_ARGS)
        for (i = 0; i < scan->numberOfKeys; i++)
        {
            ScanKey     skey = scan->keyData + i;
-           Oid         collation = skey->sk_func.fn_collation;
 
            skey->sk_func = so->giststate->consistentFn[skey->sk_attno - 1];
-           skey->sk_func.fn_collation = collation;
 
            if (skey->sk_flags & SK_ISNULL)
            {
@@ -205,16 +202,13 @@ gistrescan(PG_FUNCTION_ARGS)
         * all comparisons. The original operator is passed to the Distance
         * function in the form of its strategy number, which is available
         * from the sk_strategy field, and its subtype from the sk_subtype
-        * field.  Also, preserve sk_func.fn_collation which is the input
-        * collation for the operator.
+        * field.
         */
        for (i = 0; i < scan->numberOfOrderBys; i++)
        {
            ScanKey     skey = scan->orderByData + i;
-           Oid         collation = skey->sk_func.fn_collation;
 
            skey->sk_func = so->giststate->distanceFn[skey->sk_attno - 1];
-           skey->sk_func.fn_collation = collation;
 
            /* Check we actually have a distance function ... */
            if (!OidIsValid(skey->sk_func.fn_oid))
index b00b32d69fc8396a7dd013196ee928b1e27df2c8..ac238d9f7d803b54dce99c6f8b502ef357297d81 100644 (file)
@@ -56,7 +56,8 @@ _hash_checkqual(IndexScanDesc scan, IndexTuple itup)
        if (key->sk_flags & SK_ISNULL)
            return false;
 
-       test = FunctionCall2(&key->sk_func, datum, key->sk_argument);
+       test = FunctionCall2Coll(&key->sk_func, key->sk_collation,
+                                datum, key->sk_argument);
 
        if (!DatumGetBool(test))
            return false;
index 66af2c37c54f8f0d33145db2d5b84bb67e17b4a8..6e0db79517615a1dc13f6287fecbc17bc1e587da 100644 (file)
@@ -872,7 +872,6 @@ index_getprocinfo(Relation irel,
                 procnum, attnum, RelationGetRelationName(irel));
 
        fmgr_info_cxt(procId, locinfo, irel->rd_indexcxt);
-       fmgr_info_set_collation(irel->rd_indcollation[attnum - 1], locinfo);
    }
 
    return locinfo;
index 219f94fd0dd92c43a48b255da8deeb915000506d..d758659c314b6894149ac7a60d9742bf5f9c0ecf 100644 (file)
@@ -2043,9 +2043,10 @@ _bt_isequal(TupleDesc itupdesc, Page page, OffsetNumber offnum,
        if (isNull || (scankey->sk_flags & SK_ISNULL))
            return false;
 
-       result = DatumGetInt32(FunctionCall2(&scankey->sk_func,
-                                            datum,
-                                            scankey->sk_argument));
+       result = DatumGetInt32(FunctionCall2Coll(&scankey->sk_func,
+                                                scankey->sk_collation,
+                                                datum,
+                                                scankey->sk_argument));
 
        if (result != 0)
            return false;
index 91f8cadea52b1bca1474a05f02f17655bb1685a9..2ce2bc2f00f519d24bddde0070abaf22df57613b 100644 (file)
@@ -410,9 +410,10 @@ _bt_compare(Relation rel,
             * to flip the sign of the comparison result.  (Unless it's a DESC
             * column, in which case we *don't* flip the sign.)
             */
-           result = DatumGetInt32(FunctionCall2(&scankey->sk_func,
-                                                datum,
-                                                scankey->sk_argument));
+           result = DatumGetInt32(FunctionCall2Coll(&scankey->sk_func,
+                                                    scankey->sk_collation,
+                                                    datum,
+                                                    scankey->sk_argument));
 
            if (!(scankey->sk_flags & SK_BT_DESC))
                result = -result;
@@ -721,7 +722,7 @@ _bt_first(IndexScanDesc scan, ScanDirection dir)
                                               cur->sk_attno,
                                               InvalidStrategy,
                                               cur->sk_subtype,
-                                              cur->sk_func.fn_collation,
+                                              cur->sk_collation,
                                               procinfo,
                                               cur->sk_argument);
            }
@@ -742,7 +743,7 @@ _bt_first(IndexScanDesc scan, ScanDirection dir)
                                       cur->sk_attno,
                                       InvalidStrategy,
                                       cur->sk_subtype,
-                                      cur->sk_func.fn_collation,
+                                      cur->sk_collation,
                                       cmp_proc,
                                       cur->sk_argument);
            }
index 256a7f9f98f9aae57b69b61c717ca4e404af788b..55136e9cc4c067322b771abdc4e88166d799ef76 100644 (file)
@@ -736,9 +736,11 @@ _bt_load(BTWriteState *wstate, BTSpool *btspool, BTSpool *btspool2)
                    }
                    else
                    {
-                       compare = DatumGetInt32(FunctionCall2(&entry->sk_func,
-                                                             attrDatum1,
-                                                             attrDatum2));
+                       compare =
+                           DatumGetInt32(FunctionCall2Coll(&entry->sk_func,
+                                                           entry->sk_collation,
+                                                           attrDatum1,
+                                                           attrDatum2));
 
                        if (entry->sk_flags & SK_BT_DESC)
                            compare = -compare;
index d448ba6a502845e1f9d4d9ff75e1f0eb86768781..71bcb42c1908e85abddd73631666573dec2c2387 100644 (file)
@@ -70,8 +70,7 @@ _bt_mkscankey(Relation rel, IndexTuple itup)
 
        /*
         * We can use the cached (default) support procs since no cross-type
-        * comparison can be needed.  The cached support proc entries have the
-        * right collation for the index, too.
+        * comparison can be needed.
         */
        procinfo = index_getprocinfo(rel, i + 1, BTORDER_PROC);
        arg = index_getattr(itup, i + 1, itupdesc, &null);
@@ -81,7 +80,7 @@ _bt_mkscankey(Relation rel, IndexTuple itup)
                                       (AttrNumber) (i + 1),
                                       InvalidStrategy,
                                       InvalidOid,
-                                      procinfo->fn_collation,
+                                      rel->rd_indcollation[i],
                                       procinfo,
                                       arg);
    }
@@ -120,8 +119,7 @@ _bt_mkscankey_nodata(Relation rel)
 
        /*
         * We can use the cached (default) support procs since no cross-type
-        * comparison can be needed.  The cached support proc entries have the
-        * right collation for the index, too.
+        * comparison can be needed.
         */
        procinfo = index_getprocinfo(rel, i + 1, BTORDER_PROC);
        flags = SK_ISNULL | (indoption[i] << SK_BT_INDOPTION_SHIFT);
@@ -130,7 +128,7 @@ _bt_mkscankey_nodata(Relation rel)
                                       (AttrNumber) (i + 1),
                                       InvalidStrategy,
                                       InvalidOid,
-                                      procinfo->fn_collation,
+                                      rel->rd_indcollation[i],
                                       procinfo,
                                       (Datum) 0);
    }
@@ -604,9 +602,10 @@ _bt_compare_scankey_args(IndexScanDesc scan, ScanKey op,
     */
    if (lefttype == opcintype && righttype == optype)
    {
-       *result = DatumGetBool(FunctionCall2(&op->sk_func,
-                                            leftarg->sk_argument,
-                                            rightarg->sk_argument));
+       *result = DatumGetBool(FunctionCall2Coll(&op->sk_func,
+                                                op->sk_collation,
+                                                leftarg->sk_argument,
+                                                rightarg->sk_argument));
        return true;
    }
 
@@ -633,9 +632,10 @@ _bt_compare_scankey_args(IndexScanDesc scan, ScanKey op,
 
        if (RegProcedureIsValid(cmp_proc))
        {
-           *result = DatumGetBool(OidFunctionCall2(cmp_proc,
-                                                   leftarg->sk_argument,
-                                                   rightarg->sk_argument));
+           *result = DatumGetBool(OidFunctionCall2Coll(cmp_proc,
+                                                       op->sk_collation,
+                                                       leftarg->sk_argument,
+                                                       rightarg->sk_argument));
            return true;
        }
    }
@@ -689,6 +689,10 @@ _bt_fix_scankey_strategy(ScanKey skey, int16 *indoption)
     * Likewise, "x IS NOT NULL" is supported.  We treat that as either "less
     * than NULL" in a NULLS LAST index, or "greater than NULL" in a NULLS
     * FIRST index.
+    *
+    * Note: someday we might have to fill in sk_collation from the index
+    * column's collation.  At the moment this is a non-issue because we'll
+    * never actually call the comparison operator on a NULL.
     */
    if (skey->sk_flags & SK_ISNULL)
    {
@@ -703,6 +707,7 @@ _bt_fix_scankey_strategy(ScanKey skey, int16 *indoption)
        {
            skey->sk_strategy = BTEqualStrategyNumber;
            skey->sk_subtype = InvalidOid;
+           skey->sk_collation = InvalidOid;
        }
        else if (skey->sk_flags & SK_SEARCHNOTNULL)
        {
@@ -711,6 +716,7 @@ _bt_fix_scankey_strategy(ScanKey skey, int16 *indoption)
            else
                skey->sk_strategy = BTLessStrategyNumber;
            skey->sk_subtype = InvalidOid;
+           skey->sk_collation = InvalidOid;
        }
        else
        {
@@ -976,7 +982,8 @@ _bt_checkkeys(IndexScanDesc scan,
            return false;
        }
 
-       test = FunctionCall2(&key->sk_func, datum, key->sk_argument);
+       test = FunctionCall2Coll(&key->sk_func, key->sk_collation,
+                                datum, key->sk_argument);
 
        if (!DatumGetBool(test))
        {
@@ -1099,9 +1106,10 @@ _bt_check_rowcompare(ScanKey skey, IndexTuple tuple, TupleDesc tupdesc,
        }
 
        /* Perform the test --- three-way comparison not bool operator */
-       cmpresult = DatumGetInt32(FunctionCall2(&subkey->sk_func,
-                                               datum,
-                                               subkey->sk_argument));
+       cmpresult = DatumGetInt32(FunctionCall2Coll(&subkey->sk_func,
+                                                   subkey->sk_collation,
+                                                   datum,
+                                                   subkey->sk_argument));
 
        if (subkey->sk_flags & SK_BT_DESC)
            cmpresult = -cmpresult;
index dde301b89aa2b73254dde0a9dc4dc3c81819799d..0568a1bcf86281a9b1086d343e7027557295065c 100644 (file)
@@ -1930,8 +1930,6 @@ compute_minimal_stats(VacAttrStatsP stats,
    track_cnt = 0;
 
    fmgr_info(mystats->eqfunc, &f_cmpeq);
-   /* We always use the default collation for statistics */
-   fmgr_info_set_collation(DEFAULT_COLLATION_OID, &f_cmpeq);
 
    for (i = 0; i < samplerows; i++)
    {
@@ -1990,7 +1988,10 @@ compute_minimal_stats(VacAttrStatsP stats,
        firstcount1 = track_cnt;
        for (j = 0; j < track_cnt; j++)
        {
-           if (DatumGetBool(FunctionCall2(&f_cmpeq, value, track[j].value)))
+           /* We always use the default collation for statistics */
+           if (DatumGetBool(FunctionCall2Coll(&f_cmpeq,
+                                              DEFAULT_COLLATION_OID,
+                                              value, track[j].value)))
            {
                match = true;
                break;
@@ -2253,8 +2254,6 @@ compute_scalar_stats(VacAttrStatsP stats,
 
    SelectSortFunction(mystats->ltopr, false, &cmpFn, &cmpFlags);
    fmgr_info(cmpFn, &f_cmpfn);
-   /* We always use the default collation for statistics */
-   fmgr_info_set_collation(DEFAULT_COLLATION_OID, &f_cmpfn);
 
    /* Initial scan to find sortable values */
    for (i = 0; i < samplerows; i++)
@@ -2729,7 +2728,9 @@ compare_scalars(const void *a, const void *b, void *arg)
    CompareScalarsContext *cxt = (CompareScalarsContext *) arg;
    int32       compare;
 
+   /* We always use the default collation for statistics */
    compare = ApplySortFunction(cxt->cmpFn, cxt->cmpFlags,
+                               DEFAULT_COLLATION_OID,
                                da, false, db, false);
    if (compare != 0)
        return compare;
index 6b1ade899032f001f02de50194b74e49e8e3b3ef..ce36ea8be45c0d4f997a49305fd7215e1730af20 100644 (file)
@@ -1822,7 +1822,8 @@ ExecCallTriggerFunc(TriggerData *trigdata,
    /*
     * Call the function, passing no arguments but setting a context.
     */
-   InitFunctionCallInfoData(fcinfo, finfo, 0, (Node *) trigdata, NULL);
+   InitFunctionCallInfoData(fcinfo, finfo, 0,
+                            InvalidOid, (Node *) trigdata, NULL);
 
    pgstat_init_function_usage(&fcinfo, &fcusage);
 
index ce0086ffa17c3ec85c50270e7a199d76f470017e..d08c9bbbc5ce4699d7437639dc2154a781850ca8 100644 (file)
@@ -96,6 +96,11 @@ get_ts_parser_func(DefElem *defel, int attnum)
            break;
        case Anum_pg_ts_parser_prslextype:
            nargs = 1;
+           /*
+            * Note: because the lextype method returns type internal, it must
+            * have an internal-type argument for security reasons.  The
+            * argument is not actually used, but is just passed as a zero.
+            */
            break;
        default:
            /* should not be here */
@@ -1947,7 +1952,7 @@ getTokenTypes(Oid prsId, List *tokennames)
        elog(ERROR, "method lextype isn't defined for text search parser %u",
             prsId);
 
-   /* OidFunctionCall0 is absent */
+   /* lextype takes one dummy argument */
    list = (LexDescr *) DatumGetPointer(OidFunctionCall1(prs->lextypeOid,
                                                         (Datum) 0));
 
index 1f02428a5b3680eb0c8a2926f8a2448e3dbd2b25..bfb8cbb303c6cb3635e7f925c0eefb4f9a8d570d 100644 (file)
@@ -3,6 +3,10 @@
  * execGrouping.c
  *   executor utility routines for grouping, hashing, and aggregation
  *
+ * Note: we currently assume that equality and hashing functions are not
+ * collation-sensitive, so the code in this file has no support for passing
+ * collation settings through from callers.  That may have to change someday.
+ *
  * Portions Copyright (c) 1996-2011, PostgreSQL Global Development Group
  * Portions Copyright (c) 1994, Regents of the University of California
  *
index 5f0b58f43b79ae138f49c6198fcb6c1d333c523e..80f08d8b92e8dcfb9a5635297afd2358fef0610d 100644 (file)
@@ -1202,12 +1202,12 @@ init_fcache(Oid foid, Oid input_collation, FuncExprState *fcache,
 
    /* Set up the primary fmgr lookup information */
    fmgr_info_cxt(foid, &(fcache->func), fcacheCxt);
-   fmgr_info_set_collation(input_collation, &(fcache->func));
    fmgr_info_set_expr((Node *) fcache->xprstate.expr, &(fcache->func));
 
    /* Initialize the function call parameter struct as well */
    InitFunctionCallInfoData(fcache->fcinfo_data, &(fcache->func),
-                            list_length(fcache->args), NULL, NULL);
+                            list_length(fcache->args),
+                            input_collation, NULL, NULL);
 
    /* If function returns set, prepare expected tuple descriptor */
    if (fcache->func.fn_retset && needDescForSets)
@@ -1980,6 +1980,7 @@ ExecMakeTableFunctionResult(ExprState *funcexpr,
        returnsSet = fcache->func.fn_retset;
        InitFunctionCallInfoData(fcinfo, &(fcache->func),
                                 list_length(fcache->args),
+                                fcache->fcinfo_data.fncollation,
                                 NULL, (Node *) &rsinfo);
 
        /*
@@ -2017,7 +2018,7 @@ ExecMakeTableFunctionResult(ExprState *funcexpr,
    {
        /* Treat funcexpr as a generic expression */
        direct_function_call = false;
-       InitFunctionCallInfoData(fcinfo, NULL, 0, NULL, NULL);
+       InitFunctionCallInfoData(fcinfo, NULL, 0, InvalidOid, NULL, NULL);
    }
 
    /*
@@ -3154,6 +3155,7 @@ ExecEvalRowCompare(RowCompareExprState *rstate,
        FunctionCallInfoData locfcinfo;
 
        InitFunctionCallInfoData(locfcinfo, &(rstate->funcs[i]), 2,
+                                rstate->collations[i],
                                 NULL, NULL);
        locfcinfo.arg[0] = ExecEvalExpr(le, econtext,
                                        &locfcinfo.argnull[0], NULL);
@@ -3234,7 +3236,9 @@ ExecEvalMinMax(MinMaxExprState *minmaxExpr, ExprContext *econtext,
               bool *isNull, ExprDoneCond *isDone)
 {
    Datum       result = (Datum) 0;
-   MinMaxOp    op = ((MinMaxExpr *) minmaxExpr->xprstate.expr)->op;
+   MinMaxExpr *minmax = (MinMaxExpr *) minmaxExpr->xprstate.expr;
+   Oid         collation = minmax->inputcollid;
+   MinMaxOp    op = minmax->op;
    FunctionCallInfoData locfcinfo;
    ListCell   *arg;
 
@@ -3242,7 +3246,8 @@ ExecEvalMinMax(MinMaxExprState *minmaxExpr, ExprContext *econtext,
        *isDone = ExprSingleResult;
    *isNull = true;             /* until we get a result */
 
-   InitFunctionCallInfoData(locfcinfo, &minmaxExpr->cfunc, 2, NULL, NULL);
+   InitFunctionCallInfoData(locfcinfo, &minmaxExpr->cfunc, 2,
+                            collation, NULL, NULL);
    locfcinfo.argnull[0] = false;
    locfcinfo.argnull[1] = false;
 
@@ -4115,7 +4120,6 @@ ExecEvalArrayCoerceExpr(ArrayCoerceExprState *astate,
        /* Set up the primary fmgr lookup information */
        fmgr_info_cxt(acoerce->elemfuncid, &(astate->elemfunc),
                      econtext->ecxt_per_query_memory);
-       /* Note: coercion functions are assumed to not use collation */
        fmgr_info_set_expr((Node *) acoerce, &(astate->elemfunc));
    }
 
@@ -4124,9 +4128,11 @@ ExecEvalArrayCoerceExpr(ArrayCoerceExprState *astate,
     *
     * We pass on the desttypmod and isExplicit flags whether or not the
     * function wants them.
+    *
+    * Note: coercion functions are assumed to not use collation.
     */
    InitFunctionCallInfoData(locfcinfo, &(astate->elemfunc), 3,
-                            NULL, NULL);
+                            InvalidOid, NULL, NULL);
    locfcinfo.arg[0] = PointerGetDatum(array);
    locfcinfo.arg[1] = Int32GetDatum(acoerce->resulttypmod);
    locfcinfo.arg[2] = BoolGetDatum(acoerce->isExplicit);
@@ -4699,6 +4705,7 @@ ExecInitExpr(Expr *node, PlanState *parent)
                rstate->rargs = outlist;
                Assert(list_length(rcexpr->opfamilies) == nopers);
                rstate->funcs = (FmgrInfo *) palloc(nopers * sizeof(FmgrInfo));
+               rstate->collations = (Oid *) palloc(nopers * sizeof(Oid));
                i = 0;
                forthree(l, rcexpr->opnos, l2, rcexpr->opfamilies, l3, rcexpr->inputcollids)
                {
@@ -4726,7 +4733,7 @@ ExecInitExpr(Expr *node, PlanState *parent)
                     * does this code.
                     */
                    fmgr_info(proc, &(rstate->funcs[i]));
-                   fmgr_info_set_collation(inputcollid, &(rstate->funcs[i]));
+                   rstate->collations[i] = inputcollid;
                    i++;
                }
                state = (ExprState *) rstate;
@@ -4786,8 +4793,6 @@ ExecInitExpr(Expr *node, PlanState *parent)
                 * code.
                 */
                fmgr_info(typentry->cmp_proc, &(mstate->cfunc));
-               fmgr_info_set_collation(minmaxexpr->inputcollid,
-                                       &(mstate->cfunc));
                state = (ExprState *) mstate;
            }
            break;
index 0cbbe04d3bda1bd5511c16e36c96bd42cf63a7a9..073ef8d23b3688e3e1f2034c5badc6a51f183c01 100644 (file)
@@ -1349,9 +1349,10 @@ index_recheck_constraint(Relation index, Oid *constr_procs,
        if (existing_isnull[i])
            return false;
 
-       if (!DatumGetBool(OidFunctionCall2(constr_procs[i],
-                                          existing_values[i],
-                                          new_values[i])))
+       if (!DatumGetBool(OidFunctionCall2Coll(constr_procs[i],
+                                              index->rd_indcollation[i],
+                                              existing_values[i],
+                                              new_values[i])))
            return false;
    }
 
index 9c867bbae200b7354d0d15d980a9b9c102ec3bfd..13946cab23d88d557e103bb7490738f26ff73c0d 100644 (file)
@@ -127,7 +127,7 @@ static Node *sql_fn_param_ref(ParseState *pstate, ParamRef *pref);
 static List *init_execution_state(List *queryTree_list,
                     SQLFunctionCachePtr fcache,
                     bool lazyEvalOK);
-static void init_sql_fcache(FmgrInfo *finfo, bool lazyEvalOK);
+static void init_sql_fcache(FmgrInfo *finfo, Oid collation, bool lazyEvalOK);
 static void postquel_start(execution_state *es, SQLFunctionCachePtr fcache);
 static bool postquel_getnext(execution_state *es, SQLFunctionCachePtr fcache);
 static void postquel_end(execution_state *es);
@@ -363,7 +363,7 @@ init_execution_state(List *queryTree_list,
  * Initialize the SQLFunctionCache for a SQL function
  */
 static void
-init_sql_fcache(FmgrInfo *finfo, bool lazyEvalOK)
+init_sql_fcache(FmgrInfo *finfo, Oid collation, bool lazyEvalOK)
 {
    Oid         foid = finfo->fn_oid;
    Oid         rettype;
@@ -428,7 +428,7 @@ init_sql_fcache(FmgrInfo *finfo, bool lazyEvalOK)
     */
    fcache->pinfo = prepare_sql_fn_parse_info(procedureTuple,
                                              finfo->fn_expr,
-                                             finfo->fn_collation);
+                                             collation);
 
    /*
     * And of course we need the function body text.
@@ -798,7 +798,7 @@ fmgr_sql(PG_FUNCTION_ARGS)
    fcache = (SQLFunctionCachePtr) fcinfo->flinfo->fn_extra;
    if (fcache == NULL)
    {
-       init_sql_fcache(fcinfo->flinfo, lazyEvalOK);
+       init_sql_fcache(fcinfo->flinfo, PG_GET_COLLATION(), lazyEvalOK);
        fcache = (SQLFunctionCachePtr) fcinfo->flinfo->fn_extra;
    }
    eslist = fcache->func_state;
index 47555bab55bdc0d8bb6ce42683fb037a7dd6e5ed..13d7723480144575103dc2efb42c10296dd565f8 100644 (file)
@@ -130,6 +130,9 @@ typedef struct AggStatePerAggData
    FmgrInfo    transfn;
    FmgrInfo    finalfn;
 
+   /* Input collation derived for aggregate */
+   Oid         aggCollation;
+
    /* number of sorting columns */
    int         numSortCols;
 
@@ -430,6 +433,7 @@ advance_transition_function(AggState *aggstate,
     */
    InitFunctionCallInfoData(*fcinfo, &(peraggstate->transfn),
                             numArguments + 1,
+                            peraggstate->aggCollation,
                             (void *) aggstate, NULL);
    fcinfo->arg[0] = pergroupstate->transValue;
    fcinfo->argnull[0] = pergroupstate->transValueIsNull;
@@ -597,6 +601,8 @@ process_ordered_aggregate_single(AggState *aggstate,
 
        /*
         * If DISTINCT mode, and not distinct from prior, skip it.
+        *
+        * Note: we assume equality functions don't care about collation.
         */
        if (isDistinct &&
            haveOldVal &&
@@ -737,6 +743,7 @@ finalize_aggregate(AggState *aggstate,
        FunctionCallInfoData fcinfo;
 
        InitFunctionCallInfoData(fcinfo, &(peraggstate->finalfn), 1,
+                                peraggstate->aggCollation,
                                 (void *) aggstate, NULL);
        fcinfo.arg[0] = pergroupstate->transValue;
        fcinfo.argnull[0] = pergroupstate->transValueIsNull;
@@ -1676,16 +1683,16 @@ ExecInitAgg(Agg *node, EState *estate, int eflags)
                                &finalfnexpr);
 
        fmgr_info(transfn_oid, &peraggstate->transfn);
-       fmgr_info_set_collation(aggref->inputcollid, &peraggstate->transfn);
        fmgr_info_set_expr((Node *) transfnexpr, &peraggstate->transfn);
 
        if (OidIsValid(finalfn_oid))
        {
            fmgr_info(finalfn_oid, &peraggstate->finalfn);
-           fmgr_info_set_collation(aggref->inputcollid, &peraggstate->finalfn);
            fmgr_info_set_expr((Node *) finalfnexpr, &peraggstate->finalfn);
        }
 
+       peraggstate->aggCollation = aggref->inputcollid;
+
        get_typlenbyval(aggref->aggtype,
                        &peraggstate->resulttypeLen,
                        &peraggstate->resulttypeByVal);
@@ -1833,8 +1840,6 @@ ExecInitAgg(Agg *node, EState *estate, int eflags)
                SortGroupClause *sortcl = (SortGroupClause *) lfirst(lc);
 
                fmgr_info(get_opcode(sortcl->eqop), &peraggstate->equalfns[i]);
-               fmgr_info_set_collation(aggref->inputcollid,
-                                       &peraggstate->equalfns[i]);
                i++;
            }
            Assert(i == numDistinctCols);
index d8e59ca39e5e08a195421171c66d8b24428e14a6..dbc1467d5be26367218ff0592a36839b9fb9658a 100644 (file)
@@ -973,7 +973,7 @@ ExecIndexBuildScanKeys(PlanState *planstate, Relation index, Index scanrelid,
            this_scan_key->sk_flags = SK_ROW_HEADER;
            this_scan_key->sk_attno = first_sub_key->sk_attno;
            this_scan_key->sk_strategy = rc->rctype;
-           /* sk_subtype, sk_func not used in a header */
+           /* sk_subtype, sk_collation, sk_func not used in a header */
            this_scan_key->sk_argument = PointerGetDatum(first_sub_key);
        }
        else if (IsA(clause, ScalarArrayOpExpr))
index 4ebe0cbe033372d6016c58d2aee7f2863ef3ba77..43059664b9348ae462dcb6b1166124f4f0fde2ee 100644 (file)
@@ -327,8 +327,9 @@ heap_compare_slots(MergeAppendState *node, SlotNumber slot1, SlotNumber slot2)
        }
        else
        {
-           compare = DatumGetInt32(FunctionCall2(&scankey->sk_func,
-                                                 datum1, datum2));
+           compare = DatumGetInt32(FunctionCall2Coll(&scankey->sk_func,
+                                                     scankey->sk_collation,
+                                                     datum1, datum2));
            if (compare != 0)
            {
                if (scankey->sk_flags & SK_BT_DESC)
index d6245e61874a4e2cf44f61d41a25e9ba42e1b523..208c0fb76ac11941dc583e89e31bf6a6e92d1d9b 100644 (file)
@@ -138,11 +138,12 @@ typedef struct MergeJoinClauseData
 
    /*
     * The comparison strategy in use, and the lookup info to let us call the
-    * btree comparison support function.
+    * btree comparison support function, and the collation to use.
     */
    bool        reverse;        /* if true, negate the cmpfn's output */
    bool        nulls_first;    /* if true, nulls sort low */
    FmgrInfo    cmpfinfo;
+   Oid         collation;
 }  MergeJoinClauseData;
 
 /* Result type for MJEvalOuterValues and MJEvalInnerValues */
@@ -242,7 +243,6 @@ MJExamineQuals(List *mergeclauses,
 
        /* Set up the fmgr lookup information */
        fmgr_info(cmpproc, &(clause->cmpfinfo));
-       fmgr_info_set_collation(collation, &(clause->cmpfinfo));
 
        /* Fill the additional comparison-strategy flags */
        if (opstrategy == BTLessStrategyNumber)
@@ -254,6 +254,9 @@ MJExamineQuals(List *mergeclauses,
 
        clause->nulls_first = nulls_first;
 
+       /* ... and the collation too */
+       clause->collation = collation;
+
        iClause++;
    }
 
@@ -429,7 +432,7 @@ MJCompare(MergeJoinState *mergestate)
         * OK to call the comparison function.
         */
        InitFunctionCallInfoData(fcinfo, &(clause->cmpfinfo), 2,
-                                NULL, NULL);
+                                clause->collation, NULL, NULL);
        fcinfo.arg[0] = clause->ldatum;
        fcinfo.arg[1] = clause->rdatum;
        fcinfo.argnull[0] = false;
index 08a3017e6145e13e630c2bb88efd5f5c76b8d357..4fcb66bbb03b86d6f67a4c1bc8a21d8b90231796 100644 (file)
@@ -831,8 +831,6 @@ ExecInitSubPlan(SubPlan *subplan, PlanState *parent)
 
            /* Lookup the equality function (potentially cross-type) */
            fmgr_info(opexpr->opfuncid, &sstate->cur_eq_funcs[i - 1]);
-           fmgr_info_set_collation(opexpr->inputcollid,
-                                   &sstate->cur_eq_funcs[i - 1]);
            fmgr_info_set_expr((Node *) opexpr, &sstate->cur_eq_funcs[i - 1]);
 
            /* Look up the equality function for the RHS type */
@@ -841,8 +839,6 @@ ExecInitSubPlan(SubPlan *subplan, PlanState *parent)
                elog(ERROR, "could not find compatible hash operator for operator %u",
                     opexpr->opno);
            fmgr_info(get_opcode(rhs_eq_oper), &sstate->tab_eq_funcs[i - 1]);
-           fmgr_info_set_collation(opexpr->inputcollid,
-                                   &sstate->tab_eq_funcs[i - 1]);
 
            /* Lookup the associated hash functions */
            if (!get_op_hash_functions(opexpr->opno,
index 25d9298cefc2460e706309d1a1972c124c555a28..c90d40506f93e1304ddcd10173ff2aeba0fc86ea 100644 (file)
@@ -81,6 +81,8 @@ typedef struct WindowStatePerFuncData
 
    FmgrInfo    flinfo;         /* fmgr lookup data for window function */
 
+   Oid         winCollation;   /* collation derived for window function */
+
    /*
     * We need the len and byval info for the result of each function in order
     * to know how to copy/delete values.
@@ -289,6 +291,7 @@ advance_windowaggregate(WindowAggState *winstate,
     */
    InitFunctionCallInfoData(*fcinfo, &(peraggstate->transfn),
                             numArguments + 1,
+                            perfuncstate->winCollation,
                             (void *) winstate, NULL);
    fcinfo->arg[0] = peraggstate->transValue;
    fcinfo->argnull[0] = peraggstate->transValueIsNull;
@@ -340,6 +343,7 @@ finalize_windowaggregate(WindowAggState *winstate,
        FunctionCallInfoData fcinfo;
 
        InitFunctionCallInfoData(fcinfo, &(peraggstate->finalfn), 1,
+                                perfuncstate->winCollation,
                                 (void *) winstate, NULL);
        fcinfo.arg[0] = peraggstate->transValue;
        fcinfo.argnull[0] = peraggstate->transValueIsNull;
@@ -627,6 +631,7 @@ eval_windowfunction(WindowAggState *winstate, WindowStatePerFunc perfuncstate,
     */
    InitFunctionCallInfoData(fcinfo, &(perfuncstate->flinfo),
                             perfuncstate->numArguments,
+                            perfuncstate->winCollation,
                             (void *) perfuncstate->winobj, NULL);
    /* Just in case, make all the regular argument slots be null */
    memset(fcinfo.argnull, true, perfuncstate->numArguments);
@@ -1561,9 +1566,10 @@ ExecInitWindowAgg(WindowAgg *node, EState *estate, int eflags)
 
        fmgr_info_cxt(wfunc->winfnoid, &perfuncstate->flinfo,
                      econtext->ecxt_per_query_memory);
-       fmgr_info_set_collation(wfunc->inputcollid, &perfuncstate->flinfo);
        fmgr_info_set_expr((Node *) wfunc, &perfuncstate->flinfo);
 
+       perfuncstate->winCollation = wfunc->inputcollid;
+
        get_typlenbyval(wfunc->wintype,
                        &perfuncstate->resulttypeLen,
                        &perfuncstate->resulttypeByVal);
@@ -1801,13 +1807,11 @@ initialize_peragg(WindowAggState *winstate, WindowFunc *wfunc,
                            &finalfnexpr);
 
    fmgr_info(transfn_oid, &peraggstate->transfn);
-   fmgr_info_set_collation(wfunc->inputcollid, &peraggstate->transfn);
    fmgr_info_set_expr((Node *) transfnexpr, &peraggstate->transfn);
 
    if (OidIsValid(finalfn_oid))
    {
        fmgr_info(finalfn_oid, &peraggstate->finalfn);
-       fmgr_info_set_collation(wfunc->inputcollid, &peraggstate->finalfn);
        fmgr_info_set_expr((Node *) finalfnexpr, &peraggstate->finalfn);
    }
 
index c7ed1b6ee9d67848cfc997e830fea7cd6cf27e87..1cace6d5960eb5a15b90df67e95f2fc36b28ea3d 100644 (file)
@@ -3189,8 +3189,7 @@ prefix_quals(Node *leftop, Oid opfamily, Oid collation,
    if (oproid == InvalidOid)
        elog(ERROR, "no < operator for opfamily %u", opfamily);
    fmgr_info(get_opcode(oproid), &ltproc);
-   fmgr_info_set_collation(collation, &ltproc);
-   greaterstr = make_greater_string(prefix_const, &ltproc);
+   greaterstr = make_greater_string(prefix_const, &ltproc, collation);
    if (greaterstr)
    {
        expr = make_opclause(oproid, BOOLOID, false,
index d0a23d007aafd32afa2070fb1720c73ade297bfc..30863bb7b960cc8741fa04b148eba325ac6b4aee 100644 (file)
@@ -349,8 +349,12 @@ HandleFunctionRequest(StringInfo msgBuf)
 
    /*
     * Prepare function call info block and insert arguments.
+    *
+    * Note: for now we pass collation = InvalidOid, so collation-sensitive
+    * functions can't be called this way.  Perhaps we should pass
+    * DEFAULT_COLLATION_OID, instead?
     */
-   InitFunctionCallInfoData(fcinfo, &fip->flinfo, 0, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &fip->flinfo, 0, InvalidOid, NULL, NULL);
 
    if (PG_PROTOCOL_MAJOR(FrontendProtocol) >= 3)
        rformat = parse_fcall_arguments(msgBuf, fip, &fcinfo);
index f636c2d23b22477eb8209c69b4e804c68ded1c1b..bda8d2632d91ecab033c7b73df4d73a2527b20f1 100644 (file)
@@ -54,7 +54,7 @@ tt_setup_firstcall(FuncCallContext *funcctx, Oid prsid)
 
    st = (TSTokenTypeStorage *) palloc(sizeof(TSTokenTypeStorage));
    st->cur = 0;
-   /* OidFunctionCall0 is absent */
+   /* lextype takes one dummy argument */
    st->list = (LexDescr *) DatumGetPointer(OidFunctionCall1(prs->lextypeOid,
                                                             (Datum) 0));
    funcctx->user_fctx = (void *) st;
index 0869de66cef1e381eea189404cf7ed9a1869eebf..a234f35eb8f561650add8cfd3fa4a44a36c550f7 100644 (file)
@@ -3127,6 +3127,7 @@ array_eq(PG_FUNCTION_ARGS)
 {
    ArrayType  *array1 = PG_GETARG_ARRAYTYPE_P(0);
    ArrayType  *array2 = PG_GETARG_ARRAYTYPE_P(1);
+   Oid         collation = PG_GET_COLLATION();
    int         ndims1 = ARR_NDIM(array1);
    int         ndims2 = ARR_NDIM(array2);
    int        *dims1 = ARR_DIMS(array1);
@@ -3184,7 +3185,7 @@ array_eq(PG_FUNCTION_ARGS)
         * apply the operator to each pair of array elements.
         */
        InitFunctionCallInfoData(locfcinfo, &typentry->eq_opr_finfo, 2,
-                                NULL, NULL);
+                                collation, NULL, NULL);
 
        /* Loop over source data */
        nitems = ArrayGetNItems(ndims1, dims1);
@@ -3367,8 +3368,7 @@ array_cmp(FunctionCallInfo fcinfo)
     */
    typentry = (TypeCacheEntry *) fcinfo->flinfo->fn_extra;
    if (typentry == NULL ||
-       typentry->type_id != element_type ||
-       typentry->cmp_proc_finfo.fn_collation != collation)
+       typentry->type_id != element_type)
    {
        typentry = lookup_type_cache(element_type,
                                     TYPECACHE_CMP_PROC_FINFO);
@@ -3378,7 +3378,6 @@ array_cmp(FunctionCallInfo fcinfo)
               errmsg("could not identify a comparison function for type %s",
                      format_type_be(element_type))));
        fcinfo->flinfo->fn_extra = (void *) typentry;
-       typentry->cmp_proc_finfo.fn_collation = collation;
    }
    typlen = typentry->typlen;
    typbyval = typentry->typbyval;
@@ -3388,7 +3387,7 @@ array_cmp(FunctionCallInfo fcinfo)
     * apply the operator to each pair of array elements.
     */
    InitFunctionCallInfoData(locfcinfo, &typentry->cmp_proc_finfo, 2,
-                            NULL, NULL);
+                            collation, NULL, NULL);
 
    /* Loop over source data */
    min_nitems = Min(nitems1, nitems2);
@@ -3573,7 +3572,7 @@ hash_array(PG_FUNCTION_ARGS)
     * apply the hash function to each array element.
     */
    InitFunctionCallInfoData(locfcinfo, &typentry->hash_proc_finfo, 1,
-                            NULL, NULL);
+                            InvalidOid, NULL, NULL);
 
    /* Loop over source data */
    nitems = ArrayGetNItems(ndims, dims);
@@ -3647,8 +3646,8 @@ hash_array(PG_FUNCTION_ARGS)
  * When matchall is false, return true if any members of array1 are in array2.
  */
 static bool
-array_contain_compare(ArrayType *array1, ArrayType *array2, bool matchall,
-                     void **fn_extra)
+array_contain_compare(ArrayType *array1, ArrayType *array2, Oid collation,
+                     bool matchall, void **fn_extra)
 {
    bool        result = matchall;
    Oid         element_type = ARR_ELEMTYPE(array1);
@@ -3707,7 +3706,7 @@ array_contain_compare(ArrayType *array1, ArrayType *array2, bool matchall,
     * Apply the comparison operator to each pair of array elements.
     */
    InitFunctionCallInfoData(locfcinfo, &typentry->eq_opr_finfo, 2,
-                            NULL, NULL);
+                            collation, NULL, NULL);
 
    /* Loop over source data */
    nelems1 = ArrayGetNItems(ARR_NDIM(array1), ARR_DIMS(array1));
@@ -3811,9 +3810,10 @@ arrayoverlap(PG_FUNCTION_ARGS)
 {
    ArrayType  *array1 = PG_GETARG_ARRAYTYPE_P(0);
    ArrayType  *array2 = PG_GETARG_ARRAYTYPE_P(1);
+   Oid         collation = PG_GET_COLLATION();
    bool        result;
 
-   result = array_contain_compare(array1, array2, false,
+   result = array_contain_compare(array1, array2, collation, false,
                                   &fcinfo->flinfo->fn_extra);
 
    /* Avoid leaking memory when handed toasted input. */
@@ -3828,9 +3828,10 @@ arraycontains(PG_FUNCTION_ARGS)
 {
    ArrayType  *array1 = PG_GETARG_ARRAYTYPE_P(0);
    ArrayType  *array2 = PG_GETARG_ARRAYTYPE_P(1);
+   Oid         collation = PG_GET_COLLATION();
    bool        result;
 
-   result = array_contain_compare(array2, array1, true,
+   result = array_contain_compare(array2, array1, collation, true,
                                   &fcinfo->flinfo->fn_extra);
 
    /* Avoid leaking memory when handed toasted input. */
@@ -3845,9 +3846,10 @@ arraycontained(PG_FUNCTION_ARGS)
 {
    ArrayType  *array1 = PG_GETARG_ARRAYTYPE_P(0);
    ArrayType  *array2 = PG_GETARG_ARRAYTYPE_P(1);
+   Oid         collation = PG_GET_COLLATION();
    bool        result;
 
-   result = array_contain_compare(array1, array2, true,
+   result = array_contain_compare(array1, array2, collation, true,
                                   &fcinfo->flinfo->fn_extra);
 
    /* Avoid leaking memory when handed toasted input. */
index dcb31586b2b3e21aa817fac856d5349befadcb24..019fcaaa36f23296e9ca1482f00cf839574e238a 100644 (file)
@@ -213,7 +213,8 @@ int2vectorrecv(PG_FUNCTION_ARGS)
     * fcinfo->flinfo->fn_extra.  So we need to pass it our own flinfo
     * parameter.
     */
-   InitFunctionCallInfoData(locfcinfo, fcinfo->flinfo, 3, NULL, NULL);
+   InitFunctionCallInfoData(locfcinfo, fcinfo->flinfo, 3,
+                            InvalidOid, NULL, NULL);
 
    locfcinfo.arg[0] = PointerGetDatum(buf);
    locfcinfo.arg[1] = ObjectIdGetDatum(INT2OID);
index 0934c69ebe5682102a59554b750c865599c8f79b..09e8698af2cb88a0e29da0e92e2f25b375cb7b99 100644 (file)
@@ -174,10 +174,12 @@ Generic_Text_IC_like(text *str, text *pat, Oid collation)
    if (pg_database_encoding_max_length() > 1)
    {
        /* lower's result is never packed, so OK to use old macros here */
-       pat = DatumGetTextP(DirectFunctionCall1WithCollation(lower, collation, PointerGetDatum(pat)));
+       pat = DatumGetTextP(DirectFunctionCall1Coll(lower, collation,
+                                                   PointerGetDatum(pat)));
        p = VARDATA(pat);
        plen = (VARSIZE(pat) - VARHDRSZ);
-       str = DatumGetTextP(DirectFunctionCall1WithCollation(lower, collation, PointerGetDatum(str)));
+       str = DatumGetTextP(DirectFunctionCall1Coll(lower, collation,
+                                                   PointerGetDatum(str)));
        s = VARDATA(str);
        slen = (VARSIZE(str) - VARHDRSZ);
        if (GetDatabaseEncoding() == PG_UTF8)
index b2152a2491cf5d11f12c23316840501e7e68ba67..495b6261e62050b3e02dc918ac2fa3981b9f1e32 100644 (file)
@@ -263,7 +263,8 @@ oidvectorrecv(PG_FUNCTION_ARGS)
     * fcinfo->flinfo->fn_extra.  So we need to pass it our own flinfo
     * parameter.
     */
-   InitFunctionCallInfoData(locfcinfo, fcinfo->flinfo, 3, NULL, NULL);
+   InitFunctionCallInfoData(locfcinfo, fcinfo->flinfo, 3,
+                            InvalidOid, NULL, NULL);
 
    locfcinfo.arg[0] = PointerGetDatum(buf);
    locfcinfo.arg[1] = ObjectIdGetDatum(OIDOID);
index 4e5dd4b772d7b101cebc5b4cea7ba38dc9510a4b..fde01a0f5719aa8e3fbd90997a2e803523d80591 100644 (file)
@@ -3963,7 +3963,10 @@ ri_AttributesEqual(Oid eq_opr, Oid typeid,
                                 BoolGetDatum(false));  /* implicit coercion */
    }
 
-   /* Apply the comparison operator */
+   /*
+    * Apply the comparison operator.  We assume it doesn't
+    * care about collations.
+    */
    return DatumGetBool(FunctionCall2(&entry->eq_opr_finfo,
                                      oldvalue, newvalue));
 }
index 2c552e173bb1747898fd3046c253dcda5f3783a3..919ed9582a70a9b1d9696150c3e0116048d86afa 100644 (file)
@@ -867,6 +867,7 @@ record_cmp(FunctionCallInfo fcinfo)
    while (i1 < ncolumns1 || i2 < ncolumns2)
    {
        TypeCacheEntry *typentry;
+       Oid         collation;
        FunctionCallInfoData locfcinfo;
        int32       cmpresult;
 
@@ -898,6 +899,14 @@ record_cmp(FunctionCallInfo fcinfo)
                            format_type_be(tupdesc2->attrs[i2]->atttypid),
                            j + 1)));
 
+       /*
+        * If they're not same collation, we don't complain here, but the
+        * comparison function might.
+        */
+       collation = tupdesc1->attrs[i1]->attcollation;
+       if (collation != tupdesc2->attrs[i2]->attcollation)
+           collation = InvalidOid;
+
        /*
         * Lookup the comparison function if not done already
         */
@@ -935,7 +944,7 @@ record_cmp(FunctionCallInfo fcinfo)
 
            /* Compare the pair of elements */
            InitFunctionCallInfoData(locfcinfo, &typentry->cmp_proc_finfo, 2,
-                                    NULL, NULL);
+                                    collation, NULL, NULL);
            locfcinfo.arg[0] = values1[i1];
            locfcinfo.arg[1] = values2[i2];
            locfcinfo.argnull[0] = false;
@@ -1093,6 +1102,7 @@ record_eq(PG_FUNCTION_ARGS)
    while (i1 < ncolumns1 || i2 < ncolumns2)
    {
        TypeCacheEntry *typentry;
+       Oid         collation;
        FunctionCallInfoData locfcinfo;
        bool        oprresult;
 
@@ -1124,6 +1134,14 @@ record_eq(PG_FUNCTION_ARGS)
                            format_type_be(tupdesc2->attrs[i2]->atttypid),
                            j + 1)));
 
+       /*
+        * If they're not same collation, we don't complain here, but the
+        * equality function might.
+        */
+       collation = tupdesc1->attrs[i1]->attcollation;
+       if (collation != tupdesc2->attrs[i2]->attcollation)
+           collation = InvalidOid;
+
        /*
         * Lookup the equality function if not done already
         */
@@ -1154,7 +1172,7 @@ record_eq(PG_FUNCTION_ARGS)
 
            /* Compare the pair of elements */
            InitFunctionCallInfoData(locfcinfo, &typentry->eq_opr_finfo, 2,
-                                    NULL, NULL);
+                                    collation, NULL, NULL);
            locfcinfo.arg[0] = values1[i1];
            locfcinfo.arg[1] = values2[i2];
            locfcinfo.argnull[0] = false;
index b046bc05e6245e45c704f05491a3b12c4f9619f1..4d800f8a007980dbac1d221d48f0fa8b0e30db66 100644 (file)
@@ -285,19 +285,20 @@ var_eq_const(VariableStatData *vardata, Oid operator,
            FmgrInfo    eqproc;
 
            fmgr_info(get_opcode(operator), &eqproc);
-           fmgr_info_set_collation(DEFAULT_COLLATION_OID, &eqproc);
 
            for (i = 0; i < nvalues; i++)
            {
                /* be careful to apply operator right way 'round */
                if (varonleft)
-                   match = DatumGetBool(FunctionCall2(&eqproc,
-                                                      values[i],
-                                                      constval));
+                   match = DatumGetBool(FunctionCall2Coll(&eqproc,
+                                                          DEFAULT_COLLATION_OID,
+                                                          values[i],
+                                                          constval));
                else
-                   match = DatumGetBool(FunctionCall2(&eqproc,
-                                                      constval,
-                                                      values[i]));
+                   match = DatumGetBool(FunctionCall2Coll(&eqproc,
+                                                          DEFAULT_COLLATION_OID,
+                                                          constval,
+                                                          values[i]));
                if (match)
                    break;
            }
@@ -515,7 +516,6 @@ scalarineqsel(PlannerInfo *root, Oid operator, bool isgt,
    stats = (Form_pg_statistic) GETSTRUCT(vardata->statsTuple);
 
    fmgr_info(get_opcode(operator), &opproc);
-   fmgr_info_set_collation(DEFAULT_COLLATION_OID, &opproc);
 
    /*
     * If we have most-common-values info, add up the fractions of the MCV
@@ -598,12 +598,14 @@ mcv_selectivity(VariableStatData *vardata, FmgrInfo *opproc,
        for (i = 0; i < nvalues; i++)
        {
            if (varonleft ?
-               DatumGetBool(FunctionCall2(opproc,
-                                          values[i],
-                                          constval)) :
-               DatumGetBool(FunctionCall2(opproc,
-                                          constval,
-                                          values[i])))
+               DatumGetBool(FunctionCall2Coll(opproc,
+                                              DEFAULT_COLLATION_OID,
+                                              values[i],
+                                              constval)) :
+               DatumGetBool(FunctionCall2Coll(opproc,
+                                              DEFAULT_COLLATION_OID,
+                                              constval,
+                                              values[i])))
                mcv_selec += numbers[i];
            sumcommon += numbers[i];
        }
@@ -678,12 +680,14 @@ histogram_selectivity(VariableStatData *vardata, FmgrInfo *opproc,
            for (i = n_skip; i < nvalues - n_skip; i++)
            {
                if (varonleft ?
-                   DatumGetBool(FunctionCall2(opproc,
-                                              values[i],
-                                              constval)) :
-                   DatumGetBool(FunctionCall2(opproc,
-                                              constval,
-                                              values[i])))
+                   DatumGetBool(FunctionCall2Coll(opproc,
+                                                  DEFAULT_COLLATION_OID,
+                                                  values[i],
+                                                  constval)) :
+                   DatumGetBool(FunctionCall2Coll(opproc,
+                                                  DEFAULT_COLLATION_OID,
+                                                  constval,
+                                                  values[i])))
                    nmatch++;
            }
            result = ((double) nmatch) / ((double) (nvalues - 2 * n_skip));
@@ -802,9 +806,10 @@ ineq_histogram_selectivity(PlannerInfo *root,
                                                         NULL,
                                                         &values[probe]);
 
-               ltcmp = DatumGetBool(FunctionCall2(opproc,
-                                                  values[probe],
-                                                  constval));
+               ltcmp = DatumGetBool(FunctionCall2Coll(opproc,
+                                                      DEFAULT_COLLATION_OID,
+                                                      values[probe],
+                                                      constval));
                if (isgt)
                    ltcmp = !ltcmp;
                if (ltcmp)
@@ -1255,7 +1260,6 @@ patternsel(PG_FUNCTION_ARGS, Pattern_Type ptype, bool negate)
 
        /* Try to use the histogram entries to get selectivity */
        fmgr_info(get_opcode(operator), &opproc);
-       fmgr_info_set_collation(DEFAULT_COLLATION_OID, &opproc);
 
        selec = histogram_selectivity(&vardata, &opproc, constval, true,
                                      10, 1, &hist_size);
@@ -1705,7 +1709,6 @@ scalararraysel(PlannerInfo *root,
    if (!oprsel)
        return (Selectivity) 0.5;
    fmgr_info(oprsel, &oprselproc);
-   fmgr_info_set_collation(DEFAULT_COLLATION_OID, &oprselproc);
 
    /* deconstruct the expression */
    Assert(list_length(clause->args) == 2);
@@ -2126,7 +2129,6 @@ eqjoinsel_inner(Oid operator,
                    nmatches;
 
        fmgr_info(get_opcode(operator), &eqproc);
-       fmgr_info_set_collation(DEFAULT_COLLATION_OID, &eqproc);
        hasmatch1 = (bool *) palloc0(nvalues1 * sizeof(bool));
        hasmatch2 = (bool *) palloc0(nvalues2 * sizeof(bool));
 
@@ -2146,9 +2148,10 @@ eqjoinsel_inner(Oid operator,
            {
                if (hasmatch2[j])
                    continue;
-               if (DatumGetBool(FunctionCall2(&eqproc,
-                                              values1[i],
-                                              values2[j])))
+               if (DatumGetBool(FunctionCall2Coll(&eqproc,
+                                                  DEFAULT_COLLATION_OID,
+                                                  values1[i],
+                                                  values2[j])))
                {
                    hasmatch1[i] = hasmatch2[j] = true;
                    matchprodfreq += numbers1[i] * numbers2[j];
@@ -2349,7 +2352,6 @@ eqjoinsel_semi(Oid operator,
                    nmatches;
 
        fmgr_info(get_opcode(operator), &eqproc);
-       fmgr_info_set_collation(DEFAULT_COLLATION_OID, &eqproc);
        hasmatch1 = (bool *) palloc0(nvalues1 * sizeof(bool));
        hasmatch2 = (bool *) palloc0(nvalues2 * sizeof(bool));
 
@@ -2368,9 +2370,10 @@ eqjoinsel_semi(Oid operator,
            {
                if (hasmatch2[j])
                    continue;
-               if (DatumGetBool(FunctionCall2(&eqproc,
-                                              values1[i],
-                                              values2[j])))
+               if (DatumGetBool(FunctionCall2Coll(&eqproc,
+                                                  DEFAULT_COLLATION_OID,
+                                                  values1[i],
+                                                  values2[j])))
                {
                    hasmatch1[i] = hasmatch2[j] = true;
                    nmatches++;
@@ -4503,7 +4506,6 @@ get_variable_range(PlannerInfo *root, VariableStatData *vardata, Oid sortop,
        FmgrInfo    opproc;
 
        fmgr_info(get_opcode(sortop), &opproc);
-       fmgr_info_set_collation(DEFAULT_COLLATION_OID, &opproc);
 
        for (i = 0; i < nvalues; i++)
        {
@@ -4513,12 +4515,16 @@ get_variable_range(PlannerInfo *root, VariableStatData *vardata, Oid sortop,
                tmin_is_mcv = tmax_is_mcv = have_data = true;
                continue;
            }
-           if (DatumGetBool(FunctionCall2(&opproc, values[i], tmin)))
+           if (DatumGetBool(FunctionCall2Coll(&opproc,
+                                              DEFAULT_COLLATION_OID,
+                                              values[i], tmin)))
            {
                tmin = values[i];
                tmin_is_mcv = true;
            }
-           if (DatumGetBool(FunctionCall2(&opproc, tmax, values[i])))
+           if (DatumGetBool(FunctionCall2Coll(&opproc,
+                                              DEFAULT_COLLATION_OID,
+                                              tmax, values[i])))
            {
                tmax = values[i];
                tmax_is_mcv = true;
@@ -5183,7 +5189,6 @@ prefix_selectivity(PlannerInfo *root, VariableStatData *vardata,
    if (cmpopr == InvalidOid)
        elog(ERROR, "no >= operator for opfamily %u", opfamily);
    fmgr_info(get_opcode(cmpopr), &opproc);
-   fmgr_info_set_collation(DEFAULT_COLLATION_OID, &opproc);
 
    prefixsel = ineq_histogram_selectivity(root, vardata, &opproc, true,
                                           prefixcon->constvalue,
@@ -5205,9 +5210,8 @@ prefix_selectivity(PlannerInfo *root, VariableStatData *vardata,
    if (cmpopr == InvalidOid)
        elog(ERROR, "no < operator for opfamily %u", opfamily);
    fmgr_info(get_opcode(cmpopr), &opproc);
-   fmgr_info_set_collation(DEFAULT_COLLATION_OID, &opproc);
-
-   greaterstrcon = make_greater_string(prefixcon, &opproc);
+   greaterstrcon = make_greater_string(prefixcon, &opproc,
+                                       DEFAULT_COLLATION_OID);
    if (greaterstrcon)
    {
        Selectivity topsel;
@@ -5502,22 +5506,21 @@ pattern_selectivity(Const *patt, Pattern_Type ptype)
  * in the form of a Const node; else return NULL.
  *
  * The caller must provide the appropriate "less than" comparison function
- * for testing the strings.  In particular, ltproc->fn_collation specifies
- * the locale for comparisons.
+ * for testing the strings, along with the collation to use.
  *
  * The key requirement here is that given a prefix string, say "foo",
  * we must be able to generate another string "fop" that is greater than
  * all strings "foobar" starting with "foo".  We can test that we have
- * generated a string greater than the prefix string, but in non-C locales
+ * generated a string greater than the prefix string, but in non-C collations
  * that is not a bulletproof guarantee that an extension of the string might
  * not sort after it; an example is that "foo " is less than "foo!", but it
  * is not clear that a "dictionary" sort ordering will consider "foo!" less
  * than "foo bar". CAUTION: Therefore, this function should be used only for
- * estimation purposes when working in a non-C locale.
+ * estimation purposes when working in a non-C collation.
  *
  * To try to catch most cases where an extended string might otherwise sort
  * before the result value, we determine which of the strings "Z", "z", "y",
- * and "9" is seen as largest by the locale, and append that to the given
+ * and "9" is seen as largest by the collation, and append that to the given
  * prefix before trying to find a string that compares as larger.
  *
  * If we max out the righthand byte, truncate off the last character
@@ -5529,7 +5532,7 @@ pattern_selectivity(Const *patt, Pattern_Type ptype)
  * won't have to try more than one or two strings before succeeding.
  */
 Const *
-make_greater_string(const Const *str_const, FmgrInfo *ltproc)
+make_greater_string(const Const *str_const, FmgrInfo *ltproc, Oid collation)
 {
    Oid         datatype = str_const->consttype;
    char       *workstr;
@@ -5565,7 +5568,7 @@ make_greater_string(const Const *str_const, FmgrInfo *ltproc)
    {
        workstr = TextDatumGetCString(str_const->constvalue);
        len = strlen(workstr);
-       if (lc_collate_is_c(ltproc->fn_collation) || len == 0)
+       if (lc_collate_is_c(collation) || len == 0)
            cmpstr = str_const->constvalue;
        else
        {
@@ -5573,19 +5576,19 @@ make_greater_string(const Const *str_const, FmgrInfo *ltproc)
            static char suffixchar = 0;
            static Oid  suffixcollation = 0;
 
-           if (!suffixchar || suffixcollation != ltproc->fn_collation)
+           if (!suffixchar || suffixcollation != collation)
            {
                char       *best;
 
                best = "Z";
-               if (varstr_cmp(best, 1, "z", 1, ltproc->fn_collation) < 0)
+               if (varstr_cmp(best, 1, "z", 1, collation) < 0)
                    best = "z";
-               if (varstr_cmp(best, 1, "y", 1, ltproc->fn_collation) < 0)
+               if (varstr_cmp(best, 1, "y", 1, collation) < 0)
                    best = "y";
-               if (varstr_cmp(best, 1, "9", 1, ltproc->fn_collation) < 0)
+               if (varstr_cmp(best, 1, "9", 1, collation) < 0)
                    best = "9";
                suffixchar = *best;
-               suffixcollation = ltproc->fn_collation;
+               suffixcollation = collation;
            }
 
            /* And build the string to compare to */
@@ -5621,9 +5624,10 @@ make_greater_string(const Const *str_const, FmgrInfo *ltproc)
            else
                workstr_const = string_to_bytea_const(workstr, len);
 
-           if (DatumGetBool(FunctionCall2(ltproc,
-                                          cmpstr,
-                                          workstr_const->constvalue)))
+           if (DatumGetBool(FunctionCall2Coll(ltproc,
+                                              collation,
+                                              cmpstr,
+                                              workstr_const->constvalue)))
            {
                /* Successfully made a string larger than cmpstr */
                if (cmptxt)
index 5caa53d4d25f67d8fbc01c96ceb0a82edce974b7..350e040474b413dec4280fd1e63c1f0613c5dcde 100644 (file)
@@ -935,7 +935,7 @@ CatalogCacheInitializeCache(CatCache *cache)
        cache->cc_skey[i].sk_strategy = BTEqualStrategyNumber;
        cache->cc_skey[i].sk_subtype = InvalidOid;
        /* Currently, there are no catcaches on collation-aware data types */
-       cache->cc_skey[i].sk_func.fn_collation = InvalidOid;
+       cache->cc_skey[i].sk_collation = InvalidOid;
 
        CACHE4_elog(DEBUG2, "CatalogCacheInitializeCache %s %d %p",
                    cache->cc_relname,
index 6647fe95f729a316198039bd75dd45db6dc084dd..a650eb14a370bbe07f077fb65b319cd7be161efb 100644 (file)
@@ -71,7 +71,6 @@ typedef struct
     bool        fn_strict;  /* function is "strict" (NULL in => NULL out) */
     bool        fn_retset;  /* function returns a set (over multiple calls) */
     unsigned char fn_stats; /* collect stats if track_functions > this */
-    Oid         fn_collation;   /* collation that function should use */
     void       *fn_extra;   /* extra space for use by handler */
     MemoryContext fn_mcxt;  /* memory context to store fn_extra in */
     Node       *fn_expr;    /* expression parse tree for call, or NULL */
@@ -92,14 +91,12 @@ these values come from the function's pg_proc entry.  fn_stats is also
 set up to control whether or not to track runtime statistics for calling
 this function.
 
-fn_collation supplies the collation to use for collation-sensitive
-functions.  If the function is being called as part of a SQL expression,
-fn_expr will point to the expression parse tree for the function call; this
-can be used to extract parse-time knowledge about the actual arguments.
-Note that these two fields really are information about the arguments
-rather than information about the function, but it's proven to be more
-convenient to keep them in FmgrInfo than in FunctionCallInfoData where
-they might more logically go.
+If the function is being called as part of a SQL expression, fn_expr will
+point to the expression parse tree for the function call; this can be used
+to extract parse-time knowledge about the actual arguments.  Note that this
+field really is information about the arguments rather than information
+about the function, but it's proven to be more convenient to keep it in
+FmgrInfo than in FunctionCallInfoData where it might more logically go.
 
 
 During a call of a function, the following data structure is created
@@ -110,6 +107,7 @@ typedef struct
     FmgrInfo   *flinfo;         /* ptr to lookup info used for this call */
     Node       *context;        /* pass info about context of call */
     Node       *resultinfo;     /* pass or return extra info about result */
+    Oid         fncollation;    /* collation for function to use */
     bool        isnull;         /* function must set true if result is NULL */
     short       nargs;          /* # arguments actually passed */
     Datum       arg[FUNC_MAX_ARGS];  /* Arguments passed to function */
@@ -137,6 +135,11 @@ function that returns a set, as discussed below.)  Like the context field,
 resultinfo is a hook for expansion; fmgr itself doesn't constrain the use
 of the field.
 
+fncollation is the input collation derived by the parser, or InvalidOid
+when there are no inputs of collatable types or they don't share a common
+collation.  This is effectively a hidden additional argument, which
+collation-sensitive functions can use to determine their behavior.
+
 nargs, arg[], and argnull[] hold the arguments being passed to the function.
 Notice that all the arguments passed to a function (as well as its result
 value) will now uniformly be of type Datum.  As discussed below, callers
index 0288fbab4bc65e4d4c38bfac150815309f3c59e9..ffa19d5a2d26efce4fd4b121a0cab536b48d3aae 100644 (file)
@@ -192,7 +192,6 @@ fmgr_info_cxt_security(Oid functionId, FmgrInfo *finfo, MemoryContext mcxt,
     * elogs.
     */
    finfo->fn_oid = InvalidOid;
-   finfo->fn_collation = InvalidOid;   /* caller may set this later */
    finfo->fn_extra = NULL;
    finfo->fn_mcxt = mcxt;
    finfo->fn_expr = NULL;      /* caller may set this later */
@@ -901,7 +900,6 @@ fmgr_security_definer(PG_FUNCTION_ARGS)
 
        fmgr_info_cxt_security(fcinfo->flinfo->fn_oid, &fcache->flinfo,
                               fcinfo->flinfo->fn_mcxt, true);
-       fcache->flinfo.fn_collation = fcinfo->flinfo->fn_collation;
        fcache->flinfo.fn_expr = fcinfo->flinfo->fn_expr;
 
        tuple = SearchSysCache1(PROCOID,
@@ -1012,12 +1010,12 @@ fmgr_security_definer(PG_FUNCTION_ARGS)
  * look at FmgrInfo, since there won't be any.
  */
 Datum
-DirectFunctionCall1(PGFunction func, Datum arg1)
+DirectFunctionCall1Coll(PGFunction func, Oid collation, Datum arg1)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 1, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 1, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.argnull[0] = false;
@@ -1032,12 +1030,12 @@ DirectFunctionCall1(PGFunction func, Datum arg1)
 }
 
 Datum
-DirectFunctionCall2(PGFunction func, Datum arg1, Datum arg2)
+DirectFunctionCall2Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 2, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 2, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1054,13 +1052,13 @@ DirectFunctionCall2(PGFunction func, Datum arg1, Datum arg2)
 }
 
 Datum
-DirectFunctionCall3(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall3Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 3, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 3, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1079,13 +1077,13 @@ DirectFunctionCall3(PGFunction func, Datum arg1, Datum arg2,
 }
 
 Datum
-DirectFunctionCall4(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall4Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 4, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 4, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1106,13 +1104,13 @@ DirectFunctionCall4(PGFunction func, Datum arg1, Datum arg2,
 }
 
 Datum
-DirectFunctionCall5(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall5Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 5, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 5, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1135,14 +1133,14 @@ DirectFunctionCall5(PGFunction func, Datum arg1, Datum arg2,
 }
 
 Datum
-DirectFunctionCall6(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall6Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 6, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 6, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1167,14 +1165,14 @@ DirectFunctionCall6(PGFunction func, Datum arg1, Datum arg2,
 }
 
 Datum
-DirectFunctionCall7(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall7Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6, Datum arg7)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 7, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 7, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1201,14 +1199,14 @@ DirectFunctionCall7(PGFunction func, Datum arg1, Datum arg2,
 }
 
 Datum
-DirectFunctionCall8(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall8Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6, Datum arg7, Datum arg8)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 8, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 8, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1237,7 +1235,7 @@ DirectFunctionCall8(PGFunction func, Datum arg1, Datum arg2,
 }
 
 Datum
-DirectFunctionCall9(PGFunction func, Datum arg1, Datum arg2,
+DirectFunctionCall9Coll(PGFunction func, Oid collation, Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6, Datum arg7, Datum arg8,
                    Datum arg9)
@@ -1245,7 +1243,7 @@ DirectFunctionCall9(PGFunction func, Datum arg1, Datum arg2,
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, NULL, 9, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, NULL, 9, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1276,72 +1274,18 @@ DirectFunctionCall9(PGFunction func, Datum arg1, Datum arg2,
 }
 
 
-/*
- * These are the same as DirectFunctionCallN except that a nonzero
- * collation can be specified. No other fields of FmgrInfo are made valid.
- */
-Datum
-DirectFunctionCall1WithCollation(PGFunction func, Oid collation, Datum arg1)
-{
-   FunctionCallInfoData fcinfo;
-   FmgrInfo    flinfo;
-   Datum       result;
-
-   MemSet(&flinfo, 0, sizeof(flinfo));
-   flinfo.fn_collation = collation;
-   InitFunctionCallInfoData(fcinfo, &flinfo, 1, NULL, NULL);
-
-   fcinfo.arg[0] = arg1;
-   fcinfo.argnull[0] = false;
-
-   result = (*func) (&fcinfo);
-
-   /* Check for null result, since caller is clearly not expecting one */
-   if (fcinfo.isnull)
-       elog(ERROR, "function %p returned NULL", (void *) func);
-
-   return result;
-}
-
-Datum
-DirectFunctionCall2WithCollation(PGFunction func, Oid collation,
-                                Datum arg1, Datum arg2)
-{
-   FunctionCallInfoData fcinfo;
-   FmgrInfo    flinfo;
-   Datum       result;
-
-   MemSet(&flinfo, 0, sizeof(flinfo));
-   flinfo.fn_collation = collation;
-   InitFunctionCallInfoData(fcinfo, &flinfo, 2, NULL, NULL);
-
-   fcinfo.arg[0] = arg1;
-   fcinfo.arg[1] = arg2;
-   fcinfo.argnull[0] = false;
-   fcinfo.argnull[1] = false;
-
-   result = (*func) (&fcinfo);
-
-   /* Check for null result, since caller is clearly not expecting one */
-   if (fcinfo.isnull)
-       elog(ERROR, "function %p returned NULL", (void *) func);
-
-   return result;
-}
-
-
 /*
  * These are for invocation of a previously-looked-up function with a
  * directly-computed parameter list.  Note that neither arguments nor result
  * are allowed to be NULL.
  */
 Datum
-FunctionCall1(FmgrInfo *flinfo, Datum arg1)
+FunctionCall1Coll(FmgrInfo *flinfo, Oid collation, Datum arg1)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 1, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 1, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.argnull[0] = false;
@@ -1356,7 +1300,7 @@ FunctionCall1(FmgrInfo *flinfo, Datum arg1)
 }
 
 Datum
-FunctionCall2(FmgrInfo *flinfo, Datum arg1, Datum arg2)
+FunctionCall2Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2)
 {
    /*
     * XXX if you change this routine, see also the inlined version in
@@ -1365,7 +1309,7 @@ FunctionCall2(FmgrInfo *flinfo, Datum arg1, Datum arg2)
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 2, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 2, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1382,13 +1326,13 @@ FunctionCall2(FmgrInfo *flinfo, Datum arg1, Datum arg2)
 }
 
 Datum
-FunctionCall3(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall3Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 3, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 3, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1407,13 +1351,13 @@ FunctionCall3(FmgrInfo *flinfo, Datum arg1, Datum arg2,
 }
 
 Datum
-FunctionCall4(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall4Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3, Datum arg4)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 4, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 4, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1434,13 +1378,13 @@ FunctionCall4(FmgrInfo *flinfo, Datum arg1, Datum arg2,
 }
 
 Datum
-FunctionCall5(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall5Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 5, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 5, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1463,14 +1407,14 @@ FunctionCall5(FmgrInfo *flinfo, Datum arg1, Datum arg2,
 }
 
 Datum
-FunctionCall6(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall6Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 6, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 6, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1495,14 +1439,14 @@ FunctionCall6(FmgrInfo *flinfo, Datum arg1, Datum arg2,
 }
 
 Datum
-FunctionCall7(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall7Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6, Datum arg7)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 7, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 7, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1529,14 +1473,14 @@ FunctionCall7(FmgrInfo *flinfo, Datum arg1, Datum arg2,
 }
 
 Datum
-FunctionCall8(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall8Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6, Datum arg7, Datum arg8)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 8, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 8, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1565,7 +1509,7 @@ FunctionCall8(FmgrInfo *flinfo, Datum arg1, Datum arg2,
 }
 
 Datum
-FunctionCall9(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+FunctionCall9Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6, Datum arg7, Datum arg8,
              Datum arg9)
@@ -1573,7 +1517,7 @@ FunctionCall9(FmgrInfo *flinfo, Datum arg1, Datum arg2,
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 9, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 9, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1612,7 +1556,7 @@ FunctionCall9(FmgrInfo *flinfo, Datum arg1, Datum arg2,
  * do the fmgr_info() once and then use FunctionCallN().
  */
 Datum
-OidFunctionCall0(Oid functionId)
+OidFunctionCall0Coll(Oid functionId, Oid collation)
 {
    FmgrInfo    flinfo;
    FunctionCallInfoData fcinfo;
@@ -1620,7 +1564,7 @@ OidFunctionCall0(Oid functionId)
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 0, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 0, collation, NULL, NULL);
 
    result = FunctionCallInvoke(&fcinfo);
 
@@ -1632,7 +1576,7 @@ OidFunctionCall0(Oid functionId)
 }
 
 Datum
-OidFunctionCall1(Oid functionId, Datum arg1)
+OidFunctionCall1Coll(Oid functionId, Oid collation, Datum arg1)
 {
    FmgrInfo    flinfo;
    FunctionCallInfoData fcinfo;
@@ -1640,7 +1584,7 @@ OidFunctionCall1(Oid functionId, Datum arg1)
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 1, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 1, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.argnull[0] = false;
@@ -1655,7 +1599,7 @@ OidFunctionCall1(Oid functionId, Datum arg1)
 }
 
 Datum
-OidFunctionCall2(Oid functionId, Datum arg1, Datum arg2)
+OidFunctionCall2Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2)
 {
    FmgrInfo    flinfo;
    FunctionCallInfoData fcinfo;
@@ -1663,7 +1607,7 @@ OidFunctionCall2(Oid functionId, Datum arg1, Datum arg2)
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 2, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 2, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1680,7 +1624,7 @@ OidFunctionCall2(Oid functionId, Datum arg1, Datum arg2)
 }
 
 Datum
-OidFunctionCall3(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall3Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3)
 {
    FmgrInfo    flinfo;
@@ -1689,7 +1633,7 @@ OidFunctionCall3(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 3, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 3, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1708,7 +1652,7 @@ OidFunctionCall3(Oid functionId, Datum arg1, Datum arg2,
 }
 
 Datum
-OidFunctionCall4(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall4Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4)
 {
    FmgrInfo    flinfo;
@@ -1717,7 +1661,7 @@ OidFunctionCall4(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 4, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 4, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1738,7 +1682,7 @@ OidFunctionCall4(Oid functionId, Datum arg1, Datum arg2,
 }
 
 Datum
-OidFunctionCall5(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall5Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5)
 {
    FmgrInfo    flinfo;
@@ -1747,7 +1691,7 @@ OidFunctionCall5(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 5, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 5, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1770,7 +1714,7 @@ OidFunctionCall5(Oid functionId, Datum arg1, Datum arg2,
 }
 
 Datum
-OidFunctionCall6(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall6Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6)
 {
@@ -1780,7 +1724,7 @@ OidFunctionCall6(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 6, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 6, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1805,7 +1749,7 @@ OidFunctionCall6(Oid functionId, Datum arg1, Datum arg2,
 }
 
 Datum
-OidFunctionCall7(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall7Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6, Datum arg7)
 {
@@ -1815,7 +1759,7 @@ OidFunctionCall7(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 7, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 7, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1842,7 +1786,7 @@ OidFunctionCall7(Oid functionId, Datum arg1, Datum arg2,
 }
 
 Datum
-OidFunctionCall8(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall8Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6, Datum arg7, Datum arg8)
 {
@@ -1852,7 +1796,7 @@ OidFunctionCall8(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 8, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 8, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1881,7 +1825,7 @@ OidFunctionCall8(Oid functionId, Datum arg1, Datum arg2,
 }
 
 Datum
-OidFunctionCall9(Oid functionId, Datum arg1, Datum arg2,
+OidFunctionCall9Coll(Oid functionId, Oid collation, Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6, Datum arg7, Datum arg8,
                 Datum arg9)
@@ -1892,7 +1836,7 @@ OidFunctionCall9(Oid functionId, Datum arg1, Datum arg2,
 
    fmgr_info(functionId, &flinfo);
 
-   InitFunctionCallInfoData(fcinfo, &flinfo, 9, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, &flinfo, 9, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -1953,7 +1897,7 @@ InputFunctionCall(FmgrInfo *flinfo, char *str, Oid typioparam, int32 typmod)
 
    pushed = SPI_push_conditional();
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 3, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 3, InvalidOid, NULL, NULL);
 
    fcinfo.arg[0] = CStringGetDatum(str);
    fcinfo.arg[1] = ObjectIdGetDatum(typioparam);
@@ -2028,7 +1972,7 @@ ReceiveFunctionCall(FmgrInfo *flinfo, StringInfo buf,
 
    pushed = SPI_push_conditional();
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 3, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 3, InvalidOid, NULL, NULL);
 
    fcinfo.arg[0] = PointerGetDatum(buf);
    fcinfo.arg[1] = ObjectIdGetDatum(typioparam);
index bd5b4b0a7d7d22ebd5693c8df7fc84cd20f4e704..dcfb8f957a9bb203f4ab9ffefb140ac9141c28dc 100644 (file)
@@ -373,6 +373,7 @@ struct Tuplesortstate
    Oid         datumType;
    FmgrInfo    sortOpFn;       /* cached lookup data for sortOperator */
    int         sortFnFlags;    /* equivalent to sk_flags */
+   Oid         sortCollation;  /* equivalent to sk_collation */
    /* we need typelen and byval in order to know how to copy the Datums. */
    int         datumTypeLen;
    bool        datumTypeByVal;
@@ -582,7 +583,8 @@ tuplesort_begin_common(int workMem, bool randomAccess)
 Tuplesortstate *
 tuplesort_begin_heap(TupleDesc tupDesc,
                     int nkeys, AttrNumber *attNums,
-                 Oid *sortOperators, Oid *collations, bool *nullsFirstFlags,
+                    Oid *sortOperators, Oid *sortCollations,
+                    bool *nullsFirstFlags,
                     int workMem, bool randomAccess)
 {
    Tuplesortstate *state = tuplesort_begin_common(workMem, randomAccess);
@@ -647,7 +649,7 @@ tuplesort_begin_heap(TupleDesc tupDesc,
                               attNums[i],
                               InvalidStrategy,
                               InvalidOid,
-                              collations ? collations[i] : InvalidOid,
+                              sortCollations[i],
                               sortFunction,
                               (Datum) 0);
    }
@@ -795,8 +797,8 @@ tuplesort_begin_index_hash(Relation indexRel,
 }
 
 Tuplesortstate *
-tuplesort_begin_datum(Oid datumType,
-                   Oid sortOperator, Oid sortCollation, bool nullsFirstFlag,
+tuplesort_begin_datum(Oid datumType, Oid sortOperator, Oid sortCollation,
+                     bool nullsFirstFlag,
                      int workMem, bool randomAccess)
 {
    Tuplesortstate *state = tuplesort_begin_common(workMem, randomAccess);
@@ -837,12 +839,12 @@ tuplesort_begin_datum(Oid datumType,
        elog(ERROR, "operator %u is not a valid ordering operator",
             sortOperator);
    fmgr_info(sortFunction, &state->sortOpFn);
-   fmgr_info_set_collation(sortCollation, &state->sortOpFn);
 
-   /* set ordering flags */
+   /* set ordering flags and collation */
    state->sortFnFlags = reverse ? SK_BT_DESC : 0;
    if (nullsFirstFlag)
        state->sortFnFlags |= SK_BT_NULLS_FIRST;
+   state->sortCollation = sortCollation;
 
    /* lookup necessary attributes of the datum type */
    get_typlenbyval(datumType, &typlen, &typbyval);
@@ -2630,15 +2632,15 @@ SelectSortFunction(Oid sortOperator,
 }
 
 /*
- * Inline-able copy of FunctionCall2() to save some cycles in sorting.
+ * Inline-able copy of FunctionCall2Coll() to save some cycles in sorting.
  */
 static inline Datum
-myFunctionCall2(FmgrInfo *flinfo, Datum arg1, Datum arg2)
+myFunctionCall2Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2)
 {
    FunctionCallInfoData fcinfo;
    Datum       result;
 
-   InitFunctionCallInfoData(fcinfo, flinfo, 2, NULL, NULL);
+   InitFunctionCallInfoData(fcinfo, flinfo, 2, collation, NULL, NULL);
 
    fcinfo.arg[0] = arg1;
    fcinfo.arg[1] = arg2;
@@ -2661,7 +2663,7 @@ myFunctionCall2(FmgrInfo *flinfo, Datum arg1, Datum arg2)
  * NULLS_FIRST options are encoded in sk_flags the same way btree does it.
  */
 static inline int32
-inlineApplySortFunction(FmgrInfo *sortFunction, int sk_flags,
+inlineApplySortFunction(FmgrInfo *sortFunction, int sk_flags, Oid collation,
                        Datum datum1, bool isNull1,
                        Datum datum2, bool isNull2)
 {
@@ -2685,8 +2687,8 @@ inlineApplySortFunction(FmgrInfo *sortFunction, int sk_flags,
    }
    else
    {
-       compare = DatumGetInt32(myFunctionCall2(sortFunction,
-                                               datum1, datum2));
+       compare = DatumGetInt32(myFunctionCall2Coll(sortFunction, collation,
+                                                   datum1, datum2));
 
        if (sk_flags & SK_BT_DESC)
            compare = -compare;
@@ -2700,11 +2702,11 @@ inlineApplySortFunction(FmgrInfo *sortFunction, int sk_flags,
  * C99's brain-dead notions about how to implement inline functions...
  */
 int32
-ApplySortFunction(FmgrInfo *sortFunction, int sortFlags,
+ApplySortFunction(FmgrInfo *sortFunction, int sortFlags, Oid collation,
                  Datum datum1, bool isNull1,
                  Datum datum2, bool isNull2)
 {
-   return inlineApplySortFunction(sortFunction, sortFlags,
+   return inlineApplySortFunction(sortFunction, sortFlags, collation,
                                   datum1, isNull1,
                                   datum2, isNull2);
 }
@@ -2729,6 +2731,7 @@ comparetup_heap(const SortTuple *a, const SortTuple *b, Tuplesortstate *state)
 
    /* Compare the leading sort key */
    compare = inlineApplySortFunction(&scanKey->sk_func, scanKey->sk_flags,
+                                     scanKey->sk_collation,
                                      a->datum1, a->isnull1,
                                      b->datum1, b->isnull1);
    if (compare != 0)
@@ -2753,6 +2756,7 @@ comparetup_heap(const SortTuple *a, const SortTuple *b, Tuplesortstate *state)
        datum2 = heap_getattr(&rtup, attno, tupDesc, &isnull2);
 
        compare = inlineApplySortFunction(&scanKey->sk_func, scanKey->sk_flags,
+                                         scanKey->sk_collation,
                                          datum1, isnull1,
                                          datum2, isnull2);
        if (compare != 0)
@@ -2874,6 +2878,7 @@ comparetup_cluster(const SortTuple *a, const SortTuple *b,
    if (state->indexInfo->ii_KeyAttrNumbers[0] != 0)
    {
        compare = inlineApplySortFunction(&scanKey->sk_func, scanKey->sk_flags,
+                                         scanKey->sk_collation,
                                          a->datum1, a->isnull1,
                                          b->datum1, b->isnull1);
        if (compare != 0 || state->nKeys == 1)
@@ -2910,6 +2915,7 @@ comparetup_cluster(const SortTuple *a, const SortTuple *b,
 
            compare = inlineApplySortFunction(&scanKey->sk_func,
                                              scanKey->sk_flags,
+                                             scanKey->sk_collation,
                                              datum1, isnull1,
                                              datum2, isnull2);
            if (compare != 0)
@@ -2947,6 +2953,7 @@ comparetup_cluster(const SortTuple *a, const SortTuple *b,
        {
            compare = inlineApplySortFunction(&scanKey->sk_func,
                                              scanKey->sk_flags,
+                                             scanKey->sk_collation,
                                              l_index_values[nkey],
                                              l_index_isnull[nkey],
                                              r_index_values[nkey],
@@ -3060,6 +3067,7 @@ comparetup_index_btree(const SortTuple *a, const SortTuple *b,
 
    /* Compare the leading sort key */
    compare = inlineApplySortFunction(&scanKey->sk_func, scanKey->sk_flags,
+                                     scanKey->sk_collation,
                                      a->datum1, a->isnull1,
                                      b->datum1, b->isnull1);
    if (compare != 0)
@@ -3086,6 +3094,7 @@ comparetup_index_btree(const SortTuple *a, const SortTuple *b,
        datum2 = index_getattr(tuple2, nkey, tupDes, &isnull2);
 
        compare = inlineApplySortFunction(&scanKey->sk_func, scanKey->sk_flags,
+                                         scanKey->sk_collation,
                                          datum1, isnull1,
                                          datum2, isnull2);
        if (compare != 0)
@@ -3293,6 +3302,7 @@ comparetup_datum(const SortTuple *a, const SortTuple *b, Tuplesortstate *state)
    CHECK_FOR_INTERRUPTS();
 
    return inlineApplySortFunction(&state->sortOpFn, state->sortFnFlags,
+                                  state->sortCollation,
                                   a->datum1, a->isnull1,
                                   b->datum1, b->isnull1);
 }
index 74c1098458c2e5108e95507e6da00ca98b5534c6..06c6fa2f9c6ec66682bb05ba8fdaac94a8751882 100644 (file)
@@ -303,6 +303,8 @@ typedef struct GinState
    FmgrInfo    comparePartialFn[INDEX_MAX_KEYS];       /* optional method */
    /* canPartialMatch[i] is true if comparePartialFn[i] is valid */
    bool        canPartialMatch[INDEX_MAX_KEYS];
+   /* Collations to supply to the compareFns and comparePartialFns */
+   Oid         compareCollation[INDEX_MAX_KEYS];
 } GinState;
 
 /* XLog stuff */
index 1d0071ac2d33ae414b416515ea1f814aace0649d..a82e46ee0e1b8c5ed8ebaff11cc031ff6764358c 100644 (file)
@@ -52,16 +52,16 @@ typedef uint16 StrategyNumber;
  * the operator.  When using a ScanKey in a heap scan, these fields are not
  * used and may be set to InvalidStrategy/InvalidOid.
  *
- * If the operator is collation-sensitive, sk_func.fn_collation must be set
+ * If the operator is collation-sensitive, sk_collation must be set
  * correctly as well.
  *
  * A ScanKey can also represent a condition "column IS NULL" or "column
  * IS NOT NULL"; these cases are signaled by the SK_SEARCHNULL and
  * SK_SEARCHNOTNULL flag bits respectively.  The argument is always NULL,
- * and the sk_strategy, sk_subtype, and sk_func fields are not used (unless
- * set by the index AM).  Currently, SK_SEARCHNULL and SK_SEARCHNOTNULL are
- * supported only for index scans, not heap scans; and not all index AMs
- * support them.
+ * and the sk_strategy, sk_subtype, sk_collation, and sk_func fields are
+ * not used (unless set by the index AM).  Currently, SK_SEARCHNULL and
+ * SK_SEARCHNOTNULL are supported only for index scans, not heap scans;
+ * and not all index AMs support them.
  *
  * A ScanKey can also represent an ordering operator invocation, that is
  * an ordering requirement "ORDER BY indexedcol op constant".  This looks
@@ -70,8 +70,8 @@ typedef uint16 StrategyNumber;
  *
  * Note: in some places, ScanKeys are used as a convenient representation
  * for the invocation of an access method support procedure.  In this case
- * sk_strategy/sk_subtype are not meaningful, and sk_func may refer to a
- * function that returns something other than boolean.
+ * sk_strategy/sk_subtype are not meaningful (but sk_collation can be); and
+ * sk_func may refer to a function that returns something other than boolean.
  */
 typedef struct ScanKeyData
 {
@@ -79,6 +79,7 @@ typedef struct ScanKeyData
    AttrNumber  sk_attno;       /* table or index column number */
    StrategyNumber sk_strategy; /* operator strategy number */
    Oid         sk_subtype;     /* strategy subtype */
+   Oid         sk_collation;   /* collation to use, if needed */
    FmgrInfo    sk_func;        /* lookup info for function to call */
    Datum       sk_argument;    /* data to compare */
 } ScanKeyData;
@@ -99,7 +100,7 @@ typedef ScanKeyData *ScanKey;
  *     sk_attno = index column number for leading column of row comparison
  *     sk_strategy = btree strategy code for semantics of row comparison
  *             (ie, < <= > or >=)
- *     sk_subtype, sk_func: not used
+ *     sk_subtype, sk_collation, sk_func: not used
  *     sk_argument: pointer to subsidiary ScanKey array
  * If the header is part of a ScanKey array that's sorted by attno, it
  * must be sorted according to the leading column number.
index e81f4cb723bfb02eadf51ffd057bdb39aa1a8b06..f52ff4fb17952c0bc6bc7cd090e0bcf21701ad54 100644 (file)
@@ -54,8 +54,9 @@ do \
            break; \
        } \
  \
-       __test = FunctionCall2(&__cur_keys->sk_func, \
-                              __atp, __cur_keys->sk_argument); \
+       __test = FunctionCall2Coll(&__cur_keys->sk_func, \
+                                  __cur_keys->sk_collation, \
+                                  __atp, __cur_keys->sk_argument); \
  \
        if (!DatumGetBool(__test)) \
        { \
index e58060f834615d41181835a67b61f6054fd6d260..60d47d97b9d6e4033c73a57cc0da07794bb60471 100644 (file)
@@ -41,10 +41,10 @@ typedef Datum (*PGFunction) (FunctionCallInfo fcinfo);
  * to be called multiple times, the lookup need be done only once and the
  * info struct saved for re-use.
  *
- * Note that fn_collation and fn_expr really are parse-time-determined
- * information about the arguments, rather than about the function itself.
- * But it's convenient to store them here rather than in FunctionCallInfoData,
- * where they might more logically belong.
+ * Note that fn_expr really is parse-time-determined information about the
+ * arguments, rather than about the function itself.  But it's convenient
+ * to store it here rather than in FunctionCallInfoData, where it might more
+ * logically belong.
  */
 typedef struct FmgrInfo
 {
@@ -55,7 +55,6 @@ typedef struct FmgrInfo
    bool        fn_strict;      /* function is "strict" (NULL in => NULL out) */
    bool        fn_retset;      /* function returns a set */
    unsigned char fn_stats;     /* collect stats if track_functions > this */
-   Oid         fn_collation;   /* collation that function should use */
    void       *fn_extra;       /* extra space for use by handler */
    MemoryContext fn_mcxt;      /* memory context to store fn_extra in */
    fmNodePtr   fn_expr;        /* expression parse tree for call, or NULL */
@@ -69,6 +68,7 @@ typedef struct FunctionCallInfoData
    FmgrInfo   *flinfo;         /* ptr to lookup info used for this call */
    fmNodePtr   context;        /* pass info about context of call */
    fmNodePtr   resultinfo;     /* pass or return extra info about result */
+   Oid         fncollation;    /* collation for function to use */
    bool        isnull;         /* function must set true if result is NULL */
    short       nargs;          /* # arguments actually passed */
    Datum       arg[FUNC_MAX_ARGS];     /* Arguments passed to function */
@@ -89,9 +89,7 @@ extern void fmgr_info(Oid functionId, FmgrInfo *finfo);
 extern void fmgr_info_cxt(Oid functionId, FmgrInfo *finfo,
              MemoryContext mcxt);
 
-/* Macros for setting the fn_collation and fn_expr fields */
-#define fmgr_info_set_collation(collationId, finfo) \
-   ((finfo)->fn_collation = (collationId))
+/* Convenience macro for setting the fn_expr field */
 #define fmgr_info_set_expr(expr, finfo) \
    ((finfo)->fn_expr = (expr))
 
@@ -108,11 +106,12 @@ extern void fmgr_info_copy(FmgrInfo *dstinfo, FmgrInfo *srcinfo,
  * explicitly set each required element to false, so we don't try to zero
  * out the argnull[] array in the macro.
  */
-#define InitFunctionCallInfoData(Fcinfo, Flinfo, Nargs, Context, Resultinfo) \
+#define InitFunctionCallInfoData(Fcinfo, Flinfo, Nargs, Collation, Context, Resultinfo) \
    do { \
        (Fcinfo).flinfo = (Flinfo); \
        (Fcinfo).context = (Context); \
        (Fcinfo).resultinfo = (Resultinfo); \
+       (Fcinfo).fncollation = (Collation); \
        (Fcinfo).isnull = false; \
        (Fcinfo).nargs = (Nargs); \
    } while (0)
@@ -150,8 +149,7 @@ extern void fmgr_info_copy(FmgrInfo *dstinfo, FmgrInfo *srcinfo,
 /*
  * Get collation function should use.
  */
-#define PG_GET_COLLATION() \
-   (fcinfo->flinfo ? fcinfo->flinfo->fn_collation : InvalidOid)
+#define PG_GET_COLLATION() (fcinfo->fncollation)
 
 /*
  * Get number of arguments passed to function.
@@ -434,56 +432,68 @@ extern int no_such_variable
  * directly-computed parameter list.  Note that neither arguments nor result
  * are allowed to be NULL.
  */
-extern Datum DirectFunctionCall1(PGFunction func, Datum arg1);
-extern Datum DirectFunctionCall2(PGFunction func, Datum arg1, Datum arg2);
-extern Datum DirectFunctionCall3(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall1Coll(PGFunction func, Oid collation,
+                                    Datum arg1);
+extern Datum DirectFunctionCall2Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2);
+extern Datum DirectFunctionCall3Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3);
-extern Datum DirectFunctionCall4(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall4Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4);
-extern Datum DirectFunctionCall5(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall5Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5);
-extern Datum DirectFunctionCall6(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall6Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6);
-extern Datum DirectFunctionCall7(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall7Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6, Datum arg7);
-extern Datum DirectFunctionCall8(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall8Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6, Datum arg7, Datum arg8);
-extern Datum DirectFunctionCall9(PGFunction func, Datum arg1, Datum arg2,
+extern Datum DirectFunctionCall9Coll(PGFunction func, Oid collation,
+                                    Datum arg1, Datum arg2,
                    Datum arg3, Datum arg4, Datum arg5,
                    Datum arg6, Datum arg7, Datum arg8,
                    Datum arg9);
 
-/* The same, but passing a collation to use */
-extern Datum DirectFunctionCall1WithCollation(PGFunction func, Oid collation,
-                                Datum arg1);
-extern Datum DirectFunctionCall2WithCollation(PGFunction func, Oid collation,
-                                Datum arg1, Datum arg2);
-
 /* These are for invocation of a previously-looked-up function with a
  * directly-computed parameter list.  Note that neither arguments nor result
  * are allowed to be NULL.
  */
-extern Datum FunctionCall1(FmgrInfo *flinfo, Datum arg1);
-extern Datum FunctionCall2(FmgrInfo *flinfo, Datum arg1, Datum arg2);
-extern Datum FunctionCall3(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall1Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1);
+extern Datum FunctionCall2Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2);
+extern Datum FunctionCall3Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3);
-extern Datum FunctionCall4(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall4Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3, Datum arg4);
-extern Datum FunctionCall5(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall5Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5);
-extern Datum FunctionCall6(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall6Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6);
-extern Datum FunctionCall7(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall7Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6, Datum arg7);
-extern Datum FunctionCall8(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall8Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6, Datum arg7, Datum arg8);
-extern Datum FunctionCall9(FmgrInfo *flinfo, Datum arg1, Datum arg2,
+extern Datum FunctionCall9Coll(FmgrInfo *flinfo, Oid collation,
+                              Datum arg1, Datum arg2,
              Datum arg3, Datum arg4, Datum arg5,
              Datum arg6, Datum arg7, Datum arg8,
              Datum arg9);
@@ -494,29 +504,100 @@ extern Datum FunctionCall9(FmgrInfo *flinfo, Datum arg1, Datum arg2,
  * by FunctionCallN(). If the same function is to be invoked repeatedly,
  * do the FunctionLookup() once and then use FunctionCallN().
  */
-extern Datum OidFunctionCall0(Oid functionId);
-extern Datum OidFunctionCall1(Oid functionId, Datum arg1);
-extern Datum OidFunctionCall2(Oid functionId, Datum arg1, Datum arg2);
-extern Datum OidFunctionCall3(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall0Coll(Oid functionId, Oid collation);
+extern Datum OidFunctionCall1Coll(Oid functionId, Oid collation,
+                                 Datum arg1);
+extern Datum OidFunctionCall2Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2);
+extern Datum OidFunctionCall3Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3);
-extern Datum OidFunctionCall4(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall4Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4);
-extern Datum OidFunctionCall5(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall5Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5);
-extern Datum OidFunctionCall6(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall6Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6);
-extern Datum OidFunctionCall7(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall7Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6, Datum arg7);
-extern Datum OidFunctionCall8(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall8Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6, Datum arg7, Datum arg8);
-extern Datum OidFunctionCall9(Oid functionId, Datum arg1, Datum arg2,
+extern Datum OidFunctionCall9Coll(Oid functionId, Oid collation,
+                                 Datum arg1, Datum arg2,
                 Datum arg3, Datum arg4, Datum arg5,
                 Datum arg6, Datum arg7, Datum arg8,
                 Datum arg9);
 
+/* These macros allow the collation argument to be omitted (with a default of
+ * InvalidOid, ie, no collation).  They exist mostly for backwards
+ * compatibility of source code.
+ */
+#define DirectFunctionCall1(func, arg1) \
+   DirectFunctionCall1Coll(func, InvalidOid, arg1)
+#define DirectFunctionCall2(func, arg1, arg2) \
+   DirectFunctionCall2Coll(func, InvalidOid, arg1, arg2)
+#define DirectFunctionCall3(func, arg1, arg2, arg3) \
+   DirectFunctionCall3Coll(func, InvalidOid, arg1, arg2, arg3)
+#define DirectFunctionCall4(func, arg1, arg2, arg3, arg4) \
+   DirectFunctionCall4Coll(func, InvalidOid, arg1, arg2, arg3, arg4)
+#define DirectFunctionCall5(func, arg1, arg2, arg3, arg4, arg5) \
+   DirectFunctionCall5Coll(func, InvalidOid, arg1, arg2, arg3, arg4, arg5)
+#define DirectFunctionCall6(func, arg1, arg2, arg3, arg4, arg5, arg6) \
+   DirectFunctionCall6Coll(func, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6)
+#define DirectFunctionCall7(func, arg1, arg2, arg3, arg4, arg5, arg6, arg7) \
+   DirectFunctionCall7Coll(func, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7)
+#define DirectFunctionCall8(func, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8) \
+   DirectFunctionCall8Coll(func, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8)
+#define DirectFunctionCall9(func, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9) \
+   DirectFunctionCall9Coll(func, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9)
+#define FunctionCall1(flinfo, arg1) \
+   FunctionCall1Coll(flinfo, InvalidOid, arg1)
+#define FunctionCall2(flinfo, arg1, arg2) \
+   FunctionCall2Coll(flinfo, InvalidOid, arg1, arg2)
+#define FunctionCall3(flinfo, arg1, arg2, arg3) \
+   FunctionCall3Coll(flinfo, InvalidOid, arg1, arg2, arg3)
+#define FunctionCall4(flinfo, arg1, arg2, arg3, arg4) \
+   FunctionCall4Coll(flinfo, InvalidOid, arg1, arg2, arg3, arg4)
+#define FunctionCall5(flinfo, arg1, arg2, arg3, arg4, arg5) \
+   FunctionCall5Coll(flinfo, InvalidOid, arg1, arg2, arg3, arg4, arg5)
+#define FunctionCall6(flinfo, arg1, arg2, arg3, arg4, arg5, arg6) \
+   FunctionCall6Coll(flinfo, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6)
+#define FunctionCall7(flinfo, arg1, arg2, arg3, arg4, arg5, arg6, arg7) \
+   FunctionCall7Coll(flinfo, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7)
+#define FunctionCall8(flinfo, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8) \
+   FunctionCall8Coll(flinfo, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8)
+#define FunctionCall9(flinfo, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9) \
+   FunctionCall9Coll(flinfo, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9)
+#define OidFunctionCall0(functionId) \
+   OidFunctionCall0Coll(functionId, InvalidOid)
+#define OidFunctionCall1(functionId, arg1) \
+   OidFunctionCall1Coll(functionId, InvalidOid, arg1)
+#define OidFunctionCall2(functionId, arg1, arg2) \
+   OidFunctionCall2Coll(functionId, InvalidOid, arg1, arg2)
+#define OidFunctionCall3(functionId, arg1, arg2, arg3) \
+   OidFunctionCall3Coll(functionId, InvalidOid, arg1, arg2, arg3)
+#define OidFunctionCall4(functionId, arg1, arg2, arg3, arg4) \
+   OidFunctionCall4Coll(functionId, InvalidOid, arg1, arg2, arg3, arg4)
+#define OidFunctionCall5(functionId, arg1, arg2, arg3, arg4, arg5) \
+   OidFunctionCall5Coll(functionId, InvalidOid, arg1, arg2, arg3, arg4, arg5)
+#define OidFunctionCall6(functionId, arg1, arg2, arg3, arg4, arg5, arg6) \
+   OidFunctionCall6Coll(functionId, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6)
+#define OidFunctionCall7(functionId, arg1, arg2, arg3, arg4, arg5, arg6, arg7) \
+   OidFunctionCall7Coll(functionId, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7)
+#define OidFunctionCall8(functionId, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8) \
+   OidFunctionCall8Coll(functionId, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8)
+#define OidFunctionCall9(functionId, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9) \
+   OidFunctionCall9Coll(functionId, InvalidOid, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9)
+
+
 /* Special cases for convenient invocation of datatype I/O functions. */
 extern Datum InputFunctionCall(FmgrInfo *flinfo, char *str,
                  Oid typioparam, int32 typmod);
index 9c688c036861cca70b4bfdd4109acfb2b130cd18..16756616e5215503ba4e26889fce2b4d144d24de 100644 (file)
@@ -864,6 +864,7 @@ typedef struct RowCompareExprState
    List       *largs;          /* the left-hand input arguments */
    List       *rargs;          /* the right-hand input arguments */
    FmgrInfo   *funcs;          /* array of comparison function info */
+   Oid        *collations;     /* array of collations to use */
 } RowCompareExprState;
 
 /* ----------------
index c1b417ad8fbd3be553afcd529862c139b1e8c75f..dd38a0292fc763c6189f5133f8db681532da5347 100644 (file)
@@ -135,7 +135,8 @@ extern Pattern_Prefix_Status pattern_fixed_prefix(Const *patt,
                     Oid collation,
                     Const **prefix,
                     Const **rest);
-extern Const *make_greater_string(const Const *str_const, FmgrInfo *ltproc);
+extern Const *make_greater_string(const Const *str_const, FmgrInfo *ltproc,
+                                 Oid collation);
 
 extern Datum eqsel(PG_FUNCTION_ARGS);
 extern Datum neqsel(PG_FUNCTION_ARGS);
index a2085df8691fcbd9db29ac82f66b51d9a459c023..1ebcbfe17242a8e5262fe5d942210d0d77b1dd73 100644 (file)
@@ -60,7 +60,8 @@ typedef struct Tuplesortstate Tuplesortstate;
 
 extern Tuplesortstate *tuplesort_begin_heap(TupleDesc tupDesc,
                     int nkeys, AttrNumber *attNums,
-                 Oid *sortOperators, Oid *collations, bool *nullsFirstFlags,
+                    Oid *sortOperators, Oid *sortCollations,
+                    bool *nullsFirstFlags,
                     int workMem, bool randomAccess);
 extern Tuplesortstate *tuplesort_begin_cluster(TupleDesc tupDesc,
                        Relation indexRel,
@@ -72,7 +73,8 @@ extern Tuplesortstate *tuplesort_begin_index_hash(Relation indexRel,
                           uint32 hash_mask,
                           int workMem, bool randomAccess);
 extern Tuplesortstate *tuplesort_begin_datum(Oid datumType,
-                   Oid sortOperator, Oid sortCollation, bool nullsFirstFlag,
+                     Oid sortOperator, Oid sortCollation,
+                     bool nullsFirstFlag,
                      int workMem, bool randomAccess);
 
 extern void tuplesort_set_bound(Tuplesortstate *state, int64 bound);
@@ -125,6 +127,7 @@ extern void SelectSortFunction(Oid sortOperator, bool nulls_first,
  * reverse-sort and NULLs-ordering properly.
  */
 extern int32 ApplySortFunction(FmgrInfo *sortFunction, int sortFlags,
+                 Oid collation,
                  Datum datum1, bool isNull1,
                  Datum datum2, bool isNull2);
 
index 9f919b88c20546ed87600bc77b813c25f5724e94..535fea9257c22e5cc54dc956ebd59a6865340a55 100644 (file)
@@ -348,7 +348,7 @@ do_compile(FunctionCallInfo fcinfo,
    function->fn_xmin = HeapTupleHeaderGetXmin(procTup->t_data);
    function->fn_tid = procTup->t_self;
    function->fn_is_trigger = is_trigger;
-   function->fn_input_collation = fcinfo->flinfo->fn_collation;
+   function->fn_input_collation = fcinfo->fncollation;
    function->fn_cxt = func_cxt;
    function->out_param_varno = -1;     /* set up for no OUT param */
    function->resolve_option = plpgsql_variable_conflict;
@@ -2331,7 +2331,7 @@ compute_function_hashkey(FunctionCallInfo fcinfo,
    }
 
    /* get input collation, if known */
-   hashkey->inputCollation = fcinfo->flinfo->fn_collation;
+   hashkey->inputCollation = fcinfo->fncollation;
 
    if (procStruct->pronargs > 0)
    {