Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2016 Google LLC
*
* Licensed 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 com.google.cloud.bigquery;
import com.google.api.core.ApiFunction;
import com.google.api.services.bigquery.model.JobConfiguration;
import com.google.api.services.bigquery.model.JobStatistics2;
import com.google.api.services.bigquery.model.JobStatistics3;
import com.google.api.services.bigquery.model.JobStatistics4;
import com.google.cloud.StringEnumType;
import com.google.cloud.StringEnumValue;
import com.google.common.base.Function;
import com.google.common.base.MoreObjects;
import com.google.common.base.MoreObjects.ToStringHelper;
import com.google.common.collect.Lists;
import java.io.Serializable;
import java.util.List;
import java.util.Objects;
import org.checkerframework.checker.nullness.compatqual.NullableDecl;
/** A Google BigQuery Job statistics. */
public abstract class JobStatistics implements Serializable {
private static final long serialVersionUID = 1433024714741660399L;
private final Long creationTime;
private final Long endTime;
private final Long startTime;
private final Long numChildJobs;
private final String parentJobId;
private final ScriptStatistics scriptStatistics;
private final List reservationUsage;
private final TransactionInfo transactionInfo;
private final SessionInfo sessionInfo;
/** A Google BigQuery Copy Job statistics. */
public static class CopyStatistics extends JobStatistics {
private static final long serialVersionUID = 8218325588441660938L;
static final class Builder extends JobStatistics.Builder {
private Builder() {}
private Builder(com.google.api.services.bigquery.model.JobStatistics statisticsPb) {
super(statisticsPb);
}
@Override
CopyStatistics build() {
return new CopyStatistics(this);
}
}
private CopyStatistics(Builder builder) {
super(builder);
}
@Override
public final boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(CopyStatistics.class)
&& baseEquals((CopyStatistics) obj);
}
@Override
public final int hashCode() {
return baseHashCode();
}
static Builder newBuilder() {
return new Builder();
}
@SuppressWarnings("unchecked")
static CopyStatistics fromPb(com.google.api.services.bigquery.model.JobStatistics statisticPb) {
return new Builder(statisticPb).build();
}
}
/** A Google BigQuery Extract Job statistics. */
public static class ExtractStatistics extends JobStatistics {
private static final long serialVersionUID = -1566598819212767373L;
private final List destinationUriFileCounts;
static final class Builder extends JobStatistics.Builder {
private List destinationUriFileCounts;
private Builder() {}
private Builder(com.google.api.services.bigquery.model.JobStatistics statisticsPb) {
super(statisticsPb);
if (statisticsPb.getExtract() != null) {
this.destinationUriFileCounts = statisticsPb.getExtract().getDestinationUriFileCounts();
}
}
Builder setDestinationUriFileCounts(List destinationUriFileCounts) {
this.destinationUriFileCounts = destinationUriFileCounts;
return self();
}
@Override
ExtractStatistics build() {
return new ExtractStatistics(this);
}
}
private ExtractStatistics(Builder builder) {
super(builder);
this.destinationUriFileCounts = builder.destinationUriFileCounts;
}
/**
* Returns the number of files per destination URI or URI pattern specified in the extract job.
* These values will be in the same order as the URIs specified by {@link
* ExtractJobConfiguration#getDestinationUris()}.
*/
public List getDestinationUriFileCounts() {
return destinationUriFileCounts;
}
@Override
ToStringHelper toStringHelper() {
return super.toStringHelper().add("destinationUriFileCounts", destinationUriFileCounts);
}
@Override
public final boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(ExtractStatistics.class)
&& baseEquals((ExtractStatistics) obj);
}
@Override
public final int hashCode() {
return Objects.hash(baseHashCode(), destinationUriFileCounts);
}
@Override
com.google.api.services.bigquery.model.JobStatistics toPb() {
com.google.api.services.bigquery.model.JobStatistics statisticsPb = super.toPb();
return statisticsPb.setExtract(
new JobStatistics4().setDestinationUriFileCounts(destinationUriFileCounts));
}
static Builder newBuilder() {
return new Builder();
}
@SuppressWarnings("unchecked")
static ExtractStatistics fromPb(
com.google.api.services.bigquery.model.JobStatistics statisticPb) {
return new Builder(statisticPb).build();
}
}
/** A Google BigQuery Load Job statistics. */
public static class LoadStatistics extends JobStatistics {
private static final long serialVersionUID = -707369246536309215L;
private final Long inputBytes;
private final Long inputFiles;
private final Long outputBytes;
private final Long outputRows;
private final Long badRecords;
static final class Builder extends JobStatistics.Builder {
private Long inputBytes;
private Long inputFiles;
private Long outputBytes;
private Long outputRows;
private Long badRecords;
private Builder() {}
private Builder(com.google.api.services.bigquery.model.JobStatistics statisticsPb) {
super(statisticsPb);
if (statisticsPb.getLoad() != null) {
this.inputBytes = statisticsPb.getLoad().getInputFileBytes();
this.inputFiles = statisticsPb.getLoad().getInputFiles();
this.outputBytes = statisticsPb.getLoad().getOutputBytes();
this.outputRows = statisticsPb.getLoad().getOutputRows();
this.badRecords = statisticsPb.getLoad().getBadRecords();
}
}
Builder setInputBytes(Long inputBytes) {
this.inputBytes = inputBytes;
return self();
}
Builder setInputFiles(Long inputFiles) {
this.inputFiles = inputFiles;
return self();
}
Builder setOutputBytes(Long outputBytes) {
this.outputBytes = outputBytes;
return self();
}
Builder setOutputRows(Long outputRows) {
this.outputRows = outputRows;
return self();
}
Builder setBadRecords(Long badRecords) {
this.badRecords = badRecords;
return self();
}
@Override
LoadStatistics build() {
return new LoadStatistics(this);
}
}
private LoadStatistics(Builder builder) {
super(builder);
this.inputBytes = builder.inputBytes;
this.inputFiles = builder.inputFiles;
this.outputBytes = builder.outputBytes;
this.outputRows = builder.outputRows;
this.badRecords = builder.badRecords;
}
/** Returns the number of bytes of source data in a load job. */
public Long getInputBytes() {
return inputBytes;
}
/** Returns the number of source files in a load job. */
public Long getInputFiles() {
return inputFiles;
}
/** Returns the size of the data loaded by a load job so far, in bytes. */
public Long getOutputBytes() {
return outputBytes;
}
/** Returns the number of rows loaded by a load job so far. */
public Long getOutputRows() {
return outputRows;
}
/** Returns the number of bad records reported in a job. */
public Long getBadRecords() {
return badRecords;
}
@Override
ToStringHelper toStringHelper() {
return super.toStringHelper()
.add("inputBytes", inputBytes)
.add("inputFiles", inputFiles)
.add("outputBytes", outputBytes)
.add("outputRows", outputRows)
.add("badRecords", badRecords);
}
@Override
public final boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(LoadStatistics.class)
&& baseEquals((LoadStatistics) obj);
}
@Override
public final int hashCode() {
return Objects.hash(
baseHashCode(), inputBytes, inputFiles, outputBytes, outputRows, badRecords);
}
@Override
com.google.api.services.bigquery.model.JobStatistics toPb() {
JobStatistics3 loadStatisticsPb = new JobStatistics3();
loadStatisticsPb.setInputFileBytes(inputBytes);
loadStatisticsPb.setInputFiles(inputFiles);
loadStatisticsPb.setOutputBytes(outputBytes);
loadStatisticsPb.setOutputRows(outputRows);
loadStatisticsPb.setBadRecords(badRecords);
return super.toPb().setLoad(loadStatisticsPb);
}
static Builder newBuilder() {
return new Builder();
}
@SuppressWarnings("unchecked")
static LoadStatistics fromPb(com.google.api.services.bigquery.model.JobStatistics statisticPb) {
return new Builder(statisticPb).build();
}
}
/** A Google BigQuery Query Job statistics. */
public static class QueryStatistics extends JobStatistics {
private static final long serialVersionUID = 7539354109226732353L;
private final BiEngineStats biEngineStats;
private final Integer billingTier;
private final Boolean cacheHit;
private final String ddlOperationPerformed;
private final TableId ddlTargetTable;
private final RoutineId ddlTargetRoutine;
private final Long estimatedBytesProcessed;
private final Long numDmlAffectedRows;
private final DmlStats dmlStats;
private final List referencedTables;
private final StatementType statementType;
private final Long totalBytesBilled;
private final Long totalBytesProcessed;
private final Long totalPartitionsProcessed;
private final Long totalSlotMs;
private final List queryPlan;
private final List timeline;
private final Schema schema;
/**
* StatementType represents possible types of SQL statements reported as part of the
* QueryStatistics of a BigQuery job.
*/
public static final class StatementType extends StringEnumValue {
private static final long serialVersionUID = 818920627219751204L;
private static final ApiFunction CONSTRUCTOR =
new ApiFunction() {
@Override
public StatementType apply(String constant) {
return new StatementType(constant);
}
};
private static final StringEnumType type =
new StringEnumType(StatementType.class, CONSTRUCTOR);
public static final StatementType SELECT = type.createAndRegister("SELECT");
public static final StatementType UPDATE = type.createAndRegister("UPDATE");
public static final StatementType INSERT = type.createAndRegister("INSERT");
public static final StatementType DELETE = type.createAndRegister("DELETE");
public static final StatementType CREATE_TABLE = type.createAndRegister("CREATE_TABLE");
public static final StatementType CREATE_TABLE_AS_SELECT =
type.createAndRegister("CREATE_TABLE_AS_SELECT");
public static final StatementType CREATE_VIEW = type.createAndRegister("CREATE_VIEW");
public static final StatementType CREATE_MODEL = type.createAndRegister("CREATE_MODEL");
public static final StatementType CREATE_FUNCTION = type.createAndRegister("CREATE_FUNCTION");
public static final StatementType CREATE_PROCEDURE =
type.createAndRegister("CREATE_PROCEDURE");
public static final StatementType ALTER_TABLE = type.createAndRegister("ALTER_TABLE");
public static final StatementType ALTER_VIEW = type.createAndRegister("ALTER_VIEW");
public static final StatementType DROP_TABLE = type.createAndRegister("DROP_TABLE");
public static final StatementType DROP_VIEW = type.createAndRegister("DROP_VIEW");
public static final StatementType DROP_FUNCTION = type.createAndRegister("DROP_FUNCTION");
public static final StatementType DROP_PROCEDURE = type.createAndRegister("DROP_PROCEDURE");
public static final StatementType MERGE = type.createAndRegister("MERGE");
private StatementType(String constant) {
super(constant);
}
/**
* Get the StatementType for the given String constant, and throw an exception if the constant
* is not recognized.
*/
public static StatementType valueOfStrict(String constant) {
return type.valueOfStrict(constant);
}
/** Get the State for the given String constant, and allow unrecognized values. */
public static StatementType valueOf(String constant) {
return type.valueOf(constant);
}
/** Return the known values for State. */
public static StatementType[] values() {
return type.values();
}
}
static final class Builder extends JobStatistics.Builder {
private BiEngineStats biEngineStats;
private Integer billingTier;
private Boolean cacheHit;
private String ddlOperationPerformed;
private TableId ddlTargetTable;
private RoutineId ddlTargetRoutine;
private Long estimatedBytesProcessed;
private Long numDmlAffectedRows;
private DmlStats dmlStats;
private List referencedTables;
private StatementType statementType;
private Long totalBytesBilled;
private Long totalBytesProcessed;
private Long totalPartitionsProcessed;
private Long totalSlotMs;
private List queryPlan;
private List timeline;
private Schema schema;
private Builder() {}
private Builder(com.google.api.services.bigquery.model.JobStatistics statisticsPb) {
super(statisticsPb);
if (statisticsPb.getQuery() != null) {
if (statisticsPb.getQuery().getBiEngineStatistics() != null) {
this.biEngineStats =
BiEngineStats.fromPb(statisticsPb.getQuery().getBiEngineStatistics());
}
this.billingTier = statisticsPb.getQuery().getBillingTier();
this.cacheHit = statisticsPb.getQuery().getCacheHit();
this.ddlOperationPerformed = statisticsPb.getQuery().getDdlOperationPerformed();
if (statisticsPb.getQuery().getDdlTargetTable() != null) {
this.ddlTargetTable = TableId.fromPb(statisticsPb.getQuery().getDdlTargetTable());
}
if (statisticsPb.getQuery().getDdlTargetRoutine() != null) {
this.ddlTargetRoutine = RoutineId.fromPb(statisticsPb.getQuery().getDdlTargetRoutine());
}
this.estimatedBytesProcessed = statisticsPb.getQuery().getEstimatedBytesProcessed();
this.numDmlAffectedRows = statisticsPb.getQuery().getNumDmlAffectedRows();
this.totalBytesBilled = statisticsPb.getQuery().getTotalBytesBilled();
this.totalBytesProcessed = statisticsPb.getQuery().getTotalBytesProcessed();
this.totalPartitionsProcessed = statisticsPb.getQuery().getTotalPartitionsProcessed();
this.totalSlotMs = statisticsPb.getQuery().getTotalSlotMs();
if (statisticsPb.getQuery().getStatementType() != null) {
this.statementType = StatementType.valueOf(statisticsPb.getQuery().getStatementType());
}
if (statisticsPb.getQuery().getReferencedTables() != null) {
this.referencedTables =
Lists.transform(
statisticsPb.getQuery().getReferencedTables(), TableId.FROM_PB_FUNCTION);
}
if (statisticsPb.getQuery().getQueryPlan() != null) {
this.queryPlan =
Lists.transform(
statisticsPb.getQuery().getQueryPlan(), QueryStage.FROM_PB_FUNCTION);
}
if (statisticsPb.getQuery().getTimeline() != null) {
this.timeline =
Lists.transform(
statisticsPb.getQuery().getTimeline(), TimelineSample.FROM_PB_FUNCTION);
}
if (statisticsPb.getQuery().getSchema() != null) {
this.schema = Schema.fromPb(statisticsPb.getQuery().getSchema());
}
if (statisticsPb.getQuery().getDmlStats() != null) {
this.dmlStats = DmlStats.fromPb(statisticsPb.getQuery().getDmlStats());
}
}
}
Builder setBiEngineStats(BiEngineStats biEngineStats) {
this.biEngineStats = biEngineStats;
return self();
}
Builder setBillingTier(Integer billingTier) {
this.billingTier = billingTier;
return self();
}
Builder setCacheHit(Boolean cacheHit) {
this.cacheHit = cacheHit;
return self();
}
Builder setDDLOperationPerformed(String ddlOperationPerformed) {
this.ddlOperationPerformed = ddlOperationPerformed;
return self();
}
Builder setDDLTargetTable(TableId ddlTargetTable) {
this.ddlTargetTable = ddlTargetTable;
return self();
}
Builder setDDLTargetRoutine(RoutineId ddlTargetRoutine) {
this.ddlTargetRoutine = ddlTargetRoutine;
return self();
}
Builder setEstimatedBytesProcessed(Long estimatedBytesProcessed) {
this.estimatedBytesProcessed = estimatedBytesProcessed;
return self();
}
Builder setNumDmlAffectedRows(Long numDmlAffectedRows) {
this.numDmlAffectedRows = numDmlAffectedRows;
return self();
}
Builder setDmlStats(DmlStats dmlStats) {
this.dmlStats = dmlStats;
return self();
}
Builder setReferenceTables(List referencedTables) {
this.referencedTables = referencedTables;
return self();
}
Builder setStatementType(StatementType statementType) {
this.statementType = statementType;
return self();
}
Builder setStatementType(String strStatementType) {
this.statementType = StatementType.valueOf(strStatementType);
return self();
}
Builder setTotalBytesBilled(Long totalBytesBilled) {
this.totalBytesBilled = totalBytesBilled;
return self();
}
Builder setTotalBytesProcessed(Long totalBytesProcessed) {
this.totalBytesProcessed = totalBytesProcessed;
return self();
}
Builder setTotalPartitionsProcessed(Long totalPartitionsProcessed) {
this.totalPartitionsProcessed = totalPartitionsProcessed;
return self();
}
Builder setTotalSlotMs(Long totalSlotMs) {
this.totalSlotMs = totalSlotMs;
return self();
}
Builder setQueryPlan(List queryPlan) {
this.queryPlan = queryPlan;
return self();
}
Builder setTimeline(List timeline) {
this.timeline = timeline;
return self();
}
Builder setSchema(Schema schema) {
this.schema = schema;
return self();
}
@Override
QueryStatistics build() {
return new QueryStatistics(this);
}
}
private QueryStatistics(Builder builder) {
super(builder);
this.biEngineStats = builder.biEngineStats;
this.billingTier = builder.billingTier;
this.cacheHit = builder.cacheHit;
this.ddlOperationPerformed = builder.ddlOperationPerformed;
this.ddlTargetTable = builder.ddlTargetTable;
this.ddlTargetRoutine = builder.ddlTargetRoutine;
this.estimatedBytesProcessed = builder.estimatedBytesProcessed;
this.numDmlAffectedRows = builder.numDmlAffectedRows;
this.dmlStats = builder.dmlStats;
this.referencedTables = builder.referencedTables;
this.statementType = builder.statementType;
this.totalBytesBilled = builder.totalBytesBilled;
this.totalBytesProcessed = builder.totalBytesProcessed;
this.totalPartitionsProcessed = builder.totalPartitionsProcessed;
this.totalSlotMs = builder.totalSlotMs;
this.queryPlan = builder.queryPlan;
this.timeline = builder.timeline;
this.schema = builder.schema;
}
/** Returns query statistics specific to the use of BI Engine. */
public BiEngineStats getBiEngineStats() {
return biEngineStats;
}
/** Returns the billing tier for the job. */
public Integer getBillingTier() {
return billingTier;
}
/**
* Returns whether the query result was fetched from the query cache.
*
* @see Query Caching
*/
public Boolean getCacheHit() {
return cacheHit;
}
/** [BETA] For DDL queries, returns the operation applied to the DDL target table. */
public String getDdlOperationPerformed() {
return ddlOperationPerformed;
}
/** [BETA] For DDL queries, returns the TableID of the targeted table. */
public TableId getDdlTargetTable() {
return ddlTargetTable;
}
/** [BETA] For DDL queries, returns the RoutineId of the targeted routine. */
public RoutineId getDdlTargetRoutine() {
return ddlTargetRoutine;
}
/** The original estimate of bytes processed for the job. */
public Long getEstimatedBytesProcessed() {
return estimatedBytesProcessed;
}
/**
* The number of rows affected by a DML statement. Present only for DML statements INSERT,
* UPDATE or DELETE.
*/
public Long getNumDmlAffectedRows() {
return numDmlAffectedRows;
}
/** Detailed statistics for DML statements. */
public DmlStats getDmlStats() {
return dmlStats;
}
/**
* Referenced tables for the job. Queries that reference more than 50 tables will not have a
* complete list.
*/
public List getReferencedTables() {
return referencedTables;
}
/**
* [BETA] The type of query statement, if valid. Possible values include: SELECT INSERT UPDATE
* DELETE CREATE_TABLE CREATE_TABLE_AS_SELECT DROP_TABLE CREATE_VIEW DROP_VIEW
*/
public StatementType getStatementType() {
return statementType;
}
/** Returns the total number of bytes billed for the job. */
public Long getTotalBytesBilled() {
return totalBytesBilled;
}
/** Returns the total number of bytes processed by the job. */
public Long getTotalBytesProcessed() {
return totalBytesProcessed;
}
/** Total number of partitions processed from all partitioned tables referenced in the job. */
public Long getTotalPartitionsProcessed() {
return totalPartitionsProcessed;
}
/** Returns the slot-milliseconds consumed by the query. */
public Long getTotalSlotMs() {
return totalSlotMs;
}
/**
* Returns the query plan as a list of stages or {@code null} if a query plan is not available.
* Each stage involves a number of steps that read from data sources, perform a series of
* transformations on the input, and emit an output to a future stage (or the final result). The
* query plan is available for a completed query job and is retained for 7 days.
*
* @see Query Plan
*/
public List getQueryPlan() {
return queryPlan;
}
/**
* Return the timeline for the query, as a list of timeline samples. Each sample provides
* information about the overall progress of the query. Information includes time of the sample,
* progress reporting on active, completed, and pending units of work, as well as the cumulative
* estimation of slot-milliseconds consumed by the query.
*/
public List getTimeline() {
return timeline;
}
/**
* Returns the schema for the query result. Present only for successful dry run of non-legacy
* SQL queries.
*/
public Schema getSchema() {
return schema;
}
@Override
ToStringHelper toStringHelper() {
return super.toStringHelper()
.add("biEngineStats", biEngineStats)
.add("billingTier", billingTier)
.add("cacheHit", cacheHit)
.add("totalBytesBilled", totalBytesBilled)
.add("totalBytesProcessed", totalBytesProcessed)
.add("queryPlan", queryPlan)
.add("timeline", timeline)
.add("schema", schema);
}
@Override
public final boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(QueryStatistics.class)
&& baseEquals((QueryStatistics) obj);
}
@Override
public final int hashCode() {
return Objects.hash(
baseHashCode(),
biEngineStats,
billingTier,
cacheHit,
totalBytesBilled,
totalBytesProcessed,
queryPlan,
schema);
}
@Override
com.google.api.services.bigquery.model.JobStatistics toPb() {
JobStatistics2 queryStatisticsPb = new JobStatistics2();
if (biEngineStats != null) {
queryStatisticsPb.setBiEngineStatistics(biEngineStats.toPb());
}
queryStatisticsPb.setBillingTier(billingTier);
queryStatisticsPb.setCacheHit(cacheHit);
queryStatisticsPb.setDdlOperationPerformed(ddlOperationPerformed);
queryStatisticsPb.setEstimatedBytesProcessed(estimatedBytesProcessed);
queryStatisticsPb.setTotalBytesBilled(totalBytesBilled);
queryStatisticsPb.setTotalBytesProcessed(totalBytesProcessed);
queryStatisticsPb.setTotalPartitionsProcessed(totalPartitionsProcessed);
queryStatisticsPb.setTotalSlotMs(totalSlotMs);
if (ddlTargetTable != null) {
queryStatisticsPb.setDdlTargetTable(ddlTargetTable.toPb());
}
if (ddlTargetRoutine != null) {
queryStatisticsPb.setDdlTargetRoutine(ddlTargetRoutine.toPb());
}
if (dmlStats != null) {
queryStatisticsPb.setDmlStats(dmlStats.toPb());
}
if (referencedTables != null) {
queryStatisticsPb.setReferencedTables(
Lists.transform(referencedTables, TableId.TO_PB_FUNCTION));
}
if (statementType != null) {
queryStatisticsPb.setStatementType(statementType.toString());
}
if (queryPlan != null) {
queryStatisticsPb.setQueryPlan(Lists.transform(queryPlan, QueryStage.TO_PB_FUNCTION));
}
if (timeline != null) {
queryStatisticsPb.setTimeline(Lists.transform(timeline, TimelineSample.TO_PB_FUNCTION));
}
if (schema != null) {
queryStatisticsPb.setSchema(schema.toPb());
}
return super.toPb().setQuery(queryStatisticsPb);
}
static Builder newBuilder() {
return new Builder();
}
@SuppressWarnings("unchecked")
static QueryStatistics fromPb(
com.google.api.services.bigquery.model.JobStatistics statisticPb) {
return new Builder(statisticPb).build();
}
}
/** A Google BigQuery Script statistics. */
public static class ScriptStatistics {
private static final long serialVersionUID = 1328768324537448161L;
private final String evaluationKind;
private final List stackFrames;
public static class ScriptStackFrame {
static final Function<
com.google.api.services.bigquery.model.ScriptStackFrame, ScriptStackFrame>
FROM_PB_FUNCTION =
new Function<
com.google.api.services.bigquery.model.ScriptStackFrame, ScriptStackFrame>() {
@NullableDecl
@Override
public ScriptStackFrame apply(
@NullableDecl
com.google.api.services.bigquery.model.ScriptStackFrame stackFrame) {
return ScriptStackFrame.fromPb(stackFrame);
}
};
static final Function<
ScriptStackFrame, com.google.api.services.bigquery.model.ScriptStackFrame>
TO_PB_FUNCTION =
new Function<
ScriptStackFrame, com.google.api.services.bigquery.model.ScriptStackFrame>() {
@NullableDecl
@Override
public com.google.api.services.bigquery.model.ScriptStackFrame apply(
@NullableDecl ScriptStackFrame scriptStackFrame) {
return scriptStackFrame.toPb();
}
};
private final Integer endColumn;
private final Integer endLine;
private final String procedureId;
private final Integer startColumn;
private final Integer startLine;
private final String text;
private ScriptStackFrame(Builder builder) {
this.endColumn = builder.endColumn;
this.endLine = builder.endLine;
this.procedureId = builder.procedureId;
this.startColumn = builder.startColumn;
this.startLine = builder.startLine;
this.text = builder.text;
}
static class Builder {
private Integer endColumn;
private Integer endLine;
private String procedureId;
private Integer startColumn;
private Integer startLine;
private String text;
Builder setEndColumn(Integer endColumn) {
this.endColumn = endColumn;
return this;
}
Builder setEndLine(Integer endLine) {
this.endLine = endLine;
return this;
}
Builder setProcedureId(String procedureId) {
this.procedureId = procedureId;
return this;
}
Builder setStartColumn(Integer startColumn) {
this.startColumn = startColumn;
return this;
}
Builder setStartLine(Integer startLine) {
this.startLine = startLine;
return this;
}
Builder setText(String text) {
this.text = text;
return this;
}
ScriptStackFrame build() {
return new ScriptStackFrame(this);
}
}
/** Returns the end column. */
public Integer getEndColumn() {
return endColumn;
}
/** Returns the end line. */
public Integer getEndLine() {
return endLine;
}
/** Returns Name of the active procedure, empty if in a top-level script. */
public String getProcedureId() {
return procedureId;
}
/** Returns the start column */
public Integer getStartColumn() {
return startColumn;
}
/** Returns the start line. */
public Integer getStartLine() {
return startLine;
}
/** Returns Text of the current statement/expression. */
public String getText() {
return text;
}
ToStringHelper toStringHelper() {
return MoreObjects.toStringHelper(this)
.add("endColumn", endColumn)
.add("endLine", endLine)
.add("procedureId", procedureId)
.add("startColumn", startColumn)
.add("startLine", startLine)
.add("text", text);
}
@Override
public String toString() {
return toStringHelper().toString();
}
@Override
public boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(ScriptStackFrame.class)
&& Objects.equals(toPb(), ((ScriptStackFrame) obj).toPb());
}
@Override
public int hashCode() {
return Objects.hash(endColumn, endLine, procedureId, startColumn, startLine, text);
}
com.google.api.services.bigquery.model.ScriptStackFrame toPb() {
com.google.api.services.bigquery.model.ScriptStackFrame stackFrame =
new com.google.api.services.bigquery.model.ScriptStackFrame();
stackFrame.setEndColumn(endColumn);
stackFrame.setEndLine(endLine);
stackFrame.setProcedureId(procedureId);
stackFrame.setStartColumn(startColumn);
stackFrame.setStartLine(startLine);
stackFrame.setText(text);
return stackFrame;
}
static Builder newBuilder() {
return new Builder();
}
static ScriptStackFrame fromPb(
com.google.api.services.bigquery.model.ScriptStackFrame stackFrame) {
Builder builder = newBuilder();
if (stackFrame.getEndColumn() != null) {
builder.setEndColumn(stackFrame.getEndColumn());
}
if (stackFrame.getEndLine() != null) {
builder.setEndLine(stackFrame.getEndLine());
}
if (stackFrame.getProcedureId() != null) {
builder.setProcedureId(stackFrame.getProcedureId());
}
if (stackFrame.getStartColumn() != null) {
builder.setStartColumn(stackFrame.getStartColumn());
}
if (stackFrame.getStartLine() != null) {
builder.setStartLine(stackFrame.getStartLine());
}
if (stackFrame.getText() != null) {
builder.setText(stackFrame.getText());
}
return builder.build();
}
}
private ScriptStatistics(Builder builder) {
this.evaluationKind = builder.evaluationKind;
this.stackFrames = builder.stackFrames;
}
static class Builder {
private String evaluationKind;
private List stackFrames;
private Builder() {}
Builder setEvaluationKind(String evaluationKind) {
this.evaluationKind = evaluationKind;
return this;
}
Builder setStackFrames(List stackFrames) {
this.stackFrames = stackFrames;
return this;
}
ScriptStatistics build() {
return new ScriptStatistics(this);
}
}
static Builder newBuilder() {
return new Builder();
}
/** Returns child job was a statement or expression */
public String getEvaluationKind() {
return evaluationKind;
}
/**
* Stack trace showing the line/column/procedure name of each frame on the stack at the point
* where the current evaluation happened. The leaf frame is first, the primary script is last.
* Never empty.
*/
public List getStackFrames() {
return stackFrames;
}
ToStringHelper toStringHelper() {
return MoreObjects.toStringHelper(this)
.add("evaluationKind", evaluationKind)
.add("stackFrames", stackFrames);
}
@Override
public String toString() {
return toStringHelper().toString();
}
@Override
public boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(ScriptStatistics.class)
&& Objects.equals(toPb(), ((ScriptStatistics) obj).toPb());
}
@Override
public int hashCode() {
return Objects.hash(evaluationKind, stackFrames);
}
com.google.api.services.bigquery.model.ScriptStatistics toPb() {
com.google.api.services.bigquery.model.ScriptStatistics scriptStatistics =
new com.google.api.services.bigquery.model.ScriptStatistics();
scriptStatistics.setEvaluationKind(evaluationKind);
scriptStatistics.setStackFrames(
Lists.transform(stackFrames, ScriptStackFrame.TO_PB_FUNCTION));
return scriptStatistics;
}
static ScriptStatistics fromPb(
com.google.api.services.bigquery.model.ScriptStatistics scriptStatistics) {
Builder builder = newBuilder();
if (scriptStatistics.getEvaluationKind() != null) {
builder.setEvaluationKind(scriptStatistics.getEvaluationKind());
}
if (scriptStatistics.getStackFrames() != null) {
builder.setStackFrames(
Lists.transform(scriptStatistics.getStackFrames(), ScriptStackFrame.FROM_PB_FUNCTION));
}
return builder.build();
}
}
/** ReservationUsage contains information about a job's usage of a single reservation. */
public static class ReservationUsage {
static final Function<
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage, ReservationUsage>
FROM_PB_FUNCTION =
new Function<
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage,
ReservationUsage>() {
@Override
public ReservationUsage apply(
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage usage) {
return ReservationUsage.fromPb(usage);
}
};
static final Function<
ReservationUsage, com.google.api.services.bigquery.model.JobStatistics.ReservationUsage>
TO_PB_FUNCTION =
new Function<
ReservationUsage,
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage>() {
@Override
public com.google.api.services.bigquery.model.JobStatistics.ReservationUsage apply(
ReservationUsage usage) {
return usage.toPb();
}
};
private final String name;
private final Long slotMs;
public static class Builder {
private String name;
private Long slotMs;
private Builder() {};
Builder setName(String name) {
this.name = name;
return this;
}
Builder setSlotMs(Long slotMs) {
this.slotMs = slotMs;
return this;
}
ReservationUsage build() {
return new ReservationUsage(this);
}
}
private ReservationUsage(Builder builder) {
this.name = builder.name;
this.slotMs = builder.slotMs;
}
// Return mame indicates the utilized reservation name, or "unreserved" for ondemand usage.
public String getName() {
return name;
}
// Returns slotMs reports the slot milliseconds utilized within in the given reservation.
public Long getSlotMs() {
return slotMs;
}
static Builder newBuilder() {
return new Builder();
}
ToStringHelper toStringHelper() {
return MoreObjects.toStringHelper(this).add("name", name).add("slotMs", slotMs);
}
@Override
public String toString() {
return toStringHelper().toString();
}
@Override
public boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(ReservationUsage.class)
&& Objects.equals(toPb(), ((ReservationUsage) obj).toPb());
}
@Override
public int hashCode() {
return Objects.hash(name, slotMs);
}
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage toPb() {
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage usage =
new com.google.api.services.bigquery.model.JobStatistics.ReservationUsage();
usage.setName(name);
usage.setSlotMs(slotMs);
return usage;
}
static ReservationUsage fromPb(
com.google.api.services.bigquery.model.JobStatistics.ReservationUsage usage) {
Builder builder = newBuilder();
builder.setName(usage.getName());
builder.setSlotMs(usage.getSlotMs());
return builder.build();
}
}
// TransactionInfo contains information about a multi-statement transaction that may have
// associated with a job.
public static class TransactionInfo {
// TransactionID is the system-generated identifier for the transaction.
private final String transactionId;
public static class Builder {
private String transactionId;
private Builder() {};
Builder setTransactionId(String transactionId) {
this.transactionId = transactionId;
return this;
}
TransactionInfo build() {
return new TransactionInfo(this);
}
}
private TransactionInfo(Builder builder) {
this.transactionId = builder.transactionId;
}
public String getTransactionId() {
return transactionId;
}
static Builder newbuilder() {
return new Builder();
}
ToStringHelper toStringHelper() {
return MoreObjects.toStringHelper(this).add("transactionId", transactionId);
}
@Override
public String toString() {
return toStringHelper().toString();
}
@Override
public boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(TransactionInfo.class)
&& Objects.equals(toPb(), ((TransactionInfo) obj).toPb());
}
@Override
public int hashCode() {
return Objects.hash(transactionId);
}
com.google.api.services.bigquery.model.TransactionInfo toPb() {
com.google.api.services.bigquery.model.TransactionInfo transactionInfo =
new com.google.api.services.bigquery.model.TransactionInfo();
transactionInfo.setTransactionId(transactionId);
return transactionInfo;
}
static TransactionInfo fromPb(
com.google.api.services.bigquery.model.TransactionInfo transactionInfo) {
Builder builder = newbuilder();
builder.setTransactionId(transactionInfo.getTransactionId());
return builder.build();
}
}
// SessionInfo contains information about the session if this job is part of one.
public static class SessionInfo {
// Id of the session
private final String sessionId;
public static class Builder {
private String sessionId;
private Builder() {};
Builder setSessionId(String sessionId) {
this.sessionId = sessionId;
return this;
}
SessionInfo build() {
return new SessionInfo(this);
}
}
private SessionInfo(Builder builder) {
this.sessionId = builder.sessionId;
}
public String getSessionId() {
return sessionId;
}
static Builder newBuilder() {
return new Builder();
}
ToStringHelper toStringHelper() {
return MoreObjects.toStringHelper(this).add("sessionId", sessionId);
}
@Override
public String toString() {
return toStringHelper().toString();
}
@Override
public boolean equals(Object obj) {
return obj == this
|| obj != null
&& obj.getClass().equals(SessionInfo.class)
&& Objects.equals(toPb(), ((SessionInfo) obj).toPb());
}
@Override
public int hashCode() {
return Objects.hash(sessionId);
}
com.google.api.services.bigquery.model.SessionInfo toPb() {
com.google.api.services.bigquery.model.SessionInfo sessionInfo =
new com.google.api.services.bigquery.model.SessionInfo();
sessionInfo.setSessionId(sessionId);
return sessionInfo;
}
static SessionInfo fromPb(com.google.api.services.bigquery.model.SessionInfo sessionInfo) {
SessionInfo.Builder builder = newBuilder();
builder.setSessionId(sessionInfo.getSessionId());
return builder.build();
}
}
abstract static class Builder> {
private Long creationTime;
private Long endTime;
private Long startTime;
private Long numChildJobs;
private String parentJobId;
private ScriptStatistics scriptStatistics;
private List reservationUsage;
private TransactionInfo transactionInfo;
private SessionInfo sessionInfo;
protected Builder() {}
protected Builder(com.google.api.services.bigquery.model.JobStatistics statisticsPb) {
this.creationTime = statisticsPb.getCreationTime();
this.endTime = statisticsPb.getEndTime();
this.startTime = statisticsPb.getStartTime();
this.numChildJobs = statisticsPb.getNumChildJobs();
this.parentJobId = statisticsPb.getParentJobId();
if (statisticsPb.getScriptStatistics() != null) {
this.scriptStatistics = ScriptStatistics.fromPb(statisticsPb.getScriptStatistics());
}
if (reservationUsage != null) {
this.reservationUsage =
Lists.transform(statisticsPb.getReservationUsage(), ReservationUsage.FROM_PB_FUNCTION);
}
if (statisticsPb.getTransactionInfo() != null) {
this.transactionInfo = TransactionInfo.fromPb(statisticsPb.getTransactionInfo());
}
if (statisticsPb.getSessionInfo() != null) {
this.sessionInfo = SessionInfo.fromPb(statisticsPb.getSessionInfo());
}
}
@SuppressWarnings("unchecked")
protected B self() {
return (B) this;
}
B setCreationTimestamp(Long creationTime) {
this.creationTime = creationTime;
return self();
}
B setEndTime(Long endTime) {
this.endTime = endTime;
return self();
}
B setStartTime(Long startTime) {
this.startTime = startTime;
return self();
}
abstract T build();
}
protected JobStatistics(Builder builder) {
this.creationTime = builder.creationTime;
this.endTime = builder.endTime;
this.startTime = builder.startTime;
this.numChildJobs = builder.numChildJobs;
this.parentJobId = builder.parentJobId;
this.scriptStatistics = builder.scriptStatistics;
this.reservationUsage = builder.reservationUsage;
this.transactionInfo = builder.transactionInfo;
this.sessionInfo = builder.sessionInfo;
}
/** Returns the creation time of the job in milliseconds since epoch. */
public Long getCreationTime() {
return creationTime;
}
/**
* Returns the end time of the job in milliseconds since epoch. Returns {@code null} if the job
* has not finished yet.
*/
public Long getEndTime() {
return endTime;
}
/**
* Returns the start time of the job in milliseconds since epoch. Returns {@code null} if the job
* has not started yet.
*/
public Long getStartTime() {
return startTime;
}
/** Returns the number of child job executed. */
public Long getNumChildJobs() {
return numChildJobs;
}
/** Returns the parent job id of child job. */
public String getParentJobId() {
return parentJobId;
}
/** Returns the statistics for a child job of a script. */
public ScriptStatistics getScriptStatistics() {
return scriptStatistics;
}
/** ReservationUsage contains information about a job's usage of a single reservation. */
public List getReservationUsage() {
return reservationUsage;
}
/** Info indicates the transaction ID associated with the job, if any. */
public TransactionInfo getTransactionInfo() {
return transactionInfo;
}
/** Info of the session if this job is part of one. */
public SessionInfo getSessionInfo() {
return sessionInfo;
}
ToStringHelper toStringHelper() {
return MoreObjects.toStringHelper(this)
.add("creationTime", creationTime)
.add("endTime", endTime)
.add("startTime", startTime)
.add("numChildJobs", numChildJobs)
.add("parentJobId", parentJobId)
.add("scriptStatistics", scriptStatistics)
.add("reservationUsage", reservationUsage)
.add("transactionInfo", transactionInfo)
.add("sessionInfo", sessionInfo);
}
@Override
public String toString() {
return toStringHelper().toString();
}
final int baseHashCode() {
return Objects.hash(
creationTime,
endTime,
startTime,
numChildJobs,
parentJobId,
scriptStatistics,
reservationUsage,
transactionInfo,
sessionInfo);
}
final boolean baseEquals(JobStatistics jobStatistics) {
return Objects.equals(toPb(), jobStatistics.toPb());
}
com.google.api.services.bigquery.model.JobStatistics toPb() {
com.google.api.services.bigquery.model.JobStatistics statistics =
new com.google.api.services.bigquery.model.JobStatistics();
statistics.setCreationTime(creationTime);
statistics.setEndTime(endTime);
statistics.setStartTime(startTime);
statistics.setNumChildJobs(numChildJobs);
statistics.setParentJobId(parentJobId);
if (scriptStatistics != null) {
statistics.setScriptStatistics(scriptStatistics.toPb());
}
if (reservationUsage != null) {
statistics.setReservationUsage(
Lists.transform(reservationUsage, ReservationUsage.TO_PB_FUNCTION));
}
if (transactionInfo != null) {
statistics.setTransactionInfo(transactionInfo.toPb());
}
if (sessionInfo != null) {
statistics.setSessionInfo(sessionInfo.toPb());
}
return statistics;
}
@SuppressWarnings("unchecked")
static T fromPb(com.google.api.services.bigquery.model.Job jobPb) {
JobConfiguration jobConfigPb = jobPb.getConfiguration();
com.google.api.services.bigquery.model.JobStatistics statisticPb = jobPb.getStatistics();
if (jobConfigPb.getLoad() != null) {
return (T) LoadStatistics.fromPb(statisticPb);
} else if (jobConfigPb.getExtract() != null) {
return (T) ExtractStatistics.fromPb(statisticPb);
} else if (jobConfigPb.getQuery() != null) {
return (T) QueryStatistics.fromPb(statisticPb);
} else if (jobConfigPb.getCopy() != null) {
return (T) CopyStatistics.fromPb(statisticPb);
} else {
throw new IllegalArgumentException("unknown job configuration: " + jobConfigPb);
}
}
}