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)
ERROR
#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{
560 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
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}
bitmap_hash
uint32 bitmap_hash(const void *key, Size keysize)
Definition: bitmapset.c:1436
uint32
uint32_t uint32
Definition: c.h:538
PG_RETURN_INT32
#define PG_RETURN_INT32(x)
Definition: fmgr.h:354
PG_ARG_GETBITMAPSET
#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{
572 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
573 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
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}
bitmap_match
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
137 PG_RETURN_BITMAPSET_AS_TEXT(bms);
138}
bms_add_member
Bitmapset * bms_add_member(Bitmapset *a, int x)
Definition: bitmapset.c:815
PG_ARGISNULL
#define PG_ARGISNULL(n)
Definition: fmgr.h:209
PG_RETURN_NULL
#define PG_RETURN_NULL()
Definition: fmgr.h:345
PG_GETARG_INT32
#define PG_GETARG_INT32(n)
Definition: fmgr.h:269
PG_RETURN_BITMAPSET_AS_TEXT
#define PG_RETURN_BITMAPSET_AS_TEXT(bms)
Definition: test_bitmapset.c:101

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{
143 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
144 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
145
146 /* left input is recycled */
147 bms1 = bms_add_members(bms1, bms2);
148
149 PG_RETURN_BITMAPSET_AS_TEXT(bms1);
150}
bms_add_members
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);
489 lower = PG_GETARG_INT32(1);
490 upper = PG_GETARG_INT32(2);
491
492 bms = bms_add_range(bms, lower, upper);
493
494 PG_RETURN_BITMAPSET_AS_TEXT(bms);
495}
bms_add_range
Bitmapset * bms_add_range(Bitmapset *a, int lower, int upper)
Definition: bitmapset.c:1019
int32
int32_t int32
Definition: c.h:534
lower
Datum lower(PG_FUNCTION_ARGS)
Definition: oracle_compat.c:49
upper
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{
301 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
302 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
303 int result;
304
305 result = bms_compare(bms1, bms2);
306
307 PG_RETURN_INT32(result);
308}
bms_compare
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{
213 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
214 Bitmapset *copy_bms;
215
216 copy_bms = bms_copy(bms);
217
218 PG_RETURN_BITMAPSET_AS_TEXT(copy_bms);
219}
bms_copy
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
166 PG_RETURN_BITMAPSET_AS_TEXT(bms);
167}
bms_del_member
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{
512 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
513 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
514
515 /* left input gets recycled */
516 bms1 = bms_del_members(bms1, bms2);
517
518 PG_RETURN_BITMAPSET_AS_TEXT(bms1);
519}
bms_del_members
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{
289 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
290 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
291 Bitmapset *result_bms;
292
293 result_bms = bms_difference(bms1, bms2);
294
295 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
296}
bms_difference
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{
224 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
225 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
226 bool result;
227
228 result = bms_equal(bms1, bms2);
229
230 PG_RETURN_BOOL(result);
231}
bms_equal
bool bms_equal(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:142
PG_RETURN_BOOL
#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{
359 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
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}
bms_get_singleton_member
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{
549 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
550 uint32 hash_result;
551
552 hash_result = bms_hash_value(bms);
553
554 PG_RETURN_INT32(hash_result);
555}
bms_hash_value
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{
500 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
501 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
502
503 /* left input gets recycled */
504 bms1 = bms_int_members(bms1, bms2);
505
506 PG_RETURN_BITMAPSET_AS_TEXT(bms1);
507}
bms_int_members
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{
277 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
278 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
279 Bitmapset *result_bms;
280
281 result_bms = bms_intersect(bms1, bms2);
282
283 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
284}
bms_intersect
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{
313 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
314 bool result;
315
316 result = bms_is_empty(bms);
317
318 PG_RETURN_BOOL(result);
319}
bms_is_empty
#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}
bms_is_member
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{
324 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
325 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
326 bool result;
327
328 result = bms_is_subset(bms1, bms2);
329
330 PG_RETURN_BOOL(result);
331}
bms_is_subset
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{
536 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
537 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
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}
bms_join
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
207 PG_RETURN_BITMAPSET_AS_TEXT(bms);
208}
bms_make_singleton
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}
bms_member_index
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{
248 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
249 BMS_Membership result;
250
251 result = bms_membership(bms);
252
253 PG_RETURN_INT32((int32) result);
254}
bms_membership
BMS_Membership bms_membership(const Bitmapset *a)
Definition: bitmapset.c:781
BMS_Membership
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}
bms_next_member
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{
451 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
452 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
453 bool result;
454
455 result = bms_nonempty_difference(bms1, bms2);
456
457 PG_RETURN_BOOL(result);
458}
bms_nonempty_difference
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{
190 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
191 int result;
192
193 result = bms_num_members(bms);
194
195 PG_RETURN_INT32(result);
196}
bms_num_members
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{
393 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
394 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
395 bool result;
396
397 result = bms_overlap(bms1, bms2);
398
399 PG_RETURN_BOOL(result);
400}
bms_overlap
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}
PG_GETARG_ARRAYTYPE_P
#define PG_GETARG_ARRAYTYPE_P(n)
Definition: array.h:263
deconstruct_array
void deconstruct_array(ArrayType *array, Oid elmtype, int elmlen, bool elmbyval, char elmalign, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3632
bms_overlap_list
bool bms_overlap_list(const Bitmapset *a, const List *b)
Definition: bitmapset.c:608
i
int i
Definition: isn.c:77
lappend_int
List * lappend_int(List *list, int datum)
Definition: list.c:357
list_free
void list_free(List *list)
Definition: list.c:1546
pfree
void pfree(void *pointer)
Definition: mcxt.c:1594
NIL
#define NIL
Definition: pg_list.h:68
Datum
uint64_t Datum
Definition: postgres.h:70
DatumGetInt32
static int32 DatumGetInt32(Datum X)
Definition: postgres.h:212
List
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}
bms_prev_member
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{
524 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
525 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
526
527 /* left input gets recycled */
528 bms1 = bms_replace_members(bms1, bms2);
529
530 PG_RETURN_BITMAPSET_AS_TEXT(bms1);
531}
bms_replace_members
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{
348 Bitmapset *bms = PG_ARG_GETBITMAPSET(0);
349 int result;
350
351 result = bms_singleton_member(bms);
352
353 PG_RETURN_INT32(result);
354}
bms_singleton_member
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{
336 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
337 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
338 BMS_Comparison result;
339
340 result = bms_subset_compare(bms1, bms2);
341
342 PG_RETURN_INT32((int32) result);
343}
bms_subset_compare
BMS_Comparison bms_subset_compare(const Bitmapset *a, const Bitmapset *b)
Definition: bitmapset.c:445
BMS_Comparison
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{
236 Bitmapset *bms1 = PG_ARG_GETBITMAPSET(0);
237 Bitmapset *bms2 = PG_ARG_GETBITMAPSET(1);
238 Bitmapset *result_bms;
239
240 result_bms = bms_union(bms1, bms2);
241
242 PG_RETURN_BITMAPSET_AS_TEXT(result_bms);
243}
bms_union
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;
601 pg_prng_state state;
602 uint64 seed = GetCurrentTimestamp();
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
620 /*
621 * There can be up to "num_ops" members added. This is very unlikely,
622 * still possible if all the operations hit the "0" case during phase 4
623 * where multiple operation types are mixed together.
624 */
625 members = palloc(sizeof(int) * num_ops);
626
627 /* Phase 1: Random insertions in first set */
628 for (int i = 0; i < num_ops / 2; i++)
629 {
630 member = pg_prng_uint32(&state) % max_range + min_value;
631
632 if (!bms_is_member(member, bms1))
633 members[num_members++] = member;
634 bms1 = bms_add_member(bms1, member);
635 }
636
637 /* Phase 2: Random insertions in second set */
638 for (int i = 0; i < num_ops / 4; i++)
639 {
640 member = pg_prng_uint32(&state) % max_range + min_value;
641
642 if (!bms_is_member(member, bms2))
643 members[num_members++] = member;
644 bms2 = bms_add_member(bms2, member);
645 }
646
647 /* Test union */
648 result = bms_union(bms1, bms2);
649 EXPECT_NOT_NULL(result);
650
651 /* Verify union contains all members from first and second sets */
652 for (int i = 0; i < num_members; i++)
653 {
654 if (!bms_is_member(members[i], result))
655 elog(ERROR, "union missing member %d", members[i]);
656 }
657 bms_free(result);
658
659 /*
660 * Test intersection, checking that all the members in the result are from
661 * both the first and second sets.
662 */
663 result = bms_intersect(bms1, bms2);
664 if (result != NULL)
665 {
666 member = -1;
667
668 while ((member = bms_next_member(result, member)) >= 0)
669 {
670 if (!bms_is_member(member, bms1) || !bms_is_member(member, bms2))
671 elog(ERROR, "intersection contains invalid member %d", member);
672 }
673 bms_free(result);
674 }
675
676 /* Phase 3: Test range operations */
677 result = NULL;
678 for (int i = 0; i < num_ops; i++)
679 {
680 int lower = pg_prng_uint32(&state) % 100;
681 int upper = lower + (pg_prng_uint32(&state) % 20);
682
683 result = bms_add_range(result, lower, upper);
684 }
685 if (result != NULL)
686 {
687 EXPECT_TRUE(bms_num_members(result) > 0);
688 bms_free(result);
689 }
690
691 bms_free(bms1);
692 bms_free(bms2);
693
694 /*
695 * Phase 4: mix of operations on a single set, cross-checking a bitmap
696 * with a secondary state, "members".
697 */
698 num_members = 0;
699
700 for (int op = 0; op < num_ops; op++)
701 {
702 switch (pg_prng_uint32(&state) % 3)
703 {
704 case 0: /* add */
705 member = pg_prng_uint32(&state) % max_range + min_value;
706 if (!bms_is_member(member, bms))
707 members[num_members++] = member;
708 bms = bms_add_member(bms, member);
709 break;
710 case 1: /* delete */
711 if (num_members > 0)
712 {
713 int pos = pg_prng_uint32(&state) % num_members;
714
715 member = members[pos];
716 if (!bms_is_member(member, bms))
717 elog(ERROR, "expected %d to be a valid member", member);
718
719 bms = bms_del_member(bms, member);
720
721 /*
722 * Move the final array member at the position of the
723 * member just deleted, reducing the array size by one.
724 */
725 members[pos] = members[--num_members];
726 }
727 break;
728 case 2: /* test membership */
729 /* Verify that bitmap contains all members */
730 for (int i = 0; i < num_members; i++)
731 {
732 if (!bms_is_member(members[i], bms))
733 elog(ERROR, "missing member %d", members[i]);
734 }
735 break;
736 }
737 total_ops++;
738 }
739
740 bms_free(bms);
741 pfree(members);
742
743 PG_RETURN_INT32(total_ops);
744}
GetCurrentTimestamp
TimestampTz GetCurrentTimestamp(void)
Definition: timestamp.c:1645
bms_free
void bms_free(Bitmapset *a)
Definition: bitmapset.c:239
uint64
uint64_t uint64
Definition: c.h:539
elog
#define elog(elevel,...)
Definition: elog.h:226
palloc
void * palloc(Size size)
Definition: mcxt.c:1365
pg_prng_uint32
uint32 pg_prng_uint32(pg_prng_state *state)
Definition: pg_prng.c:227
pg_prng_seed
void pg_prng_seed(pg_prng_state *state, uint64 seed)
Definition: pg_prng.c:89
pg_prng_state
Definition: pg_prng.h:20
state
Definition: regguts.h:323
EXPECT_TRUE
#define EXPECT_TRUE(expr)
Definition: test_bitmapset.c:70
EXPECT_NOT_NULL
#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.