PostgreSQL Source Code git master
Macros | Functions | Variables
test_bitmapset.c File Reference
#include "postgres.h"
#include <stddef.h>
#include "catalog/pg_type.h"
#include "common/pg_prng.h"
#include "utils/array.h"
#include "fmgr.h"
#include "nodes/bitmapset.h"
#include "nodes/nodes.h"
#include "nodes/pg_list.h"
#include "utils/builtins.h"
#include "utils/timestamp.h"
Include dependency graph for test_bitmapset.c:

Go to the source code of this file.

Macros

#define  EXPECT_TRUE(expr)
 
#define  EXPECT_NOT_NULL(expr)
 
#define  BITMAPSET_TO_TEXT(bms)   cstring_to_text(nodeToString(bms))
 
#define  TEXT_TO_BITMAPSET(str)   ((Bitmapset *) stringToNode(text_to_cstring(str)))
 
#define  PG_ARG_GETBITMAPSET(n)    (PG_ARGISNULL(n) ? NULL : TEXT_TO_BITMAPSET(PG_GETARG_TEXT_PP(n)))
 
#define  PG_RETURN_BITMAPSET_AS_TEXT(bms)    PG_RETURN_TEXT_P(BITMAPSET_TO_TEXT(bms))
 

Functions

  PG_FUNCTION_INFO_V1 (test_bms_make_singleton)
 
  PG_FUNCTION_INFO_V1 (test_bms_add_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_del_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_is_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_num_members)
 
  PG_FUNCTION_INFO_V1 (test_bms_copy)
 
  PG_FUNCTION_INFO_V1 (test_bms_equal)
 
  PG_FUNCTION_INFO_V1 (test_bms_compare)
 
  PG_FUNCTION_INFO_V1 (test_bms_is_subset)
 
  PG_FUNCTION_INFO_V1 (test_bms_subset_compare)
 
  PG_FUNCTION_INFO_V1 (test_bms_union)
 
  PG_FUNCTION_INFO_V1 (test_bms_intersect)
 
  PG_FUNCTION_INFO_V1 (test_bms_difference)
 
  PG_FUNCTION_INFO_V1 (test_bms_is_empty)
 
  PG_FUNCTION_INFO_V1 (test_bms_membership)
 
  PG_FUNCTION_INFO_V1 (test_bms_singleton_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_get_singleton_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_next_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_prev_member)
 
  PG_FUNCTION_INFO_V1 (test_bms_hash_value)
 
  PG_FUNCTION_INFO_V1 (test_bms_overlap)
 
  PG_FUNCTION_INFO_V1 (test_bms_overlap_list)
 
  PG_FUNCTION_INFO_V1 (test_bms_nonempty_difference)
 
  PG_FUNCTION_INFO_V1 (test_bms_member_index)
 
  PG_FUNCTION_INFO_V1 (test_bms_add_range)
 
  PG_FUNCTION_INFO_V1 (test_bms_add_members)
 
  PG_FUNCTION_INFO_V1 (test_bms_int_members)
 
  PG_FUNCTION_INFO_V1 (test_bms_del_members)
 
  PG_FUNCTION_INFO_V1 (test_bms_replace_members)
 
  PG_FUNCTION_INFO_V1 (test_bms_join)
 
  PG_FUNCTION_INFO_V1 (test_bitmap_hash)
 
  PG_FUNCTION_INFO_V1 (test_bitmap_match)
 
  PG_FUNCTION_INFO_V1 (test_random_operations)
 
Datum  test_bms_add_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_add_members (PG_FUNCTION_ARGS)
 
Datum  test_bms_del_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_is_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_num_members (PG_FUNCTION_ARGS)
 
Datum  test_bms_make_singleton (PG_FUNCTION_ARGS)
 
Datum  test_bms_copy (PG_FUNCTION_ARGS)
 
Datum  test_bms_equal (PG_FUNCTION_ARGS)
 
Datum  test_bms_union (PG_FUNCTION_ARGS)
 
Datum  test_bms_membership (PG_FUNCTION_ARGS)
 
Datum  test_bms_next_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_intersect (PG_FUNCTION_ARGS)
 
Datum  test_bms_difference (PG_FUNCTION_ARGS)
 
Datum  test_bms_compare (PG_FUNCTION_ARGS)
 
Datum  test_bms_is_empty (PG_FUNCTION_ARGS)
 
Datum  test_bms_is_subset (PG_FUNCTION_ARGS)
 
