登录
首页 >  Golang >  Go问答

请问 go gin 框架内如何像 php 一样在任意的地方随时结束一个请求处理?

来源:SegmentFault

时间:2023-01-11 18:44:17 135浏览 收藏

知识点掌握了,还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战,手把手教大家学习《请问 go gin 框架内如何像 php 一样在任意的地方随时结束一个请求处理?》,在实现功能的过程中也带大家重新温习相关知识点,温故而知新,回头看看说不定又有不一样的感悟!

问题内容

例如使用 php 开发一个 api 接口时,当我们验证请求的参数时候,如果发现参数不符合规则,则可以直接 exit(json) 直接返回输出并将程序终止,那么在 go 中该如何处理呢?exit 肯定是不行的,总不能一直要 return 吧?

我现在的代码如下(这是一个 controller 文件):

package user

import (
    userRequest "diary/request/user"
    "fmt"
    "github.com/gin-gonic/gin"
)

func Login(c *gin.Context) {
    userRequest.LoginVerify(c)
    fmt.Println("come on")
}

diary/request/user.go 文件中代码如下:

func LoginVerify(c *gin.Context) {
    var form Login
    if err := c.Bind(&form); err != nil {
        c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"errcode": 101, "msg": err.Error()})
        //c.JSON(http.StatusBadRequest, gin.H{"errcode": 101, "msg": err.Error()})
        //return
        fmt.Println("hello b")
    }
}

代码还是会执行到

fmt.Println("come on")

如何才能让他在 diary/request/user.go 文件中就可以自行结束呢?感谢各位大神的指点!

正确答案

你可以上官方issue tracker里看一眼,或者就谷歌一下

https://github.com/gin-gonic/...

总结就是:用

Abort

ctx.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"Message": "Unauthorized"})

中间件方式实现

package main

import (
    "log"

    "github.com/gin-gonic/gin"
)

func Middleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        // 你的 LoginVerify
        if c.Request.PostFormValue("flag") == "1" {
            c.AbortWithStatus(400)
            return
        }

        // before request
        c.Next()
        // after request
    }
}

func main() {
    r := gin.New()
    r.Use(Middleware())

    r.POST("/test", func(c *gin.Context) {
        log.Println("Hello world!")
        c.JSONP(200, gin.H{
            "hello": "world",
        })
    })

    // Listen and serve on 0.0.0.0:8080
    r.Run(":8080")
}

通过 panic 随时中断请求处理

package main

import (
    "fmt"

    "github.com/gin-gonic/gin"
)

type CustomErrorResponse struct {
    Status  int
    Code    int
    Message string
}

func (c *CustomErrorResponse) Error() string {
    return fmt.Sprintf("%d %d %s", c.Status, c.Code, c.Message)
}

func Middleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        defer func() {
            if e := recover(); e != nil {
                if err, ok := e.(*CustomErrorResponse); ok {
                    c.AbortWithStatusJSON(err.Status, gin.H{
                        "Code":    err.Code,
                        "Message": err.Message,
                    })
                }
            }
        }()

        // before request
        c.Next()
        // after request
    }
}

func main() {
    r := gin.New()
    r.Use(Middleware())

    r.POST("/test", func(c *gin.Context) {
        // 你的 VerifyLogin
        if c.Request.PostFormValue("flag") == "1" {
            panic(&CustomErrorResponse{
                Status:  400,
                Code:    1,
                Message: "flag is 1",
            })
        }

        c.JSONP(200, gin.H{
            "hello": "world",
        })
    })

    // Listen and serve on 0.0.0.0:8080
    r.Run(":8080")
}

到这里,我们也就讲完了《请问 go gin 框架内如何像 php 一样在任意的地方随时结束一个请求处理?》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于golang的知识点!

声明:本文转载于:SegmentFault 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>