func (db *ETCDDB) UpdateDesiredLRP(logger lager.Logger, processGuid string, update *models.DesiredLRPUpdate) (*models.DesiredLRP, error) {
logger.Info("starting")
defer logger.Info("complete")
var schedulingInfo *models.DesiredLRPSchedulingInfo
var err error
var beforeDesiredLRP *models.DesiredLRP
for i := 0; i < 2; i++ {
var index uint64
beforeDesiredLRP, index, err = db.rawDesiredLRPByProcessGuid(logger, processGuid)
if err != nil {
logger.Error("failed-to-fetch-desired-lrp", err)
break
}
schedulingInfoValue := beforeDesiredLRP.DesiredLRPSchedulingInfo()
schedulingInfo = &schedulingInfoValue
schedulingInfo.ApplyUpdate(update)
err = db.updateDesiredLRPSchedulingInfo(logger, schedulingInfo, index)
if err != nil {
logger.Error("update-scheduling-info-failed", err)
modelErr := models.ConvertError(err)
if modelErr != models.ErrResourceConflict {
break
}
// Retry on CAS fail
continue
}
break
}
if err != nil {
return nil, err
}
return beforeDesiredLRP, nil
}
_, actualLRPKey, actualLRPInstanceKey, actualLrpNetInfo, ttl := fakeEvacuationDB.EvacuateActualLRPArgsForCall(0)
Expect(*actualLRPKey).To(Equal(actual.ActualLRPKey))
Expect(*actualLRPInstanceKey).To(Equal(actual.ActualLRPInstanceKey))
Expect(*actualLrpNetInfo).To(Equal(actual.ActualLRPNetInfo))
Expect(ttl).To(BeEquivalentTo(60))
})
It("unclaims the lrp and requests an auction", func() {
Expect(fakeActualLRPDB.UnclaimActualLRPCallCount()).To(Equal(1))
_, actualLRPKey, actualLRPInstanceKey, actualLrpNetInfo, ttl := fakeEvacuationDB.EvacuateActualLRPArgsForCall(0)
Expect(*actualLRPKey).To(Equal(actual.ActualLRPKey))
Expect(*actualLRPInstanceKey).To(Equal(actual.ActualLRPInstanceKey))
Expect(*actualLrpNetInfo).To(Equal(actual.ActualLRPNetInfo))
Expect(ttl).To(BeEquivalentTo(60))
schedulingInfo := desiredLRP.DesiredLRPSchedulingInfo()
expectedStartRequest := auctioneer.NewLRPStartRequestFromSchedulingInfo(&schedulingInfo, int(actual.Index))
Expect(fakeAuctioneerClient.RequestLRPAuctionsCallCount()).To(Equal(1))
startRequests := fakeAuctioneerClient.RequestLRPAuctionsArgsForCall(0)
Expect(startRequests).To(Equal([]*auctioneer.LRPStartRequest{&expectedStartRequest}))
})
It("emits events to the hub", func() {
Eventually(actualHub.EmitCallCount).Should(Equal(2))
for i := 0; i < actualHub.EmitCallCount(); i++ {
switch event := actualHub.EmitArgsForCall(i).(type) {
case *models.ActualLRPCreatedEvent:
Expect(event.ActualLrpGroup).To(Equal(&models.ActualLRPGroup{Evacuating: afterActual}))
case *models.ActualLRPChangedEvent: