diff --git a/router/router.go b/router/router.go index 4eafe07..32143b6 100644 --- a/router/router.go +++ b/router/router.go @@ -13,6 +13,8 @@ func Setup(engine *gin.Engine) { hub := service.ExUserService // 测试方法,获取signed jwt user.POST("/login", app.HandlerFunc(hub.Login)) + user.Use(middleware.JWT(1)) + user.GET("/me", app.HandlerFunc(hub.Me)) } checkin := engine.Group("/checkin") { diff --git a/service/user.go b/service/user.go index fe5248d..876af7f 100644 --- a/service/user.go +++ b/service/user.go @@ -28,3 +28,6 @@ func (u UserService) Login(aw *app.Wrapper) app.Result { } return aw.Success(util.GenerateJWT(user.ID, user.RoleID)) } +func (u UserService) Me(aw *app.Wrapper) app.Result { + return aw.Success(userDAO.FindUserByID(aw.ExtractUserClaims().UserID)) +}