2015-01-14 01:31:27 +02:00
|
|
|
package client
|
|
|
|
|
2015-05-21 20:24:57 +02:00
|
|
|
import (
|
|
|
|
"github.com/myodc/go-micro/transport"
|
2015-05-23 12:53:40 +02:00
|
|
|
"golang.org/x/net/context"
|
2015-05-21 20:24:57 +02:00
|
|
|
)
|
|
|
|
|
2015-01-14 01:31:27 +02:00
|
|
|
type Client interface {
|
|
|
|
NewRequest(string, string, interface{}) Request
|
|
|
|
NewProtoRequest(string, string, interface{}) Request
|
|
|
|
NewJsonRequest(string, string, interface{}) Request
|
2015-05-23 12:53:40 +02:00
|
|
|
Call(context.Context, Request, interface{}) error
|
|
|
|
CallRemote(context.Context, string, Request, interface{}) error
|
2015-01-14 01:31:27 +02:00
|
|
|
}
|
|
|
|
|
2015-05-21 20:24:57 +02:00
|
|
|
type options struct {
|
|
|
|
transport transport.Transport
|
|
|
|
}
|
|
|
|
|
2015-05-21 20:28:57 +02:00
|
|
|
type Option func(*options)
|
2015-05-21 20:24:57 +02:00
|
|
|
|
2015-01-14 01:31:27 +02:00
|
|
|
var (
|
2015-05-23 18:40:53 +02:00
|
|
|
DefaultClient Client = newRpcClient()
|
2015-01-14 01:31:27 +02:00
|
|
|
)
|
|
|
|
|
2015-05-21 20:28:57 +02:00
|
|
|
func Transport(t transport.Transport) Option {
|
|
|
|
return func(o *options) {
|
|
|
|
o.transport = t
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-05-23 12:53:40 +02:00
|
|
|
func Call(ctx context.Context, request Request, response interface{}) error {
|
|
|
|
return DefaultClient.Call(ctx, request, response)
|
2015-01-14 01:31:27 +02:00
|
|
|
}
|
|
|
|
|
2015-05-23 12:53:40 +02:00
|
|
|
func CallRemote(ctx context.Context, address string, request Request, response interface{}) error {
|
|
|
|
return DefaultClient.CallRemote(ctx, address, request, response)
|
2015-01-14 01:31:27 +02:00
|
|
|
}
|
|
|
|
|
2015-05-23 21:04:16 +02:00
|
|
|
func NewClient(opt ...Option) Client {
|
2015-05-23 18:40:53 +02:00
|
|
|
return newRpcClient(opt...)
|
|
|
|
}
|
|
|
|
|
2015-01-14 01:31:27 +02:00
|
|
|
func NewRequest(service, method string, request interface{}) Request {
|
2015-05-21 20:24:57 +02:00
|
|
|
return DefaultClient.NewRequest(service, method, request)
|
2015-01-14 01:31:27 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func NewProtoRequest(service, method string, request interface{}) Request {
|
2015-05-21 20:24:57 +02:00
|
|
|
return DefaultClient.NewProtoRequest(service, method, request)
|
2015-01-14 01:31:27 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func NewJsonRequest(service, method string, request interface{}) Request {
|
2015-05-21 20:24:57 +02:00
|
|
|
return DefaultClient.NewJsonRequest(service, method, request)
|
2015-01-14 01:31:27 +02:00
|
|
|
}
|