• 身份认证

    身份认证

    示例: 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的值，否则调用会出错