micro/util/wrapper/wrapper.go

26 lines
707 B
Go
Raw Normal View History

package wrapper
2015-12-21 02:50:16 +03:00
import (
2018-03-03 14:53:52 +03:00
"context"
"github.com/micro/go-micro/v3/client"
2015-12-21 02:50:16 +03:00
)
type staticClient struct {
address string
client.Client
}
func (s *staticClient) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
return s.Client.Call(ctx, req, rsp, append(opts, client.WithAddress(s.address))...)
}
func (s *staticClient) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) {
return s.Client.Stream(ctx, req, append(opts, client.WithAddress(s.address))...)
}
// StaticClient sets an address on every call
func StaticClient(address string, c client.Client) client.Client {
return &staticClient{address, c}
}