Merge pull request #51 from netlify/salvador/connect-correct-replset

[mongo] Use the primary node only if it belongs to the given replica set

Author: sgirones

mongo/db.go

@@ -24,13 +24,15 @@ type Config struct {
 	TLS         *nftls.Config `mapstructure:"tls_conf"`
 	DB          string        `mapstructure:"db"`
 	Servers     []string      `mapstructure:"servers"`
+	ReplSetName string        `mapstructure:"replset_name"`
 	ConnTimeout int64         `mapstructure:"conn_timeout"`
 }
 
 func Connect(config *Config, log *logrus.Entry) (*mgo.Database, error) {
 	info := &mgo.DialInfo{
-		Addrs:   config.Servers,
-		Timeout: time.Second * time.Duration(config.ConnTimeout),
+		Addrs:          config.Servers,
+		ReplicaSetName: config.ReplSetName,
+		Timeout:        time.Second * time.Duration(config.ConnTimeout),
 	}
 
 	if config.TLS != nil {
@@ -53,7 +55,11 @@ func Connect(config *Config, log *logrus.Entry) (*mgo.Database, error) {
 		log.Debug("Skipping TLS config")
 	}
 
-	log.WithField("servers", strings.Join(info.Addrs, ",")).Debug("Dialing database")
+	log.WithFields(logrus.Fields{
+		"servers":     strings.Join(info.Addrs, ","),
+		"replica_set": config.ReplSetName,
+	}).Debug("Dialing database")
+
 	sess, err := mgo.DialWithInfo(info)
 	if err != nil {
 		return nil, err