Fix tsvector_update_trigger() to be domain-friendly: it needs to allow all

the columns it works with to be domains over the expected type, not just
exactly the expected type.  In passing, fix ts_stat() the same way.
Per report from Markus Wollny.

diff --git a/src/backend/utils/adt/tsvector_op.c b/src/backend/utils/adt/tsvector_op.c
index 0624c00ea91f9b35f6601809e782238775ea70f9..539a9be9789cea96957530fd3588e5b6b0e99f37 100644 (file)
--- a/src/backend/utils/adt/tsvector_op.c
+++ b/src/backend/utils/adt/tsvector_op.c
@@ -7,7 +7,7 @@
  *
  *
  * IDENTIFICATION
- *       $PostgreSQL: pgsql/src/backend/utils/adt/tsvector_op.c,v 1.14 2008/03/25 22:42:44 tgl Exp $
+ *       $PostgreSQL: pgsql/src/backend/utils/adt/tsvector_op.c,v 1.15 2008/04/08 18:20:29 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -66,6 +66,39 @@ typedef struct
 static Datum tsvector_update_trigger(PG_FUNCTION_ARGS, bool config_column);
 
 
+/*
+ * Check if datatype is the specified type or equivalent to it.
+ *
+ * Note: we could just do getBaseType() unconditionally, but since that's
+ * a relatively expensive catalog lookup that most users won't need, we
+ * try the straight comparison first.
+ */
+static bool
+is_expected_type(Oid typid, Oid expected_type)
+{
+       if (typid == expected_type)
+               return true;
+       typid = getBaseType(typid);
+       if (typid == expected_type)
+               return true;
+       return false;
+}
+
+/* Check if datatype is TEXT or binary-equivalent to it */
+static bool
+is_text_type(Oid typid)
+{
+       /* varchar(n) and char(n) are binary-compatible with text */
+       if (typid == TEXTOID || typid == VARCHAROID || typid == BPCHAROID)
+               return true;
+       /* Allow domains over these types, too */
+       typid = getBaseType(typid);
+       if (typid == TEXTOID || typid == VARCHAROID || typid == BPCHAROID)
+               return true;
+       return false;
+}
+
+
 /*
  * Order: haspos, len, word, for all positions (pos, weight)
  */
@@ -1102,7 +1135,8 @@ ts_stat_sql(text *txt, text *ws)
 
        if (SPI_tuptable == NULL ||
                SPI_tuptable->tupdesc->natts != 1 ||
-               SPI_gettypeid(SPI_tuptable->tupdesc, 1) != TSVECTOROID)
+               !is_expected_type(SPI_gettypeid(SPI_tuptable->tupdesc, 1),
+                                                 TSVECTOROID))
                ereport(ERROR,
                                (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                                 errmsg("ts_stat query must return one tsvector column")));
@@ -1227,21 +1261,6 @@ ts_stat2(PG_FUNCTION_ARGS)
 }
 
 
-/* Check if datatype is TEXT or binary-equivalent to it */
-static bool
-istexttype(Oid typid)
-{
-       /* varchar(n) and char(n) are binary-compatible with text */
-       if (typid == TEXTOID || typid == VARCHAROID || typid == BPCHAROID)
-               return true;
-       /* Allow domains over these types, too */
-       typid = getBaseType(typid);
-       if (typid == TEXTOID || typid == VARCHAROID || typid == BPCHAROID)
-               return true;
-       return false;
-}
-
-
 /*
  * Triggers for automatic update of a tsvector column from text column(s)
  *
@@ -1309,7 +1328,8 @@ tsvector_update_trigger(PG_FUNCTION_ARGS, bool config_column)
                                (errcode(ERRCODE_UNDEFINED_COLUMN),
                                 errmsg("tsvector column \"%s\" does not exist",
                                                trigger->tgargs[0])));
-       if (SPI_gettypeid(rel->rd_att, tsvector_attr_num) != TSVECTOROID)
+       if (!is_expected_type(SPI_gettypeid(rel->rd_att, tsvector_attr_num),
+                                                 TSVECTOROID))
                ereport(ERROR,
                                (errcode(ERRCODE_DATATYPE_MISMATCH),
                                 errmsg("column \"%s\" is not of tsvector type",
@@ -1326,7 +1346,8 @@ tsvector_update_trigger(PG_FUNCTION_ARGS, bool config_column)
                                        (errcode(ERRCODE_UNDEFINED_COLUMN),
                                         errmsg("configuration column \"%s\" does not exist",
                                                        trigger->tgargs[1])));
-               if (SPI_gettypeid(rel->rd_att, config_attr_num) != REGCONFIGOID)
+               if (!is_expected_type(SPI_gettypeid(rel->rd_att, config_attr_num),
+                                                         REGCONFIGOID))
                        ereport(ERROR,
                                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                                         errmsg("column \"%s\" is not of regconfig type",
@@ -1371,7 +1392,7 @@ tsvector_update_trigger(PG_FUNCTION_ARGS, bool config_column)
                                        (errcode(ERRCODE_UNDEFINED_COLUMN),
                                         errmsg("column \"%s\" does not exist",
                                                        trigger->tgargs[i])));
-               if (!istexttype(SPI_gettypeid(rel->rd_att, numattr)))
+               if (!is_text_type(SPI_gettypeid(rel->rd_att, numattr)))
                        ereport(ERROR,
                                        (errcode(ERRCODE_DATATYPE_MISMATCH),
                                         errmsg("column \"%s\" is not of character type",