SK-2274 add retry tokens fn

v3/src/main/java/com/skyflow/utils/validations/Validations.java

@@ -81,7 +81,7 @@ public static void validateDetokenizeRequest(DetokenizeRequest request) throws S
             ));
             throw new SkyflowException(ErrorCode.INVALID_INPUT.getCode(), ErrorMessage.DetokenizeRequestNull.getMessage());
         }
-        ArrayList<String> tokens = request.getTokens();
+        List<String> tokens = request.getTokens();
         if (tokens == null || tokens.isEmpty()) {
             LogUtil.printErrorLog(Utils.parameterizedString(
                     ErrorLogs.EMPTY_DETOKENIZE_DATA.getLog(), InterfaceName.DETOKENIZE.getName()
@@ -96,7 +96,7 @@ public static void validateDetokenizeRequest(DetokenizeRequest request) throws S
                 throw new SkyflowException(ErrorCode.INVALID_INPUT.getCode(), ErrorMessage.EmptyTokenInDetokenizeData.getMessage());
             }
         }
-        ArrayList<TokenGroupRedactions> groupRedactions = request.getTokenGroupRedactions();
+        List<TokenGroupRedactions> groupRedactions = request.getTokenGroupRedactions();
         if (groupRedactions != null && !groupRedactions.isEmpty()) {
             for (TokenGroupRedactions group : groupRedactions) {
                 if (group == null) {

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

@@ -196,7 +196,7 @@ private com.skyflow.vault.data.InsertResponse processSync(
 
     private DetokenizeResponse processDetokenizeSync(
             com.skyflow.generated.rest.resources.recordservice.requests.DetokenizeRequest detokenizeRequest,
-            ArrayList<String> originalTokens
+            List<String> originalTokens
     ) throws ExecutionException, InterruptedException, SkyflowException {
         LogUtil.printInfoLog(InfoLogs.PROCESSING_BATCHES.getLog());
         List<ErrorRecord> errorTokens = Collections.synchronizedList(new ArrayList<>());

v3/src/main/java/com/skyflow/vault/data/DetokenizeRequest.java

@@ -1,6 +1,7 @@
 package com.skyflow.vault.data;
 
 import java.util.ArrayList;
+import java.util.List;
 
 public class DetokenizeRequest {
     private final DetokenizeRequestBuilder builder;
@@ -11,23 +12,23 @@ private DetokenizeRequest(DetokenizeRequestBuilder builder){
     public static DetokenizeRequestBuilder builder(){
         return new DetokenizeRequestBuilder();
     }
-    public ArrayList<String> getTokens(){
+    public List<String> getTokens(){
         return this.builder.tokens;
     }
-    public ArrayList<TokenGroupRedactions> getTokenGroupRedactions(){
+    public List<TokenGroupRedactions> getTokenGroupRedactions(){
         return this.builder.tokenGroupRedactions;
     }
 
     public static final class DetokenizeRequestBuilder{
-        private ArrayList<String> tokens;
+        private List<String> tokens;
 
-        private ArrayList<TokenGroupRedactions> tokenGroupRedactions;
+        private List<TokenGroupRedactions> tokenGroupRedactions;
 
-        public DetokenizeRequestBuilder tokens(ArrayList<String> tokens){
+        public DetokenizeRequestBuilder tokens(List<String> tokens){
             this.tokens = tokens;
             return this;
         }
-        public DetokenizeRequestBuilder tokenGroupRedactions(ArrayList<TokenGroupRedactions> tokenGroupRedactions){
+        public DetokenizeRequestBuilder tokenGroupRedactions(List<TokenGroupRedactions> tokenGroupRedactions){
             this.tokenGroupRedactions = tokenGroupRedactions;
             return this;
         }

v3/src/main/java/com/skyflow/vault/data/DetokenizeResponse.java

@@ -7,6 +7,7 @@
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
+import java.util.stream.Collectors;
 
 public class DetokenizeResponse {
     @Expose(serialize = true)
@@ -18,8 +19,7 @@ public class DetokenizeResponse {
     private List<ErrorRecord> errors;
 
     private List<String> originalPayload;
-    private ArrayList<HashMap<String, Object>> recordsToRetry;
-
+    private List<String> recordsToRetry;
 
     public DetokenizeResponse(List<DetokenizeResponseObject> success, List<ErrorRecord> errors) {
         this.success = success;
@@ -34,6 +34,17 @@ public DetokenizeResponse(List<DetokenizeResponseObject> success, List<ErrorReco
         this.summary = new DetokenizeSummary(this.originalPayload.size(), this.success.size(), this.errors.size());
     }
 
+    public List<String> getTokensToRetry() {
+        if (recordsToRetry == null) {
+            recordsToRetry = new ArrayList<>();
+            recordsToRetry = errors.stream()
+                    .filter(error -> (error.getCode() >= 500 && error.getCode() <= 599) && error.getCode() != 529)
+                    .map(errorRecord -> originalPayload.get(errorRecord.getIndex()))
+                    .collect(Collectors.toList());
+        }
+        return recordsToRetry;
+    }
+
     public List<DetokenizeResponseObject> getSuccess() {
         return success;
     }