From 07e7416e08602773bcb9e639586db6fed7d49c00 Mon Sep 17 00:00:00 2001
From: "bastian.b" <bastian.b@webinterpret.com>
Date: Thu, 16 Aug 2018 18:59:13 +0200
Subject: [PATCH] Add availibity to add default attrs

---
 bridge/bridge.go | 2 +-
 bridge/types.go  | 1 +
 bridge/util.go   | 9 ++++++++-
 registrator.go   | 2 ++
 4 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/bridge/bridge.go b/bridge/bridge.go
index f02ca99c9..28eda5fea 100644
--- a/bridge/bridge.go
+++ b/bridge/bridge.go
@@ -266,7 +266,7 @@ func (b *Bridge) newService(port ServicePort, isgroup bool) *Service {
 		port.HostIP = b.config.HostIp
 	}
 
-	metadata, metadataFromPort := serviceMetaData(container.Config, port.ExposedPort)
+	metadata, metadataFromPort := serviceMetaData(container.Config, port.ExposedPort, b.config.ForceAttrs)
 
 	ignore := mapDefault(metadata, "ignore", "")
 	if ignore != "" {
diff --git a/bridge/types.go b/bridge/types.go
index e643ed3f5..0600435c2 100644
--- a/bridge/types.go
+++ b/bridge/types.go
@@ -25,6 +25,7 @@ type Config struct {
 	Explicit        bool
 	UseIpFromLabel  string
 	ForceTags       string
+	ForceAttrs      string
 	RefreshTtl      int
 	RefreshInterval int
 	DeregisterCheck string
diff --git a/bridge/util.go b/bridge/util.go
index e3c295cfb..e3186f5f0 100644
--- a/bridge/util.go
+++ b/bridge/util.go
@@ -54,13 +54,20 @@ func combineTags(tagParts ...string) []string {
 	return tags
 }
 
-func serviceMetaData(config *dockerapi.Config, port string) (map[string]string, map[string]bool) {
+func serviceMetaData(config *dockerapi.Config, port string, default_attrs string) (map[string]string, map[string]bool) {
 	meta := config.Env
 	for k, v := range config.Labels {
 		meta = append(meta, k+"="+v)
 	}
 	metadata := make(map[string]string)
 	metadataFromPort := make(map[string]bool)
+
+	for _, kv := range strings.Split(default_attrs, ",") {
+		kvp := strings.SplitN(kv, "=", 2)
+		key := strings.ToLower(kvp[0])
+		metadata[key] = kvp[1]
+	}
+
 	for _, kv := range meta {
 		kvp := strings.SplitN(kv, "=", 2)
 		if strings.HasPrefix(kvp[0], "SERVICE_") && len(kvp) > 1 {
diff --git a/registrator.go b/registrator.go
index e26d3397a..9d47b0a97 100644
--- a/registrator.go
+++ b/registrator.go
@@ -26,6 +26,7 @@ var useIpFromLabel = flag.String("useIpFromLabel", "", "Use IP which is stored i
 var refreshInterval = flag.Int("ttl-refresh", 0, "Frequency with which service TTLs are refreshed")
 var refreshTtl = flag.Int("ttl", 0, "TTL for services (default is no expiry)")
 var forceTags = flag.String("tags", "", "Append tags for all registered services")
+var forceAttrs = flag.String("attrs", "", "Append attrs (ServiceMeta) for all registered services (only for consul)")
 var resyncInterval = flag.Int("resync", 0, "Frequency with which services are resynchronized")
 var deregister = flag.String("deregister", "always", "Deregister exited services \"always\" or \"on-success\"")
 var retryAttempts = flag.Int("retry-attempts", 0, "Max retry attempts to establish a connection with the backend. Use -1 for infinite retries")
@@ -108,6 +109,7 @@ func main() {
 		Explicit:        *explicit,
 		UseIpFromLabel:  *useIpFromLabel,
 		ForceTags:       *forceTags,
+		ForceAttrs:      *forceAttrs,
 		RefreshTtl:      *refreshTtl,
 		RefreshInterval: *refreshInterval,
 		DeregisterCheck: *deregister,