|
37 | 37 | import com.google.api.gax.grpc.GrpcCallSettings; |
38 | 38 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
39 | 39 | import com.google.api.gax.rpc.ClientContext; |
| 40 | +import com.google.api.gax.rpc.RequestParamsExtractor; |
40 | 41 | import com.google.api.gax.rpc.UnaryCallable; |
| 42 | +import com.google.common.collect.ImmutableMap; |
41 | 43 | import com.google.longrunning.CancelOperationRequest; |
42 | 44 | import com.google.longrunning.DeleteOperationRequest; |
43 | 45 | import com.google.longrunning.GetOperationRequest; |
|
48 | 50 | import io.grpc.MethodDescriptor; |
49 | 51 | import io.grpc.protobuf.ProtoUtils; |
50 | 52 | import java.io.IOException; |
| 53 | +import java.util.Map; |
51 | 54 | import java.util.concurrent.TimeUnit; |
52 | 55 |
|
53 | 56 | /** |
@@ -146,19 +149,55 @@ protected GrpcOperationsStub( |
146 | 149 | GrpcCallSettings<GetOperationRequest, Operation> getOperationTransportSettings = |
147 | 150 | GrpcCallSettings.<GetOperationRequest, Operation>newBuilder() |
148 | 151 | .setMethodDescriptor(getOperationMethodDescriptor) |
| 152 | + .setParamsExtractor( |
| 153 | + new RequestParamsExtractor<GetOperationRequest>() { |
| 154 | + @Override |
| 155 | + public Map<String, String> extract(GetOperationRequest request) { |
| 156 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 157 | + params.put("name", String.valueOf(request.getName())); |
| 158 | + return params.build(); |
| 159 | + } |
| 160 | + }) |
149 | 161 | .build(); |
150 | 162 | GrpcCallSettings<ListOperationsRequest, ListOperationsResponse> |
151 | 163 | listOperationsTransportSettings = |
152 | 164 | GrpcCallSettings.<ListOperationsRequest, ListOperationsResponse>newBuilder() |
153 | 165 | .setMethodDescriptor(listOperationsMethodDescriptor) |
| 166 | + .setParamsExtractor( |
| 167 | + new RequestParamsExtractor<ListOperationsRequest>() { |
| 168 | + @Override |
| 169 | + public Map<String, String> extract(ListOperationsRequest request) { |
| 170 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 171 | + params.put("name", String.valueOf(request.getName())); |
| 172 | + return params.build(); |
| 173 | + } |
| 174 | + }) |
154 | 175 | .build(); |
155 | 176 | GrpcCallSettings<CancelOperationRequest, Empty> cancelOperationTransportSettings = |
156 | 177 | GrpcCallSettings.<CancelOperationRequest, Empty>newBuilder() |
157 | 178 | .setMethodDescriptor(cancelOperationMethodDescriptor) |
| 179 | + .setParamsExtractor( |
| 180 | + new RequestParamsExtractor<CancelOperationRequest>() { |
| 181 | + @Override |
| 182 | + public Map<String, String> extract(CancelOperationRequest request) { |
| 183 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 184 | + params.put("name", String.valueOf(request.getName())); |
| 185 | + return params.build(); |
| 186 | + } |
| 187 | + }) |
158 | 188 | .build(); |
159 | 189 | GrpcCallSettings<DeleteOperationRequest, Empty> deleteOperationTransportSettings = |
160 | 190 | GrpcCallSettings.<DeleteOperationRequest, Empty>newBuilder() |
161 | 191 | .setMethodDescriptor(deleteOperationMethodDescriptor) |
| 192 | + .setParamsExtractor( |
| 193 | + new RequestParamsExtractor<DeleteOperationRequest>() { |
| 194 | + @Override |
| 195 | + public Map<String, String> extract(DeleteOperationRequest request) { |
| 196 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 197 | + params.put("name", String.valueOf(request.getName())); |
| 198 | + return params.build(); |
| 199 | + } |
| 200 | + }) |
162 | 201 | .build(); |
163 | 202 |
|
164 | 203 | this.getOperationCallable = |
|
0 commit comments