Golang - 如何获得 mux 的授权？

学习Golang要努力，但是不要急！今天的这篇文章《Golang - 如何获得 mux 的授权？》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

我是 golang 的新手，我开始使用 jwt 学习 gorilla/mux。

我刚刚成功注册用户到 mongodb，然后登录。 我用postman来测试。但是当我尝试浏览受保护的路线时。 我在验证中收到错误“缺少身份验证令牌”。

邮递员如何获得浏览受保护路由的授权？

下面是注册和登录的代码：

路由器功能

func router() *mux.router {

    router := mux.newrouter().strictslash(true)
  router.use(commonmiddleware)

  //router.handlefunc("/", middleware.testapi).methods("get", "options")
  router.handlefunc("/register", middleware.registerhandler).methods("post", "options")
  router.handlefunc("/login", middleware.loginhandler).methods("post", "options")

  secure := router.pathprefix("/auth").subrouter()
  secure.use(auth.jwtverify)

    secure.handlefunc("/api/task", middleware.getalltask).methods("get", "options")
    secure.handlefunc("/api/task", middleware.createtask).methods("post", "options")
    secure.handlefunc("/api/task/{id}", middleware.taskcomplete).methods("put", "options")
    secure.handlefunc("/api/undotask/{id}", middleware.undotask).methods("put", "options")
    secure.handlefunc("/api/deletetask/{id}", middleware.deletetask).methods("delete", "options")
    secure.handlefunc("/api/deletealltask", middleware.deletealltask).methods("delete", "options")
    return router
}

func commonmiddleware(next http.handler) http.handler {
  return http.handlerfunc(func(w http.responsewriter, r *http.request) {
    w.header().add("content-type", "application/json")
    w.header().set("access-control-allow-origin", "*")
    w.header().set("access-control-allow-methods", "post, get, options, put, delete")
    w.header().set("access-control-allow-headers", "accept, content-type, content-length, accept-encoding, x-csrf-token, authorization, access-control-request-headers, access-control-request-method, connection, host, origin, user-agent, referer, cache-control, x-header")
    next.servehttp(w, r)
  })
}

注册功能

func registerhandler(w http.responsewriter, r *http.request) {
  w.header().set("content-type", "application/json")
  var user models.user
  body, _ := ioutil.readall(r.body)
  err := json.unmarshal(body, &user)
  var res models.responseresult
  if err != nil {
    res.error = err.error()
    json.newencoder(w).encode(res)
    return
  }

  var result models.user
  err = usercollection.findone(context.todo(), bson.d{{"username", user.username}}).decode(&result)

  if err != nil {
    if err.error() == "mongo: no documents in result" {
      hash, err := bcrypt.generatefrompassword([]byte(user.password), 5)

      if err != nil {
        res.error = "error while hashing password, try again"
        json.newencoder(w).encode(res)
        return
      }
      user.password = string(hash)

      _, err = usercollection.insertone(context.todo(), user)
      if err != nil {
        res.error = "error while creating user, try again"
        json.newencoder(w).encode(res)
        return
      }
      res.result = "register successful"
      json.newencoder(w).encode(res)
      return
    }
    res.error = err.error()
    json.newencoder(w).encode(res)
    return
  }
  res.result = "username already exists!!"
  json.newencoder(w).encode(res)
  return
}

登录功能

func loginhandler(w http.responsewriter, r *http.request) {
  w.header().set("content-type", "application/json")
  var user models.user
  body, _ := ioutil.readall(r.body)
  err := json.unmarshal(body, &user)

  if err != nil {
    log.fatal(err)
  }
  //var resulttoken models.token
  var resultuser models.user
  var res models.responseresult

  err = usercollection.findone(context.todo(), bson.d{{"username", user.username}}).decode(&resultuser)

  if err != nil {
    res.error = "invalid username"
    json.newencoder(w).encode(res)
    return
  }

  expiresat := time.now().add(time.minute * 100000).unix()

  errf := bcrypt.comparehashandpassword([]byte(resultuser.password), []byte(user.password))

  if errf != nil && errf == bcrypt.errmismatchedhashandpassword {
    var  res = map[string]interface{}{"status": false, "message": "invalid login credential. please try again"}
    json.newencoder(w).encode(res)
    return
  }
  tk := &models.token{
    username: user.username,
    standardclaims: &jwt.standardclaims{
      expiresat: expiresat,
    },
  }

  token := jwt.newwithclaims(jwt.getsigningmethod("hs256"), tk)

  tokenstring, err := token.signedstring([]byte("secret"))

  if err != nil {
    res.error = "error while generating token, try again"
    json.newencoder(w).encode(res)
    return
  }
  var resp = map[string]interface{}{"status": false, "message": "logged in"}
  resp["token"] = tokenstring
  resp["tk"] = tk

  json.newencoder(w).encode(resp)
}

验证功能

func JwtVerify(next http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    var header = r.Header.Get("x-access-token")

    json.NewEncoder(w).Encode(r)
    header = strings.TrimSpace(header)

    if header == "" {
      w.WriteHeader(http.StatusForbidden)
      json.NewEncoder(w).Encode(Exception{Message: "Missing auth token"})
      return
    }

    tk := &models.Token{}

    _, err := jwt.ParseWithClaims(header, tk, func(token *jwt.Token) (interface{}, error) {
      return []byte("secret"), nil
    })

    if err != nil {
      w.WriteHeader(http.StatusForbidden)
      json.NewEncoder(w).Encode(Exception{Message: err.Error()})
      return
    }

    ctx := context.WithValue(r.Context(), "user", tk)
    next.ServeHTTP(w, r.WithContext(ctx))
  })
}

解决方案

由于您正在检查请求标头中的令牌 x-access-token，因此需要在发送请求时添加相同的令牌。这可以在 postman 中轻松完成，如下所示 -

我使用的路由器是 -

主包

func router() *mux.router {
    router := mux.newrouter().strictslash(true)
    secure := router.pathprefix("/auth").subrouter()
    secure.use(auth.jwtverify)
    secure.handlefunc("/api", middleware.apihandler).methods("get")
    return router
}

func main() {
    r := router()
    http.listenandserve(":8080", r)
}

我使用的中间件是 -

包授权

func jwtverify(next http.handler) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        var header = r.header.get("x-access-token")

        json.newencoder(w).encode(r)
        header = strings.trimspace(header)

        if header == "" {
            w.writeheader(http.statusforbidden)
            json.newencoder(w).encode("missing auth token")
            return
        } else {
            json.newencoder(w).encode(fmt.sprintf("token found. value %s", header))
        }
        next.servehttp(w, r)
    })
}

处理程序是 -

封装中间件

func ApiHandler(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusOK)
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode("SUCCESS!")
    return
}

您可以在授权部分传递令牌：

以上就是《Golang - 如何获得 mux 的授权？》的详细内容，更多关于的资料请关注golang学习网公众号！