// MaybeInitiateMongoServer checks for an existing mongo configuration.
// If no existing configuration is found one is created using Initiate.
func MaybeInitiateMongoServer(p InitiateMongoParams) error {
logger.Debugf("Initiating mongo replicaset; dialInfo %#v; memberHostport %q; user %q; password %q", p.DialInfo, p.MemberHostPort, p.User, p.Password)
defer logger.Infof("finished MaybeInitiateMongoServer")
if len(p.DialInfo.Addrs) > 1 {
logger.Infof("more than one member; replica set must be already initiated")
return nil
}
p.DialInfo.Direct = true
// TODO(rog) remove this code when we no longer need to upgrade
// from pre-HA-capable environments.
if p.User != "" {
p.DialInfo.Username = p.User
p.DialInfo.Password = p.Password
}
session, err := mgo.DialWithInfo(p.DialInfo)
if err != nil {
return fmt.Errorf("can't dial mongo to initiate replicaset: %v", err)
}
defer session.Close()
cfg, err := replicaset.CurrentConfig(session)
if err == nil && len(cfg.Members) > 0 {
logger.Infof("replica set configuration already found: %#v", cfg)
return nil
}
if err != nil && err != mgo.ErrNotFound {
return fmt.Errorf("cannot get replica set configuration: %v", err)
}
err = replicaset.Initiate(
session,
p.MemberHostPort,
mongo.ReplicaSetName,
map[string]string{
jujuMachineTag: agent.BootstrapMachineId,
},
)
if err != nil {
return fmt.Errorf("cannot initiate replica set: %v", err)
}
return nil
}
// startReplicaSet starts up a replica set with n mongo instances.
func startReplicaSet(n int) (_ []*testing.MgoInstance, err error) {
insts := make([]*testing.MgoInstance, 0, n)
root, err := newMongoInstance()
if err != nil {
return nil, err
}
insts = append(insts, root)
defer func() {
if err == nil {
return
}
for _, inst := range insts {
inst.Destroy()
}
}()
dialInfo := root.DialInfo()
dialInfo.Direct = true
dialInfo.Timeout = 60 * time.Second
session, err := root.DialDirect()
if err != nil {
return nil, fmt.Errorf("cannot dial root instance: %v", err)
}
defer session.Close()
logger.Infof("dialled root instance")
if err := replicaset.Initiate(session, root.Addr(), replicaSetName, nil); err != nil {
return nil, fmt.Errorf("cannot initiate replica set: %v", err)
}
var members []replicaset.Member
for i := 1; i < n; i++ {
inst, err := newMongoInstance()
if err != nil {
return nil, err
}
insts = append(insts, inst)
members = append(members, replicaset.Member{
Address: inst.Addr(),
Priority: newFloat64(0.1),
Id: i + 1,
})
}
attempt := utils.AttemptStrategy{
Total: 60 * time.Second,
Delay: 1 * time.Second,
}
for a := attempt.Start(); a.Next(); {
err := replicaset.Add(session, members...)
if err == nil {
break
}
logger.Errorf("cannot add members: %v", err)
if !a.HasNext() {
return nil, fmt.Errorf("timed out trying to add members")
}
logger.Errorf("retrying")
}
return insts, err
}
