Skip to content

Commit 9bce8f2

Browse files
Merge pull request #259 from skyflowapi/release/25.11.2
SK-2370/Release/25.11.2
2 parents 7c5c85e + 712a030 commit 9bce8f2

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

v3/src/main/java/com/skyflow/vault/controller/VaultController.java

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,13 @@
22

33
import com.google.gson.Gson;
44
import com.google.gson.GsonBuilder;
5+
import com.google.gson.JsonObject;
56
import com.skyflow.VaultClient;
67
import com.skyflow.config.Credentials;
78
import com.skyflow.config.VaultConfig;
89
import com.skyflow.errors.SkyflowException;
910
import com.skyflow.generated.rest.core.ApiClientApiException;
11+
import com.skyflow.generated.rest.core.RequestOptions;
1012
import com.skyflow.generated.rest.types.InsertRecordData;
1113
import com.skyflow.generated.rest.types.InsertResponse;
1214
import com.skyflow.generated.rest.types.Upsert;
@@ -31,6 +33,7 @@
3133

3234
public final class VaultController extends VaultClient {
3335
private static final Gson gson = new GsonBuilder().serializeNulls().create();
36+
private JsonObject metrics = Utils.getMetrics();
3437
private int insertBatchSize;
3538
private int insertConcurrencyLimit;
3639
private int detokenizeBatchSize;
@@ -262,7 +265,10 @@ private List<CompletableFuture<DetokenizeResponse>> detokenizeBatchFutures(Execu
262265
}
263266

264267
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);
266272
}
267273

268274
private List<CompletableFuture<com.skyflow.vault.data.InsertResponse>>
@@ -305,7 +311,10 @@ private InsertResponse insertBatch(List<InsertRecordData> batch, String tableNam
305311
req.tableName(tableName);
306312
}
307313
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);
309318
}
310319

311320
private void configureInsertConcurrencyAndBatchSize(int totalRequests) {

0 commit comments

Comments
 (0)