Datum  test_bms_subset_compare (PG_FUNCTION_ARGS)
 
Datum  test_bms_singleton_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_get_singleton_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_prev_member (PG_FUNCTION_ARGS)
 
Datum  test_bms_overlap (PG_FUNCTION_ARGS)
 
Datum  test_bms_overlap_list (PG_FUNCTION_ARGS)
 
Datum  test_bms_nonempty_difference (PG_FUNCTION_ARGS)
 
Datum  test_bms_member_index (PG_FUNCTION_ARGS)
 
Datum  test_bms_add_range (PG_FUNCTION_ARGS)
 
Datum  test_bms_int_members (PG_FUNCTION_ARGS)
 
Datum  test_bms_del_members (PG_FUNCTION_ARGS)
 
Datum  test_bms_replace_members (PG_FUNCTION_ARGS)
 
Datum  test_bms_join (PG_FUNCTION_ARGS)
 
Datum  test_bms_hash_value (PG_FUNCTION_ARGS)
 
Datum  test_bitmap_hash (PG_FUNCTION_ARGS)
 
Datum  test_bitmap_match (PG_FUNCTION_ARGS)
 
Datum  test_random_operations (PG_FUNCTION_ARGS)
 

Variables

  PG_MODULE_MAGIC
 

Macro Definition Documentation

◆ BITMAPSET_TO_TEXT

#define BITMAPSET_TO_TEXT (   bms )    cstring_to_text(nodeToString(bms))

Definition at line 87 of file test_bitmapset.c.

◆ EXPECT_NOT_NULL

