问题描述
我有以下格式的 id 和角色映射
{
"ra": [
{
"id": 168,"code": "AFAP"
},{
"id": 180,"code": "ABC"
},"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 微秒内找到 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
]
}
}