|
24 | 24 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
25 | 25 | import com.google.api.gax.rpc.ClientContext; |
26 | 26 | import com.google.api.gax.rpc.OperationCallable; |
| 27 | +import com.google.api.gax.rpc.RequestParamsExtractor; |
27 | 28 | import com.google.api.gax.rpc.UnaryCallable; |
28 | 29 | import com.google.cloud.redis.v1.CreateInstanceRequest; |
29 | 30 | import com.google.cloud.redis.v1.DeleteInstanceRequest; |
|
34 | 35 | import com.google.cloud.redis.v1.ListInstancesResponse; |
35 | 36 | import com.google.cloud.redis.v1.OperationMetadata; |
36 | 37 | import com.google.cloud.redis.v1.UpdateInstanceRequest; |
| 38 | +import com.google.common.collect.ImmutableMap; |
37 | 39 | import com.google.longrunning.Operation; |
38 | 40 | import com.google.longrunning.stub.GrpcOperationsStub; |
39 | 41 | import com.google.protobuf.Empty; |
40 | 42 | import io.grpc.MethodDescriptor; |
41 | 43 | import io.grpc.protobuf.ProtoUtils; |
42 | 44 | import java.io.IOException; |
| 45 | +import java.util.Map; |
43 | 46 | import java.util.concurrent.TimeUnit; |
44 | 47 | import javax.annotation.Generated; |
45 | 48 |
|
@@ -170,26 +173,80 @@ protected GrpcCloudRedisStub( |
170 | 173 | GrpcCallSettings<ListInstancesRequest, ListInstancesResponse> listInstancesTransportSettings = |
171 | 174 | GrpcCallSettings.<ListInstancesRequest, ListInstancesResponse>newBuilder() |
172 | 175 | .setMethodDescriptor(listInstancesMethodDescriptor) |
| 176 | + .setParamsExtractor( |
| 177 | + new RequestParamsExtractor<ListInstancesRequest>() { |
| 178 | + @Override |
| 179 | + public Map<String, String> extract(ListInstancesRequest request) { |
| 180 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 181 | + params.put("parent", String.valueOf(request.getParent())); |
| 182 | + return params.build(); |
| 183 | + } |
| 184 | + }) |
173 | 185 | .build(); |
174 | 186 | GrpcCallSettings<GetInstanceRequest, Instance> getInstanceTransportSettings = |
175 | 187 | GrpcCallSettings.<GetInstanceRequest, Instance>newBuilder() |
176 | 188 | .setMethodDescriptor(getInstanceMethodDescriptor) |
| 189 | + .setParamsExtractor( |
| 190 | + new RequestParamsExtractor<GetInstanceRequest>() { |
| 191 | + @Override |
| 192 | + public Map<String, String> extract(GetInstanceRequest request) { |
| 193 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 194 | + params.put("name", String.valueOf(request.getName())); |
| 195 | + return params.build(); |
| 196 | + } |
| 197 | + }) |
177 | 198 | .build(); |
178 | 199 | GrpcCallSettings<CreateInstanceRequest, Operation> createInstanceTransportSettings = |
179 | 200 | GrpcCallSettings.<CreateInstanceRequest, Operation>newBuilder() |
180 | 201 | .setMethodDescriptor(createInstanceMethodDescriptor) |
| 202 | + .setParamsExtractor( |
| 203 | + new RequestParamsExtractor<CreateInstanceRequest>() { |
| 204 | + @Override |
| 205 | + public Map<String, String> extract(CreateInstanceRequest request) { |
| 206 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 207 | + params.put("parent", String.valueOf(request.getParent())); |
| 208 | + return params.build(); |
| 209 | + } |
| 210 | + }) |
181 | 211 | .build(); |
182 | 212 | GrpcCallSettings<UpdateInstanceRequest, Operation> updateInstanceTransportSettings = |
183 | 213 | GrpcCallSettings.<UpdateInstanceRequest, Operation>newBuilder() |
184 | 214 | .setMethodDescriptor(updateInstanceMethodDescriptor) |
| 215 | + .setParamsExtractor( |
| 216 | + new RequestParamsExtractor<UpdateInstanceRequest>() { |
| 217 | + @Override |
| 218 | + public Map<String, String> extract(UpdateInstanceRequest request) { |
| 219 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 220 | + params.put("instance.name", String.valueOf(request.getInstance().getName())); |
| 221 | + return params.build(); |
| 222 | + } |
| 223 | + }) |
185 | 224 | .build(); |
186 | 225 | GrpcCallSettings<DeleteInstanceRequest, Operation> deleteInstanceTransportSettings = |
187 | 226 | GrpcCallSettings.<DeleteInstanceRequest, Operation>newBuilder() |
188 | 227 | .setMethodDescriptor(deleteInstanceMethodDescriptor) |
| 228 | + .setParamsExtractor( |
| 229 | + new RequestParamsExtractor<DeleteInstanceRequest>() { |
| 230 | + @Override |
| 231 | + public Map<String, String> extract(DeleteInstanceRequest request) { |
| 232 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 233 | + params.put("name", String.valueOf(request.getName())); |
| 234 | + return params.build(); |
| 235 | + } |
| 236 | + }) |
189 | 237 | .build(); |
190 | 238 | GrpcCallSettings<FailoverInstanceRequest, Operation> failoverInstanceTransportSettings = |
191 | 239 | GrpcCallSettings.<FailoverInstanceRequest, Operation>newBuilder() |
192 | 240 | .setMethodDescriptor(failoverInstanceMethodDescriptor) |
| 241 | + .setParamsExtractor( |
| 242 | + new RequestParamsExtractor<FailoverInstanceRequest>() { |
| 243 | + @Override |
| 244 | + public Map<String, String> extract(FailoverInstanceRequest request) { |
| 245 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 246 | + params.put("name", String.valueOf(request.getName())); |
| 247 | + return params.build(); |
| 248 | + } |
| 249 | + }) |
193 | 250 | .build(); |
194 | 251 |
|
195 | 252 | this.listInstancesCallable = |
|
0 commit comments