#define EXPECT_NOT_NULL (   expr )
Value:
do { \
if ((expr) == NULL) \
elog(ERROR, \
"%s was unexpectedly true in file \"%s\" line %u", \
#expr, __FILE__, __LINE__); \
} while (0)
#define ERROR
Definition: elog.h:39

Definition at line 78 of file test_bitmapset.c.

◆ EXPECT_TRUE

#define EXPECT_TRUE (   expr )
Value:
do { \
if (!(expr)) \
elog(ERROR, \
"%s was unexpectedly false in file \"%s\" line %u", \
#expr, __FILE__, __LINE__); \
} while (0)

Definition at line 70 of file test_bitmapset.c.

◆ PG_ARG_GETBITMAPSET

#define PG_ARG_GETBITMAPSET (   n )     (PG_ARGISNULL(n) ? NULL : TEXT_TO_BITMAPSET(PG_GETARG_TEXT_PP(n)))

Definition at line 94 of file test_bitmapset.c.

◆ PG_RETURN_BITMAPSET_AS_TEXT

#define PG_RETURN_BITMAPSET_AS_TEXT (   bms )     PG_RETURN_TEXT_P(BITMAPSET_TO_TEXT(bms))

Definition at line 101 of file test_bitmapset.c.

◆ TEXT_TO_BITMAPSET

#define TEXT_TO_BITMAPSET (   str )    ((Bitmapset *) stringToNode(text_to_cstring(str)))

Definition at line 88 of file test_bitmapset.c.

Function Documentation

◆ PG_FUNCTION_INFO_V1() [1/33]

PG_FUNCTION_INFO_V1 ( test_bitmap_hash  )

◆ PG_FUNCTION_INFO_V1() [2/33]

PG_FUNCTION_INFO_V1 ( test_bitmap_match  )

◆ PG_FUNCTION_INFO_V1() [3/33]

PG_FUNCTION_INFO_V1 ( test_bms_add_member  )

◆ PG_FUNCTION_INFO_V1() [4/33]

PG_FUNCTION_INFO_V1 ( test_bms_add_members  )

◆ PG_FUNCTION_INFO_V1() [5/33]

PG_FUNCTION_INFO_V1 ( test_bms_add_range  )

◆ PG_FUNCTION_INFO_V1() [6/33]

PG_FUNCTION_INFO_V1 ( test_bms_compare  )

◆ PG_FUNCTION_INFO_V1() [7/33]

PG_FUNCTION_INFO_V1 ( test_bms_copy  )

◆ PG_FUNCTION_INFO_V1() [8/33]

PG_FUNCTION_INFO_V1 ( test_bms_del_member  )

◆ PG_FUNCTION_INFO_V1() [9/33]

PG_FUNCTION_INFO_V1 ( test_bms_del_members  )

◆ PG_FUNCTION_INFO_V1() [10/33]

PG_FUNCTION_INFO_V1 ( test_bms_difference  )

◆ PG_FUNCTION_INFO_V1() [11/33]

PG_FUNCTION_INFO_V1 ( test_bms_equal  )

◆ PG_FUNCTION_INFO_V1() [12/33]

PG_FUNCTION_INFO_V1 ( test_bms_get_singleton_member  )

◆ PG_FUNCTION_INFO_V1() [13/33]

PG_FUNCTION_INFO_V1 ( test_bms_hash_value  )

◆ PG_FUNCTION_INFO_V1() [14/33]

PG_FUNCTION_INFO_V1 ( test_bms_int_members  )

◆ PG_FUNCTION_INFO_V1() [15/33]

PG_FUNCTION_INFO_V1 ( test_bms_intersect  )

◆ PG_FUNCTION_INFO_V1() [16/33]

PG_FUNCTION_INFO_V1 ( test_bms_is_empty  )

◆ PG_FUNCTION_INFO_V1() [17/33]

PG_FUNCTION_INFO_V1 ( test_bms_is_member  )

◆ PG_FUNCTION_INFO_V1() [18/33]

PG_FUNCTION_INFO_V1 ( test_bms_is_subset  )

◆ PG_FUNCTION_INFO_V1() [19/33]

PG_FUNCTION_INFO_V1 ( test_bms_join  )

◆ PG_FUNCTION_INFO_V1() [20/33]

PG_FUNCTION_INFO_V1 ( test_bms_make_singleton  )

◆ PG_FUNCTION_INFO_V1() [21/33]

PG_FUNCTION_INFO_V1 ( test_bms_member_index  )

◆ PG_FUNCTION_INFO_V1() [22/33]

PG_FUNCTION_INFO_V1 ( test_bms_membership  )

◆ PG_FUNCTION_INFO_V1() [23/33]

PG_FUNCTION_INFO_V1 ( test_bms_next_member  )

◆ PG_FUNCTION_INFO_V1() [24/33]

PG_FUNCTION_INFO_V1 ( test_bms_nonempty_difference  )

◆ PG_FUNCTION_INFO_V1() [25/33]

PG_FUNCTION_INFO_V1 ( test_bms_num_members  )

◆ PG_FUNCTION_INFO_V1() [26/33]

PG_FUNCTION_INFO_V1 ( test_bms_overlap  )

◆ PG_FUNCTION_INFO_V1() [27/33]

PG_FUNCTION_INFO_V1 ( test_bms_overlap_list  )

◆ PG_FUNCTION_INFO_V1() [28/33]

PG_FUNCTION_INFO_V1 ( test_bms_prev_member  )

◆ PG_FUNCTION_INFO_V1() [29/33]

PG_FUNCTION_INFO_V1 ( test_bms_replace_members  )

◆ PG_FUNCTION_INFO_V1() [30/33]

PG_FUNCTION_INFO_V1 ( test_bms_singleton_member  )

◆ PG_FUNCTION_INFO_V1() [31/33]

PG_FUNCTION_INFO_V1 ( test_bms_subset_compare  )

◆ PG_FUNCTION_INFO_V1() [32/33]

PG_FUNCTION_INFO_V1 ( test_bms_union  )

◆ PG_FUNCTION_INFO_V1() [33/33]

PG_FUNCTION_INFO_V1 ( test_random_operations  )

◆ test_bitmap_hash()

Datum test_bitmap_hash ( PG_FUNCTION_ARGS  )

Definition at line 558 of file test_bitmapset.c.

559{
561 uint32 hash_result;
562
563 /* Call bitmap_hash */
564 hash_result = bitmap_hash(&bms, sizeof(Bitmapset *));
565
566 PG_RETURN_INT32(hash_result);
567}
uint32 bitmap_hash(const void *key, Size keysize)
Definition: bitmapset.c:1436
uint32_t uint32
Definition: c.h:538
#define PG_RETURN_INT32(x)
Definition: fmgr.h:354
#define PG_ARG_GETBITMAPSET(n)
Definition: test_bitmapset.c:94

References bitmap_hash(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bitmap_match()

Datum test_bitmap_match ( PG_FUNCTION_ARGS  )

Definition at line 570 of file test_bitmapset.c.

571{
574 int match_result;
575
576 /* Call bitmap_match with addresses of the Bitmapset pointers */
577 match_result = bitmap_match(&bms1, &bms2, sizeof(Bitmapset *));
578
579 PG_RETURN_INT32(match_result);
580}
int bitmap_match(const void *key1, const void *key2, Size keysize)
Definition: bitmapset.c:1446

References bitmap_match(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_add_member()

Datum test_bms_add_member ( PG_FUNCTION_ARGS  )

Definition at line 124 of file test_bitmapset.c.

125{
126 Bitmapset *bms;
127 int member;
128
129 if (PG_ARGISNULL(1))
130 PG_RETURN_NULL(); /* invalid input */
131
132 bms = PG_ARG_GETBITMAPSET(0);
133 member = PG_GETARG_INT32(1);
134
135 bms = bms_add_member(bms, member);
136
138}
Bitmapset * bms_add_member(Bitmapset *a, int x)
Definition: bitmapset.c:815
#define PG_ARGISNULL(n)
Definition: fmgr.h:209
#define PG_RETURN_NULL()
Definition: fmgr.h:345
#define PG_GETARG_INT32(n)
Definition: fmgr.h:269
#define PG_RETURN_BITMAPSET_AS_TEXT(bms)

References bms_add_member(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BITMAPSET_AS_TEXT, and PG_RETURN_NULL.

◆ test_bms_add_members()

Datum test_bms_add_members ( PG_FUNCTION_ARGS  )

Definition at line 141 of file test_bitmapset.c.

142{
145
146 /* left input is recycled */
147 bms1 = bms_add_members(bms1, bms2);
148
150}
Bitmapset * bms_add_members(Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:917

References bms_add_members(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_add_range()

Datum test_bms_add_range ( PG_FUNCTION_ARGS  )

Definition at line 479 of file test_bitmapset.c.

480{
481 Bitmapset *bms;
482 int32 lower,
483 upper;
484
485 if (PG_ARGISNULL(1) || PG_ARGISNULL(2))
486 PG_RETURN_NULL(); /* invalid input */
487
488 bms = PG_ARG_GETBITMAPSET(0);
491
492 bms = bms_add_range(bms, lower, upper);
493
495}
Bitmapset * bms_add_range(Bitmapset *a, int lower, int upper)
Definition: bitmapset.c:1019
int32_t int32
Definition: c.h:534
Datum lower(PG_FUNCTION_ARGS)
Definition: oracle_compat.c:49
Datum upper(PG_FUNCTION_ARGS)
Definition: oracle_compat.c:80

References bms_add_range(), lower(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BITMAPSET_AS_TEXT, PG_RETURN_NULL, and upper().

◆ test_bms_compare()

Datum test_bms_compare ( PG_FUNCTION_ARGS  )

Definition at line 299 of file test_bitmapset.c.

300{
303 int result;
304
305 result = bms_compare(bms1, bms2);
306
307 PG_RETURN_INT32(result);
308}
int bms_compare(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:183

References bms_compare(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_copy()

Datum test_bms_copy ( PG_FUNCTION_ARGS  )

Definition at line 211 of file test_bitmapset.c.

212{
214 Bitmapset *copy_bms;
215
216 copy_bms = bms_copy(bms);
217
219}
Bitmapset * bms_copy(const Bitmapset *a)
Definition: bitmapset.c:122

References bms_copy(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_del_member()

Datum test_bms_del_member ( PG_FUNCTION_ARGS  )

Definition at line 153 of file test_bitmapset.c.

154{
155 Bitmapset *bms;
156 int32 member;
157
158 if (PG_ARGISNULL(1))
159 PG_RETURN_NULL(); /* invalid input */
160
161 bms = PG_ARG_GETBITMAPSET(0);
162 member = PG_GETARG_INT32(1);
163
164 bms = bms_del_member(bms, member);
165
167}
Bitmapset * bms_del_member(Bitmapset *a, int x)
Definition: bitmapset.c:868

References bms_del_member(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BITMAPSET_AS_TEXT, and PG_RETURN_NULL.

◆ test_bms_del_members()

Datum test_bms_del_members ( PG_FUNCTION_ARGS  )

Definition at line 510 of file test_bitmapset.c.

511{
514
515 /* left input gets recycled */
516 bms1 = bms_del_members(bms1, bms2);
517
519}
Bitmapset * bms_del_members(Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:1161

References bms_del_members(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_difference()

Datum test_bms_difference ( PG_FUNCTION_ARGS  )

Definition at line 287 of file test_bitmapset.c.

288{
291 Bitmapset *result_bms;
292
293 result_bms = bms_difference(bms1, bms2);
294
295 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
296}
Bitmapset * bms_difference(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:346

References bms_difference(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_equal()

Datum test_bms_equal ( PG_FUNCTION_ARGS  )

Definition at line 222 of file test_bitmapset.c.

223{
226 bool result;
227
228 result = bms_equal(bms1, bms2);
229
230 PG_RETURN_BOOL(result);
231}
bool bms_equal(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:142
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:359

References bms_equal(), PG_ARG_GETBITMAPSET, and PG_RETURN_BOOL.

◆ test_bms_get_singleton_member()

Datum test_bms_get_singleton_member ( PG_FUNCTION_ARGS  )

Definition at line 357 of file test_bitmapset.c.

358{
360 int member;
361
362 /*
363 * Keep this simple. Return -1 when we detect the set is not a singleton
364 * set, otherwise return the singleton member.
365 */
366 if (!bms_get_singleton_member(bms, &member))
367 member = -1;
368
369 PG_RETURN_INT32(member);
370}
bool bms_get_singleton_member(const Bitmapset *a, int *member)
Definition: bitmapset.c:715

References bms_get_singleton_member(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_hash_value()

Datum test_bms_hash_value ( PG_FUNCTION_ARGS  )

Definition at line 547 of file test_bitmapset.c.

548{
550 uint32 hash_result;
551
552 hash_result = bms_hash_value(bms);
553
554 PG_RETURN_INT32(hash_result);
555}
uint32 bms_hash_value(const Bitmapset *a)
Definition: bitmapset.c:1420

References bms_hash_value(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_int_members()

Datum test_bms_int_members ( PG_FUNCTION_ARGS  )

Definition at line 498 of file test_bitmapset.c.

499{
502
503 /* left input gets recycled */
504 bms1 = bms_int_members(bms1, bms2);
505
507}
Bitmapset * bms_int_members(Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:1109

References bms_int_members(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_intersect()

Datum test_bms_intersect ( PG_FUNCTION_ARGS  )

Definition at line 275 of file test_bitmapset.c.

276{
279 Bitmapset *result_bms;
280
281 result_bms = bms_intersect(bms1, bms2);
282
283 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
284}
Bitmapset * bms_intersect(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:292

References bms_intersect(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_is_empty()

Datum test_bms_is_empty ( PG_FUNCTION_ARGS  )

Definition at line 311 of file test_bitmapset.c.

312{
314 bool result;
315
316 result = bms_is_empty(bms);
317
318 PG_RETURN_BOOL(result);
319}
#define bms_is_empty(a)
Definition: bitmapset.h:118

References bms_is_empty, PG_ARG_GETBITMAPSET, and PG_RETURN_BOOL.

◆ test_bms_is_member()

Datum test_bms_is_member ( PG_FUNCTION_ARGS  )

Definition at line 170 of file test_bitmapset.c.

171{
172 Bitmapset *bms;
173 int32 member;
174 bool result;
175
176 if (PG_ARGISNULL(1))
177 PG_RETURN_NULL(); /* invalid input */
178
179 bms = PG_ARG_GETBITMAPSET(0);
180 member = PG_GETARG_INT32(1);
181
182 result = bms_is_member(member, bms);
183
184 PG_RETURN_BOOL(result);
185}
bool bms_is_member(int x, const Bitmapset *a)
Definition: bitmapset.c:510

References bms_is_member(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_BOOL, and PG_RETURN_NULL.

◆ test_bms_is_subset()

Datum test_bms_is_subset ( PG_FUNCTION_ARGS  )

Definition at line 322 of file test_bitmapset.c.

323{
326 bool result;
327
328 result = bms_is_subset(bms1, bms2);
329
330 PG_RETURN_BOOL(result);
331}
bool bms_is_subset(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:412

References bms_is_subset(), PG_ARG_GETBITMAPSET, and PG_RETURN_BOOL.

◆ test_bms_join()

Datum test_bms_join ( PG_FUNCTION_ARGS  )

Definition at line 534 of file test_bitmapset.c.

535{
538 Bitmapset *result_bms;
539
540 /* either input can be recycled */
541 result_bms = bms_join(bms1, bms2);
542
543 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
544}
Bitmapset * bms_join(Bitmapset *a, Bitmapset *b)
Definition: bitmapset.c:1230

References bms_join(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_make_singleton()

Datum test_bms_make_singleton ( PG_FUNCTION_ARGS  )

Definition at line 199 of file test_bitmapset.c.

200{
201 Bitmapset *bms;
202 int32 member;
203
204 member = PG_GETARG_INT32(0);
205 bms = bms_make_singleton(member);
206
208}
Bitmapset * bms_make_singleton(int x)
Definition: bitmapset.c:216

References bms_make_singleton(), PG_GETARG_INT32, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_member_index()

Datum test_bms_member_index ( PG_FUNCTION_ARGS  )

Definition at line 461 of file test_bitmapset.c.

462{
463 Bitmapset *bms;
464 int32 member;
465 int result;
466
467 if (PG_ARGISNULL(1))
468 PG_RETURN_NULL(); /* invalid input */
469
470 bms = PG_ARG_GETBITMAPSET(0);
471 member = PG_GETARG_INT32(1);
472
473 result = bms_member_index(bms, member);
474
475 PG_RETURN_INT32(result);
476}
int bms_member_index(Bitmapset *a, int x)
Definition: bitmapset.c:539

References bms_member_index(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_INT32, and PG_RETURN_NULL.

◆ test_bms_membership()

Datum test_bms_membership ( PG_FUNCTION_ARGS  )

Definition at line 246 of file test_bitmapset.c.

247{
249 BMS_Membership result;
250
251 result = bms_membership(bms);
252
253 PG_RETURN_INT32((int32) result);
254}
BMS_Membership bms_membership(const Bitmapset *a)
Definition: bitmapset.c:781
BMS_Membership
Definition: bitmapset.h:70

References bms_membership(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_next_member()

Datum test_bms_next_member ( PG_FUNCTION_ARGS  )

Definition at line 257 of file test_bitmapset.c.

258{
259 Bitmapset *bms;
260 int32 prevmember;
261 int result;
262
263 if (PG_ARGISNULL(1))
264 PG_RETURN_NULL(); /* invalid input */
265
266 bms = PG_ARG_GETBITMAPSET(0);
267 prevmember = PG_GETARG_INT32(1);
268
269 result = bms_next_member(bms, prevmember);
270
271 PG_RETURN_INT32(result);
272}
int bms_next_member(const Bitmapset *a, int prevbit)
Definition: bitmapset.c:1306

References bms_next_member(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_INT32, and PG_RETURN_NULL.

◆ test_bms_nonempty_difference()

Datum test_bms_nonempty_difference ( PG_FUNCTION_ARGS  )

Definition at line 449 of file test_bitmapset.c.

450{
453 bool result;
454
455 result = bms_nonempty_difference(bms1, bms2);
456
457 PG_RETURN_BOOL(result);
458}
bool bms_nonempty_difference(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:641

References bms_nonempty_difference(), PG_ARG_GETBITMAPSET, and PG_RETURN_BOOL.

◆ test_bms_num_members()

Datum test_bms_num_members ( PG_FUNCTION_ARGS  )

Definition at line 188 of file test_bitmapset.c.

189{
191 int result;
192
193 result = bms_num_members(bms);
194
195 PG_RETURN_INT32(result);
196}
int bms_num_members(const Bitmapset *a)
Definition: bitmapset.c:751

References bms_num_members(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_overlap()

Datum test_bms_overlap ( PG_FUNCTION_ARGS  )

Definition at line 391 of file test_bitmapset.c.

392{
395 bool result;
396
397 result = bms_overlap(bms1, bms2);
398
399 PG_RETURN_BOOL(result);
400}
bool bms_overlap(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:582

References bms_overlap(), PG_ARG_GETBITMAPSET, and PG_RETURN_BOOL.

◆ test_bms_overlap_list()

Datum test_bms_overlap_list ( PG_FUNCTION_ARGS  )

Definition at line 403 of file test_bitmapset.c.

404{
405 Bitmapset *bms;
406 ArrayType *array;
407 List *int_list = NIL;
408 bool result;
409 Datum *elem_datums = NULL;
410 bool *elem_nulls = NULL;
411 int elem_count;
412 int i;
413
414 bms = PG_ARG_GETBITMAPSET(0);
415
416 if (!PG_ARGISNULL(1))
417 {
418 array = PG_GETARG_ARRAYTYPE_P(1);
419
420 deconstruct_array(array,
421 INT4OID, sizeof(int32), true, 'i',
422 &elem_datums, &elem_nulls, &elem_count);
423
424 for (i = 0; i < elem_count; i++)
425 {
426 if (!elem_nulls[i])
427 {
428 int32 member = DatumGetInt32(elem_datums[i]);
429
430 int_list = lappend_int(int_list, member);
431 }
432 }
433 }
434
435 result = bms_overlap_list(bms, int_list);
436
437 list_free(int_list);
438
439 if (elem_datums)
440 pfree(elem_datums);
441
442 if (elem_nulls)
443 pfree(elem_nulls);
444
445 PG_RETURN_BOOL(result);
446}
#define PG_GETARG_ARRAYTYPE_P(n)
Definition: array.h:263
void deconstruct_array(ArrayType *array, Oid elmtype, int elmlen, bool elmbyval, char elmalign, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3632
bool bms_overlap_list(const Bitmapset *a, const List *b)
Definition: bitmapset.c:608
i
int i
Definition: isn.c:77
List * lappend_int(List *list, int datum)
Definition: list.c:357
void list_free(List *list)
Definition: list.c:1546
void pfree(void *pointer)
Definition: mcxt.c:1594
#define NIL
Definition: pg_list.h:68
uint64_t Datum
Definition: postgres.h:70
static int32 DatumGetInt32(Datum X)
Definition: postgres.h:212
Definition: array.h:93
Definition: pg_list.h:54

References bms_overlap_list(), DatumGetInt32(), deconstruct_array(), i, lappend_int(), list_free(), NIL, pfree(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_ARRAYTYPE_P, and PG_RETURN_BOOL.

◆ test_bms_prev_member()

Datum test_bms_prev_member ( PG_FUNCTION_ARGS  )

Definition at line 373 of file test_bitmapset.c.

374{
375 Bitmapset *bms;
376 int32 prevmember;
377 int result;
378
379 if (PG_ARGISNULL(1))
380 PG_RETURN_NULL(); /* invalid input */
381
382 bms = PG_ARG_GETBITMAPSET(0);
383 prevmember = PG_GETARG_INT32(1);
384
385 result = bms_prev_member(bms, prevmember);
386
387 PG_RETURN_INT32(result);
388}
int bms_prev_member(const Bitmapset *a, int prevbit)
Definition: bitmapset.c:1367

References bms_prev_member(), PG_ARG_GETBITMAPSET, PG_ARGISNULL, PG_GETARG_INT32, PG_RETURN_INT32, and PG_RETURN_NULL.

◆ test_bms_replace_members()

Datum test_bms_replace_members ( PG_FUNCTION_ARGS  )

Definition at line 522 of file test_bitmapset.c.

523{
526
527 /* left input gets recycled */
528 bms1 = bms_replace_members(bms1, bms2);
529
531}
Bitmapset * bms_replace_members(Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:972

References bms_replace_members(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_bms_singleton_member()

Datum test_bms_singleton_member ( PG_FUNCTION_ARGS  )

Definition at line 346 of file test_bitmapset.c.

347{
349 int result;
350
351 result = bms_singleton_member(bms);
352
353 PG_RETURN_INT32(result);
354}
int bms_singleton_member(const Bitmapset *a)
Definition: bitmapset.c:672

References bms_singleton_member(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_subset_compare()

Datum test_bms_subset_compare ( PG_FUNCTION_ARGS  )

Definition at line 334 of file test_bitmapset.c.

335{
338 BMS_Comparison result;
339
340 result = bms_subset_compare(bms1, bms2);
341
342 PG_RETURN_INT32((int32) result);
343}
BMS_Comparison bms_subset_compare(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:445
BMS_Comparison
Definition: bitmapset.h:61

References bms_subset_compare(), PG_ARG_GETBITMAPSET, and PG_RETURN_INT32.

◆ test_bms_union()

Datum test_bms_union ( PG_FUNCTION_ARGS  )

Definition at line 234 of file test_bitmapset.c.

235{
238 Bitmapset *result_bms;
239
240 result_bms = bms_union(bms1, bms2);
241
242 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
243}
Bitmapset * bms_union(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:251

References bms_union(), PG_ARG_GETBITMAPSET, and PG_RETURN_BITMAPSET_AS_TEXT.

◆ test_random_operations()

Datum test_random_operations ( PG_FUNCTION_ARGS  )

Definition at line 595 of file test_bitmapset.c.

596{
597 Bitmapset *bms1 = NULL;
598 Bitmapset *bms2 = NULL;
599 Bitmapset *bms = NULL;
600 Bitmapset *result = NULL;
603 int num_ops;
604 int max_range;
605 int min_value;
606 int member;
607 int *members;
608 int num_members = 0;
609 int total_ops = 0;
610
611 if (PG_GETARG_INT32(0) > 0)
612 seed = PG_GETARG_INT32(0);
613
614 num_ops = PG_GETARG_INT32(1);
615 max_range = PG_GETARG_INT32(2);
616 min_value = PG_GETARG_INT32(3);
617
618 pg_prng_seed(&state, seed);
619 members = palloc(sizeof(int) * num_ops);
620
621 /* Phase 1: Random insertions */
622 for (int i = 0; i < num_ops / 2; i++)
623 {
624 member = pg_prng_uint32(&state) % max_range + min_value;
625
626 if (!bms_is_member(member, bms1))
627 {
628 members[num_members++] = member;
629 bms1 = bms_add_member(bms1, member);
630 }
631 }
632
633 /* Phase 2: Random set operations */
634 for (int i = 0; i < num_ops / 4; i++)
635 {
636 member = pg_prng_uint32(&state) % max_range + min_value;
637
638 bms2 = bms_add_member(bms2, member);
639 }
640
641 /* Test union */
642 result = bms_union(bms1, bms2);
643 EXPECT_NOT_NULL(result);
644
645 /* Verify union contains all members from first set */
646 for (int i = 0; i < num_members; i++)
647 {
648 if (!bms_is_member(members[i], result))
649 elog(ERROR, "union missing member %d", members[i]);
650 }
651 bms_free(result);
652
653 /* Test intersection */
654 result = bms_intersect(bms1, bms2);
655 if (result != NULL)
656 {
657 member = -1;
658
659 while ((member = bms_next_member(result, member)) >= 0)
660 {
661 if (!bms_is_member(member, bms1) || !bms_is_member(member, bms2))
662 elog(ERROR, "intersection contains invalid member %d", member);
663 }
664 bms_free(result);
665 }
666
667 /* Phase 3: Test range operations */
668 result = NULL;
669 for (int i = 0; i < num_ops; i++)
670 {
671 int lower = pg_prng_uint32(&state) % 100;
672 int upper = lower + (pg_prng_uint32(&state) % 20);
673
674 result = bms_add_range(result, lower, upper);
675 }
676 if (result != NULL)
677 {
678 EXPECT_TRUE(bms_num_members(result) > 0);
679 bms_free(result);
680 }
681
682 pfree(members);
683 bms_free(bms1);
684 bms_free(bms2);
685
686 for (int i = 0; i < num_ops; i++)
687 {
688 member = pg_prng_uint32(&state) % max_range + min_value;
689 switch (pg_prng_uint32(&state) % 3)
690 {
691 case 0: /* add */
692 bms = bms_add_member(bms, member);
693 break;
694 case 1: /* delete */
695 if (bms != NULL)
696 {
697 bms = bms_del_member(bms, member);
698 }
699 break;
700 case 2: /* test membership */
701 if (bms != NULL)
702 {
703 bms_is_member(member, bms);
704 }
705 break;
706 }
707 total_ops++;
708 }
709
710 bms_free(bms);
711
712 PG_RETURN_INT32(total_ops);
713}
TimestampTz GetCurrentTimestamp(void)
Definition: timestamp.c:1645
void bms_free(Bitmapset *a)
Definition: bitmapset.c:239
uint64_t uint64
Definition: c.h:539
#define elog(elevel,...)
Definition: elog.h:226
void * palloc(Size size)
Definition: mcxt.c:1365
uint32 pg_prng_uint32(pg_prng_state *state)
Definition: pg_prng.c:227
void pg_prng_seed(pg_prng_state *state, uint64 seed)
Definition: pg_prng.c:89
Definition: pg_prng.h:20
Definition: regguts.h:323
#define EXPECT_TRUE(expr)
Definition: test_bitmapset.c:70
#define EXPECT_NOT_NULL(expr)
Definition: test_bitmapset.c:78

References bms_add_member(), bms_add_range(), bms_del_member(), bms_free(), bms_intersect(), bms_is_member(), bms_next_member(), bms_num_members(), bms_union(), elog, ERROR, EXPECT_NOT_NULL, EXPECT_TRUE, GetCurrentTimestamp(), i, lower(), palloc(), pfree(), PG_GETARG_INT32, pg_prng_seed(), pg_prng_uint32(), PG_RETURN_INT32, and upper().

Variable Documentation

◆ PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 30 of file test_bitmapset.c.

AltStyle によって変換されたページ (->オリジナル) /