diff --git a/src/models/invite.go b/src/models/invite.go index a2569875e00a72afb6d298938c88c37741ec9ee6..7a1c6d776525af482616930e8db9f72845e90474 100644 --- a/src/models/invite.go +++ b/src/models/invite.go @@ -25,6 +25,20 @@ func InviteGet(where string, args ...interface{}) (*Invite, error) { return &obj, nil } +func InviteMustGet(where string, args ...interface{}) (*Invite, error) { + var obj Invite + has, err := DB["rdb"].Where(where, args...).Get(&obj) + if err != nil { + return nil, err + } + + if !has { + return nil, fmt.Errorf("invite not found") + } + + return &obj, nil +} + func InviteNew(token, creator string) error { now := time.Now().Unix() obj := Invite{ diff --git a/src/modules/server/http/router_user.go b/src/modules/server/http/router_user.go index d3f08d8cf482c8600bfd51e461975a9c0fb07ed0..1bbf558005cff181e867aecebccc756bf2637e76 100644 --- a/src/modules/server/http/router_user.go +++ b/src/modules/server/http/router_user.go @@ -320,7 +320,7 @@ func userInvitePost(c *gin.Context) { return err } - inv, err := models.InviteGet("token=?", f.Token) + inv, err := models.InviteMustGet("token=?", f.Token) if err != nil { return err }