Auth

Example:auth

For the security, only authorized clients can invoke services.

Clients must set an authorization token which can gets from OAuth/OAuth2 or granted access token.

Servers receive requests, first check the auth token. They reject requests with invalid tokens.

  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. }

Server must define a AuthFunc to validate auth token. In the above example, only requests with token bearer tGzv3JOkF0XG5Qx2TlKWIA are valid.

Client must set the token:

  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. }

Notice: you must set a map[string]string as share.ReqMetaDataKey otherwise it will report an error

By smallnest updated 2018-12-04 11:47:26