身份认证

示例: auth

出于安全的考虑， 很多场景下只有授权的客户端才可以调用服务。

客户端必须设置一个 token, 这个token可以从其他的 OAuth/OAuth2 服务器获得，或者由服务提供者分配。

服务接收到请求后，需要验证这个token。如果是token是从 OAuth2服务器中申请到，则服务需要到OAuth2服务中去验证， 如果是自己分配的，则需要和自己的记录进行对别。

因为rpcx提供的是一个身份验证的框架，所以具体的身份验证需要自己集成和验证。

  1. func main() {
  2.     flag.Parse()
  4.     s := server.NewServer()
  5.     s.RegisterName("Arith", new(example.Arith), "")
  6.     s.AuthFunc = auth
  7.     s.Serve("reuseport", *addr)
  8. }
  10. func auth(ctx context.Context, req *protocol.Message, token string) error {
  12.     if token == "bearer tGzv3JOkF0XG5Qx2TlKWIA" {
  13.         return nil
  14.     }
  16.     return errors.New("invalid token")
  17. }

服务器必须定义 AuthFunc 来验证token。在上面的例子中， 只有token为bearer tGzv3JOkF0XG5Qx2TlKWIA 才是合法的客户端。

客户端必须设置这个toekn:

  2. func main() {
  3.     flag.Parse()
  5.     d := client.NewPeer2PeerDiscovery("tcp@"+*addr, "")
  7.     option := client.DefaultOption
  8.     option.ReadTimeout = 10 * time.Second
  10.     xclient := client.NewXClient("Arith", client.Failtry, client.RandomSelect, d, option)
  11.     defer xclient.Close()
  13.     //xclient.Auth("bearer tGzv3JOkF0XG5Qx2TlKWIA")
  14.     xclient.Auth("bearer abcdefg1234567")
  16.     args := &example.Args{
  17.         A: 10,
  18.         B: 20,
  19.     }
  21.     reply := &example.Reply{}
  22.     ctx := context.WithValue(context.Background(), share.ReqMetaDataKey, make(map[string]string))
  23.     err := xclient.Call(ctx, "Mul", args, reply)
  24.     if err != nil {
  25.         log.Fatalf("failed to call: %v", err)
  26.     }
  28.     log.Printf("%d * %d = %d", args.A, args.B, reply.C)
  30. }

注意: 你必须设置 map[string]string 为 share.ReqMetaDataKey的值，否则调用会出错