int getQueueInfo(LibYarnClient_t *client, char *queue, bool includeApps, bool includeChildQueues, bool recursive, LibYarnQueueInfo_t **queueInfo) { string queueStr(queue); QueueInfo queueInfoCpp; int result = client->getQueueInfo(queueStr, includeApps, includeChildQueues,recursive, queueInfoCpp); if (result == FUNCTION_SUCCEEDED) { *queueInfo = (LibYarnQueueInfo_t *) malloc(sizeof(LibYarnQueueInfo_t)); (*queueInfo)->queueName = strdup(queueInfoCpp.getQueueName().c_str()); (*queueInfo)->capacity = queueInfoCpp.getCapacity(); (*queueInfo)->maximumCapacity = queueInfoCpp.getMaximumCapacity(); (*queueInfo)->currentCapacity = queueInfoCpp.getCurrentCapacity(); (*queueInfo)->state = queueInfoCpp.getQueueState(); list<QueueInfo> childQueueInfos = queueInfoCpp.getChildQueues(); (*queueInfo)->childQueueNameArraySize = childQueueInfos.size(); (*queueInfo)->childQueueNameArray = (char**) malloc(sizeof(char *) * childQueueInfos.size()); int i = 0; for (list<QueueInfo>::iterator it = childQueueInfos.begin(); it != childQueueInfos.end(); it++) { (*queueInfo)->childQueueNameArray[i] = strdup(it->getQueueName().c_str()); i++; } return FUNCTION_SUCCEEDED; } else { setErrorMessage(client->getErrorMessage()); return FUNCTION_FAILED; } }
TEST_F(TestApplicationClient,TestGetQueueInfo){ string queue = ""; QueueInfo queueInfo = client->getQueueInfo(queue,true,true,true); EXPECT_EQ(queueInfo.getQueueName(), "default"); EXPECT_FLOAT_EQ(queueInfo.getCapacity(), 0.67); EXPECT_FLOAT_EQ(queueInfo.getMaximumCapacity(), 0.95); EXPECT_FLOAT_EQ(queueInfo.getCurrentCapacity(), 0.5); EXPECT_EQ(queueInfo.getQueueState(), QueueState::Q_RUNNING); list<QueueInfo> child = queueInfo.getChildQueues(); EXPECT_EQ(child.size(), 1); list<QueueInfo>::iterator it = child.begin(); EXPECT_EQ(it->getQueueName(), "hawq-queue"); EXPECT_FLOAT_EQ(it->getCapacity(), 0.33); EXPECT_FLOAT_EQ(it->getMaximumCapacity(), 0.5); EXPECT_FLOAT_EQ(it->getCurrentCapacity(), 0.25); list<ApplicationReport> appReportList = queueInfo.getApplicationReports(); list<ApplicationReport>::iterator itAppReport = appReportList.begin(); EXPECT_EQ(itAppReport->getApplicationId().getId(), 100); EXPECT_EQ(itAppReport->getUser(), "postgres"); }
int main(void) { string rmHost("localhost"); string rmPort("8032"); string schedHost("localhost"); string schedPort("8030"); string amHost("localhost"); int32_t amPort = 0; //cluster node1 /* string rmHost("10.37.7.101"); string rmPort("8032"); string schedHost("10.37.7.101"); string schedPort("8030"); string amHost("10.34.0.134"); int32_t amPort = 0; */ string am_tracking_url("url"); LibYarnClient client(rmHost, rmPort, schedHost, schedPort, amHost, amPort, am_tracking_url); //1. createJob string jobName("libyarn"); string queue("default"); string jobId = client.createJob(jobName, queue); cout<<"jobId:"<<jobId<<endl; //2. allocate ResourceRequest resRequest; string host("*"); resRequest.setResourceName(host); Resource capability; capability.setVirtualCores(1); capability.setMemory(1024); resRequest.setCapability(capability); resRequest.setNumContainers(3); resRequest.setRelaxLocality(true); Priority priority; priority.setPriority(1); resRequest.setPriority(priority); string resGroupId("group_1"); list<string> blackListAdditions; list<string> blackListRemovals; list<Container> allocatedResources = client.allocateResources(jobId, resGroupId, resRequest, blackListAdditions, blackListRemovals); for (list<Container>::iterator it = allocatedResources.begin(); it != allocatedResources.end(); it++) { cout << "allocate: host:" << (*it).getNodeId().getHost() << ", port:" << (*it).getNodeId().getPort() << ", cid:" << (*it).getId().getId() << ", vcores:" << (*it).getResource().getVirtualCores() << ", mem:" << (*it).getResource().getMemory() << endl; } //3. active client.activeResources(jobId, resGroupId); sleep(1); //4. release client.releaseResources(jobId, resGroupId); sleep(2); //5. getQueueInfo QueueInfo queueInfo = client.getQueueInfo(queue, true, true, true); cout << "queueName:" << queueInfo.getQueueName() << ", capacity:" << queueInfo.getCapacity() << ", maximumCapacity:" << queueInfo.getMaximumCapacity() << ", currentCapacity:" << queueInfo.getCurrentCapacity() << ", state:" << queueInfo.getQueueState() << endl; //6. getCluster list<NodeState> nodeStates; nodeStates.push_back(NodeState::NS_RUNNING); list<NodeReport> nodeReports = client.getClusterNodes(nodeStates); for (list<NodeReport>::iterator it = nodeReports.begin(); it != nodeReports.end(); it++) { cout << "host:" << it->getNodeId().getHost() << ", port:" << it->getNodeId().getPort() << ", httpAddress:" << it->getHttpAddress() << ", rackName:" << it->getRackName() << ", used:[vcore:" << it->getUsedResource().getVirtualCores() << ", mem:" << it->getUsedResource().getMemory() << "], capability:[vcore:" << it->getResourceCapability().getVirtualCores() << ", mem:" << it->getResourceCapability().getMemory() << "], numContainers:" << it->getNumContainers() << ", node_state:" << it->getNodeState() << ", health_report:" << it->getHealthReport() << ", last_health_report_time:" << it->getLastHealthReportTime() << endl; } //7. finish cout << "to finish: jobId:" << jobId << endl; client.finishJob(jobId, FinalApplicationStatus::APP_SUCCEEDED); sleep(1); return 0; }