Open Policy Agent -提高分组理解的性能

Question

我有一个id和角色映射，格式如下

{
    "ra": [
      {
        "id": 168,
        "code": "AFAP"
      },
      {
        "id": 180,
        "code": "ABC"
      },
      {
        "id": 180,
        "code": "ABCMND"
      }
   ]
}

我需要输出如下所示

{
    "roleactions": {
        "168": [
            "AFAP"
        ],
        "180": [
            "ABC",
            "ABCMND",
            "DCRMP"
        ]
    }
}

所以我写了下面的代码

roleactions = r_map {
    r := data.ra
    r_map := {id: list |
        some i
        id := r[i].id
        list := [obj |
            some j
            r[j].id == id
            obj := r[j].code
        ]
    }
}

但是当我运行它的时候几乎需要5-6秒

​

​

在5682526.465µs内找到1个结果。

有人可以指导如何编写此策略映射以提高性能吗？

Answer 1

OPA可以在线性时间内评估这样的理解：https://www.openpolicyagent.org/docs/latest/policy-performance/#comprehension-indexing。这种情况下的问题是，当单独考虑理解时，局部变量r是不安全的。

如果你像下面这样重构理解，运行时应该是线性的：

roleactions := r_map {
    r_map := {id: list |
        some i
        id := data.ra[i].id
        list := [obj |
            some j
            data.ra[j].id == id
            obj := data.ra[j].code
        ]
    }
}