22
22
import com .google .api .gax .grpc .GrpcCallSettings ;
23
23
import com .google .api .gax .grpc .GrpcStubCallableFactory ;
24
24
import com .google .api .gax .rpc .ClientContext ;
25
+ import com .google .api .gax .rpc .RequestParamsBuilder ;
25
26
import com .google .api .gax .rpc .ServerStreamingCallable ;
26
27
import com .google .api .gax .rpc .UnaryCallable ;
28
+ import com .google .api .pathtemplate .PathTemplate ;
27
29
import com .google .bigtable .v2 .CheckAndMutateRowRequest ;
28
30
import com .google .bigtable .v2 .CheckAndMutateRowResponse ;
29
31
import com .google .bigtable .v2 .MutateRowRequest ;
36
38
import com .google .bigtable .v2 .ReadRowsResponse ;
37
39
import com .google .bigtable .v2 .SampleRowKeysRequest ;
38
40
import com .google .bigtable .v2 .SampleRowKeysResponse ;
39
- import com .google .common .collect .ImmutableMap ;
40
41
import com .google .longrunning .stub .GrpcOperationsStub ;
41
42
import io .grpc .MethodDescriptor ;
42
43
import io .grpc .protobuf .ProtoUtils ;
@@ -123,6 +124,31 @@ public class GrpcBigtableStub extends BigtableStub {
123
124
private final GrpcOperationsStub operationsStub ;
124
125
private final GrpcStubCallableFactory callableFactory ;
125
126
127
+ private static final PathTemplate READ_ROWS_0_PATH_TEMPLATE =
128
+ PathTemplate .create ("{table_name=projects/*/instances/*/tables/*}" );
129
+ private static final PathTemplate READ_ROWS_1_PATH_TEMPLATE =
130
+ PathTemplate .create ("{app_profile_id=**}" );
131
+ private static final PathTemplate SAMPLE_ROW_KEYS_0_PATH_TEMPLATE =
132
+ PathTemplate .create ("{table_name=projects/*/instances/*/tables/*}" );
133
+ private static final PathTemplate SAMPLE_ROW_KEYS_1_PATH_TEMPLATE =
134
+ PathTemplate .create ("{app_profile_id=**}" );
135
+ private static final PathTemplate MUTATE_ROW_0_PATH_TEMPLATE =
136
+ PathTemplate .create ("{table_name=projects/*/instances/*/tables/*}" );
137
+ private static final PathTemplate MUTATE_ROW_1_PATH_TEMPLATE =
138
+ PathTemplate .create ("{app_profile_id=**}" );
139
+ private static final PathTemplate MUTATE_ROWS_0_PATH_TEMPLATE =
140
+ PathTemplate .create ("{table_name=projects/*/instances/*/tables/*}" );
141
+ private static final PathTemplate MUTATE_ROWS_1_PATH_TEMPLATE =
142
+ PathTemplate .create ("{app_profile_id=**}" );
143
+ private static final PathTemplate CHECK_AND_MUTATE_ROW_0_PATH_TEMPLATE =
144
+ PathTemplate .create ("{table_name=projects/*/instances/*/tables/*}" );
145
+ private static final PathTemplate CHECK_AND_MUTATE_ROW_1_PATH_TEMPLATE =
146
+ PathTemplate .create ("{app_profile_id=**}" );
147
+ private static final PathTemplate READ_MODIFY_WRITE_ROW_0_PATH_TEMPLATE =
148
+ PathTemplate .create ("{table_name=projects/*/instances/*/tables/*}" );
149
+ private static final PathTemplate READ_MODIFY_WRITE_ROW_1_PATH_TEMPLATE =
150
+ PathTemplate .create ("{app_profile_id=**}" );
151
+
126
152
public static final GrpcBigtableStub create (BigtableStubSettings settings ) throws IOException {
127
153
return new GrpcBigtableStub (settings , ClientContext .create (settings ));
128
154
}
@@ -163,39 +189,48 @@ protected GrpcBigtableStub(
163
189
.setMethodDescriptor (readRowsMethodDescriptor )
164
190
.setParamsExtractor (
165
191
request -> {
166
- ImmutableMap .Builder <String , String > params = ImmutableMap .builder ();
167
- params .put ("table_name" , String .valueOf (request .getTableName ()));
168
- return params .build ();
192
+ RequestParamsBuilder builder = RequestParamsBuilder .create ();
193
+ builder .add (request .getTableName (), "table_name" , READ_ROWS_0_PATH_TEMPLATE );
194
+ builder .add (
195
+ request .getAppProfileId (), "app_profile_id" , READ_ROWS_1_PATH_TEMPLATE );
196
+ return builder .build ();
169
197
})
170
198
.build ();
171
199
GrpcCallSettings <SampleRowKeysRequest , SampleRowKeysResponse > sampleRowKeysTransportSettings =
172
200
GrpcCallSettings .<SampleRowKeysRequest , SampleRowKeysResponse >newBuilder ()
173
201
.setMethodDescriptor (sampleRowKeysMethodDescriptor )
174
202
.setParamsExtractor (
175
203
request -> {
176
- ImmutableMap .Builder <String , String > params = ImmutableMap .builder ();
177
- params .put ("table_name" , String .valueOf (request .getTableName ()));
178
- return params .build ();
204
+ RequestParamsBuilder builder = RequestParamsBuilder .create ();
205
+ builder .add (
206
+ request .getTableName (), "table_name" , SAMPLE_ROW_KEYS_0_PATH_TEMPLATE );
207
+ builder .add (
208
+ request .getAppProfileId (), "app_profile_id" , SAMPLE_ROW_KEYS_1_PATH_TEMPLATE );
209
+ return builder .build ();
179
210
})
180
211
.build ();
181
212
GrpcCallSettings <MutateRowRequest , MutateRowResponse > mutateRowTransportSettings =
182
213
GrpcCallSettings .<MutateRowRequest , MutateRowResponse >newBuilder ()
183
214
.setMethodDescriptor (mutateRowMethodDescriptor )
184
215
.setParamsExtractor (
185
216
request -> {
186
- ImmutableMap .Builder <String , String > params = ImmutableMap .builder ();
187
- params .put ("table_name" , String .valueOf (request .getTableName ()));
188
- return params .build ();
217
+ RequestParamsBuilder builder = RequestParamsBuilder .create ();
218
+ builder .add (request .getTableName (), "table_name" , MUTATE_ROW_0_PATH_TEMPLATE );
219
+ builder .add (
220
+ request .getAppProfileId (), "app_profile_id" , MUTATE_ROW_1_PATH_TEMPLATE );
221
+ return builder .build ();
189
222
})
190
223
.build ();
191
224
GrpcCallSettings <MutateRowsRequest , MutateRowsResponse > mutateRowsTransportSettings =
192
225
GrpcCallSettings .<MutateRowsRequest , MutateRowsResponse >newBuilder ()
193
226
.setMethodDescriptor (mutateRowsMethodDescriptor )
194
227
.setParamsExtractor (
195
228
request -> {
196
- ImmutableMap .Builder <String , String > params = ImmutableMap .builder ();
197
- params .put ("table_name" , String .valueOf (request .getTableName ()));
198
- return params .build ();
229
+ RequestParamsBuilder builder = RequestParamsBuilder .create ();
230
+ builder .add (request .getTableName (), "table_name" , MUTATE_ROWS_0_PATH_TEMPLATE );
231
+ builder .add (
232
+ request .getAppProfileId (), "app_profile_id" , MUTATE_ROWS_1_PATH_TEMPLATE );
233
+ return builder .build ();
199
234
})
200
235
.build ();
201
236
GrpcCallSettings <CheckAndMutateRowRequest , CheckAndMutateRowResponse >
@@ -204,9 +239,16 @@ protected GrpcBigtableStub(
204
239
.setMethodDescriptor (checkAndMutateRowMethodDescriptor )
205
240
.setParamsExtractor (
206
241
request -> {
207
- ImmutableMap .Builder <String , String > params = ImmutableMap .builder ();
208
- params .put ("table_name" , String .valueOf (request .getTableName ()));
209
- return params .build ();
242
+ RequestParamsBuilder builder = RequestParamsBuilder .create ();
243
+ builder .add (
244
+ request .getTableName (),
245
+ "table_name" ,
246
+ CHECK_AND_MUTATE_ROW_0_PATH_TEMPLATE );
247
+ builder .add (
248
+ request .getAppProfileId (),
249
+ "app_profile_id" ,
250
+ CHECK_AND_MUTATE_ROW_1_PATH_TEMPLATE );
251
+ return builder .build ();
210
252
})
211
253
.build ();
212
254
GrpcCallSettings <ReadModifyWriteRowRequest , ReadModifyWriteRowResponse >
@@ -215,9 +257,16 @@ protected GrpcBigtableStub(
215
257
.setMethodDescriptor (readModifyWriteRowMethodDescriptor )
216
258
.setParamsExtractor (
217
259
request -> {
218
- ImmutableMap .Builder <String , String > params = ImmutableMap .builder ();
219
- params .put ("table_name" , String .valueOf (request .getTableName ()));
220
- return params .build ();
260
+ RequestParamsBuilder builder = RequestParamsBuilder .create ();
261
+ builder .add (
262
+ request .getTableName (),
263
+ "table_name" ,
264
+ READ_MODIFY_WRITE_ROW_0_PATH_TEMPLATE );
265
+ builder .add (
266
+ request .getAppProfileId (),
267
+ "app_profile_id" ,
268
+ READ_MODIFY_WRITE_ROW_1_PATH_TEMPLATE );
269
+ return builder .build ();
221
270
})
222
271
.build ();
223
272
0 commit comments