// registerService registers a Service with Consul func (c *ConsulService) registerService(service *structs.Service, task *structs.Task, alloc *structs.Allocation) error { var mErr multierror.Error host, port := task.FindHostAndPortFor(service.PortLabel) if host == "" || port == 0 { return fmt.Errorf("consul: the port:%q marked for registration of service: %q couldn't be found", service.PortLabel, service.Name) } serviceID := alloc.Services[service.Name] c.serviceStates[serviceID] = service.Hash() asr := &consul.AgentServiceRegistration{ ID: serviceID, Name: service.Name, Tags: service.Tags, Port: port, Address: host, } if err := c.client.ServiceRegister(asr); err != nil { c.printLogMessage("[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(serviceID, check, host, port) if err := c.registerCheck(cr); err != nil { c.printLogMessage("[DEBUG] consul: error while registering check %v with consul: %v", check.Name, err) mErr.Errors = append(mErr.Errors, err) } } return mErr.ErrorOrNil() }
// registerService registers a Service with Consul func (c *ConsulService) registerService(service *structs.Service, task *structs.Task, allocID string) error { var mErr multierror.Error service.Id = fmt.Sprintf("%s-%s", allocID, service.Name) host, port := task.FindHostAndPortFor(service.PortLabel) if host == "" || port == 0 { return fmt.Errorf("consul: The port:%s marked for registration of service: %s couldn't be found", service.PortLabel, service.Name) } c.serviceStates[service.Id] = service.Hash() asr := &consul.AgentServiceRegistration{ ID: service.Id, Name: service.Name, Tags: service.Tags, Port: port, Address: host, } if err := c.client.ServiceRegister(asr); err != nil { c.logger.Printf("[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("[ERROR] consul: Error while registerting check %v with Consul: %v", check.Name, err) mErr.Errors = append(mErr.Errors, err) } } return mErr.ErrorOrNil() }
func TestConsul_Service_Should_Be_Re_Reregistered_On_Change(t *testing.T) { c := newConsulService() task := newTask() s1 := structs.Service{ Name: "example-cache-redis", Tags: []string{"global"}, PortLabel: "db", } task.Services = append(task.Services, &s1) alloc := mock.Alloc() serviceID := alloc.Services[s1.Name] c.Register(task, alloc) s1.Tags = []string{"frontcache"} c.performSync() if len(c.serviceStates) != 1 { t.Fatal("We should be tracking one service") } if c.serviceStates[serviceID] != s1.Hash() { t.Fatalf("Hash is %v, expected %v", c.serviceStates[serviceID], s1.Hash()) } }