@Override public boolean cancel(String projectId, String jobId, String location) { try { bigquery.jobs().cancel(projectId, jobId).setLocation(location).execute(); return true; } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.getCode() == HTTP_NOT_FOUND) { return false; } throw serviceException; } }
@Override public Job getJob(String projectId, String jobId, String location, Map<Option, ?> options) { try { return bigquery .jobs() .get(projectId, jobId) .setLocation(location) .setFields(Option.FIELDS.getString(options)) .execute(); } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.getCode() == HTTP_NOT_FOUND) { return null; } throw serviceException; } }
@Override public Job create(Job job, Map<Option, ?> options) { try { String projectId = job.getJobReference() != null ? job.getJobReference().getProjectId() : this.options.getProjectId(); return bigquery .jobs() .insert(projectId, job) .setFields(Option.FIELDS.getString(options)) .execute(); } catch (IOException ex) { throw translate(ex); } }
@Override public GetQueryResultsResponse getQueryResults( String projectId, String jobId, String location, Map<Option, ?> options) { try { return bigquery .jobs() .getQueryResults(projectId, jobId) .setLocation(location) .setMaxResults(Option.MAX_RESULTS.getLong(options)) .setPageToken(Option.PAGE_TOKEN.getString(options)) .setStartIndex( Option.START_INDEX.getLong(options) != null ? BigInteger.valueOf(Option.START_INDEX.getLong(options)) : null) .setTimeoutMs(Option.TIMEOUT.getLong(options)) .execute(); } catch (IOException ex) { throw translate(ex); } }
JobList jobsList = bigquery .jobs() .list(projectId) .setAllUsers(Option.ALL_USERS.getBoolean(options))
void submitJob(String projectId, Job job) throws IOException { client.jobs() .insert(projectId, job) .execute(); }
Job jobStatus(String projectId, String jobId) throws IOException { return client.jobs() .get(projectId, jobId) .execute(); }
@Override public QueryResponse query(QueryRequest request) { try { return bigquery.jobs().query(this.options.projectId(), request).execute(); } catch (IOException ex) { throw translate(ex); } }
@Override public QueryResponse query(QueryRequest request) { try { return bigquery.jobs().query(this.options.projectId(), request).execute(); } catch (IOException ex) { throw translate(ex); } }
@Before public void setUp() throws IOException { MockitoAnnotations.initMocks(this); when(mockBigqueryClient.jobs()).thenReturn(mockJobs); when(mockJobs.query(anyString(), any(QueryRequest.class))).thenReturn(mockQuery); PowerMockito.mockStatic(BigqueryClient.class); when(BigqueryClient.getNewBigquerryClient(anyString())).thenReturn(mockBigqueryClient); bqClient = spy(new BigqueryClient("test-app")); }
@Override public boolean cancel(String jobId) { try { bigquery.jobs().cancel(this.options.projectId(), jobId).execute(); return true; } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.code() == HTTP_NOT_FOUND) { return false; } throw serviceException; } }
@Override public boolean cancel(String jobId) { try { bigquery.jobs().cancel(this.options.projectId(), jobId).execute(); return true; } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.code() == HTTP_NOT_FOUND) { return false; } throw serviceException; } }
@Override public boolean cancel(String projectId, String jobId, String location) { try { bigquery.jobs().cancel(projectId, jobId).setLocation(location).execute(); return true; } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.getCode() == HTTP_NOT_FOUND) { return false; } throw serviceException; } }
public List<Jobs> listAllJobs(HttpServletResponse resp, Bigquery bigquery) throws IOException { Bigquery.Projects.List projectRequest = bigquery.projects().list(); ProjectList projectResponse = projectRequest.execute(); Bigquery.Jobs.List jobsRequest = bigquery.jobs().list(projectResponse.getProjects().get(0).getId()); JobList jobsResponse = jobsRequest.execute(); return jobsResponse.getJobs(); }
public Job listJob(HttpServletResponse resp, Bigquery bigquery, String jobId) throws IOException { Bigquery.Projects.List projectRequest = bigquery.projects().list(); ProjectList projectResponse = projectRequest.execute(); if (projectResponse.getTotalItems() == 0) { return null; } Projects project = projectResponse.getProjects().get(0); Get jobsRequest = bigquery.jobs().get(project.getId(), jobId); Job j = jobsRequest.execute(); return j; } }
@Override public Job create(Job job, Map<Option, ?> options) { try { return bigquery.jobs() .insert(this.options.projectId(), job) .setFields(FIELDS.getString(options)) .execute(); } catch (IOException ex) { throw translate(ex); } }
@Override public Job create(Job job, Map<Option, ?> options) { try { return bigquery.jobs() .insert(this.options.projectId(), job) .setFields(FIELDS.getString(options)) .execute(); } catch (IOException ex) { throw translate(ex); } }
@Override public Job getJob(String jobId, Map<Option, ?> options) { try { return bigquery.jobs() .get(this.options.projectId(), jobId) .setFields(FIELDS.getString(options)) .execute(); } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.code() == HTTP_NOT_FOUND) { return null; } throw serviceException; } }
@Override public Job getJob(String jobId, Map<Option, ?> options) { try { return bigquery.jobs() .get(this.options.projectId(), jobId) .setFields(FIELDS.getString(options)) .execute(); } catch (IOException ex) { BigQueryException serviceException = translate(ex); if (serviceException.code() == HTTP_NOT_FOUND) { return null; } throw serviceException; } }
@Override public Job create(Job job, Map<Option, ?> options) { try { String projectId = job.getJobReference() != null ? job.getJobReference().getProjectId() : this.options.getProjectId(); return bigquery .jobs() .insert(projectId, job) .setFields(Option.FIELDS.getString(options)) .execute(); } catch (IOException ex) { throw translate(ex); } }