open-policy-agent - 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
        ]
    }
}