zhangqian
2024-08-01 fc3313955a083c9480e4ea74398f72f9ba6addcd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package opa
 
import (
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/open-policy-agent/opa/rego"
)
 
func OpaCheck(ctx *gin.Context, input map[string]interface{}, query string) bool {
    opa, err := rego.New(
        rego.Load([]string{"conf/opatest.rego"}, nil),
        rego.Query("data.opa."+query),
    ).PrepareForEval(ctx)
    if err != nil {
        fmt.Println("PrepareForEval err: ", err)
        return false
    }
    res, err := opa.Eval(ctx, rego.EvalInput(input))
    if err != nil {
        fmt.Println("Eval err: ", err)
        return false
    }
    return res.Allowed()
}