处理函数模块-验证器-CSDN博客

本文链接：https://blog.csdn.net/qq_34893654/article/details/130147388

该代码示例展示了如何在Go中创建一个RequestValidator结构体，用于验证HTTP请求的合法性。它检查请求头中的必要字段，如Authorization，验证URL路径是否匹配特定模式，并在POST、PUT或PATCH方法时检查请求体内容，例如解析JSON并确保包含用户名。

摘要由CSDN通过智能技术生成

请求进行安全验证，包括验证请求头、请求体格式、URL路径等是否符合要求，还需要验证请求参数的合法性，比如是否缺少必须的参数，参数类型是否符合要求,对请求的访问权限进行验证，以确保请求的合法性。

type RequestValidator struct {
    // 需要验证的请求头
    requiredHeaders map[string]string
    // URL路径正则表达式
    pathRegex *regexp.Regexp
    // 验证请求体的函数
    bodyValidator func([]byte) error
}

func NewRequestValidator(requiredHeaders map[string]string, pathRegex *regexp.Regexp, bodyValidator func([]byte) error) *RequestValidator {
    return &RequestValidator{
        requiredHeaders: requiredHeaders,
        pathRegex: pathRegex,
        bodyValidator: bodyValidator,
    }
}

// 验证请求是否合法
func (rv *RequestValidator) Validate(req *http.Request) error {
    // 验证请求头是否符合要求
    for key, value := range rv.requiredHeaders {
        if req.Header.Get(key) != value {
            return fmt.Errorf("missing required header: %s", key)
        }
    }

    // 验证URL路径是否符合要求
    if !rv.pathRegex.MatchString(req.URL.Path) {
        return fmt.Errorf("invalid URL path")
    }

    // 验证请求体是否符合要求
    if req.Method == "POST" || req.Method == "PUT" || req.Method == "PATCH" {
        body, err := ioutil.ReadAll(req.Body)
        if err != nil {
            return fmt.Errorf("failed to read request body")
        }
        if err := rv.bodyValidator(body); err != nil {
            return fmt.Errorf("invalid request body: %v", err)
        }
    }

    return nil
}

示例

rv := NewRequestValidator(map[string]string{
    "Authorization": "Bearer some_token",
}, regexp.MustCompile("^/users/[0-9]+$"), func(body []byte) error {
    var user User
    if err := json.Unmarshal(body, &user); err != nil {
        return err
    }
    if user.Name == "" {
        return errors.New("missing user name")
    }
    return nil
})

// 在处理请求时调用Validate函数进行验证
if err := rv.Validate(req); err != nil {
    // 处理验证失败的情况
}