func (i *importer) makeUnitDoc(s description.Service, u description.Unit) (*unitDoc, error) {
// NOTE: if we want to support units having different charms deployed
// than the service recomments and migrate that, then we should serialize
// the charm url for each unit rather than grabbing the services charm url.
// Currently the units charm url matching the service is a precondiation
// to migration.
charmUrl, err := charm.ParseURL(s.CharmURL())
if err != nil {
return nil, errors.Trace(err)
}
var subordinates []string
if subs := u.Subordinates(); len(subs) > 0 {
for _, s := range subs {
subordinates = append(subordinates, s.Id())
}
}
return &unitDoc{
Name: u.Name(),
Service: s.Name(),
Series: s.Series(),
CharmURL: charmUrl,
Principal: u.Principal().Id(),
Subordinates: subordinates,
// StorageAttachmentCount int `bson:"storageattachmentcount"`
MachineId: u.Machine().Id(),
Tools: i.makeTools(u.Tools()),
Life: Alive,
PasswordHash: u.PasswordHash(),
}, nil
}
func (i *importer) makeServiceDoc(s description.Service) (*serviceDoc, error) {
charmUrl, err := charm.ParseURL(s.CharmURL())
if err != nil {
return nil, errors.Trace(err)
}
return &serviceDoc{
Name: s.Name(),
Series: s.Series(),
Subordinate: s.Subordinate(),
CharmURL: charmUrl,
Channel: s.Channel(),
CharmModifiedVersion: s.CharmModifiedVersion(),
ForceCharm: s.ForceCharm(),
Life: Alive,
UnitCount: len(s.Units()),
RelationCount: i.relationCount(s.Name()),
Exposed: s.Exposed(),
MinUnits: s.MinUnits(),
MetricCredentials: s.MetricsCredentials(),
}, nil
}
func (i *importer) service(s description.Service) error {
// Import this service, then soon, its units.
i.logger.Debugf("importing service %s", s.Name())
// 1. construct a serviceDoc
sdoc, err := i.makeServiceDoc(s)
if err != nil {
return errors.Trace(err)
}
// 2. construct a statusDoc
status := s.Status()
if status == nil {
return errors.NotValidf("missing status")
}
statusDoc := i.makeStatusDoc(status)
// TODO: update never set malarky... maybe...
ops := addServiceOps(i.st, addServiceOpsArgs{
serviceDoc: sdoc,
statusDoc: statusDoc,
constraints: i.constraints(s.Constraints()),
// networks TODO,
// storage TODO,
settings: s.Settings(),
settingsRefCount: s.SettingsRefCount(),
leadershipSettings: s.LeadershipSettings(),
})
if err := i.st.runTransaction(ops); err != nil {
return errors.Trace(err)
}
svc := newService(i.st, sdoc)
if annotations := s.Annotations(); len(annotations) > 0 {
if err := i.st.SetAnnotations(svc, annotations); err != nil {
return errors.Trace(err)
}
}
if err := i.importStatusHistory(svc.globalKey(), s.StatusHistory()); err != nil {
return errors.Trace(err)
}
for _, unit := range s.Units() {
if err := i.unit(s, unit); err != nil {
return errors.Trace(err)
}
}
if s.Leader() != "" {
if err := i.st.LeadershipClaimer().ClaimLeadership(
s.Name(),
s.Leader(),
initialLeaderClaimTime); err != nil {
return errors.Trace(err)
}
}
return nil
}