Client

Example: function

通常我们将方法注册为服务的方法，这些方法必须满足以下的要求：

  • 必须是可导出类型的方法
  • 接受3个参数，第一个是 context.Context类型，其他2个都是可导出（或内置）的类型。
  • 第3个参数是一个指针
  • 有一个 error 类型的返回值

Rpcx 也支持将纯函数注册为服务，函数必须满足以下的要求：

  • 函数可以是可导出的或者不可导出的
  • 接受3个参数，第一个是 context.Context类型，其他2个都是可导出（或内置）的类型。
  • 第3个参数是一个指针
  • 有一个 error 类型的返回值

下面有一个例子。

服务端必须使用 RegisterFunction 来注册一个函数并且提供一个服务名。

  1. // server.go
  2. type Args struct {
  3.     A int
  4.     B int
  5. }
  7. type Reply struct {
  8.     C int
  9. }
  11. func mul(ctx context.Context, args *Args, reply *Reply) error {
  12.     reply.C = args.A * args.B
  13.     return nil
  14. }
  16. func main() {
  17.     flag.Parse()
  19.     s := server.NewServer()
  20.     s.RegisterFunction("a.fake.service", mul, "")
  21.     s.Serve("tcp", *addr)
  22. }

客户端可以通过服务名和函数名来调用服务：

  1. // client.go
  2. d := client.NewPeer2PeerDiscovery("tcp@"+*addr, "")
  3. xclient := client.NewXClient("a.fake.service", client.Failtry, client.RandomSelect, d, client.DefaultOption)
  4. defer xclient.Close()
  6. args := &example.Args{
  7.     A: 10,
  8.     B: 20,
  9. }
  11. reply := &example.Reply{}
  12. err := xclient.Call(context.Background(), "mul", args, reply)
  13. if err != nil {
  14.     log.Fatalf("failed to call: %v", err)
  15. }
  17. log.Printf("%d * %d = %d", args.A, args.B, reply.C)