2015-11-26 15:23:42 +03:00
|
|
|
/*
|
|
|
|
The client package provides a method to make synchronous, asynchronous and
|
|
|
|
streaming requests to services. By default json and protobuf codecs are
|
|
|
|
supported.
|
2015-12-03 04:02:14 +03:00
|
|
|
|
|
|
|
import "github.com/micro/go-micro/client"
|
|
|
|
|
|
|
|
c := client.NewClient()
|
|
|
|
|
2015-12-03 04:05:16 +03:00
|
|
|
req := c.NewRequest("go.micro.srv.greeter", "Greeter.Hello", &greeter.Request{
|
2015-12-03 04:02:14 +03:00
|
|
|
Name: "John",
|
|
|
|
})
|
|
|
|
|
|
|
|
rsp := &greeter.Response{}
|
|
|
|
|
|
|
|
if err := c.Call(context.Background(), req, rsp); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
fmt.Println(rsp.Msg)
|
2015-11-26 15:23:42 +03:00
|
|
|
*/
|
2015-01-14 02:31:27 +03:00
|
|
|
package client
|
|
|
|
|
2015-05-21 21:24:57 +03:00
|
|
|
import (
|
2015-05-23 13:53:40 +03:00
|
|
|
"golang.org/x/net/context"
|
2015-05-21 21:24:57 +03:00
|
|
|
)
|
|
|
|
|
2015-01-14 02:31:27 +03:00
|
|
|
type Client interface {
|
2016-01-02 22:12:17 +03:00
|
|
|
Init(...Option) error
|
|
|
|
Options() Options
|
2015-06-12 21:52:27 +03:00
|
|
|
NewPublication(topic string, msg interface{}) Publication
|
2015-12-17 23:37:35 +03:00
|
|
|
NewRequest(service, method string, req interface{}, reqOpts ...RequestOption) Request
|
|
|
|
NewProtoRequest(service, method string, req interface{}, reqOpts ...RequestOption) Request
|
|
|
|
NewJsonRequest(service, method string, req interface{}, reqOpts ...RequestOption) Request
|
2015-12-08 22:25:42 +03:00
|
|
|
Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error
|
|
|
|
CallRemote(ctx context.Context, addr string, req Request, rsp interface{}, opts ...CallOption) error
|
2015-12-17 23:37:35 +03:00
|
|
|
Stream(ctx context.Context, req Request, opts ...CallOption) (Streamer, error)
|
|
|
|
StreamRemote(ctx context.Context, addr string, req Request, opts ...CallOption) (Streamer, error)
|
2015-12-08 22:25:42 +03:00
|
|
|
Publish(ctx context.Context, p Publication, opts ...PublishOption) error
|
2015-12-20 00:56:14 +03:00
|
|
|
String() string
|
2015-06-12 21:52:27 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
type Publication interface {
|
|
|
|
Topic() string
|
|
|
|
Message() interface{}
|
|
|
|
ContentType() string
|
|
|
|
}
|
|
|
|
|
|
|
|
type Request interface {
|
|
|
|
Service() string
|
|
|
|
Method() string
|
|
|
|
ContentType() string
|
|
|
|
Request() interface{}
|
2015-12-17 23:37:35 +03:00
|
|
|
// indicates whether the request will be a streaming one rather than unary
|
|
|
|
Stream() bool
|
2015-06-01 20:55:27 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
type Streamer interface {
|
2015-12-17 23:37:35 +03:00
|
|
|
Context() context.Context
|
2015-06-01 20:55:27 +03:00
|
|
|
Request() Request
|
2015-12-17 23:37:35 +03:00
|
|
|
Send(interface{}) error
|
|
|
|
Recv(interface{}) error
|
2015-06-01 20:55:27 +03:00
|
|
|
Error() error
|
|
|
|
Close() error
|
2015-01-14 02:31:27 +03:00
|
|
|
}
|
|
|
|
|
2015-12-31 21:11:46 +03:00
|
|
|
type Option func(*Options)
|
|
|
|
type CallOption func(*CallOptions)
|
|
|
|
type PublishOption func(*PublishOptions)
|
|
|
|
type RequestOption func(*RequestOptions)
|
2015-05-21 21:24:57 +03:00
|
|
|
|
2015-01-14 02:31:27 +03:00
|
|
|
var (
|
2015-05-23 19:40:53 +03:00
|
|
|
DefaultClient Client = newRpcClient()
|
2015-01-14 02:31:27 +03:00
|
|
|
)
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Makes a synchronous call to a service using the default client
|
2015-12-08 22:25:42 +03:00
|
|
|
func Call(ctx context.Context, request Request, response interface{}, opts ...CallOption) error {
|
|
|
|
return DefaultClient.Call(ctx, request, response, opts...)
|
2015-01-14 02:31:27 +03:00
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Makes a synchronous call to the specified address using the default client
|
2015-12-08 22:25:42 +03:00
|
|
|
func CallRemote(ctx context.Context, address string, request Request, response interface{}, opts ...CallOption) error {
|
|
|
|
return DefaultClient.CallRemote(ctx, address, request, response, opts...)
|
2015-01-14 02:31:27 +03:00
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a streaming connection with a service and returns responses on the
|
|
|
|
// channel passed in. It's upto the user to close the streamer.
|
2015-12-17 23:37:35 +03:00
|
|
|
func Stream(ctx context.Context, request Request, opts ...CallOption) (Streamer, error) {
|
|
|
|
return DefaultClient.Stream(ctx, request, opts...)
|
2015-06-01 20:55:27 +03:00
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a streaming connection to the address specified.
|
2015-12-17 23:37:35 +03:00
|
|
|
func StreamRemote(ctx context.Context, address string, request Request, opts ...CallOption) (Streamer, error) {
|
|
|
|
return DefaultClient.StreamRemote(ctx, address, request, opts...)
|
2015-06-01 20:55:27 +03:00
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Publishes a publication using the default client. Using the underlying broker
|
|
|
|
// set within the options.
|
2015-06-12 21:52:27 +03:00
|
|
|
func Publish(ctx context.Context, p Publication) error {
|
|
|
|
return DefaultClient.Publish(ctx, p)
|
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a new client with the options passed in
|
2015-05-23 22:04:16 +03:00
|
|
|
func NewClient(opt ...Option) Client {
|
2015-05-23 19:40:53 +03:00
|
|
|
return newRpcClient(opt...)
|
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a new publication using the default client
|
2015-06-12 21:52:27 +03:00
|
|
|
func NewPublication(topic string, message interface{}) Publication {
|
|
|
|
return DefaultClient.NewPublication(topic, message)
|
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a new request using the default client. Content Type will
|
|
|
|
// be set to the default within options and use the appropriate codec
|
2015-12-17 23:37:35 +03:00
|
|
|
func NewRequest(service, method string, request interface{}, reqOpts ...RequestOption) Request {
|
|
|
|
return DefaultClient.NewRequest(service, method, request, reqOpts...)
|
2015-01-14 02:31:27 +03:00
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a new protobuf request using the default client
|
2015-12-17 23:37:35 +03:00
|
|
|
func NewProtoRequest(service, method string, request interface{}, reqOpts ...RequestOption) Request {
|
|
|
|
return DefaultClient.NewProtoRequest(service, method, request, reqOpts...)
|
2015-01-14 02:31:27 +03:00
|
|
|
}
|
|
|
|
|
2015-11-26 15:21:00 +03:00
|
|
|
// Creates a new json request using the default client
|
2015-12-17 23:37:35 +03:00
|
|
|
func NewJsonRequest(service, method string, request interface{}, reqOpts ...RequestOption) Request {
|
|
|
|
return DefaultClient.NewJsonRequest(service, method, request, reqOpts...)
|
2015-01-14 02:31:27 +03:00
|
|
|
}
|
2015-12-20 00:56:14 +03:00
|
|
|
|
|
|
|
func String() string {
|
|
|
|
return DefaultClient.String()
|
|
|
|
}
|