|
19 | 19 | import static com.google.common.base.Preconditions.checkNotNull; |
20 | 20 |
|
21 | 21 | import com.google.api.services.bigquery.model.JobReference; |
22 | | - |
| 22 | +import com.google.auto.value.AutoValue; |
23 | 23 | import java.io.Serializable; |
24 | | -import java.util.Objects; |
25 | 24 | import java.util.UUID; |
| 25 | +import javax.annotation.Nullable; |
26 | 26 |
|
27 | | -/** |
28 | | - * Google BigQuery Job identity. |
29 | | - */ |
30 | | -public final class JobId implements Serializable { |
31 | | - |
32 | | - private static final long serialVersionUID = 1225914835379688976L; |
| 27 | +/** Google BigQuery Job identity. */ |
| 28 | +@AutoValue |
| 29 | +public abstract class JobId implements Serializable { |
33 | 30 |
|
34 | | - private final String project; |
35 | | - private final String job; |
| 31 | + private static final long serialVersionUID = 1225914835379688977L; |
36 | 32 |
|
| 33 | + JobId() { |
| 34 | + // Users cannot extend this, but AutoValue can. |
| 35 | + } |
37 | 36 |
|
38 | 37 | /** |
39 | | - * Returns project's user-defined id. |
| 38 | + * Returns job's project id. |
| 39 | + * |
| 40 | + * <p>When sending requests with null project, the client will attempt to infer the project name |
| 41 | + * from the environment. |
40 | 42 | */ |
41 | | - public String getProject() { |
42 | | - return project; |
43 | | - } |
| 43 | + @Nullable |
| 44 | + public abstract String getProject(); |
44 | 45 |
|
| 46 | + /** |
| 47 | + * Returns the job's id. |
| 48 | + * |
| 49 | + * <p>The server returns null job id for dry-run queries. |
| 50 | + */ |
| 51 | + @Nullable |
| 52 | + public abstract String getJob(); |
45 | 53 |
|
46 | 54 | /** |
47 | | - * Returns the job's user-defined id. |
| 55 | + * Returns the job's location. |
| 56 | + * |
| 57 | + * <p>When sending requests, the location must be specified for jobs whose location not "US" or |
| 58 | + * "EU". |
48 | 59 | */ |
49 | | - public String getJob() { |
50 | | - return job; |
| 60 | + @Nullable |
| 61 | + abstract String getLocation(); |
| 62 | + |
| 63 | + public abstract Builder toBuilder(); |
| 64 | + |
| 65 | + public static Builder newBuilder() { |
| 66 | + return new AutoValue_JobId.Builder(); |
51 | 67 | } |
52 | 68 |
|
53 | | - private JobId(String project, String job) { |
54 | | - this.project = project; |
55 | | - this.job = job; |
| 69 | + @AutoValue.Builder |
| 70 | + public abstract static class Builder { |
| 71 | + public abstract Builder setProject(String project); |
| 72 | + |
| 73 | + public abstract Builder setJob(String job); |
| 74 | + |
| 75 | + /** {@code setJob} to a pseudo-random string. */ |
| 76 | + public Builder setRandomJob() { |
| 77 | + return setJob(UUID.randomUUID().toString()); |
| 78 | + } |
| 79 | + |
| 80 | + abstract Builder setLocation(String location); |
| 81 | + |
| 82 | + public abstract JobId build(); |
56 | 83 | } |
57 | 84 |
|
58 | 85 | /** |
59 | 86 | * Creates a job identity given project's and job's user-defined id. |
60 | 87 | */ |
61 | 88 | public static JobId of(String project, String job) { |
62 | | - return new JobId(checkNotNull(project), checkNotNull(job)); |
| 89 | + return newBuilder().setProject(checkNotNull(project)).setJob(checkNotNull(job)).build(); |
63 | 90 | } |
64 | 91 |
|
65 | 92 | /** |
66 | 93 | * Creates a job identity given only its user-defined id. |
67 | 94 | */ |
68 | 95 | public static JobId of(String job) { |
69 | | - return new JobId(null, checkNotNull(job)); |
| 96 | + return newBuilder().setJob(checkNotNull(job)).build(); |
70 | 97 | } |
71 | 98 |
|
72 | 99 | /** |
73 | 100 | * Creates a job identity with autogenerated id and no project specified. |
74 | 101 | */ |
75 | 102 | public static JobId of() { |
76 | | - return new JobId(null, UUID.randomUUID().toString()); |
77 | | - } |
78 | | - |
79 | | - @Override |
80 | | - public boolean equals(Object obj) { |
81 | | - return obj == this |
82 | | - || obj instanceof JobId |
83 | | - && Objects.equals(toPb(), ((JobId) obj).toPb()); |
84 | | - } |
85 | | - |
86 | | - @Override |
87 | | - public int hashCode() { |
88 | | - return Objects.hash(project, job); |
89 | | - } |
90 | | - |
91 | | - @Override |
92 | | - public String toString() { |
93 | | - return toPb().toString(); |
| 103 | + return newBuilder().setRandomJob().build(); |
94 | 104 | } |
95 | 105 |
|
96 | 106 | JobId setProjectId(String projectId) { |
97 | | - return getProject() != null ? this : JobId.of(projectId, getJob()); |
| 107 | + return getProject() != null ? this : toBuilder().setProject(projectId).build(); |
98 | 108 | } |
99 | 109 |
|
100 | 110 | JobReference toPb() { |
101 | | - return new JobReference().setProjectId(project).setJobId(job); |
| 111 | + return new JobReference() |
| 112 | + .setProjectId(getProject()) |
| 113 | + .setJobId(getJob()) |
| 114 | + .setLocation(getLocation()); |
102 | 115 | } |
103 | 116 |
|
104 | 117 | static JobId fromPb(JobReference jobRef) { |
105 | | - return new JobId(jobRef.getProjectId(), jobRef.getJobId()); |
| 118 | + return newBuilder() |
| 119 | + .setProject(jobRef.getProjectId()) |
| 120 | + .setJob(jobRef.getJobId()) |
| 121 | + .setLocation(jobRef.getLocation()) |
| 122 | + .build(); |
106 | 123 | } |
107 | 124 | } |
0 commit comments