|
2 | 2 |
|
3 | 3 | import com.google.gson.Gson; |
4 | 4 | import com.google.gson.GsonBuilder; |
| 5 | +import com.google.gson.JsonObject; |
5 | 6 | import com.skyflow.VaultClient; |
6 | 7 | import com.skyflow.config.Credentials; |
7 | 8 | import com.skyflow.config.VaultConfig; |
8 | 9 | import com.skyflow.errors.SkyflowException; |
9 | 10 | import com.skyflow.generated.rest.core.ApiClientApiException; |
| 11 | +import com.skyflow.generated.rest.core.RequestOptions; |
10 | 12 | import com.skyflow.generated.rest.types.InsertRecordData; |
11 | 13 | import com.skyflow.generated.rest.types.InsertResponse; |
12 | 14 | import com.skyflow.generated.rest.types.Upsert; |
|
31 | 33 |
|
32 | 34 | public final class VaultController extends VaultClient { |
33 | 35 | private static final Gson gson = new GsonBuilder().serializeNulls().create(); |
| 36 | + private JsonObject metrics = Utils.getMetrics(); |
34 | 37 | private int insertBatchSize; |
35 | 38 | private int insertConcurrencyLimit; |
36 | 39 | private int detokenizeBatchSize; |
@@ -262,7 +265,10 @@ private List<CompletableFuture<DetokenizeResponse>> detokenizeBatchFutures(Execu |
262 | 265 | } |
263 | 266 |
|
264 | 267 | private com.skyflow.generated.rest.types.DetokenizeResponse processDetokenizeBatch(com.skyflow.generated.rest.resources.recordservice.requests.DetokenizeRequest batch) { |
265 | | - return this.getRecordsApi().detokenize(batch); |
| 268 | + RequestOptions requestOptions = RequestOptions.builder() |
| 269 | + .addHeader(Constants.SDK_METRICS_HEADER_KEY, metrics.toString()) |
| 270 | + .build(); |
| 271 | + return this.getRecordsApi().detokenize(batch, requestOptions); |
266 | 272 | } |
267 | 273 |
|
268 | 274 | private List<CompletableFuture<com.skyflow.vault.data.InsertResponse>> |
@@ -305,7 +311,10 @@ private InsertResponse insertBatch(List<InsertRecordData> batch, String tableNam |
305 | 311 | req.tableName(tableName); |
306 | 312 | } |
307 | 313 | com.skyflow.generated.rest.resources.recordservice.requests.InsertRequest request = req.build(); |
308 | | - return this.getRecordsApi().insert(request); |
| 314 | + RequestOptions requestOptions = RequestOptions.builder() |
| 315 | + .addHeader(Constants.SDK_METRICS_HEADER_KEY, metrics.toString()) |
| 316 | + .build(); |
| 317 | + return this.getRecordsApi().insert(request, requestOptions); |
309 | 318 | } |
310 | 319 |
|
311 | 320 | private void configureInsertConcurrencyAndBatchSize(int totalRequests) { |
|
0 commit comments