@Override public String apply(StringRequest input) { return input.getString(); } })), Sets.newHashSet("a-50", "f-50"));
@Override public ResourceRequirement estimateRequirement(StringRequest s, ResourcePool resourcePool) { double memory = Double.parseDouble(s.getString().split("-")[1]); this.queriedRequests.add(s.getString()); return resourcePool.getResourceRequirementBuilder().setRequirement(MEMORY, memory).build(); } }
@Test public void testAllocateRequests() throws Exception { RequestAllocatorConfig<StringRequest> configuration = RequestAllocatorConfig.builder(new StringRequest.StringRequestEstimator()) .withPrioritizer(new StringRequest.StringRequestComparator()).build(); BruteForceAllocator<StringRequest> allocator = new BruteForceAllocator<>(configuration); ResourcePool pool = ResourcePool.builder().maxResource(StringRequest.MEMORY, 100.).build(); List<Requestor<StringRequest>> requests = Lists.<Requestor<StringRequest>>newArrayList( new StringRequestor("r1", "a-50", "f-50", "k-20"), new StringRequestor("r2", "j-10", "b-20", "e-20"), new StringRequestor("r3", "g-20", "c-200", "d-30")); AllocatedRequestsIterator<StringRequest> result = allocator.allocateRequests(requests.iterator(), pool); List<StringRequest> resultList = Lists.newArrayList(result); Assert.assertEquals(resultList.size(), 4); Assert.assertEquals(resultList.get(0).getString(), "a-50"); Assert.assertEquals(resultList.get(1).getString(), "b-20"); // No c because it is too large to fit Assert.assertEquals(resultList.get(2).getString(), "d-30"); Assert.assertEquals(resultList.get(3).getString(), "e-20"); } }
@Test public void testAllocateRequests() throws Exception { StringRequest.StringRequestEstimator estimator = new StringRequest.StringRequestEstimator(); RequestAllocatorConfig<StringRequest> configuration = RequestAllocatorConfig.builder(estimator).withPrioritizer(new StringRequest.StringRequestComparator()).build(); PreOrderAllocator<StringRequest> allocator = new PreOrderAllocator<>(configuration); ResourcePool pool = ResourcePool.builder().maxResource(StringRequest.MEMORY, 100.).build(); List<Requestor<StringRequest>> requests = Lists.<Requestor<StringRequest>>newArrayList( new StringRequestor("r1", "a-50", "f-50", "k-20"), new StringRequestor("r2", "j-10", "b-20", "e-20"), new StringRequestor("r3", "g-20", "c-200", "d-30")); AllocatedRequestsIterator<StringRequest> result = allocator.allocateRequests(requests.iterator(), pool); List<StringRequest> resultList = Lists.newArrayList(result); Assert.assertEquals(resultList.size(), 3); Assert.assertEquals(resultList.get(0).getString(), "a-50"); Assert.assertEquals(resultList.get(1).getString(), "b-20"); // No c because it is too large to fit Assert.assertEquals(resultList.get(2).getString(), "d-30"); Assert.assertTrue(estimator.getQueriedRequests().contains("a-50")); Assert.assertTrue(estimator.getQueriedRequests().contains("b-20")); Assert.assertTrue(estimator.getQueriedRequests().contains("c-200")); Assert.assertTrue(estimator.getQueriedRequests().contains("d-30")); Assert.assertFalse(estimator.getQueriedRequests().contains("e-20")); Assert.assertFalse(estimator.getQueriedRequests().contains("f-50")); }