func (i *importer) unit(s description.Service, 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) ops := addUnitOps(i.st, addUnitOpsArgs{ unitDoc: udoc, agentStatusDoc: agentStatusDoc, workloadStatusDoc: workloadStatusDoc, meterStatusDoc: &meterStatusDoc{ Code: u.MeterStatusCode(), Info: u.MeterStatusInfo(), }, }) // 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 { 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) } return nil }
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 }