mirror of
https://github.com/alibaba/higress.git
synced 2026-02-21 06:20:54 +08:00
326 lines
11 KiB
Go
326 lines
11 KiB
Go
package workflow
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
"api-workflow/utils"
|
||
|
||
"github.com/alibaba/higress/plugins/wasm-go/pkg/wrapper"
|
||
"github.com/tidwall/gjson"
|
||
"github.com/tidwall/sjson"
|
||
)
|
||
|
||
const (
|
||
TaskTypeHTTP string = "http"
|
||
TaskStart string = "start"
|
||
TaskEnd string = "end"
|
||
TaskContinue string = "continue"
|
||
UseContextFlag string = "||"
|
||
AllFlag string = "@all"
|
||
)
|
||
|
||
type PluginConfig struct {
|
||
|
||
// @Title zh-CN 工作流
|
||
// @Description zh-CN 工作流的具体描述
|
||
Workflow Workflow `json:"workflow" yaml:"workflow"`
|
||
// @Title zh-CN 环境变量
|
||
// @Description zh-CN 用来定义整个工作流的环境变量
|
||
Env Env `json:"env" yaml:"env"`
|
||
}
|
||
|
||
type Env struct {
|
||
// @Title zh-CN 超时时间
|
||
// @Description zh-CN 用来定义工作流的超时时间，单位是毫秒
|
||
Timeout uint32 `json:"timeout" yaml:"timeout"`
|
||
// @Title zh-CN 最大迭代深度
|
||
// @Description zh-CN 用来定义工作流最大的迭代深度，默认是100
|
||
MaxDepth uint32 `json:"max_depth" yaml:"max_depth"`
|
||
}
|
||
type Workflow struct {
|
||
// @Title zh-CN 边的列表
|
||
// @Description zh-CN 边的列表
|
||
Edges []Edge `json:"edges" yaml:"edges"`
|
||
// @Title zh-CN 节点的列表
|
||
// @Description zh-CN 节点的列表
|
||
Nodes map[string]Node `json:"nodes" yaml:"nodes"`
|
||
// @Title zh-CN 工作流的状态
|
||
// @Description zh-CN 工作流的执行状态，用于记录node之间的相互依赖和执行情况
|
||
WorkflowExecStatus map[string]int `json:"-" yaml:"-"`
|
||
}
|
||
|
||
type Edge struct {
|
||
// @Title zh-CN 上一步节点
|
||
// @Description zh-CN 上一步节点，必须是定义node的name，或者初始化工作流的start
|
||
Source string `json:"source" yaml:"source"`
|
||
// @Title zh-CN 当前执行的节点
|
||
// @Description zh-CN 当前执行节点，必须是定义的node的name，或者结束工作流的关键字 end continue
|
||
Target string `json:"target" yaml:"target"`
|
||
// @Title zh-CN 执行操作
|
||
// @Description zh-CN 执行单元，里面实时封装需要的数据
|
||
Task *Task
|
||
// @Title zh-CN 判断表达式
|
||
// @Description zh-CN 是否执行下一步的判断条件
|
||
Conditional string `json:"conditional" yaml:"conditional"`
|
||
}
|
||
|
||
type Task struct {
|
||
Cluster wrapper.Cluster `json:"-" yaml:"-"`
|
||
ServicePath string `json:"service_path" yaml:"service_path"`
|
||
ServicePort int64 `json:"service_port" yaml:"service_port"`
|
||
ServiceKey string `json:"service_key" yaml:"service_key"`
|
||
Body []byte `json:"-" yaml:"-"`
|
||
Headers [][2]string `json:"headers" yaml:"headers"`
|
||
Method string `json:"method" yaml:"method"`
|
||
TaskType string `json:"task_type" yaml:"task_type"`
|
||
}
|
||
|
||
type Node struct {
|
||
// @Title zh-CN 节点名称
|
||
// @Description zh-CN 节点名称全局唯一
|
||
Name string `json:"name" yaml:"name"`
|
||
// @Title zh-CN 服务名称
|
||
// @Description zh-CN 带服务类型的完整名称，例如 my.dns or foo.static
|
||
ServiceName string `json:"service_name" yaml:"service_name"`
|
||
// @Title zh-CN 服务端口
|
||
// @Description zh-CN static类型默认是80
|
||
ServicePort int64 `json:"service_port" yaml:"service_port"`
|
||
// @Title zh-CN 服务域名
|
||
// @Description zh-CN 服务域名，例如 dashscope.aliyuncs.com
|
||
ServiceDomain string `json:"service_domain" yaml:"service_domain"`
|
||
// @Title zh-CN http访问路径
|
||
// @Description zh-CN http访问路径，默认是 /
|
||
ServicePath string `json:"service_path" yaml:"service_path"`
|
||
// @Title zh-CN http 方法
|
||
// @Description zh-CN http方法，支持所有可用方法 GET，POST等
|
||
ServiceMethod string `json:"service_method" yaml:"service_method"`
|
||
// @Title zh-CN http 请求头文件
|
||
// @Description zh-CN 请求头文件
|
||
ServiceHeaders []ServiceHeader `json:"service_headers" yaml:"service_headers"`
|
||
// @Title zh-CN http 请求body模板
|
||
// @Description zh-CN 请求body模板，用来构造请求
|
||
ServiceBodyTmpl string `json:"service_body_tmpl" yaml:"service_body_tmpl"`
|
||
// @Title zh-CN http 请求body模板替换键值对
|
||
// @Description zh-CN 请求body模板替换键值对，用来构造请求。to表示填充的位置，from表示数据从哪里，
|
||
// 标识表达式基于 [GJSON PATH](https://github.com/tidwall/gjson/blob/master/SYNTAX.md) 语法提取字符串
|
||
ServiceBodyReplaceKeys []BodyReplaceKeyPair `json:"service_body_replace_keys" yaml:"service_body_replace_keys"`
|
||
}
|
||
type BodyReplaceKeyPair struct {
|
||
// @Title zh-CN from表示数据从哪里，
|
||
// @Description zh-CN from表示数据从哪里
|
||
// 标识表达式基于 [GJSON PATH](https://github.com/tidwall/gjson/blob/master/SYNTAX.md) 语法提取字符串
|
||
From string `json:"from" yaml:"from"`
|
||
// @Title zh-CN to表示填充的位置
|
||
// @Description zh-CN to表示填充的位置，
|
||
// 标识表达式基于 [GJSON PATH](https://github.com/tidwall/gjson/blob/master/SYNTAX.md) 语法提取字符串
|
||
To string `json:"to" yaml:"to"`
|
||
}
|
||
type ServiceHeader struct {
|
||
Key string `json:"key" yaml:"key"`
|
||
Value string `json:"value" yaml:"value"`
|
||
}
|
||
|
||
func (w *Edge) IsEnd() bool {
|
||
if w.Target == TaskEnd {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
func (w *Edge) IsContinue() bool {
|
||
if w.Target == TaskContinue {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
func (e *Edge) IsPass(ctx wrapper.HttpContext) (bool, error) {
|
||
// 执行判断Conditional
|
||
if e.Conditional != "" {
|
||
|
||
var err error
|
||
// 获取模板里的表达式
|
||
|
||
e.Conditional, err = e.WrapperDataByTmplStr(e.Conditional, ctx)
|
||
if err != nil {
|
||
return false, fmt.Errorf("workflow WrapperDateByTmplStr %s failed: %v", e.Conditional, err)
|
||
}
|
||
ok, err := e.ExecConditional()
|
||
if err != nil {
|
||
|
||
return false, fmt.Errorf("wl exec conditional %s failed: %v", e.Conditional, err)
|
||
}
|
||
return !ok, nil
|
||
|
||
}
|
||
return false, nil
|
||
}
|
||
|
||
func (w *Edge) WrapperTask(config PluginConfig, ctx wrapper.HttpContext) error {
|
||
|
||
// 判断 node 是否存在
|
||
node, isTool := config.Workflow.Nodes[w.Target]
|
||
|
||
if isTool {
|
||
w.Task.TaskType = TaskTypeHTTP
|
||
} else {
|
||
return fmt.Errorf("do not find target :%s", w.Target)
|
||
}
|
||
|
||
switch w.Task.TaskType {
|
||
default:
|
||
return fmt.Errorf("unknown node type :%s", w.Task.TaskType)
|
||
case TaskTypeHTTP:
|
||
err := w.wrapperNodeTask(node, ctx)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
}
|
||
return nil
|
||
|
||
}
|
||
|
||
func (w *Edge) wrapperBody(requestBodyTemplate string, keyPairs []BodyReplaceKeyPair, ctx wrapper.HttpContext) error {
|
||
|
||
requestBody, err := w.WrapperDataByTmplStrAndKeys(requestBodyTemplate, keyPairs, ctx)
|
||
if err != nil {
|
||
return fmt.Errorf("wrapper date by tmpl str is %s ,find err: %v", requestBodyTemplate, err)
|
||
}
|
||
|
||
w.Task.Body = requestBody
|
||
return nil
|
||
}
|
||
|
||
func (w *Edge) wrapperNodeTask(node Node, ctx wrapper.HttpContext) error {
|
||
// 封装cluster
|
||
w.Task.Cluster = wrapper.FQDNCluster{
|
||
Host: node.ServiceDomain,
|
||
FQDN: node.ServiceName,
|
||
Port: node.ServicePort,
|
||
}
|
||
|
||
// 封装请求body
|
||
err := w.wrapperBody(node.ServiceBodyTmpl, node.ServiceBodyReplaceKeys, ctx)
|
||
if err != nil {
|
||
return fmt.Errorf("wrapper body parse failed: %v", err)
|
||
}
|
||
|
||
// 封装请求Method path headers
|
||
w.Task.Method = node.ServiceMethod
|
||
w.Task.ServicePath = node.ServicePath
|
||
w.Task.Headers = make([][2]string, 0)
|
||
if len(node.ServiceHeaders) > 0 {
|
||
for _, header := range node.ServiceHeaders {
|
||
w.Task.Headers = append(w.Task.Headers, [2]string{header.Key, header.Value})
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// 利用模板和替换键值对构造请求，使用`||`分隔，str1代表使用node是执行结果。tr2代表如何取数据，使用gjson的表达式，`@all`代表全都要
|
||
func (w *Edge) WrapperDataByTmplStrAndKeys(tmpl string, keyPairs []BodyReplaceKeyPair, ctx wrapper.HttpContext) ([]byte, error) {
|
||
var err error
|
||
// 不需要替换 node.service_body_replace_keys 为空
|
||
if len(keyPairs) == 0 {
|
||
return []byte(tmpl), nil
|
||
}
|
||
|
||
for _, keyPair := range keyPairs {
|
||
|
||
jsonPath := keyPair.From
|
||
target := keyPair.To
|
||
var contextValueRaw []byte
|
||
// 获取上下文数据
|
||
if strings.Contains(jsonPath, UseContextFlag) {
|
||
pathStr := strings.Split(jsonPath, UseContextFlag)
|
||
if len(pathStr) == 2 {
|
||
contextKey := pathStr[0]
|
||
contextBody := ctx.GetContext(contextKey)
|
||
if contextValue, ok := contextBody.([]byte); ok {
|
||
contextValueRaw = contextValue
|
||
jsonPath = pathStr[1]
|
||
} else {
|
||
return nil, fmt.Errorf("context value is not []byte,key is %s", contextKey)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 执行封装 ， `@all`代表全都要
|
||
requestBody := gjson.ParseBytes(contextValueRaw)
|
||
if jsonPath == AllFlag {
|
||
|
||
tmpl, err = sjson.SetRaw(tmpl, target, requestBody.Raw)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("wrapper body parse failed: %v", err)
|
||
}
|
||
continue
|
||
}
|
||
requestBodyJson := requestBody.Get(jsonPath)
|
||
if requestBodyJson.Exists() {
|
||
tmpl, err = sjson.SetRaw(tmpl, target, requestBodyJson.Raw)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("wrapper body parse failed: %v", err)
|
||
}
|
||
|
||
} else {
|
||
return nil, fmt.Errorf("wrapper body parse failed: not exists %s", jsonPath)
|
||
}
|
||
}
|
||
return []byte(tmpl), nil
|
||
|
||
}
|
||
|
||
// 变量使用`{{str1||str2}}`包裹，使用`||`分隔，str1代表使用node是执行结果。tr2代表如何取数据，使用gjson的表达式，`@all`代表全都要
|
||
func (w *Edge) WrapperDataByTmplStr(tmpl string, ctx wrapper.HttpContext) (string, error) {
|
||
var body []byte
|
||
// 获取模板里的表达式
|
||
TmplKeyAndPath := utils.ParseTmplStr(tmpl)
|
||
if len(TmplKeyAndPath) == 0 {
|
||
return tmpl, nil
|
||
}
|
||
// 解析表达式 { "{{str1||str2}}":"str1||str2" }
|
||
for k, path := range TmplKeyAndPath {
|
||
// 变量使用`{{str1||str2}}`包裹，使用`||`分隔，str1代表使用前面命名为name的数据()。
|
||
if strings.Contains(path, UseContextFlag) {
|
||
pathStr := strings.Split(path, UseContextFlag)
|
||
if len(pathStr) == 2 {
|
||
contextKey := pathStr[0]
|
||
contextBody := ctx.GetContext(contextKey)
|
||
if contextValue, ok := contextBody.([]byte); ok {
|
||
body = contextValue
|
||
path = pathStr[1]
|
||
} else {
|
||
return tmpl, fmt.Errorf("context value is not []byte,key is %s", contextKey)
|
||
}
|
||
}
|
||
// 执行封装 ， `@all`代表全都要
|
||
requestBody := gjson.ParseBytes(body)
|
||
if path == AllFlag {
|
||
tmpl = strings.Replace(tmpl, k, utils.TrimQuote(requestBody.Raw), -1)
|
||
continue
|
||
}
|
||
requestBodyJson := requestBody.Get(path)
|
||
if requestBodyJson.Exists() {
|
||
tmpl = utils.ReplacedStr(tmpl, map[string]string{k: utils.TrimQuote(requestBodyJson.Raw)})
|
||
} else {
|
||
return tmpl, fmt.Errorf("use path {{%s}} get value is not exists,json is:%s", path, requestBody.Raw)
|
||
}
|
||
} else {
|
||
return "", fmt.Errorf("tmpl parse find error: || is not exists %s", path)
|
||
}
|
||
|
||
}
|
||
return tmpl, nil
|
||
}
|
||
|
||
func (w *Edge) ExecConditional() (bool, error) {
|
||
|
||
ConditionalResult, err := utils.ExecConditionalStr(w.Conditional)
|
||
if err != nil {
|
||
return false, fmt.Errorf("exec conditional failed: %v", err)
|
||
}
|
||
return ConditionalResult, nil
|
||
|
||
}
|