// Configure sets up the db connector using the options in opts. It parses the
// connection string and then sets up the dial information using the default
// dial timeout.
func (self *VanillaDBConnector) Configure(opts options.ToolOptions) error {
// create the addresses to be used to connect
connectionAddrs := util.CreateConnectionAddrs(opts.Host, opts.Port)
timeout := time.Duration(opts.Timeout) * time.Second
// set up the dial info
self.dialInfo = &mgo.DialInfo{
Addrs: connectionAddrs,
Timeout: timeout,
Direct: opts.Direct,
ReplicaSetName: opts.ReplicaSetName,
Username: opts.Auth.Username,
Password: opts.Auth.Password,
Source: opts.GetAuthenticationDatabase(),
Mechanism: opts.Auth.Mechanism,
}
kerberos.AddKerberosOpts(opts, self.dialInfo)
return nil
}
// Configure the connector to connect to the server over ssl. Parses the
// connection string, and sets up the correct function to dial the server
// based on the ssl options passed in.
func (self *SSLDBConnector) Configure(opts options.ToolOptions) error {
// create the addresses to be used to connect
connectionAddrs := util.CreateConnectionAddrs(opts.Host, opts.Port)
var err error
self.ctx, err = setupCtx(opts)
if err != nil {
return fmt.Errorf("openssl configuration: %v", err)
}
var flags openssl.DialFlags
flags = 0
if opts.SSLAllowInvalidCert || opts.SSLAllowInvalidHost {
flags = openssl.InsecureSkipHostVerification
}
// create the dialer func that will be used to connect
dialer := func(addr *mgo.ServerAddr) (net.Conn, error) {
conn, err := openssl.Dial("tcp", addr.String(), self.ctx, flags)
self.dialError = err
return conn, err
}
timeout := time.Duration(opts.Timeout) * time.Second
// set up the dial info
self.dialInfo = &mgo.DialInfo{
Addrs: connectionAddrs,
Timeout: timeout,
Direct: opts.Direct,
ReplicaSetName: opts.ReplicaSetName,
DialServer: dialer,
Username: opts.Auth.Username,
Password: opts.Auth.Password,
Source: opts.GetAuthenticationDatabase(),
Mechanism: opts.Auth.Mechanism,
}
kerberos.AddKerberosOpts(opts, self.dialInfo)
return nil
}