Adds external-source Metadata when register consul-esm as a service

hashicorp · Jun 1, 2022 · 07b8246 · 07b8246
1 parent c7b53f4
commit 07b8246
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/agent.go b/agent.go
@@ -170,6 +170,12 @@ func (a *Agent) serviceID() string {
 	return fmt.Sprintf("%s:%s", a.config.Service, a.id)
 }
 
+func (a *Agent) serviceMeta() map[string]string {
+	return map[string]string{
+		"external-source": "consul-esm",
+	}
+}
+
 type alreadyExistsError struct {
 	serviceID string
 }
@@ -188,6 +194,7 @@ func (a *Agent) register() error {
 	service := &api.AgentServiceRegistration{
 		ID:   a.serviceID(),
 		Name: a.config.Service,
+		Meta: a.serviceMeta(),
 	}
 	if a.config.Tag != "" {
 		service.Tags = []string{a.config.Tag}

diff --git a/agent_test.go b/agent_test.go
@@ -79,6 +79,9 @@ func TestAgent_registerServiceAndCheck(t *testing.T) {
 		if got, want := services[0].ServiceTags, []string{"test"}; !reflect.DeepEqual(got, want) {
 			r.Fatalf("got %q, want %q", got, want)
 		}
+		if got, want := services[0].ServiceMeta, map[string]string{"external-source": "consul-esm"}; !reflect.DeepEqual(got, want) {
+			r.Fatalf("got %q, want %q", got, want)
+		}
 
 		checks, _, err := agent.client.Health().Checks(agent.config.Service, nil)
 		if err != nil {