使用 SQLite 进行位掩码分组

Question

当每个表都有一个用于保存位掩码枚举的字段时，我有一个 sqlite 表。

我想执行一个查询，它将返回所有结果并将所有枚举分组到一个字段中：

服务：TEXT 名称、INTEGER 服务、INTEGER 掩码

SELECT service,XXX FROM Services GROUP BY service

基本上，我希望 XXX 是每个服务的所有掩码的位掩码 OR (|) 的结果。

数据：

'a1',1,1
'a2',1,2,
'a3',1,4,
'a4',1,8,

'b1',2,1,
'b2',2,3,
'b3',2,2

因此，我想得到以下行：

1,15 (1|2|4|8)
2,3 (1|3|2)

谢谢

编辑： 在我最初的问题中，我忘了提到每个掩码不是单个位，而是可以是多个位的掩码（我已更改第二个示例以反映这一点）。

Answer 1

SQLite 支持自定义聚合函数；根据您的设置，您可以register a custom function 轻松完成此操作。使用 C API：

void bitwise_or_step(sqlite3_context *context, int argc, sqlite3_value** argv)
{
    int *buffer = (int *)sqlite3_aggregate_context(context, sizeof(int));
    int x = sqlite3_value_int(argv[0]);
    *buffer |= x;
}

void bitwise_or_final(sqlite3_context *context)
{
    int *buffer = (int *)sqlite3_aggregate_context(context, sizeof(int));
    sqlite3_result_int(context, *buffer);
}

sqlite3_create_function_v2(db, "BITWISE_OR", 1, SQLITE_ANY, NULL,
    NULL, bitwise_or_step, bitwise_or_final, NULL);

然后在你的 SQL 中，你应该能够做到：

SELECT service,BITWISE_OR(mask) FROM Services GROUP BY service

如果你使用 PHP，你也可以define a custom aggregate function from PHP。

Answer 2

SQL 中整数的位掩码 OR (|) 操作很简单，只需将每个键的 2 个值的唯一幂相加即可：

todd$ sqlite3 ex1
SQLite version 3.7.5
Enter ".help" for instructions
Enter SQL statements terminated with a ";"
sqlite> create table Services(Name varchar(100), Service int, mask int);
sqlite> insert into Services values("a1", 1, 1);
sqlite> insert into Services values("a2", 1, 2);
sqlite> insert into Services values("a3", 1, 4);
sqlite> insert into Services values("a4", 1, 8);
sqlite> insert into Services values("b1", 2, 1);
sqlite> insert into Services values("b2", 2, 3);
sqlite> insert into Services values("b3", 2, 2);
sqlite> select * from Services;
a1|1|1
a2|1|2
a3|1|4
a4|1|8
b1|2|1
b2|2|3
b3|2|2

编辑：当您知道位图域时，您可以将值分成其部分并求和：

sqlite> select Service, max(mask&1) + max(mask&2) +  max(mask&4) + max(mask&8) from Services group by Service;
1|15
2|3
sqlite> 

您可以为存储在位图掩码中的 2 的所有已知幂扩展 max(mask&bit) 逻辑。