Support multiple Nats URIs

Signed-off-by: Phan Le <ple@pivotallabs.com>

cfcomponent/config.go

@@ -3,17 +3,17 @@ package cfcomponent
 import (
 	"errors"
 	"fmt"
+
 	"github.com/cloudfoundry/gosteno"
 	"github.com/cloudfoundry/yagnats"
-	"strconv"
 )
 
 type Config struct {
 	Syslog     string
 	VarzPort   uint32
 	VarzUser   string
 	VarzPass   string
-	NatsHost   string
+	NatsHosts  []string
 	NatsPort   int
 	NatsUser   string
 	NatsPass   string
@@ -27,17 +27,28 @@ var DefaultYagnatsClientProvider = func(logger *gosteno.Logger) yagnats.NATSClie
 }
 
 func (c *Config) Validate(logger *gosteno.Logger) (err error) {
-	c.MbusClient = DefaultYagnatsClientProvider(logger)
+	members := []yagnats.ConnectionProvider{}
+
+	for _, natsHost := range c.NatsHosts {
+		members = append(members, &yagnats.ConnectionInfo{
+			Addr:     fmt.Sprintf("%s:%d", natsHost, c.NatsPort),
+			Username: c.NatsUser,
+			Password: c.NatsPass,
+		})
+	}
 
-	addr := c.NatsHost + ":" + strconv.Itoa(c.NatsPort)
-	info := &yagnats.ConnectionInfo{
-		Addr:     addr,
-		Username: c.NatsUser,
-		Password: c.NatsPass,
+	connectionInfo := &yagnats.ConnectionCluster{
+		Members: members,
 	}
-	err = c.MbusClient.Connect(info)
+
+	natsClient := yagnats.NewClient()
+
+	err = natsClient.Connect(connectionInfo)
+
 	if err != nil {
 		return errors.New(fmt.Sprintf("Could not connect to NATS: %v", err.Error()))
 	}
+
+	c.MbusClient = natsClient
 	return
 }