Michael Crosby
08c25433ac
Signed-off-by: Michael Crosby <crosbymichael@gmail.com> Upstream-commit: ad2b34f2056ca7d6edab17967a797fb19bf4ed5a Component: engine
24 lines
557 B
Go
24 lines
557 B
Go
package ttrpc
|
|
|
|
import "github.com/pkg/errors"
|
|
|
|
type serverConfig struct {
|
|
handshaker Handshaker
|
|
}
|
|
|
|
type ServerOpt func(*serverConfig) error
|
|
|
|
// WithServerHandshaker can be passed to NewServer to ensure that the
|
|
// handshaker is called before every connection attempt.
|
|
//
|
|
// Only one handshaker is allowed per server.
|
|
func WithServerHandshaker(handshaker Handshaker) ServerOpt {
|
|
return func(c *serverConfig) error {
|
|
if c.handshaker != nil {
|
|
return errors.New("only one handshaker allowed per server")
|
|
}
|
|
c.handshaker = handshaker
|
|
return nil
|
|
}
|
|
}
|