额外参数新增拼接处理类型

2026-04-23 00:17:04 +08:00 · 2023-08-14 16:58:43 +08:00
parent 6c79e8795b
commit c894228ff8
5 changed files with 245 additions and 153 deletions
@@ -0,0 +1,172 @@
+package dynamic_params
+
+import (
+	"errors"
+	"fmt"
+	"strconv"
+	"strings"
+
+	http_service "github.com/eolinker/eosc/eocontext/http-context"
+	"github.com/ohler55/ojg/jp"
+	"github.com/ohler55/ojg/oj"
+
+	"github.com/eolinker/eosc/log"
+)
+
+const (
+	positionCurrent = iota
+
+	positionHeader
+	positionQuery
+	positionBody
+
+	positionSystem
+)
+
+type Param struct {
+	name  string
+	value []*ParamValue
+}
+
+type ParamValue struct {
+	key      string
+	position int
+	optional bool
+}
+
+func NewParam(name string, value []string) *Param {
+	vs := make([]*ParamValue, 0, len(value))
+	for _, v := range value {
+		v = strings.TrimSpace(v)
+		vLen := len(v)
+		if vLen > 0 {
+			if v[0] == '{' && v[vLen-1] == '}' {
+				vars := strings.Split(v[1:vLen-1], ".")
+				position := positionBody
+				variable := vars[0]
+				if len(vars) > 1 {
+					variable = vars[1]
+					switch vars[0] {
+					case "header":
+						position = positionHeader
+					case "query":
+						position = positionQuery
+					}
+				}
+				vs = append(vs, &ParamValue{
+					key:      variable,
+					position: position,
+				})
+			} else if v[0] == '#' {
+				vars := strings.Split(v[1:], ".")
+				position := positionBody
+				variable := vars[0]
+				if len(vars) > 1 {
+					variable = vars[1]
+					switch vars[0] {
+					case "header":
+						position = positionHeader
+					case "query":
+						position = positionQuery
+					}
+				}
+				vs = append(vs, &ParamValue{
+					key:      variable,
+					position: position,
+					optional: true,
+				})
+			} else if vLen > 3 && v[0] == '$' && v[1] == '{' && v[vLen-1] == '}' {
+				// 使用系统变量
+				vs = append(vs, &ParamValue{
+					key:      v[2 : vLen-1],
+					position: positionSystem,
+				})
+			} else {
+				vs = append(vs, &ParamValue{
+					key: v,
+				})
+			}
+		}
+	}
+	return &Param{
+		name:  name,
+		value: vs,
+	}
+}
+
+func (m *Param) Name() string {
+	return m.name
+}
+
+func (m *Param) Generate(ctx http_service.IHttpContext, contentType string, args ...interface{}) (interface{}, error) {
+	builder := strings.Builder{}
+	var params interface{}
+	if contentType == "application/json" {
+		if len(args) < 1 {
+			return nil, errors.New("missing args")
+		}
+		params = args[0]
+	}
+	for _, v := range m.value {
+		if v.key == "" {
+			continue
+		}
+		builder.WriteString(retrieveParam(ctx, contentType, params, v))
+	}
+
+	return builder.String(), nil
+}
+
+func retrieveParam(ctx http_service.IHttpContext, contentType string, body interface{}, value *ParamValue) string {
+	switch value.position {
+	case positionCurrent:
+		return value.key
+	case positionHeader:
+		return ctx.Proxy().Header().Headers().Get(value.key)
+	case positionQuery:
+		return ctx.Proxy().URI().GetQuery(value.key)
+	case positionBody:
+
+		if contentType == "application/x-www-form-urlencoded" {
+			if !value.optional {
+				return ctx.Proxy().Body().GetForm(value.key)
+			}
+			form, _ := ctx.Proxy().Body().BodyForm()
+			if _, ok := form[value.key]; ok {
+				return value.key
+			}
+		} else if contentType == "application/json" {
+			key := value.key
+			if !strings.HasPrefix(key, "$.") {
+				key = "$." + key
+			}
+
+			x, err := jp.ParseString(key)
+			if err != nil {
+				log.Errorf("parse json path(%s) error: %v", key, err)
+				return ""
+			}
+			result := x.Get(body)
+
+			if len(result) > 0 {
+				if value.optional {
+					return value.key
+				}
+
+				switch r := result[0].(type) {
+				case string:
+					return r
+				case float32, float64:
+					return fmt.Sprintf("%.f", r)
+				case bool:
+					return strconv.FormatBool(r)
+				default:
+					return oj.JSON(r)
+				}
+			}
+		}
+	case positionSystem:
+		return ctx.GetLabel(value.key)
+	}
+	return ""
+}