unistack-org/micro
4
0
Fork 3
Code Issues 20 Pull Requests 1 Actions Packages Projects Releases 281 Wiki Activity

add default port where none is specified

Browse Source
This commit is contained in:
Asim 2015-11-16 12:11:03 +00:00
parent fa3b66a5dd
commit 9b23729eac
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
registry/consul_registry.go
View File

@ -3,6 +3,8 @@ package registry
import ( import (
"encoding/json" "encoding/json"
"errors" "errors"
"fmt"
"net"
"sync" "sync"
consul "github.com/hashicorp/consul/api" consul "github.com/hashicorp/consul/api"
@ -73,7 +75,13 @@ func decodeMetadata(tags []string) map[string]string {
func newConsulRegistry(addrs []string, opts ...Option) Registry { func newConsulRegistry(addrs []string, opts ...Option) Registry {
config := consul.DefaultConfig() config := consul.DefaultConfig()
if len(addrs) > 0 { if len(addrs) > 0 {
config.Address = addrs[0] addr, port, err := net.SplitHostPort(addrs[0])
if ae, ok := err.(*net.AddrError); ok && ae.Err == "missing port in address" {
port = "8500"
config.Address = fmt.Sprintf("%s:%s", addr, port)
} else if err == nil {
config.Address = fmt.Sprintf("%s:%s", addr, port)
}
} }
client, _ := consul.NewClient(config) client, _ := consul.NewClient(config)