在 Golang GRPC 中集中处理请求变量的编写方法

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《在 Golang GRPC 中集中处理请求变量的编写方法》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

我有一个简单的问题。是否有机会以某种方式保存请求中的变量。 这是我的协议文件，我可以在其中生成客户端和服务器的代码。

service rectask {
  rpc allpath(usertoken) returns (usertoken) {
    option (google.api.http) = {
      get: "/*"
    };
  }
  rpc auth(google.protobuf.empty) returns (tokenrender) {
    option (google.api.http) = {
      get: "/auth"
    };
  }
  rpc starttracking(usertoken) returns (usertoken) {
    option (google.api.http) = {
      get: "/auth/{user_token}"
    };
  }
}

message usertoken {
  string user_token = 1;
  int32 user_counter = 2;
}

我只想统计“aut​​h”之后的用户请求

下面两个函数：第一个将计数器初始化为 0，另一个尝试根据请求进行更新：

func (s *RecTask) StartTracking(ctx context.Context, req *pb.UserToken) (*pb.UserToken, error) {
    if contains(tokens, req.UserToken) {
        response := &pb.UserToken{
            UserCounter: 0,
        }
        return response, nil

func (s *RecTask) AllPath(ctx context.Context, req *pb.UserToken) (res *pb.UserToken, err error) {
    response := &pb.UserToken{
        UserCounter: +1,
    }

如果有人能告诉我如何将用户的当前计数器存储在变量中，例如“counter”，我将非常感激，最好

正确答案

一种选择是将用户计数器变量放入自定义 rectask

type RecTask struct {
   userCounter map[string]int32
}

func (r *RecTask) Auth (ctx context.Context, req *pb.UserToken) (res *pb.UserToken, err error) {
    // handle auth logic

    // increase user count by token
    if _, ok := r.userCounter[req.Token]; ok {
        r.userCounter[req.Token]++
    } else {
        r.userCounter[req.Token] = 1
    }
}

好了，本文到此结束，带大家了解了《在 Golang GRPC 中集中处理请求变量的编写方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！