use of org.apache.airavata.cloud.aurora.client.sdk.Response in project airavata by apache.
the class AuroraThriftClient method getJobList.
/**
* Gets the job list.
*
* @param ownerRole the owner role
* @return the job list
* @throws Exception the exception
*/
public GetJobsResponseBean getJobList(String ownerRole) throws Exception {
GetJobsResponseBean response = null;
// try till we get response or scheduler connection not found
while (response == null) {
try {
Response jobListResponse = this.readOnlySchedulerClient.getJobs(ownerRole);
response = (GetJobsResponseBean) AuroraThriftClientUtil.getResponseBean(jobListResponse, ResponseResultType.GET_JOBS);
} catch (Exception ex) {
if (ex instanceof TTransportException) {
// if re-connection success, retry command
if (this.reconnectWithAuroraScheduler()) {
continue;
}
}
logger.error(ex.getMessage(), ex);
throw ex;
}
}
return response;
}
use of org.apache.airavata.cloud.aurora.client.sdk.Response in project airavata by apache.
the class AuroraThriftClient method killTasks.
/**
* Kill tasks.
*
* @param jobKeyBean the job key bean
* @param instances the instances
* @return the response bean
* @throws Exception the exception
*/
public ResponseBean killTasks(JobKeyBean jobKeyBean, Set<Integer> instances) throws Exception {
ResponseBean response = null;
// try till we get response or scheduler connection not found
while (response == null) {
try {
if (jobKeyBean != null) {
JobKey jobKey = AuroraThriftClientUtil.getAuroraJobKey(jobKeyBean);
Response killTaskResponse = this.auroraSchedulerManagerClient.killTasks(jobKey, instances);
response = AuroraThriftClientUtil.getResponseBean(killTaskResponse, ResponseResultType.KILL_TASKS);
}
} catch (Exception ex) {
if (ex instanceof TTransportException) {
// if re-connection success, retry command
if (this.reconnectWithAuroraScheduler()) {
continue;
}
}
logger.error(ex.getMessage(), ex);
throw ex;
}
}
return response;
}
use of org.apache.airavata.cloud.aurora.client.sdk.Response in project airavata by apache.
the class AuroraThriftClient method getJobDetails.
/**
* Gets the job details.
*
* @param jobKeyBean the job key bean
* @return the job details
* @throws Exception the exception
*/
public JobDetailsResponseBean getJobDetails(JobKeyBean jobKeyBean) throws Exception {
JobDetailsResponseBean response = null;
// try till we get response or scheduler connection not found
while (response == null) {
try {
if (jobKeyBean != null) {
JobKey jobKey = AuroraThriftClientUtil.getAuroraJobKey(jobKeyBean);
Set<JobKey> jobKeySet = new HashSet<>();
jobKeySet.add(jobKey);
TaskQuery query = new TaskQuery();
query.setJobKeys(jobKeySet);
Response jobDetailsResponse = this.readOnlySchedulerClient.getTasksStatus(query);
response = (JobDetailsResponseBean) AuroraThriftClientUtil.getResponseBean(jobDetailsResponse, ResponseResultType.GET_JOB_DETAILS);
}
} catch (Exception ex) {
if (ex instanceof TTransportException) {
// if re-connection success, retry command
if (this.reconnectWithAuroraScheduler()) {
continue;
}
}
logger.error(ex.getMessage(), ex);
throw ex;
}
}
return response;
}
use of org.apache.airavata.cloud.aurora.client.sdk.Response in project airavata by apache.
the class AuroraThriftClient method getPendingReasonForJob.
/**
* Gets the pending reason for job.
*
* @param jobKeyBean the job key bean
* @return the pending reason for job
* @throws Exception the exception
*/
public PendingJobReasonBean getPendingReasonForJob(JobKeyBean jobKeyBean) throws Exception {
PendingJobReasonBean response = null;
// try till we get response or scheduler connection not found
while (response == null) {
try {
JobKey jobKey = AuroraThriftClientUtil.getAuroraJobKey(jobKeyBean);
Set<JobKey> jobKeySet = new HashSet<>();
jobKeySet.add(jobKey);
TaskQuery query = new TaskQuery();
query.setJobKeys(jobKeySet);
Response pendingReasonResponse = this.readOnlySchedulerClient.getPendingReason(query);
response = (PendingJobReasonBean) AuroraThriftClientUtil.getResponseBean(pendingReasonResponse, ResponseResultType.GET_PENDING_JOB_REASON);
} catch (Exception ex) {
if (ex instanceof TTransportException) {
// if re-connection success, retry command
if (this.reconnectWithAuroraScheduler()) {
continue;
}
}
logger.error(ex.getMessage(), ex);
throw ex;
}
}
return response;
}
use of org.apache.airavata.cloud.aurora.client.sdk.Response in project airavata by apache.
the class AuroraThriftClient method createJob.
/**
* Creates the job.
*
* @param jobConfigBean the job config bean
* @return the response bean
* @throws Exception the exception
*/
public ResponseBean createJob(JobConfigBean jobConfigBean) throws Exception {
ResponseBean response = null;
// try till we get response or scheduler connection not found
while (response == null) {
try {
if (jobConfigBean != null) {
JobConfiguration jobConfig = AuroraThriftClientUtil.getAuroraJobConfig(jobConfigBean);
Response createJobResponse = this.auroraSchedulerManagerClient.createJob(jobConfig);
response = AuroraThriftClientUtil.getResponseBean(createJobResponse, ResponseResultType.CREATE_JOB);
}
} catch (Exception ex) {
if (ex instanceof TTransportException) {
// if re-connection success, retry command
if (this.reconnectWithAuroraScheduler()) {
continue;
}
}
logger.error(ex.getMessage(), ex);
throw ex;
}
}
return response;
}
Aggregations