mirror of https://github.com/hashicorp/consul
27 lines
955 B
Go
27 lines
955 B
Go
package gocql
|
|
|
|
import "net"
|
|
|
|
// AddressTranslator provides a way to translate node addresses (and ports) that are
|
|
// discovered or received as a node event. This can be useful in an ec2 environment,
|
|
// for instance, to translate public IPs to private IPs.
|
|
type AddressTranslator interface {
|
|
// Translate will translate the provided address and/or port to another
|
|
// address and/or port. If no translation is possible, Translate will return the
|
|
// address and port provided to it.
|
|
Translate(addr net.IP, port int) (net.IP, int)
|
|
}
|
|
|
|
type AddressTranslatorFunc func(addr net.IP, port int) (net.IP, int)
|
|
|
|
func (fn AddressTranslatorFunc) Translate(addr net.IP, port int) (net.IP, int) {
|
|
return fn(addr, port)
|
|
}
|
|
|
|
// IdentityTranslator will do nothing but return what it was provided. It is essentially a no-op.
|
|
func IdentityTranslator() AddressTranslator {
|
|
return AddressTranslatorFunc(func(addr net.IP, port int) (net.IP, int) {
|
|
return addr, port
|
|
})
|
|
}
|