Add support for SpawnDaemon().

cluster/cluster.go

@@ -148,10 +148,10 @@ func Restart(ctx context.Context) error {
 }
 
 // StartWith a local cluster with specific addresses
-func StartWith(localPeers []gubernator.PeerInfo) error {
+func StartWith(localPeers []gubernator.PeerInfo, opts ...option) error {
 	for _, peer := range localPeers {
 		ctx, cancel := context.WithTimeout(context.Background(), clock.Second*10)
-		d, err := gubernator.SpawnDaemon(ctx, gubernator.DaemonConfig{
+		cfg := gubernator.DaemonConfig{
 			Logger:            logrus.WithField("instance", peer.GRPCAddress),
 			InstanceID:        peer.GRPCAddress,
 			GRPCListenAddress: peer.GRPCAddress,
@@ -163,7 +163,11 @@ func StartWith(localPeers []gubernator.PeerInfo) error {
 				GlobalTimeout:  clock.Second * 5,
 				BatchTimeout:   clock.Second * 5,
 			},
-		})
+		}
+		for _, opt := range opts {
+			opt.Apply(&cfg)
+		}
+		d, err := gubernator.SpawnDaemon(ctx, cfg)
 		cancel()
 		if err != nil {
 			return errors.Wrapf(err, "while starting server for addr '%s'", peer.GRPCAddress)
@@ -196,3 +200,20 @@ func Stop() {
 	peers = nil
 	daemons = nil
 }
+
+type option interface {
+	Apply(cfg *gubernator.DaemonConfig)
+}
+
+type eventChannelOption struct {
+	eventChannel chan<- gubernator.HitEvent
+}
+
+func (o *eventChannelOption) Apply(cfg *gubernator.DaemonConfig) {
+	cfg.EventChannel = o.eventChannel
+}
+
+// WithEventChannel sets EventChannel to Gubernator config.
+func WithEventChannel(eventChannel chan<- gubernator.HitEvent) option {
+	return &eventChannelOption{eventChannel: eventChannel}
+}

config.go

@@ -256,6 +256,9 @@ type DaemonConfig struct {
 	// (Optional) TraceLevel sets the tracing level, this controls the number of spans included in a single trace.
 	//  Valid options are (tracing.InfoLevel, tracing.DebugLevel) Defaults to tracing.InfoLevel
 	TraceLevel tracing.Level
+
+	// (Optional) EventChannel receives hit events
+	EventChannel chan<- HitEvent
 }
 
 func (d *DaemonConfig) ClientTLS() *tls.Config {

daemon.go

@@ -156,6 +156,7 @@ func (s *Daemon) Start(ctx context.Context) error {
 		CacheSize:     s.conf.CacheSize,
 		Workers:       s.conf.Workers,
 		InstanceID:    s.conf.InstanceID,
+		EventChannel:  s.conf.EventChannel,
 	}
 
 	s.V1Server, err = NewV1Instance(s.instanceConf)