コード例 #1
0
func (s *BinPackScheduler) MatchOffers(offers []*mesos.Offer, lrpAuctions []auctiontypes.LRPAuction, taskAuctions []auctiontypes.TaskAuction) map[string] /*slaveId*/ *OfferMatch {
	sort.Sort(auctionrunner.SortableLRPAuctions(lrpAuctions))
	sort.Sort(auctionrunner.SortableTaskAuctions(taskAuctions))
	offerAggregates := aggregateOffersBySlave(offers)
	sort.Sort(binpackSortableOfferAggregates{offerAggregates: offerAggregates, registry: s.registry})

	matches := make(map[string]*OfferMatch)
	for _, offers := range offerAggregates {
		matches[offers[0].SlaveId.GetValue()] = &OfferMatch{Offers: offers}
	}
	matches[""] = &OfferMatch{}

	for _, lrpAuction := range lrpAuctions {
		_, offers := matchLrpAuction(offerAggregates, &lrpAuction)
		if offers != nil {
			matches[offers[0].SlaveId.GetValue()].LrpAuctions = append(matches[offers[0].SlaveId.GetValue()].LrpAuctions, lrpAuction)
		} else {
			matches[""].LrpAuctions = append(matches[""].LrpAuctions, lrpAuction)
		}
	}
	for _, taskAuction := range taskAuctions {
		_, offers := matchTaskAuction(offerAggregates, &taskAuction)
		if offers != nil {
			matches[offers[0].SlaveId.GetValue()].TaskAuctions = append(matches[offers[0].SlaveId.GetValue()].TaskAuctions, taskAuction)
		} else {
			matches[""].TaskAuctions = append(matches[""].TaskAuctions, taskAuction)
		}
	}

	return matches
}
コード例 #2
0
func (s *SpreadScheduler) MatchOffers(offers []*mesos.Offer, lrpAuctions []auctiontypes.LRPAuction, taskAuctions []auctiontypes.TaskAuction) map[string]*OfferMatch {
	s.pendingRegistry = NewTaskRegistry()

	sort.Sort(auctionrunner.SortableLRPAuctions(lrpAuctions))
	sort.Sort(auctionrunner.SortableTaskAuctions(taskAuctions))
	offerAggregates := aggregateOffersBySlave(offers)

	matches := make(map[string]*OfferMatch)
	for _, offers := range offerAggregates {
		matches[offers[0].SlaveId.GetValue()] = &OfferMatch{Offers: offers}
	}
	matches[""] = &OfferMatch{}

	for _, lrpAuction := range lrpAuctions {
		sort.Sort(spreadSortableOfferAggregatesForLrp{offerAggregates: offerAggregates, registry: s.registry, pendingRegistry: s.pendingRegistry, lrpGuid: lrpAuction.ProcessGuid})
		_, offers := matchLrpAuction(offerAggregates, &lrpAuction)
		if offers != nil {
			matches[offers[0].SlaveId.GetValue()].LrpAuctions = append(matches[offers[0].SlaveId.GetValue()].LrpAuctions, lrpAuction)
			s.pendingRegistry.AddLrp(offers[0].SlaveId.GetValue(), lrpAuction.ProcessGuid, lrpAuction.Index, mesos.TaskState_TASK_STAGING)
		} else {
			matches[""].LrpAuctions = append(matches[""].LrpAuctions, lrpAuction)
		}
	}
	for _, taskAuction := range taskAuctions {
		sort.Sort(spreadSortableOfferAggregatesForTask{offerAggregates: offerAggregates, registry: s.registry, pendingRegistry: s.pendingRegistry})
		_, offers := matchTaskAuction(offerAggregates, &taskAuction)
		if offers != nil {
			matches[offers[0].SlaveId.GetValue()].TaskAuctions = append(matches[offers[0].SlaveId.GetValue()].TaskAuctions, taskAuction)
			s.pendingRegistry.AddTask(offers[0].SlaveId.GetValue(), taskAuction.TaskGuid, mesos.TaskState_TASK_STAGING)
		} else {
			matches[""].TaskAuctions = append(matches[""].TaskAuctions, taskAuction)
		}
	}

	s.pendingRegistry = nil
	return matches
}
コード例 #3
0
			It("sorts by index", func() {
				for i := 0; i < len(lrps); i++ {
					Expect(lrps[i].Index).To(BeEquivalentTo(i))
				}
			})
		})
	})

	Describe("Task Auctions", func() {
		var tasks []auctiontypes.TaskAuction

		BeforeEach(func() {
			tasks = []auctiontypes.TaskAuction{
				BuildTaskAuction(BuildTask("tg-6", "domain", "linux", 10, 10), time.Time{}),
				BuildTaskAuction(BuildTask("tg-7", "domain", "linux", 20, 10), time.Time{}),
				BuildTaskAuction(BuildTask("tg-8", "domain", "linux", 30, 10), time.Time{}),
				BuildTaskAuction(BuildTask("tg-9", "domain", "linux", 40, 10), time.Time{}),
			}

			sort.Sort(auctionrunner.SortableTaskAuctions(tasks))
		})

		It("sorts boulders before pebbles", func() {
			Expect(tasks[0].Task.TaskGuid).To((Equal("tg-9")))
			Expect(tasks[1].Task.TaskGuid).To((Equal("tg-8")))
			Expect(tasks[2].Task.TaskGuid).To((Equal("tg-7")))
			Expect(tasks[3].Task.TaskGuid).To((Equal("tg-6")))
		})
	})
})