Add availibity to add default attrs

Author: gokuatkai

README.md

@@ -44,6 +44,7 @@ Guide. Typically, running Registrator looks like this:
 Usage of /bin/registrator:
   /bin/registrator [options] <registry URI>
 
+  -attrs="": Append attrs (ServiceMeta) for all registered services (only for consul)
   -cleanup=false: Remove dangling services
   -deregister="always": Deregister exited services "always" or "on-success"
   -internal=false: Use internal ports instead of published ones
@@ -52,8 +53,8 @@ Usage of /bin/registrator:
   -retry-attempts=0: Max retry attempts to establish a connection with the backend. Use -1 for infinite retries
   -retry-interval=2000: Interval (in millisecond) between retry-attempts.
   -tags="": Append tags for all registered services
-  -ttl=0: TTL for services (default is no expiry)
   -ttl-refresh=0: Frequency with which service TTLs are refreshed
+  -ttl=0: TTL for services (default is no expiry)
 ```
 
 ## Contributing

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 != "" {

bridge/types.go

@@ -25,6 +25,7 @@ type Config struct {
 	Explicit        bool
 	UseIpFromLabel  string
 	ForceTags       string
+	ForceAttrs      string
 	RefreshTtl      int
 	RefreshInterval int
 	DeregisterCheck string

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 {

docs/user/run.md

@@ -41,6 +41,7 @@ Option                           | Since | Description
 `-retry-attempts <number>`       | v7    | Max retry attempts to establish a connection with the backend
 `-retry-interval <milliseconds>` | v7    | Interval (in millisecond) between retry-attempts
 `-tags <tags>`                   | v5    | Force comma-separated tags on all registered services
+`-attrs <attrs>`                 | v?    | Force comma-separated attrs on all registered services (Consul Only)
 `-ttl <seconds>`                 |       | TTL for services. Default: 0, no expiry (supported backends only)
 `-ttl-refresh <seconds>`         |       | Frequency service TTLs are refreshed (supported backends only)
 `-useIpFromLabel <label>`        |       | Uses the IP address stored in the given label, which is assigned to a container, for registration with Consul

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,