diff --git a/agent/consul/server.go b/agent/consul/server.go index 3792ec45b..de7b737d8 100644 --- a/agent/consul/server.go +++ b/agent/consul/server.go @@ -294,9 +294,8 @@ func NewServerLogger(config *Config, logger *log.Logger, tokens *token.Store) (* shutdownCh: shutdownCh, } - // Set up autopilot - apDelegate := &AutopilotDelegate{s} - s.autopilot = autopilot.NewAutopilot(logger, apDelegate, config.AutopilotInterval, config.ServerHealthInterval) + // Run divergent OSS/Enterprise setup + s.startServerEnterprise(config) // Initialize the stats fetcher that autopilot will use. s.statsFetcher = NewStatsFetcher(logger, s.connPool, s.config.Datacenter) diff --git a/agent/consul/server_oss.go b/agent/consul/server_oss.go index 05c02e46c..5038b7fd5 100644 --- a/agent/consul/server_oss.go +++ b/agent/consul/server_oss.go @@ -1,5 +1,7 @@ package consul +import "github.com/hashicorp/consul/agent/consul/autopilot" + func init() { registerEndpoint(func(s *Server) interface{} { return &ACL{s} }) registerEndpoint(func(s *Server) interface{} { return &Catalog{s} }) @@ -13,3 +15,9 @@ func init() { registerEndpoint(func(s *Server) interface{} { return &Status{s} }) registerEndpoint(func(s *Server) interface{} { return &Txn{s} }) } + +func (s *Server) startServerEnterprise(config *Config) { + // Set up autopilot + apDelegate := &AutopilotDelegate{s} + s.autopilot = autopilot.NewAutopilot(s.logger, apDelegate, config.AutopilotInterval, config.ServerHealthInterval) +} \ No newline at end of file