Adding raw byte[] variants

smallrye · Jan 21, 2025 · 96c14bd · 96c14bd
1 parent 4d44a72
commit 96c14bd
Showing 1 changed file with 65 additions and 1 deletion.
diff --git a/common/src/main/java/io/smallrye/config/common/utils/StringUtil.java b/common/src/main/java/io/smallrye/config/common/utils/StringUtil.java
@@ -54,6 +54,13 @@ private static char replacementOf(char c) {
         return (char) (((int) NON_ALPHANUMERIC_UNDERSCORE_REPLACEMENTS[c & 0xFF]) & 0xFF);
     }
 
+    private static byte rawReplacementOf(char c) {
+        if (c > 255) {
+            return '_';
+        }
+        return NON_ALPHANUMERIC_UNDERSCORE_REPLACEMENTS[c & 0xFF];
+    }
+
     private static final String[] NO_STRINGS = new String[0];
 
     private static final Pattern ITEM_PATTERN = Pattern.compile("(,+)|([^\\\\,]+)|\\\\(.)");
@@ -118,7 +125,21 @@ public static String[] split(String text) {
     }
 
     public static String replaceNonAlphanumericByUnderscores(final String name) {
-        return replaceNonAlphanumericByUnderscores(name, new StringBuilder(name.length()));
+        // size it accounting for worst case scenario
+        byte[] usAsciiResult = new byte[name.length() + 1];
+        int length = name.length();
+        // bogus value
+        char c = 0;
+        for (int i = 0; i < length; i++) {
+            c = name.charAt(i);
+            usAsciiResult[i] = rawReplacementOf(c);
+        }
+        if (c == '"') {
+            usAsciiResult[length] = '_';
+            return new String(usAsciiResult, 0, 0, usAsciiResult.length);
+        } else {
+            return new String(usAsciiResult, 0, 0, usAsciiResult.length - 1);
+        }
     }
 
     public static String replaceNonAlphanumericByUnderscores(final String name, final StringBuilder sb) {
@@ -135,6 +156,49 @@ public static String replaceNonAlphanumericByUnderscores(final String name, fina
         return sb.toString();
     }
 
+    public static final class ResizableByteArray {
+
+        private byte[] array;
+
+        public ResizableByteArray(int initialSize) {
+            this.array = new byte[initialSize];
+        }
+
+        public void set(int index, byte value) {
+            array[index] = value;
+        }
+
+        public void ensureCapacity(int capacity) {
+            if (array.length < capacity) {
+                byte[] newArray = new byte[capacity];
+                System.arraycopy(array, 0, newArray, 0, array.length);
+                array = newArray;
+            }
+        }
+
+        public String toUsAsciiString(int length) {
+            return new String(array, 0, 0, length);
+        }
+    }
+
+    public static String replaceNonAlphanumericByUnderscores(final String name, final ResizableByteArray sb) {
+        // size it accounting for worst case scenario
+        int length = name.length();
+        sb.ensureCapacity(length + 1);
+        // bogus value
+        char c = 0;
+        for (int i = 0; i < length; i++) {
+            c = name.charAt(i);
+            sb.set(i, rawReplacementOf(c));
+        }
+        if (c == '"') {
+            sb.set(length, (byte) '_');
+            return sb.toUsAsciiString(length + 1);
+        } else {
+            return sb.toUsAsciiString(length);
+        }
+    }
+
     public static String toLowerCaseAndDotted(final String name) {
         int length = name.length();