使用pg_roaringbitmap插件

创建插件

CREATE EXTENSION if not exists roaringbitmap;
CREATE EXTENSION

查看插件版本

postgres=# \dx
                    List of installed extensions
     Name      | Version |   Schema   |         Description
---------------+---------+------------+------------------------------
 plpgsql       | 1.0     | pg_catalog | PL/pgSQL procedural language
 roaringbitmap | 0.5     | public     | support for Roaring Bitmaps
(2 rows)

输入和输出格式

PolarDB目前仅支持array和bytea两种输入输出格式。

• 输入格式
  • array

    postgres=# select roaringbitmap('{1,100,10}');
                     roaringbitmap                  
    ------------------------------------------------
     \x3a30000001000000000002001000000001000a006400
    (1 row)

  • bytea

    postgres=# select '\x3a30000001000000000002001000000001000a006400'::roaringbitmap;
                     roaringbitmap                  
    ------------------------------------------------
     \x3a30000001000000000002001000000001000a006400
    (1 row)

• 输出格式
  说明 输出格式默认为bytea，您可以通过roaringbitmap.output_format修改输出格式。

  postgres=# set roaringbitmap.output_format='bytea';
  SET
  postgres=# select '{1}'::roaringbitmap;
               roaringbitmap              
  ----------------------------------------
   \x3a3000000100000000000000100000000100
  (1 row)
  
  postgres=# set roaringbitmap.output_format='array';
  SET
  postgres=# select '{1}'::roaringbitmap;
   roaringbitmap 
  ---------------
   {1}
  (1 row)

创建表

CREATE TABLE t1 (id integer, bitmap roaringbitmap);

整数集合创建位图

INSERT INTO t1 SELECT 1,rb_build(ARRAY[1,2,3,4,5,6,7,8,9,200]);

INSERT INTO t1 SELECT 2,rb_build_agg(e) FROM generate_series(1,100) e;

位图计算函数

位图计算函数包含OR、AND、XOR、ANDNOT。

SELECT roaringbitmap('{1,2,3}') | roaringbitmap('{3,4,5}');
SELECT roaringbitmap('{1,2,3}') & roaringbitmap('{3,4,5}');
SELECT roaringbitmap('{1,2,3}') # roaringbitmap('{3,4,5}');
SELECT roaringbitmap('{1,2,3}') - roaringbitmap('{3,4,5}');

位图聚合函数

位图聚合函数包含OR、AND、XOR、BUILD。

SELECT rb_or_agg(bitmap) FROM t1;
SELECT rb_and_agg(bitmap) FROM t1;
SELECT rb_xor_agg(bitmap) FROM t1;
SELECT rb_build_agg(e) FROM generate_series(1,100) e;

位图基数计算函数

SELECT rb_cardinality('{1,2,3}');

位图转换整数集合

SELECT rb_to_array(bitmap) FROM t1 WHERE id = 1;

位图转换整数

SELECT unnest(rb_to_array('{1,2,3}'::roaringbitmap));

或

SELECT rb_iterate('{1,2,3}'::roaringbitmap);

操作列表

功能函数列表

聚合函数列表