diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java index 2ccd673733..0ed8a4ed00 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java @@ -39,6 +39,7 @@ import javax.servlet.http.HttpServletResponse; import javax.ws.rs.Consumes; import javax.ws.rs.DELETE; +import javax.ws.rs.DefaultValue; import javax.ws.rs.FormParam; import javax.ws.rs.GET; import javax.ws.rs.POST; @@ -82,6 +83,8 @@ import org.apache.hadoop.yarn.api.protocolrecords.MoveApplicationAcrossQueuesRequest; import org.apache.hadoop.yarn.api.protocolrecords.RenewDelegationTokenRequest; import org.apache.hadoop.yarn.api.protocolrecords.RenewDelegationTokenResponse; +import org.apache.hadoop.yarn.api.protocolrecords.ReservationListRequest; +import org.apache.hadoop.yarn.api.protocolrecords.ReservationListResponse; import org.apache.hadoop.yarn.api.protocolrecords.SubmitApplicationRequest; import org.apache.hadoop.yarn.api.protocolrecords.SubmitApplicationResponse; import org.apache.hadoop.yarn.api.protocolrecords.UpdateApplicationPriorityRequest; @@ -156,6 +159,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationDefinitionInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationDeleteRequestInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationDeleteResponseInfo; +import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationListInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationRequestInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationRequestsInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationSubmissionResponseInfo; @@ -2168,4 +2172,56 @@ private ReservationDeleteRequest createReservationDeleteRequest( return request; } + /** + * Function to retrieve a list of all the reservations. + */ + @GET + @Path("/reservation/list") + @Produces({ MediaType.APPLICATION_JSON, MediaType.APPLICATION_XML }) + public Response listReservation( + @QueryParam("queue") @DefaultValue("default") String queue, + @QueryParam("reservation-id") @DefaultValue("") String reservationId, + @QueryParam("start-time") @DefaultValue("0") long startTime, + @QueryParam("end-time") @DefaultValue("-1") long endTime, + @QueryParam("include-resource-allocations") @DefaultValue("false") + boolean includeResourceAllocations, @Context HttpServletRequest hsr) + throws Exception { + init(); + + final ReservationListRequest request = ReservationListRequest.newInstance( + queue, reservationId, startTime, endTime, includeResourceAllocations); + + UserGroupInformation callerUGI = getCallerUserGroupInformation(hsr, true); + if (callerUGI == null) { + throw new AuthorizationException("Unable to obtain user name, " + + "user not authenticated"); + } + if (UserGroupInformation.isSecurityEnabled() && isStaticUser(callerUGI)) { + String msg = "The default static user cannot carry out this operation."; + return Response.status(Status.FORBIDDEN).entity(msg).build(); + } + + ReservationListResponse resRespInfo; + try { + resRespInfo = callerUGI.doAs( + new PrivilegedExceptionAction() { + @Override + public ReservationListResponse run() throws IOException, + YarnException { + return rm.getClientRMService().listReservations(request); + } + }); + } catch (UndeclaredThrowableException ue) { + if (ue.getCause() instanceof YarnException) { + throw new BadRequestException(ue.getCause().getMessage()); + } + LOG.info("List reservation request failed", ue); + throw ue; + } + + ReservationListInfo resResponse = new ReservationListInfo(resRespInfo, + includeResourceAllocations); + return Response.status(Status.OK).entity(resResponse).build(); + } + } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationDefinitionInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationDefinitionInfo.java index ff82e7510d..71ee924e3d 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationDefinitionInfo.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationDefinitionInfo.java @@ -18,6 +18,8 @@ package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; +import org.apache.hadoop.yarn.api.records.ReservationDefinition; + import javax.xml.bind.annotation.XmlAccessType; import javax.xml.bind.annotation.XmlAccessorType; import javax.xml.bind.annotation.XmlElement; @@ -46,6 +48,14 @@ public ReservationDefinitionInfo() { } + public ReservationDefinitionInfo(ReservationDefinition definition) { + arrival = definition.getArrival(); + deadline = definition.getDeadline(); + reservationName = definition.getReservationName(); + reservationRequests = new ReservationRequestsInfo(definition + .getReservationRequests()); + } + public long getArrival() { return arrival; } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationIdInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationIdInfo.java new file mode 100644 index 0000000000..3a2596a1c8 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationIdInfo.java @@ -0,0 +1,64 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; + +import org.apache.hadoop.yarn.api.records.ReservationId; + +import javax.xml.bind.annotation.XmlAccessType; +import javax.xml.bind.annotation.XmlAccessorType; +import javax.xml.bind.annotation.XmlElement; +import javax.xml.bind.annotation.XmlRootElement; + +/** + * Simple class that represent a reservation ID. + */ +@XmlRootElement +@XmlAccessorType(XmlAccessType.FIELD) +public class ReservationIdInfo { + @XmlElement(name = "cluster-timestamp") + private long clusterTimestamp; + + @XmlElement(name = "reservation-id") + private long reservationId; + + public ReservationIdInfo() { + this.clusterTimestamp = 0; + this.reservationId = 0; + } + + public ReservationIdInfo(ReservationId reservationId) { + this.clusterTimestamp = reservationId.getClusterTimestamp(); + this.reservationId = reservationId.getId(); + } + + public long getClusterTimestamp() { + return this.clusterTimestamp; + } + + public void setClusterTimestamp(long newClusterTimestamp) { + this.clusterTimestamp = newClusterTimestamp; + } + + public long getReservationId() { + return this.reservationId; + } + + public void setReservationId(long newReservationId) { + this.reservationId = newReservationId; + } +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationInfo.java new file mode 100644 index 0000000000..1a31a8b17d --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationInfo.java @@ -0,0 +1,105 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; + +import org.apache.hadoop.yarn.api.records.ReservationAllocationState; +import org.apache.hadoop.yarn.api.records.ResourceAllocationRequest; +import org.apache.hadoop.yarn.server.resourcemanager.reservation.ReservationInterval; + +import javax.xml.bind.annotation.XmlAccessType; +import javax.xml.bind.annotation.XmlAccessorType; +import javax.xml.bind.annotation.XmlElement; +import javax.xml.bind.annotation.XmlRootElement; +import java.util.ArrayList; +import java.util.List; + +/** + * Simple class that represent a reservation. + */ +@XmlRootElement +@XmlAccessorType(XmlAccessType.FIELD) +public class ReservationInfo { + + @XmlElement(name = "acceptance-time") + private long acceptanceTime; + + private String user; + + @XmlElement(name = "resource-allocations") + private List resourceAllocations = new ArrayList<>(); + + @XmlElement(name = "reservation-id") + private ReservationIdInfo reservationId; + + @XmlElement(name = "reservation-definition") + private ReservationDefinitionInfo reservationDefinition; + + + public ReservationInfo() { + acceptanceTime = 0; + user = ""; + + reservationDefinition = new ReservationDefinitionInfo(); + reservationId = new ReservationIdInfo(); + } + + public ReservationInfo(ReservationAllocationState allocation, boolean + includeResourceAllocations) throws + Exception { + acceptanceTime = allocation.getAcceptanceTime(); + user = allocation.getUser(); + + if (includeResourceAllocations) { + List requests = allocation + .getResourceAllocationRequests(); + for (ResourceAllocationRequest request : requests) { + resourceAllocations.add(new ResourceAllocationInfo(new + ReservationInterval(request.getStartTime(), request + .getEndTime()), request.getCapability())); + } + } + + reservationId = new ReservationIdInfo(allocation.getReservationId()); + reservationDefinition = new ReservationDefinitionInfo( + allocation.getReservationDefinition()); + } + + public String getUser() { + return user; + } + + public void setUser(String newUser) { + user = newUser; + } + + public long getAcceptanceTime() { + return acceptanceTime; + } + + public List getResourceAllocations() { + return resourceAllocations; + } + + public ReservationIdInfo getReservationId() { + return reservationId; + } + + public ReservationDefinitionInfo getReservationDefinition() { + return reservationDefinition; + } +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationListInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationListInfo.java new file mode 100644 index 0000000000..25df67a085 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationListInfo.java @@ -0,0 +1,53 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; + +import org.apache.hadoop.yarn.api.protocolrecords.ReservationListResponse; +import org.apache.hadoop.yarn.api.records.ReservationAllocationState; + +import javax.xml.bind.annotation.XmlAccessType; +import javax.xml.bind.annotation.XmlAccessorType; +import javax.xml.bind.annotation.XmlElement; +import javax.xml.bind.annotation.XmlRootElement; +import java.util.ArrayList; +import java.util.List; + +/** + * Simple class that represent a list of reservations. + */ +@XmlRootElement +@XmlAccessorType(XmlAccessType.FIELD) +public class ReservationListInfo { + @XmlElement(name = "reservations") + private List reservations; + + public ReservationListInfo() { + reservations = new ArrayList<>(); + } + + public ReservationListInfo(ReservationListResponse response, + boolean includeResourceAllocations) throws Exception { + this(); + + for (ReservationAllocationState allocation : + response.getReservationAllocationState()) { + reservations.add(new ReservationInfo(allocation, + includeResourceAllocations)); + } + } +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestInfo.java index ab82f3f375..e1d6fa310b 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestInfo.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestInfo.java @@ -18,6 +18,8 @@ package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; +import org.apache.hadoop.yarn.api.records.ReservationRequest; + import javax.xml.bind.annotation.XmlAccessType; import javax.xml.bind.annotation.XmlAccessorType; import javax.xml.bind.annotation.XmlElement; @@ -43,6 +45,13 @@ public ReservationRequestInfo() { } + public ReservationRequestInfo(ReservationRequest request) { + capability = new ResourceInfo(request.getCapability()); + minConcurrency = request.getConcurrency(); + duration = request.getDuration(); + numContainers = request.getNumContainers(); + } + public ResourceInfo getCapability() { return capability; } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestsInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestsInfo.java index 42cdb0ecc9..f37da1b1af 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestsInfo.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ReservationRequestsInfo.java @@ -18,6 +18,9 @@ package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; +import org.apache.hadoop.yarn.api.records.ReservationRequest; +import org.apache.hadoop.yarn.api.records.ReservationRequests; + import java.util.ArrayList; import javax.xml.bind.annotation.XmlAccessType; @@ -42,6 +45,14 @@ public ReservationRequestsInfo() { } + public ReservationRequestsInfo(ReservationRequests requests) { + reservationRequest = new ArrayList<>(); + for (ReservationRequest request : requests.getReservationResources()) { + reservationRequest.add(new ReservationRequestInfo(request)); + } + reservationRequestsInterpreter = requests.getInterpreter().ordinal(); + } + public int getReservationRequestsInterpreter() { return reservationRequestsInterpreter; } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ResourceAllocationInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ResourceAllocationInfo.java new file mode 100644 index 0000000000..9ed9f51cd7 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ResourceAllocationInfo.java @@ -0,0 +1,72 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; + +import org.apache.hadoop.yarn.api.records.Resource; +import org.apache.hadoop.yarn.server.resourcemanager.reservation.ReservationInterval; + +import javax.xml.bind.annotation.XmlAccessType; +import javax.xml.bind.annotation.XmlAccessorType; +import javax.xml.bind.annotation.XmlRootElement; + +/** + * Simple class that represent a resource allocation. + */ +@XmlRootElement(name = "resource-allocation") +@XmlAccessorType(XmlAccessType.FIELD) +public class ResourceAllocationInfo { + private ResourceInfo resource; + private long startTime; + private long endTime; + + public ResourceAllocationInfo() { + resource = new ResourceInfo(); + startTime = -1; + endTime = -1; + } + + public ResourceAllocationInfo(ReservationInterval interval, Resource res) { + this.resource = new ResourceInfo(res); + this.startTime = interval.getStartTime(); + this.endTime = interval.getEndTime(); + } + + public long getStartTime() { + return startTime; + } + + public void setStartTime(long newStartTime) { + this.startTime = newStartTime; + } + + public long getEndTime() { + return endTime; + } + + public void setEndTime(long newEndTime) { + this.endTime = newEndTime; + } + + public ResourceInfo getResource() { + return resource; + } + + public void setResource(ResourceInfo newResource) { + this.resource = newResource; + } +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesReservation.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesReservation.java index bb0db81161..72705d4f4d 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesReservation.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesReservation.java @@ -21,6 +21,7 @@ import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertNull; +import static org.junit.Assert.assertTrue; import java.io.File; import java.io.FileWriter; @@ -54,8 +55,11 @@ import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationDeleteRequestInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationSubmissionRequestInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ReservationUpdateRequestInfo; +import org.apache.hadoop.yarn.util.Clock; +import org.apache.hadoop.yarn.util.UTCClock; import org.apache.hadoop.yarn.webapp.GenericExceptionHandler; import org.apache.hadoop.yarn.webapp.JerseyTestBase; +import org.codehaus.jettison.json.JSONArray; import org.codehaus.jettison.json.JSONException; import org.codehaus.jettison.json.JSONObject; import org.junit.After; @@ -82,17 +86,22 @@ @RunWith(Parameterized.class) public class TestRMWebServicesReservation extends JerseyTestBase { - private static MockRM rm; - private static Injector injector; private String webserviceUserName = "testuser"; - private boolean setAuthFilter = false; + private static MockRM rm; + private static Injector injector; + + private static final int MINIMUM_RESOURCE_DURATION = 1000000; + private static final Clock clock = new UTCClock(); private static final String TEST_DIR = new File(System.getProperty( "test.build.data", "/tmp")).getAbsolutePath(); private static final String FS_ALLOC_FILE = new File(TEST_DIR, "test-fs-queues.xml").getAbsolutePath(); + // This is what is used in the test resource files. + private static final String DEFAULT_QUEUE = "dedicated"; + private static final String LIST_RESERVATION_PATH = "reservation/list"; public static class GuiceServletConfig extends GuiceServletContextListener { @@ -318,13 +327,9 @@ public void tearDown() throws Exception { } @Test - public void testSubmitReservation() throws JSONException, Exception { + public void testSubmitReservation() throws Exception { rm.start(); - for (int i = 0; i < 100; i++) { - MockNM amNodeManager = - rm.registerNode("127.0.0." + i + ":1234", 100 * 1024); - amNodeManager.nodeHeartbeat(true); - } + setupCluster(100); ReservationId rid = testSubmissionReservationHelper("reservation/submit", MediaType.APPLICATION_JSON); @@ -335,14 +340,10 @@ public void testSubmitReservation() throws JSONException, Exception { } @Test - public void testFailedSubmitReservation() throws JSONException, Exception { + public void testFailedSubmitReservation() throws Exception { rm.start(); // setup a cluster too small to accept the reservation - for (int i = 0; i < 1; i++) { - MockNM amNodeManager = - rm.registerNode("127.0.0." + i + ":1234", 100 * 1024); - amNodeManager.nodeHeartbeat(true); - } + setupCluster(1); ReservationId rid = testSubmissionReservationHelper("reservation/submit", MediaType.APPLICATION_JSON); @@ -353,11 +354,7 @@ public void testFailedSubmitReservation() throws JSONException, Exception { @Test public void testUpdateReservation() throws JSONException, Exception { rm.start(); - for (int i = 0; i < 100; i++) { - MockNM amNodeManager = - rm.registerNode("127.0.0." + i + ":1234", 100 * 1024); - amNodeManager.nodeHeartbeat(true); - } + setupCluster(100); ReservationId rid = testSubmissionReservationHelper("reservation/submit", MediaType.APPLICATION_JSON); @@ -370,6 +367,452 @@ public void testUpdateReservation() throws JSONException, Exception { rm.stop(); } + @Test + public void testTimeIntervalRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("start-time", Long.toString((long) (time * 0.9))) + .queryParam("end-time", Long.toString(time + (long) (0.9 * + MINIMUM_RESOURCE_DURATION))) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + String reservationName = reservations.getJSONObject + ("reservation-definition").getString("reservation-name"); + assertEquals(reservationName, "res_1"); + + rm.stop(); + } + + @Test + public void testSameTimeIntervalRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + String timeParam = Long.toString(time + MINIMUM_RESOURCE_DURATION / 2); + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("start-time", timeParam) + .queryParam("end-time", timeParam) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + String reservationName = reservations.getJSONObject + ("reservation-definition").getString("reservation-name"); + assertEquals(reservationName, "res_1"); + + rm.stop(); + } + @Test + public void testInvalidTimeIntervalRequestListReservation() throws + Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + + ReservationId res1 = testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + ReservationId res2 = testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + WebResource resource; + resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("start-time", "-100") + .queryParam("end-time", "-100") + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONArray reservations = json.getJSONArray("reservations"); + + assertEquals(2, reservations.length()); + + testRDLHelper(reservations.getJSONObject(0)); + testRDLHelper(reservations.getJSONObject(1)); + + rm.stop(); + } + + @Test + public void testInvalidEndTimeRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("start-time", Long.toString((long) (time + + MINIMUM_RESOURCE_DURATION * 1.3))) + .queryParam("end-time", "-1") + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + String reservationName = reservations.getJSONObject + ("reservation-definition").getString("reservation-name"); + assertEquals(reservationName, "res_2"); + + rm.stop(); + } + + @Test + public void testEmptyEndTimeRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("start-time", new Long((long) (time + + MINIMUM_RESOURCE_DURATION * 1.3)).toString()) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + String reservationName = reservations.getJSONObject + ("reservation-definition").getString("reservation-name"); + assertEquals(reservationName, "res_2"); + + rm.stop(); + } + + @Test + public void testInvalidStartTimeRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("start-time", "-1") + .queryParam("end-time", new Long((long)(time + + MINIMUM_RESOURCE_DURATION * 0.9)).toString()) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + // only res_1 should fall into the time interval given in the request json. + String reservationName = reservations.getJSONObject + ("reservation-definition").getString("reservation-name"); + assertEquals(reservationName, "res_1"); + + rm.stop(); + } + + @Test + public void testEmptyStartTimeRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + long time = clock.getTime() + MINIMUM_RESOURCE_DURATION; + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time, "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, time + MINIMUM_RESOURCE_DURATION, + "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("end-time", new Long((long)(time + + MINIMUM_RESOURCE_DURATION * 0.9)).toString()) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + // only res_1 should fall into the time interval given in the request json. + String reservationName = reservations.getJSONObject + ("reservation-definition").getString("reservation-name"); + assertEquals(reservationName, "res_1"); + + rm.stop(); + } + + @Test + public void testQueueOnlyRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("queue", DEFAULT_QUEUE); + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + assertEquals(json.getJSONArray("reservations").length(), 2); + + testRDLHelper(json.getJSONArray("reservations").getJSONObject(0)); + testRDLHelper(json.getJSONArray("reservations").getJSONObject(1)); + + rm.stop(); + } + + @Test + public void testEmptyQueueRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH); + + testListReservationHelper(resource, Status.BAD_REQUEST); + + rm.stop(); + } + + @Test + public void testNonExistentQueueRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("queue", DEFAULT_QUEUE + "_invalid"); + + testListReservationHelper(resource, Status.BAD_REQUEST); + + rm.stop(); + } + + @Test + public void testReservationIdRequestListReservation() throws Exception { + rm.start(); + setupCluster(100); + + ReservationId id1 = testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_2", 2); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + if (id1 != null) { + resource = resource.queryParam("reservation-id", id1.toString()); + } + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + long reservationId = reservations.getJSONObject("reservation-id") + .getLong("reservation-id"); + assertEquals(id1.getId(), reservationId); + + rm.stop(); + } + + @Test + public void testInvalidReservationIdRequestListReservation() throws + Exception { + rm.start(); + setupCluster(100); + + ReservationId id1 = testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("queue", DEFAULT_QUEUE); + + if (id1 != null) { + resource = resource.queryParam("reservation-id", + "invalid" + id1.toString()); + } + + JSONObject response = testListReservationHelper(resource, Status.NOT_FOUND); + + rm.stop(); + } + + @Test + public void testIncludeResourceAllocations() throws Exception { + rm.start(); + setupCluster(100); + + ReservationId id1 = testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("include-resource-allocations", "true") + .queryParam("queue", DEFAULT_QUEUE); + + if (id1 != null) { + resource = resource.queryParam("reservation-id", id1.toString()); + } + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + long reservationId = reservations.getJSONObject("reservation-id") + .getLong("reservation-id"); + assertEquals(id1.getId(), reservationId); + + assertTrue(reservations.has("resource-allocations")); + + rm.stop(); + } + + @Test + public void testExcludeResourceAllocations() throws Exception { + rm.start(); + setupCluster(100); + + ReservationId id1 = testSubmissionReservationHelper("reservation/submit", + MediaType.APPLICATION_JSON, clock.getTime(), "res_1", 1); + + WebResource resource = constructWebResource(LIST_RESERVATION_PATH) + .queryParam("include-resource-allocations", "false") + .queryParam("queue", DEFAULT_QUEUE); + + if (id1 != null) { + resource = resource.queryParam("reservation-id", id1.toString()); + } + + JSONObject json = testListReservationHelper(resource); + + if (!this.isAuthenticationEnabled() && json == null) { + return; + } + + JSONObject reservations = json.getJSONObject("reservations"); + + testRDLHelper(reservations); + + long reservationId = reservations.getJSONObject("reservation-id") + .getLong("reservation-id"); + assertEquals(id1.getId(), reservationId); + + assertTrue(!reservations.has("resource-allocations")); + + rm.stop(); + } + @Test public void testDeleteReservation() throws JSONException, Exception { rm.start(); @@ -391,10 +834,27 @@ public void testDeleteReservation() throws JSONException, Exception { } private ReservationId testSubmissionReservationHelper(String path, - String media) throws JSONException, Exception { + String media) throws Exception { + long arrival = clock.getTime() + MINIMUM_RESOURCE_DURATION; + return testSubmissionReservationHelper(path, media, arrival, "res_1", 1); + } + + private ReservationId testSubmissionReservationHelper(String path, + String media, Long arrival, String reservationName, int expectedId) + throws Exception { String reservationJson = loadJsonFile("submit-reservation.json"); + String reservationJsonRequest = String.format(reservationJson, arrival, + arrival + MINIMUM_RESOURCE_DURATION, reservationName); + + return submitAndVerifyReservation(path, media, reservationJsonRequest, + expectedId); + } + + private ReservationId submitAndVerifyReservation(String path, String media, + String reservationJson, int expectedId) throws Exception { + JSONJAXBContext jc = new JSONJAXBContext(JSONConfiguration.mapped() .build(), ReservationSubmissionRequestInfo.class); @@ -421,7 +881,7 @@ private ReservationId testSubmissionReservationHelper(String path, ReservationId rid = null; try { rid = ReservationId.parseReservationId(json.getString("reservation-id")); - assertEquals("incorrect return value", rid.getId(), 1); + assertEquals("incorrect return value", rid.getId(), expectedId); } catch (JSONException j) { // failure is possible and is checked outside } @@ -511,7 +971,45 @@ private void testDeleteReservationHelper(String path, System.out.println("RESPONSE:" + response); assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); assertEquals(Status.OK, response.getClientResponseStatus()); - } + private void testRDLHelper(JSONObject json) throws JSONException { + JSONObject requests = json.getJSONObject("reservation-definition") + .getJSONObject("reservation-requests"); + String type = requests.getString + ("reservation-request-interpreter"); + + assertEquals("0", type); + assertEquals(60, requests.getJSONArray("reservation-request") + .getJSONObject(0).getInt("duration")); + } + + private JSONObject testListReservationHelper(WebResource resource) throws + Exception { + return testListReservationHelper(resource, Status.OK); + } + + private JSONObject testListReservationHelper(WebResource resource, Status + status) throws Exception { + Thread.sleep(1000); + ClientResponse response = resource.get(ClientResponse.class); + + if (!this.isAuthenticationEnabled()) { + assertEquals(Status.UNAUTHORIZED, response.getClientResponseStatus()); + return null; + } + + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + assertEquals(status, response.getClientResponseStatus()); + + return response.getEntity(JSONObject.class); + } + + private void setupCluster(int nodes) throws Exception { + for (int i = 0; i < nodes; i++) { + MockNM amNodeManager = + rm.registerNode("127.0.0." + i + ":1234", 100 * 1024); + amNodeManager.nodeHeartbeat(true); + } + } } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/resources/submit-reservation.json b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/resources/submit-reservation.json index 573d317d23..0a243a288d 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/resources/submit-reservation.json +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/resources/submit-reservation.json @@ -1,9 +1,9 @@ { "queue" : "dedicated", "reservation-definition" : { - "arrival" : 1765541532000, - "deadline" : 1765542252000, - "reservation-name" : "res_1", + "arrival" : %s, + "deadline" : %s, + "reservation-name" : "%s", "reservation-requests" : { "reservation-request-interpreter" : 0, "reservation-request" : [