contrib/btree_gist/btree_text.c

   1 /*
   2  * $PostgreSQL:$
   3  */
   4 #include "btree_gist.h"
   5 #include "btree_utils_var.h"
   6 #include "utils/builtins.h"
   7
   8 /*
   9 ** Text ops
  10 */
  11 PG_FUNCTION_INFO_V1(gbt_text_compress);
  12 PG_FUNCTION_INFO_V1(gbt_bpchar_compress);
  13 PG_FUNCTION_INFO_V1(gbt_text_union);
  14 PG_FUNCTION_INFO_V1(gbt_text_picksplit);
  15 PG_FUNCTION_INFO_V1(gbt_text_consistent);
  16 PG_FUNCTION_INFO_V1(gbt_bpchar_consistent);
  17 PG_FUNCTION_INFO_V1(gbt_text_penalty);
  18 PG_FUNCTION_INFO_V1(gbt_text_same);
  19
  20 Datum           gbt_text_compress(PG_FUNCTION_ARGS);
  21 Datum           gbt_bpchar_compress(PG_FUNCTION_ARGS);
  22 Datum           gbt_text_union(PG_FUNCTION_ARGS);
  23 Datum           gbt_text_picksplit(PG_FUNCTION_ARGS);
  24 Datum           gbt_text_consistent(PG_FUNCTION_ARGS);
  25 Datum           gbt_bpchar_consistent(PG_FUNCTION_ARGS);
  26 Datum           gbt_text_penalty(PG_FUNCTION_ARGS);
  27 Datum           gbt_text_same(PG_FUNCTION_ARGS);
  28
  29
  30 /* define for comparison */
  31
  32 static bool
  33 gbt_textgt(const void *a, const void *b)
  34 {
  35         return (DatumGetBool(DirectFunctionCall2(text_gt, PointerGetDatum(a), PointerGetDatum(b))));
  36 }
  37
  38 static bool
  39 gbt_textge(const void *a, const void *b)
  40 {
  41         return (DatumGetBool(DirectFunctionCall2(text_ge, PointerGetDatum(a), PointerGetDatum(b))));
  42 }
  43
  44 static bool
  45 gbt_texteq(const void *a, const void *b)
  46 {
  47         return (DatumGetBool(DirectFunctionCall2(texteq, PointerGetDatum(a), PointerGetDatum(b))));
  48 }
  49
  50 static bool
  51 gbt_textle(const void *a, const void *b)
  52 {
  53         return (DatumGetBool(DirectFunctionCall2(text_le, PointerGetDatum(a), PointerGetDatum(b))));
  54 }
  55
  56 static bool
  57 gbt_textlt(const void *a, const void *b)
  58 {
  59         return (DatumGetBool(DirectFunctionCall2(text_lt, PointerGetDatum(a), PointerGetDatum(b))));
  60 }
  61
  62 static int32
  63 gbt_textcmp(const bytea *a, const bytea *b)
  64 {
  65         return DatumGetInt32(DirectFunctionCall2(bttextcmp, PointerGetDatum(a), PointerGetDatum(b)));
  66 }
  67
  68 static gbtree_vinfo tinfo =
  69 {
  70         gbt_t_text,
  71         0,
  72         FALSE,
  73         gbt_textgt,
  74         gbt_textge,
  75         gbt_texteq,
  76         gbt_textle,
  77         gbt_textlt,
  78         gbt_textcmp,
  79         NULL
  80 };
  81
  82
  83 /**************************************************
  84  * Text ops
  85  **************************************************/
  86
  87
  88 Datum
  89 gbt_text_compress(PG_FUNCTION_ARGS)
  90 {
  91         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
  92
  93         if (tinfo.eml == 0)
  94         {
  95                 tinfo.eml = pg_database_encoding_max_length();
  96         }
  97
  98         PG_RETURN_POINTER(gbt_var_compress(entry, &tinfo));
  99 }
 100
 101 Datum
 102 gbt_bpchar_compress(PG_FUNCTION_ARGS)
 103 {
 104
 105         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 106         GISTENTRY  *retval;
 107
 108         if (tinfo.eml == 0)
 109         {
 110                 tinfo.eml = pg_database_encoding_max_length();
 111         }
 112
 113         if (entry->leafkey)
 114         {
 115
 116                 Datum           d = DirectFunctionCall1(rtrim1, entry->key);
 117                 GISTENTRY       trim;
 118
 119                 gistentryinit(trim, d,
 120                                           entry->rel, entry->page,
 121                                           entry->offset, TRUE);
 122                 retval = gbt_var_compress(&trim, &tinfo);
 123         }
 124         else
 125                 retval = entry;
 126
 127         PG_RETURN_POINTER(retval);
 128 }
 129
 130
 131
 132 Datum
 133 gbt_text_consistent(PG_FUNCTION_ARGS)
 134 {
 135         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 136         void       *query = (void *) DatumGetTextP(PG_GETARG_DATUM(1));
 137         StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
 138         /* Oid          subtype = PG_GETARG_OID(3); */
 139         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 140         bool            retval;
 141         GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
 142         GBT_VARKEY_R r = gbt_var_key_readable(key);
 143
 144         /* All cases served by this function are exact */
 145         *recheck = false;
 146
 147         if (tinfo.eml == 0)
 148         {
 149                 tinfo.eml = pg_database_encoding_max_length();
 150         }
 151
 152         retval = gbt_var_consistent(&r, query, &strategy, GIST_LEAF(entry), &tinfo);
 153
 154         PG_RETURN_BOOL(retval);
 155 }
 156
 157
 158 Datum
 159 gbt_bpchar_consistent(PG_FUNCTION_ARGS)
 160 {
 161         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 162         void       *query = (void *) DatumGetPointer(PG_DETOAST_DATUM(PG_GETARG_DATUM(1)));
 163         StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
 164         /* Oid          subtype = PG_GETARG_OID(3); */
 165         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 166         bool            retval;
 167         GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
 168         GBT_VARKEY_R r = gbt_var_key_readable(key);
 169         void       *trim = (void *) DatumGetPointer(DirectFunctionCall1(rtrim1, PointerGetDatum(query)));
 170
 171         /* All cases served by this function are exact */
 172         *recheck = false;
 173
 174         if (tinfo.eml == 0)
 175         {
 176                 tinfo.eml = pg_database_encoding_max_length();
 177         }
 178
 179         retval = gbt_var_consistent(&r, trim, &strategy, GIST_LEAF(entry), &tinfo);
 180         PG_RETURN_BOOL(retval);
 181 }
 182
 183
 184 Datum
 185 gbt_text_union(PG_FUNCTION_ARGS)
 186 {
 187         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 188         int32      *size = (int *) PG_GETARG_POINTER(1);
 189
 190         PG_RETURN_POINTER(gbt_var_union(entryvec, size, &tinfo));
 191 }
 192
 193
 194 Datum
 195 gbt_text_picksplit(PG_FUNCTION_ARGS)
 196 {
 197         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 198         GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
 199
 200         gbt_var_picksplit(entryvec, v, &tinfo);
 201         PG_RETURN_POINTER(v);
 202 }
 203
 204 Datum
 205 gbt_text_same(PG_FUNCTION_ARGS)
 206 {
 207         Datum           d1 = PG_GETARG_DATUM(0);
 208         Datum           d2 = PG_GETARG_DATUM(1);
 209         bool       *result = (bool *) PG_GETARG_POINTER(2);
 210
 211         PG_RETURN_POINTER(gbt_var_same(result, d1, d2, &tinfo));
 212 }
 213
 214
 215 Datum
 216 gbt_text_penalty(PG_FUNCTION_ARGS)
 217 {
 218         GISTENTRY  *o = (GISTENTRY *) PG_GETARG_POINTER(0);
 219         GISTENTRY  *n = (GISTENTRY *) PG_GETARG_POINTER(1);
 220         float      *result = (float *) PG_GETARG_POINTER(2);
 221
 222         PG_RETURN_POINTER(gbt_var_penalty(result, o, n, &tinfo));
 223 }