rpc编写与调用

在前面我们完善了对用户进行登录，用户查询图书等接口协议，但是用户在查询图书时没有做任何用户校验，如果当前用户是一个不存在的用户则我们不允许其查阅图书信息，从上文信息我们可以得知，需要user服务提供一个方法来获取用户信息供search服务使用，因此我们就需要创建一个user rpc服务，并提供一个getUser方法。

编译proto文件


  package user;
  option go_package = "user";
  message IdReq{
    int64 id = 1;
  }
  message UserInfoReply{
    int64 id = 1;
    string name = 2;
    string number = 3;
    string gender = 4;
  }
  service user {
    rpc getUser(IdReq) returns(UserInfoReply);
  }

生成rpc服务代码

$ cd service/user/rpc
$ goctl rpc proto -src user.proto -dir .

添加配置及完善yaml配置项

  $ vim service/user/rpc/internal/config/config.go

  type Config struct {
      zrpc.RpcServerConf
      Mysql struct {
          DataSource string
      }
      CacheRedis cache.CacheConf
  }

  $ vim /service/user/rpc/etc/user.yaml

  Name: user.rpc
  ListenOn: 127.0.0.1:8080
  Etcd:
  Hosts:
  - $etcdHost
    Key: user.rpc
  CacheRedis:
  - Host: $host
    Pass: $pass
    Type: node

添加资源依赖

  $ vim service/user/rpc/internal/svc/servicecontext.go

添加rpc逻辑

  $ service/user/rpc/internal/logic/getuserlogic.go

  func (l *GetUserLogic) GetUser(in *user.IdReq) (*user.UserInfoReply, error) {
      one, err := l.svcCtx.UserModel.FindOne(in.Id)
      if err != nil {
          return nil, err
      }
      return &user.UserInfoReply{
          Id:     one.Id,
          Name:   one.Name,
          Number: one.Number,
          Gender: one.Gender,
      }, nil
  }

接下来我们在search服务中调用user rpc

添加UserRpc配置及yaml配置项

  $ vim service/search/api/internal/config/config.go

  type Config struct {
      rest.RestConf
      Auth struct {
          AccessSecret string
          AccessExpire int64
      }
      UserRpc zrpc.RpcClientConf
  }

  $ vim service/search/api/etc/search-api.yaml

  Name: search-api
  Host: 0.0.0.0
  Port: 8889
  Auth:
    AccessSecret: $AccessSecret
    AccessExpire: $AccessExpire
  UserRpc:
    Etcd:
      Hosts:
        - $etcdHost

添加依赖

补充逻辑

  $ vim /service/search/api/internal/logic/searchlogic.go

  func (l *SearchLogic) Search(req types.SearchReq) (*types.SearchReply, error) {
      userIdNumber := json.Number(fmt.Sprintf("%v", l.ctx.Value("userId")))
      logx.Infof("userId: %s", userIdNumber)
      userId, err := userIdNumber.Int64()
      if err != nil {
          return nil, err
      }
      // 使用user rpc
      _, err = l.svcCtx.UserRpc.GetUser(l.ctx, &userclient.IdReq{
          Id: userId,
      })
      if err != nil {
          return nil, err
      }
      return &types.SearchReply{
          Name:  req.Name,
          Count: 100,
      }, nil
  }

启动etcd、redis、mysql

启动user rpc

  $ cd /service/user/rpc
  $ go run user.go -f etc/user.yaml

  Starting rpc server at 127.0.0.1:8080...

启动search api

$ cd service/search/api
$ go run search.go -f etc/search-api.yaml

验证服务

  $ curl -i -X GET \
    'http://127.0.0.1:8889/search/do?name=%E8%A5%BF%E6%B8%B8%E8%AE%B0' \
    -H 'authorization: eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2MTI4NjcwNzQsImlhdCI6MTYxMjc4MDY3NCwidXNlcklkIjoxfQ.JKa83g9BlEW84IiCXFGwP2aSd0xF3tMnxrOzVebbt80'

  HTTP/1.1 200 OK
  Content
  -Type: application/json
  Date: Tue, 09 Feb 2021 06:05:52 GMT
  Content-Length: 32
  {"name":"西游记","count":100}

rpc服务编写与调用

rpc编写与调用

猜你想看