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
 139         /* Oid          subtype = PG_GETARG_OID(3); */
 140         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 141         bool            retval;
 142         GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
 143         GBT_VARKEY_R r = gbt_var_key_readable(key);
 144
 145         /* All cases served by this function are exact */
 146         *recheck = false;
 147
 148         if (tinfo.eml == 0)
 149         {
 150                 tinfo.eml = pg_database_encoding_max_length();
 151         }
 152
 153         retval = gbt_var_consistent(&r, query, &strategy, GIST_LEAF(entry), &tinfo);
 154
 155         PG_RETURN_BOOL(retval);
 156 }
 157
 158
 159 Datum
 160 gbt_bpchar_consistent(PG_FUNCTION_ARGS)
 161 {
 162         GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
 163         void       *query = (void *) DatumGetPointer(PG_DETOAST_DATUM(PG_GETARG_DATUM(1)));
 164         StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
 165
 166         /* Oid          subtype = PG_GETARG_OID(3); */
 167         bool       *recheck = (bool *) PG_GETARG_POINTER(4);
 168         bool            retval;
 169         GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
 170         GBT_VARKEY_R r = gbt_var_key_readable(key);
 171         void       *trim = (void *) DatumGetPointer(DirectFunctionCall1(rtrim1, PointerGetDatum(query)));
 172
 173         /* All cases served by this function are exact */
 174         *recheck = false;
 175
 176         if (tinfo.eml == 0)
 177         {
 178                 tinfo.eml = pg_database_encoding_max_length();
 179         }
 180
 181         retval = gbt_var_consistent(&r, trim, &strategy, GIST_LEAF(entry), &tinfo);
 182         PG_RETURN_BOOL(retval);
 183 }
 184
 185
 186 Datum
 187 gbt_text_union(PG_FUNCTION_ARGS)
 188 {
 189         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 190         int32      *size = (int *) PG_GETARG_POINTER(1);
 191
 192         PG_RETURN_POINTER(gbt_var_union(entryvec, size, &tinfo));
 193 }
 194
 195
 196 Datum
 197 gbt_text_picksplit(PG_FUNCTION_ARGS)
 198 {
 199         GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
 200         GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
 201
 202         gbt_var_picksplit(entryvec, v, &tinfo);
 203         PG_RETURN_POINTER(v);
 204 }
 205
 206 Datum
 207 gbt_text_same(PG_FUNCTION_ARGS)
 208 {
 209         Datum           d1 = PG_GETARG_DATUM(0);
 210         Datum           d2 = PG_GETARG_DATUM(1);
 211         bool       *result = (bool *) PG_GETARG_POINTER(2);
 212
 213         PG_RETURN_POINTER(gbt_var_same(result, d1, d2, &tinfo));
 214 }
 215
 216
 217 Datum
 218 gbt_text_penalty(PG_FUNCTION_ARGS)
 219 {
 220         GISTENTRY  *o = (GISTENTRY *) PG_GETARG_POINTER(0);
 221         GISTENTRY  *n = (GISTENTRY *) PG_GETARG_POINTER(1);
 222         float      *result = (float *) PG_GETARG_POINTER(2);
 223
 224         PG_RETURN_POINTER(gbt_var_penalty(result, o, n, &tinfo));
 225 }