我可以在 PostgreSQL 中将一堆布尔列转换为单个位图吗？

Question

我想将查询转换为位掩码，例如：

SELECT BoolA, BoolB, BoolC, BoolD FROM MyTable;

转换为位掩码，其中位由上面的值定义。

例如，如果 BoolA 和 BoolD 为 true，我需要 1001 或 9。

我的想法是：

SELECT
   CASE WHEN BoolD THEN 2^0 ELSE 0 END +
   CASE WHEN BoolC THEN 2^1 ELSE 0 END +
   CASE WHEN BoolB THEN 2^2 ELSE 0 END +
   CASE WHEN BoolA THEN 2^3 ELSE 0 END
FROM MyTable;

但我不确定这是否是最好的方法，而且看起来相当冗长。有没有简单的方法可以做到这一点？

Answer 1

对于位掩码，类型 bitstring< /strong> 将是更好的选择。可能看起来像这样：

SELECT BoolD::int::bit
    || BoolC::int::bit
    || BoolB::int::bit
    || BoolA::int::bit
FROM tbl;

true 转换为 1，false 转换为 0。您可以简单地将位连接到位串。

将 bit(n) 转换为整数

似乎您需要一个整数作为结果 - 有一个简单的 &快速方法：

SELECT (BoolD::int::bit
     || BoolC::int::bit
     || BoolB::int::bit
     || BoolA::int::bit)::bit(4)::int
FROM tbl;

请务必阅读“位字符串函数”一章中的细则和手册的操作员”。

---

我又提出了两个想法，并用 10k 行进行了快速测试/参考，以总结所有内容。

测试设置：

CREATE TEMP TABLE t (boola bool, boolb bool, boolc bool, boold bool);
INSERT INTO t
SELECT random()::int::bool
     , random()::int::bool
     , random()::int::bool
     , random()::int::bool
FROM   generate_series(1,10000);

演示：

SELECT  CASE WHEN boold THEN 1 ELSE 0 END
     + (CASE WHEN boolc THEN 1 ELSE 0 END << 1)
     + (CASE WHEN boolb THEN 1 ELSE 0 END << 2)
     + (CASE WHEN boola THEN 1 ELSE 0 END << 3) AS andriy

     ,  boold::int
     + (boolc::int << 1)
     + (boolb::int << 2)
     + (boola::int << 3) AS mike

     , (boola::int::bit
     || boolb::int::bit
     || boolc::int::bit
     || boold::int::bit)::bit(4)::int AS erwin1

     ,  boold::int
     | (boolc::int << 1)
     | (boolb::int << 2)
     | (boola::int << 3) AS erwin2

     , (((
       boola::int << 1)
     | boolb::int << 1)
     | boolc::int << 1)
     | boold::int        AS erwin3
FROM   t
LIMIT  15;

您还可以使用 |（按位 OR）代替 + 运算符。
单独的测试运行显示所有五种方法的性能基本相同。

Answer 2

也许是这样的：

SELECT
  (CASE WHEN BoolA THEN 1 ELSE 0 END << 0) +
  (CASE WHEN BoolB THEN 1 ELSE 0 END << 1) +
  (CASE WHEN BoolC THEN 1 ELSE 0 END << 2) +
  (CASE WHEN BoolD THEN 1 ELSE 0 END << 3) AS BitMask
FROM MyTable;

其中 << 是 按位左移运算符。

Answer 3

我也想出了这个方法。这是我能找到的最简洁的，除了编写自定义函数之外。我会接受这个答案，除非有人有更聪明的东西。

SELECT
  (BoolD::int << 0) +
  (BoolC::int << 1) +
  (BoolB::int << 2) +
  (BoolA::int << 3)
from MyTable;