create res, check if lease exists, remove and error if it doesnt

RackHD · Nov 28, 2016 · 687f7dc · 687f7dc
1 parent 5c63ec3
commit 687f7dc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/ipam/reservations.go b/ipam/reservations.go
@@ -40,10 +40,16 @@ func (ipam *Ipam) CreateReservation(reservation models.Reservation) error {
 		return err
 	}
 
-	return session.DB(IpamDatabase).C(IpamCollectionLeases).Update(
+	err = session.DB(IpamDatabase).C(IpamCollectionLeases).Update(
 		bson.M{"reservation": nil, "subnet": reservation.Subnet},
 		bson.M{"$set": bson.M{"reservation": reservation.ID}},
 	)
+	if err != nil {
+		session.DB(IpamDatabase).C(IpamCollectionReservations).Remove(reservation)
+		return err
+	}
+
+	return nil
 }
 
 // UpdateReservation updates a Reservation.