如何实现类似继承的效果

问答 / 0 / 8 / 创建于 4年前 / 更新于 4年前

在iris的mvc架构中，我有一组XxxController结构体，每个结构体都需要实现一个auth()方法，为了简便，我尝试所有结构体引入一个名为BaseController的结构体，这样我就可以在各个XxxController里使用这个方法了，代码如下：

//BaseController 根结构体
type BaseController struct {
    Ctx iris.Context
    UserClaims middleware.UserClaims
}

func (b *BaseController) auth() middleware.UserClaims {
    if b.UserClaims.ID > 0 {
        return b.UserClaims
    }
    mc := b.Ctx.Values().Get("jwt").(*jwt.Token).Claims.(jwt.MapClaims)
    return middleware.UserClaims{
        Nickname:  mc["nickname"].(string),
        ID:        uint(mc["id"].(float64)),
        Issuer:    mc["iss"].(string),
        IssuedAt:  int64(mc["iat"].(float64)),
        JwtId:     mc["jti"].(string),
        NotBefore: int64(mc["nbf"].(float64)),
        ExpiresAt: int64(mc["exp"].(float64)),
    }
}


//TestController 子结构体
type TestController struct {
   BaseController
  Service *services.TestService
}

func NewTestController() *TestController {
   return &TestController{Service: services.NewTestService()}
}

func (t *TestController) PostSave() (r models.Result) {
   fmt.Println("PostSave")
   var test models.Test
   if err := t.Ctx.ReadJSON(&test); err != nil {
      t.Ctx.StopWithError(iris.StatusBadRequest, err)
      return models.Result{
         Code: models.CodeFail,
         Msg:  "fail",
      }
   }
   r.Data=test
   r.Auth=t.auth()//直接在子结构体里调用父结构体的方法
   return 
}

我以为这样调用是可行的，但事实上请求会返回一个400错误，返回内容如下：

