wip

Author: agilelab-tmnd1991

server/core/src/main/java/io/whitefox/core/services/DeltaSharingCapabilities.java

@@ -0,0 +1,104 @@
+package io.whitefox.core.services;
+
+import java.util.Arrays;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
+
+public class DeltaSharingCapabilities {
+    private final Map<String, String> values;
+
+    public DeltaSharingCapabilities(Map<String, String> values) {
+        this.values = Map.copyOf(values);
+    }
+
+    public Map<String, String> values() {
+        return values;
+    }
+
+    public DeltaSharingResponseFormat getResponseFormat() {
+        var value = values.get(DELTA_SHARING_RESPONSE_FORMAT);
+        if (value == null) {
+            return DeltaSharingResponseFormat.PARQUET;
+        } else {
+            return DeltaSharingResponseFormat.valueOf(value.toUpperCase());
+        }
+    }
+
+    public Set<DeltaSharingReaderFeatures> getReaderFeatures() {
+        var value = values.get(DELTA_SHARING_READER_FEATURES);
+        if (value == null) {
+            return Set.of();
+        } else {
+            return Arrays.stream(value.split(","))
+                    .map(String::trim)
+                    .map(DeltaSharingReaderFeatures::fromString)
+                    .collect(Collectors.toSet());
+        }
+    }
+
+    public final static String DELTA_SHARING_RESPONSE_FORMAT = "responseformat";
+    public final static String DELTA_SHARING_READER_FEATURES = "readerfeatures";
+
+    public enum DeltaSharingResponseFormat {
+        DELTA, PARQUET
+    }
+
+    public enum DeltaSharingReaderFeatures {
+        DELETION_VECTORS("deletionvectors"),
+        COLUMN_MAPPING("columnmapping"),
+        TIMESTAMP_NTZ("timestampntz"),
+        DOMAIN_METADATA("domainmetadata"),
+        V2CHECKPOINT("v2checkpoint"),
+        CHECK_CONSTRAINTS("checkconstraints"),
+        GENERATED_COLUMNS("generatedcolumns"),
+        ALLOW_COLUMN_DEFAULTS("allowcolumndefaults"),
+        IDENTITY_COLUMNS("identitycolumns");
+
+        DeltaSharingReaderFeatures(String stringRepresentation) {
+            this.stringRepresentation = stringRepresentation;
+        }
+
+        private final String stringRepresentation;
+
+        public String stringRepresentation() {
+            return stringRepresentation;
+        }
+
+        public static DeltaSharingReaderFeatures fromString(String s) {
+            switch (s) {
+                case DELTA_SHARING_READER_FEATURE_DELETION_VECTOR:
+                    return DeltaSharingReaderFeatures.DELETION_VECTORS;
+                case DELTA_SHARING_READER_FEATURE_COLUMN_MAPPING:
+                    return DeltaSharingReaderFeatures.COLUMN_MAPPING;
+                case DELTA_SHARING_READER_FEATURE_TIMESTAMP_NTZ:
+                    return DeltaSharingReaderFeatures.TIMESTAMP_NTZ;
+                case DELTA_SHARING_READER_FEATURE_DOMAIN_METADATA:
+                    return DeltaSharingReaderFeatures.DOMAIN_METADATA;
+                case DELTA_SHARING_READER_FEATURE_V2CHECKPOINT:
+                    return DeltaSharingReaderFeatures.V2CHECKPOINT;
+                case DELTA_SHARING_READER_FEATURE_CHECK_CONSTRAINTS:
+                    return DeltaSharingReaderFeatures.CHECK_CONSTRAINTS;
+                case DELTA_SHARING_READER_FEATURE_GENERATED_COLUMNS:
+                    return DeltaSharingReaderFeatures.GENERATED_COLUMNS;
+                case DELTA_SHARING_READER_FEATURE_ALLOW_COLUMN_DEFAULTS:
+                    return DeltaSharingReaderFeatures.ALLOW_COLUMN_DEFAULTS;
+                case DELTA_SHARING_READER_FEATURE_IDENTITY_COLUMNS:
+                    return DeltaSharingReaderFeatures.IDENTITY_COLUMNS;
+                default:
+                    throw new IllegalArgumentException("Unknown reader feature: " + s);
+            }
+        }
+
+        private final static String DELTA_SHARING_READER_FEATURE_DELETION_VECTOR = "deletionvectors";
+        private final static String DELTA_SHARING_READER_FEATURE_COLUMN_MAPPING = "columnmapping";
+        private final static String DELTA_SHARING_READER_FEATURE_TIMESTAMP_NTZ = "timestampntz";
+        private final static String DELTA_SHARING_READER_FEATURE_DOMAIN_METADATA = "domainmetadata";
+        private final static String DELTA_SHARING_READER_FEATURE_V2CHECKPOINT = "v2checkpoint";
+        private final static String DELTA_SHARING_READER_FEATURE_CHECK_CONSTRAINTS = "checkconstraints";
+        private final static String DELTA_SHARING_READER_FEATURE_GENERATED_COLUMNS = "generatedcolumns";
+        private final static String DELTA_SHARING_READER_FEATURE_ALLOW_COLUMN_DEFAULTS = "allowcolumndefaults";
+        private final static String DELTA_SHARING_READER_FEATURE_IDENTITY_COLUMNS = "identitycolumns";
+
+    }
+}