Replace proxy options

This commit is contained in:
Asim Aslam
2019-12-16 14:55:47 +00:00
parent a1ddfa827e
commit 03700ae6c0
5 changed files with 88 additions and 95 deletions

51
proxy/options.go Normal file
View File

@@ -0,0 +1,51 @@
// Package proxy is a transparent proxy built on the go-micro/server
package proxy
import (
"github.com/micro/go-micro/client"
"github.com/micro/go-micro/router"
)
type Options struct {
// Specific endpoint to always call
Endpoint string
// The default client to use
Client client.Client
// The default router to use
Router router.Router
// Extra links for different clients
Links map[string]client.Client
}
type Option func(o *Options)
// WithEndpoint sets a proxy endpoint
func WithEndpoint(e string) Option {
return func(o *Options) {
o.Endpoint = e
}
}
// WithClient sets the client
func WithClient(c client.Client) Option {
return func(o *Options) {
o.Client = c
}
}
// WithRouter specifies the router to use
func WithRouter(r router.Router) Option {
return func(o *Options) {
o.Router = r
}
}
// WithLink sets a link for outbound requests
func WithLink(name string, c client.Client) Option {
return func(o *Options) {
if o.Links == nil {
o.Links = make(map[string]client.Client)
}
o.Links[name] = c
}
}