readObjectStart: expect { or n, but found ▯ , error found in #0 byte of ...||..., bigger context ...||...

是否有大神知道这种情况怎么解

2021年10月27日10点03分：新进展
我发现iris的配置里DisableBodyConsumptionOnUnmarshal设置为true后我上面的模式也可以实现，这个配置百度翻译是：如果此字段设置为true，则将创建一个新的缓冲区来读取和请求正文。不会更改正文，并且不会使用context.UnmarshalBody/ReadJSON/ReadXML之前的现有数据,依然感谢各位大神的指点，‘@滚球兽进化’的方案仍旧是极好的

go iris controller

世界最好语言的追随者

renxiaotu

377 声望

i++

0 人点赞

推荐文章：

更多推荐...

博客

php-go面经-北京多家大厂面试总结 25 / 4 |

博客

分享一本一直更新的Gitbook，[为你自己学Go] 23 / 17 |

博客

lancet: 一个全面、高效、可复用的go语言工具函数库 12 / 14 |

博客

一： 2023 go岗位-跳槽-学习规划（求补充） 19 / 16 |

博客

Go 实现端口扫描器 16 / 5 |

博客

里程碑！用自己的编程语言实现了一个网站 13 / 9 |

我们只希望世界和平

课程读者 398 声望

最佳答案

iris 应该有中间件。或者你自己去丰富这个请求的生命周期。你的需求是在每个控制器的逻辑执行前执行指定逻辑。其实就是 1 -> {} -> 逻辑 -> end 。你从这个角度看的话，实现的方法就多种多样了。我用gin写的时候因为觉得用 context 上设置 user 不太好，就写了个函数把传递过去的控制器方法处理成高阶函数，然后把控制器改造成想要的 request 参数

4年前评论

renxiaotu （楼主）

是有中间件，但中间件只接受context，如你所说我也不希望在context上设置user，你的第二个方案我不太明白，但似乎有些复杂

我们只希望世界和平（作者）

@renxiaotu

高阶函数，这个写一到两个就够了

func upHandleAuth(action func(request Controllers.Request) Controllers.Response) func(c *gin.Context) {
    return func(c *gin.Context) {
        userIdFromContext, ok := c.Get("userId")
        if !ok {
            c.JSON(http.StatusOK, gin.H{
                "message": "un Login",
            })
            return
        }
        userId, ok := userIdFromContext.(uint64)
        request := Controllers.Request{}
        request.Context = c
        request.UserId = userId
        response := action(request)
        c.JSON(response.Code, response.Data)
    }
}

完整的路由，你也可以二次封装

r.Group("api").Use(Middleware.JWTAuth()).
        Any("/userInfo", upHandleAuth(userController.UserInfoV2))

完整的控制器

func (UserController) UserInfoV2(request Request) Response {
    user, err := request.GetUser()
    if err != nil {
        return FailResponse("账号异常")
    }
    return SuccessResponse(user)
}

封装的 request 和 response


type Request struct {
    TraceId  string
    UserId   uint64
    userSet  bool
    userInfo User.User
    Context  *gin.Context
}

func (r *Request) GetUser() (User.User, error) {
    if r.userSet != false {
        return r.userInfo, nil
    }
    user, err := User.Repository().GetById(r.UserId)
    if err != nil {
        return r.userInfo, err
    }
    r.userSet = true
    r.userInfo = user
    return r.userInfo, nil
}

type Response struct {
    Code int
    Data interface{}
}

func BuildResponse(code int, data interface{}) Response {
    return Response{code, data}
}

func SuccessResponse(data interface{}) Response {
    return BuildResponse(http.StatusOK,
        map[string]interface{}{
            "msg":  nil,
            "data": data,
            "code": SUCCESS,
        },
    )
}

func FailResponse(msg string) Response {
    return BuildResponse(http.StatusOK,
        map[string]interface{}{
            "msg":  msg,
            "data": nil,
            "code": FAIL,
        },
    )
}

renxiaotu （楼主）

@滚球兽进化太棒了，这就是我想要的

讨论数量: 8

我们只希望世界和平

课程读者 398 声望

iris 应该有中间件。或者你自己去丰富这个请求的生命周期。你的需求是在每个控制器的逻辑执行前执行指定逻辑。其实就是 1 -> {} -> 逻辑 -> end 。你从这个角度看的话，实现的方法就多种多样了。我用gin写的时候因为觉得用 context 上设置 user 不太好，就写了个函数把传递过去的控制器方法处理成高阶函数，然后把控制器改造成想要的 request 参数

4年前评论

renxiaotu （楼主）

是有中间件，但中间件只接受context，如你所说我也不希望在context上设置user，你的第二个方案我不太明白，但似乎有些复杂

我们只希望世界和平（作者）

@renxiaotu

高阶函数，这个写一到两个就够了

func upHandleAuth(action func(request Controllers.Request) Controllers.Response) func(c *gin.Context) {
    return func(c *gin.Context) {
        userIdFromContext, ok := c.Get("userId")
        if !ok {
            c.JSON(http.StatusOK, gin.H{
                "message": "un Login",
            })
            return
        }
        userId, ok := userIdFromContext.(uint64)
        request := Controllers.Request{}
        request.Context = c
        request.UserId = userId
        response := action(request)
        c.JSON(response.Code, response.Data)
    }
}

完整的路由，你也可以二次封装

r.Group("api").Use(Middleware.JWTAuth()).
        Any("/userInfo", upHandleAuth(userController.UserInfoV2))

完整的控制器

func (UserController) UserInfoV2(request Request) Response {
    user, err := request.GetUser()
    if err != nil {
        return FailResponse("账号异常")
    }
    return SuccessResponse(user)
}

封装的 request 和 response


type Request struct {
    TraceId  string
    UserId   uint64
    userSet  bool
    userInfo User.User
    Context  *gin.Context
}

func (r *Request) GetUser() (User.User, error) {
    if r.userSet != false {
        return r.userInfo, nil
    }
    user, err := User.Repository().GetById(r.UserId)
    if err != nil {
        return r.userInfo, err
    }
    r.userSet = true
    r.userInfo = user
    return r.userInfo, nil
}

type Response struct {
    Code int
    Data interface{}
}

func BuildResponse(code int, data interface{}) Response {
    return Response{code, data}
}

func SuccessResponse(data interface{}) Response {
    return BuildResponse(http.StatusOK,
        map[string]interface{}{
            "msg":  nil,
            "data": data,
            "code": SUCCESS,
        },
    )
}

func FailResponse(msg string) Response {
    return BuildResponse(http.StatusOK,
        map[string]interface{}{
            "msg":  msg,
            "data": nil,
            "code": FAIL,
        },
    )
}

renxiaotu （楼主）

@滚球兽进化太棒了，这就是我想要的

deatil

见习助教 779 声望

你说的这个报错，从给的来看看不出来是哪的问题

4年前评论

renxiaotu （楼主）

因为含特殊字符，有一段粘贴的时候缺失，我已经补上了（不过也只有一小段），是t.Ctx.ReadJSON(&test)返回的错误，也许是路由写入context到控制器的时候失败

wuyan94zl

课程读者 174 声望 / 村长 @ 农村

博客：gin + gorm 实现单模型的curd的父类控制器
也许能帮到你

4年前评论

renxiaotu （楼主）

是个好方法，如果是用的普通路由模型，这个方式在iris框架里应该也能用，但iris的mvc路由接受的是controller interface{},就不能用这个方案了

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容，与人为善，比聪明更重要！

帮助

如何实现类似继承的效果

推荐文章：

社区赞助商

关于 LearnKu

资源推荐

服务提供商

其他信息

如何实现类似继承的效果

推荐文章：

社区赞助商

关于 LearnKu

资源推荐

服务提供商

其他信息

请登录