From 6af01d2d6f216f7d284fc8ea940ae4ba64dfc3a0 Mon Sep 17 00:00:00 2001 From: Diptanu Choudhury Date: Wed, 9 Dec 2015 15:24:44 -0800 Subject: [PATCH] Muted the consul debug messages --- client/alloc_runner_test.go | 4 ++-- client/client.go | 2 +- client/consul.go | 18 +++++++++++++----- client/consul_test.go | 2 +- client/task_runner_test.go | 4 ++-- 5 files changed, 19 insertions(+), 11 deletions(-) diff --git a/client/alloc_runner_test.go b/client/alloc_runner_test.go index 6b49fcb53..8e0c2c33c 100644 --- a/client/alloc_runner_test.go +++ b/client/alloc_runner_test.go @@ -31,7 +31,7 @@ func testAllocRunner(restarts bool) (*MockAllocStateUpdater, *AllocRunner) { conf.AllocDir = os.TempDir() upd := &MockAllocStateUpdater{} alloc := mock.Alloc() - consulClient, _ := NewConsulService(logger, "127.0.0.1:8500", "", "", false, false) + consulClient, _ := NewConsulService(logger, "127.0.0.1:8500", "", "", false, false, &structs.Node{}) if !restarts { alloc.Job.Type = structs.JobTypeBatch *alloc.Job.LookupTaskGroup(alloc.TaskGroup).RestartPolicy = structs.RestartPolicy{Attempts: 0} @@ -142,7 +142,7 @@ func TestAllocRunner_SaveRestoreState(t *testing.T) { } // Create a new alloc runner - consulClient, err := NewConsulService(ar.logger, "127.0.0.1:8500", "", "", false, false) + consulClient, err := NewConsulService(ar.logger, "127.0.0.1:8500", "", "", false, false, &structs.Node{}) ar2 := NewAllocRunner(ar.logger, ar.config, upd.Update, &structs.Allocation{ID: ar.alloc.ID}, consulClient) err = ar2.RestoreState() diff --git a/client/client.go b/client/client.go index c41b1b0ed..6f88f7144 100644 --- a/client/client.go +++ b/client/client.go @@ -157,7 +157,7 @@ func (c *Client) setupConsulService() error { auth := c.config.Read("consul.auth") enableSSL := c.config.ReadBoolDefault("consul.ssl", false) verifySSL := c.config.ReadBoolDefault("consul.verifyssl", true) - if consulService, err = NewConsulService(c.logger, addr, token, auth, enableSSL, verifySSL); err != nil { + if consulService, err = NewConsulService(c.logger, addr, token, auth, enableSSL, verifySSL, c.config.Node); err != nil { return err } c.consulService = consulService diff --git a/client/consul.go b/client/consul.go index 271918729..2eb54b1e1 100644 --- a/client/consul.go +++ b/client/consul.go @@ -72,6 +72,7 @@ type ConsulService struct { client consulApi logger *log.Logger shutdownCh chan struct{} + node *structs.Node trackedTasks map[string]*trackedTask serviceStates map[string]string @@ -80,7 +81,7 @@ type ConsulService struct { // A factory method to create new consul service func NewConsulService(logger *log.Logger, consulAddr string, token string, - auth string, enableSSL bool, verifySSL bool) (*ConsulService, error) { + auth string, enableSSL bool, verifySSL bool, node *structs.Node) (*ConsulService, error) { var err error var c *consul.Client cfg := consul.DefaultConfig() @@ -122,6 +123,7 @@ func NewConsulService(logger *log.Logger, consulAddr string, token string, consulService := ConsulService{ client: &consulApiClient{client: c}, logger: logger, + node: node, trackedTasks: make(map[string]*trackedTask), serviceStates: make(map[string]string), shutdownCh: make(chan struct{}), @@ -161,7 +163,7 @@ func (c *ConsulService) Deregister(task *structs.Task, allocID string) error { } c.logger.Printf("[INFO] consul: deregistering service %v with consul", service.Name) if err := c.deregisterService(service.Id); err != nil { - c.logger.Printf("[DEBUG] consul: error in deregistering service %v from consul", service.Name) + c.printDebugMessage("[DEBUG] consul: error in deregistering service %v from consul", service.Name) mErr.Errors = append(mErr.Errors, err) } } @@ -273,13 +275,13 @@ func (c *ConsulService) registerService(service *structs.Service, task *structs. } if err := c.client.ServiceRegister(asr); err != nil { - c.logger.Printf("[DEBUG] consul: error while registering service %v with consul: %v", service.Name, err) + c.printDebugMessage("[DEBUG] consul: error while registering service %v with consul: %v", service.Name, err) mErr.Errors = append(mErr.Errors, err) } for _, check := range service.Checks { cr := c.makeCheck(service, check, host, port) if err := c.registerCheck(cr); err != nil { - c.logger.Printf("[DEBUG] consul: error while registerting check %v with consul: %v", check.Name, err) + c.printDebugMessage("[DEBUG] consul: error while registerting check %v with consul: %v", check.Name, err) mErr.Errors = append(mErr.Errors, err) } @@ -289,7 +291,7 @@ func (c *ConsulService) registerService(service *structs.Service, task *structs. // registerCheck registers a check with Consul func (c *ConsulService) registerCheck(check *consul.AgentCheckRegistration) error { - c.logger.Printf("[INFO] consul: registering Check with ID: %v for service: %v", check.ID, check.ServiceID) + c.logger.Printf("[INFO] consul: registering check with ID: %v for service: %v", check.ID, check.ServiceID) return c.client.CheckRegister(check) } @@ -336,3 +338,9 @@ func (c *ConsulService) makeCheck(service *structs.Service, check *structs.Servi } return cr } + +func (c *ConsulService) printDebugMessage(message string, v ...interface{}) { + if _, ok := c.node.Attributes["consul.version"]; ok { + c.logger.Printf(message, v) + } +} diff --git a/client/consul_test.go b/client/consul_test.go index f756c078c..9cb38ede7 100644 --- a/client/consul_test.go +++ b/client/consul_test.go @@ -46,7 +46,7 @@ func (a *mockConsulApiClient) Checks() (map[string]*consul.AgentCheck, error) { func newConsulService() *ConsulService { logger := log.New(os.Stdout, "logger: ", log.Lshortfile) - c, _ := NewConsulService(logger, "", "", "", false, false) + c, _ := NewConsulService(logger, "", "", "", false, false, &structs.Node{}) c.client = &mockConsulApiClient{} return c } diff --git a/client/task_runner_test.go b/client/task_runner_test.go index 4557b6b55..96da69d4e 100644 --- a/client/task_runner_test.go +++ b/client/task_runner_test.go @@ -32,7 +32,7 @@ func testTaskRunner(restarts bool) (*MockTaskStateUpdater, *TaskRunner) { upd := &MockTaskStateUpdater{} alloc := mock.Alloc() task := alloc.Job.TaskGroups[0].Tasks[0] - consulClient, _ := NewConsulService(logger, "127.0.0.1:8500", "", "", false, false) + consulClient, _ := NewConsulService(logger, "127.0.0.1:8500", "", "", false, false, &structs.Node{}) // Initialize the port listing. This should be done by the offer process but // we have a mock so that doesn't happen. task.Resources.Networks[0].ReservedPorts = []structs.Port{{"", 80}} @@ -164,7 +164,7 @@ func TestTaskRunner_SaveRestoreState(t *testing.T) { } // Create a new task runner - consulClient, _ := NewConsulService(tr.logger, "127.0.0.1:8500", "", "", false, false) + consulClient, _ := NewConsulService(tr.logger, "127.0.0.1:8500", "", "", false, false, &structs.Node{}) tr2 := NewTaskRunner(tr.logger, tr.config, upd.Update, tr.ctx, tr.allocID, &structs.Task{Name: tr.task.Name}, tr.state, tr.restartTracker, consulClient)