概要

实现从header获取信息

请求约定

在上一节中，我们学会了怎么使用jwt进行鉴权，但是/user/info接口并没有定义请求参数，我们本节后规定用户信息通过请求头中通过x-user-id携带到服务器

格式如下

curl -X GET \
  http://127.0.0.1:8888/user/info \
  -H 'authorization: eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2MDM2MzU0OTMsImlhdCI6MTYwMzU0OTA5M30.fGNe-sAEL6NuWDPWpfVi840qsamPA3fC9h4iO3rF9v0' \
  -H 'x-user-id: 1'

完善获取用户信息逻辑

1、编辑userinfohandler.go,从header中获取用户id

$  vi ~/book/user/api/internal/handler/userinfohandler.go

userinfohandler.go

package handler

import (
  "net/http"

  "book/user/api/internal/logic"
  "book/user/api/internal/svc"
  "github.com/tal-tech/go-zero/rest/httpx"
)

func userInfoHandler(ctx *svc.ServiceContext) http.HandlerFunc {
  return func(w http.ResponseWriter, r *http.Request) {
    userId:=r.Header.Get("x-user-id")
    l := logic.NewUserInfoLogic(r.Context(), ctx)
    resp, err := l.UserInfo(userId)
    if err != nil {
      httpx.Error(w, err)
    } else {
      httpx.OkJson(w, resp)
    }
  }
}

2、编辑error.go,添加errirUserNotFound错误类型

$ vi ~/book/user/api/internal/logic/error.go


errorUserNotFound       = shared.NewDefaultError("用户不存在")

2、编辑userinfologic.go,填充获取用户信息逻辑

$ vi ~/book/user/api/internal/logic/userinfologic.go

userinfologic.go

package logic

import (
  "book/user/model"
  "context"
  "strconv"

  "book/user/api/internal/svc"
  "book/user/api/internal/types"

  "github.com/tal-tech/go-zero/core/logx"
)

type UserInfoLogic struct {
  logx.Logger
  ctx    context.Context
  svcCtx *svc.ServiceContext
}

func NewUserInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) UserInfoLogic {
  return UserInfoLogic{
    Logger: logx.WithContext(ctx),
    ctx:    ctx,
    svcCtx: svcCtx,
  }
}

func (l *UserInfoLogic) UserInfo(userId string) (*types.UserReply, error) {
  userInt, err := strconv.ParseInt(userId, 10, 64)
  if err != nil {
    return nil, err
  }

  userInfo, err := l.svcCtx.UserModel.FindOne(userInt)
  switch err {
  case nil:
    return &types.UserReply{
      Id:       userInfo.Id,
      Username: userInfo.Name,
      Mobile:   userInfo.Mobile,
      Nickname: userInfo.Nickname,
      Gender:   userInfo.Gender,
    }, nil
  case model.ErrNotFound:
    return nil, errorUserNotFound
  default:
    return nil, err
  }
}

验证

$ curl -i -X GET \
  http://127.0.0.1:8888/user/info \
  -H 'authorization: eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2MDM2MzU0OTMsImlhdCI6MTYwMzU0OTA5M30.fGNe-sAEL6NuWDPWpfVi840qsamPA3fC9h4iO3rF9v0' \
  -H 'x-user-id: 1'