func (i *importer) unit(s description.Application, u description.Unit) error {
i.logger.Debugf("importing unit %s", u.Name())
// 1. construct a unitDoc
udoc, err := i.makeUnitDoc(s, u)
if err != nil {
return errors.Trace(err)
}
// 2. construct a statusDoc for the workload status and agent status
agentStatus := u.AgentStatus()
if agentStatus == nil {
return errors.NotValidf("missing agent status")
}
agentStatusDoc := i.makeStatusDoc(agentStatus)
workloadStatus := u.WorkloadStatus()
if workloadStatus == nil {
return errors.NotValidf("missing workload status")
}
workloadStatusDoc := i.makeStatusDoc(workloadStatus)
workloadVersion := u.WorkloadVersion()
versionStatus := status.Active
if workloadVersion == "" {
versionStatus = status.Unknown
}
workloadVersionDoc := statusDoc{
Status: versionStatus,
StatusInfo: workloadVersion,
}
ops, err := addUnitOps(i.st, addUnitOpsArgs{
unitDoc: udoc,
agentStatusDoc: agentStatusDoc,
workloadStatusDoc: workloadStatusDoc,
workloadVersionDoc: workloadVersionDoc,
meterStatusDoc: &meterStatusDoc{
Code: u.MeterStatusCode(),
Info: u.MeterStatusInfo(),
},
})
if err != nil {
return errors.Trace(err)
}
// If the unit is a principal, add it to its machine.
if u.Principal().Id() == "" {
ops = append(ops, txn.Op{
C: machinesC,
Id: u.Machine().Id(),
Assert: txn.DocExists,
Update: bson.M{"$addToSet": bson.M{"principals": u.Name()}},
})
}
// We should only have constraints for principal agents.
// We don't encode that business logic here, if there are constraints
// in the imported model, we put them in the database.
if cons := u.Constraints(); cons != nil {
agentGlobalKey := unitAgentGlobalKey(u.Name())
ops = append(ops, createConstraintsOp(i.st, agentGlobalKey, i.constraints(cons)))
}
if err := i.st.runTransaction(ops); err != nil {
i.logger.Debugf("failed ops: %#v", ops)
return errors.Trace(err)
}
unit := newUnit(i.st, udoc)
if annotations := u.Annotations(); len(annotations) > 0 {
if err := i.st.SetAnnotations(unit, annotations); err != nil {
return errors.Trace(err)
}
}
if err := i.importStatusHistory(unit.globalKey(), u.WorkloadStatusHistory()); err != nil {
return errors.Trace(err)
}
if err := i.importStatusHistory(unit.globalAgentKey(), u.AgentStatusHistory()); err != nil {
return errors.Trace(err)
}
if err := i.importStatusHistory(unit.globalWorkloadVersionKey(), u.WorkloadVersionHistory()); err != nil {
return errors.Trace(err)
}
if err := i.importUnitPayloads(unit, u.Payloads()); err != nil {
return errors.Trace(err)
}
return nil
}
