micro/proxy/proxy.go

57 lines
1.4 KiB
Go
Raw Normal View History

2019-06-06 11:50:25 +03:00
// Package proxy is a transparent proxy built on the go-micro/server
package proxy
2019-06-06 19:49:41 +03:00
import (
"context"
"github.com/micro/go-micro/client"
2019-06-12 14:45:42 +03:00
"github.com/micro/go-micro/config/options"
2019-08-05 19:44:33 +03:00
"github.com/micro/go-micro/router"
2019-06-06 19:49:41 +03:00
"github.com/micro/go-micro/server"
)
// Proxy can be used as a proxy server for go-micro services
type Proxy interface {
options.Options
// ProcessMessage handles inbound messages
ProcessMessage(context.Context, server.Message) error
// ServeRequest handles inbound requests
2019-06-06 19:58:21 +03:00
ServeRequest(context.Context, server.Request, server.Response) error
2019-06-06 19:49:41 +03:00
}
var (
DefaultEndpoint = "localhost:9090"
)
// WithEndpoint sets a proxy endpoint
func WithEndpoint(e string) options.Option {
return options.WithValue("proxy.endpoint", e)
}
// WithClient sets the client
func WithClient(c client.Client) options.Option {
return options.WithValue("proxy.client", c)
}
// WithRouter specifies the router to use
func WithRouter(r router.Router) options.Option {
return options.WithValue("proxy.router", r)
}
2019-08-23 16:05:11 +03:00
// WithLink sets a link for outbound requests
func WithLink(name string, c client.Client) options.Option {
return func(o *options.Values) error {
var links map[string]client.Client
v, ok := o.Get("proxy.links")
if ok {
links = v.(map[string]client.Client)
} else {
links = map[string]client.Client{}
}
links[name] = c
// save the links
o.Set("proxy.links", links)
return nil
}
}