CharBuffer API: indexOfAny and lastIndexOfAny

Author: maxim5

src/main/java/io/routekit/util/CharBuffer.java

@@ -192,6 +192,25 @@ public int indexOf(char ch, int from, int def) {
         return def;
     }
 
+    public int indexOfAny(char ch1, char ch2) {
+        return indexOfAny(ch1, ch2, 0, -1);
+    }
+
+    public int indexOfAny(char ch1, char ch2, int from) {
+        return indexOfAny(ch1, ch2, from, -1);
+    }
+
+    public int indexOfAny(char ch1, char ch2, int from, int def) {
+        assert from >= 0 : "From index can't be negative: %d".formatted(from);
+        assert def < 0 || def >= length() : "Default index can't be within buffer bounds: %d".formatted(def);
+        for (int i = start + from; i < end; ++i) {
+            if (chars[i] == ch1 || chars[i] == ch2) {
+                return i - start;
+            }
+        }
+        return def;
+    }
+
     public int lastIndexOf(char ch) {
         return lastIndexOf(ch, end - 1, -1);
     }
@@ -211,6 +230,25 @@ public int lastIndexOf(char ch, int from, int def) {
         return def;
     }
 
+    public int lastIndexOfAny(char ch1, char ch2) {
+        return lastIndexOfAny(ch1, ch2, end - 1, -1);
+    }
+
+    public int lastIndexOfAny(char ch1, char ch2, int from) {
+        return lastIndexOfAny(ch1, ch2, from, -1);
+    }
+
+    public int lastIndexOfAny(char ch1, char ch2, int from, int def) {
+        assert from >= 0 : "From index can't be negative: %d".formatted(from);
+        assert def < 0 || def >= length() : "Default index can't be within buffer bounds: %d".formatted(def);
+        for (int i = from; i >= start; --i) {
+            if (chars[i] == ch1 || chars[i] == ch2) {
+                return i - start;
+            }
+        }
+        return def;
+    }
+
     // Returns the length of the common prefix
     public int commonPrefix(CharBuffer buf) {
         int index = Arrays.mismatch(chars, start, end, buf.chars, buf.start, buf.end);

src/test/java/io/routekit/util/CharBufferTest.java

@@ -91,6 +91,34 @@ public void indexOf() {
         Assertions.assertEquals(11, buffer.indexOf('-', 8, buffer.length()));
     }
 
+    @Test
+    public void indexOfAny() {
+        CharBuffer buffer = new CharBuffer("foo-bar-baz");
+
+        Assertions.assertEquals(0, buffer.indexOfAny('f', 'o'));
+        Assertions.assertEquals(1, buffer.indexOfAny('o', 'o'));
+        Assertions.assertEquals(3, buffer.indexOfAny('a', '-'));
+        Assertions.assertEquals(5, buffer.indexOfAny('a', 'z'));
+        Assertions.assertEquals(10, buffer.indexOfAny('z', 'w'));
+        Assertions.assertEquals(-1, buffer.indexOfAny('x', 'y'));
+
+        Assertions.assertEquals(-1, buffer.indexOfAny('f', 'g', 1));
+        Assertions.assertEquals(-2, buffer.indexOfAny('f', 'g', 1, -2));
+        Assertions.assertEquals(11, buffer.indexOfAny('f', 'g', 1, buffer.length()));
+        Assertions.assertEquals(1, buffer.indexOfAny('o', 'a', 1));
+        Assertions.assertEquals(1, buffer.indexOfAny('o', 'a', 1, -2));
+        Assertions.assertEquals(3, buffer.indexOfAny('-', 'a', 3));
+        Assertions.assertEquals(3, buffer.indexOfAny('-', 'a', 3, buffer.length()));
+        Assertions.assertEquals(7, buffer.indexOfAny('z', '-', 4));
+        Assertions.assertEquals(7, buffer.indexOfAny('z', '-', 4, buffer.length()));
+        Assertions.assertEquals(7, buffer.indexOfAny('z', '-', 4, -2));
+        Assertions.assertEquals(7, buffer.indexOfAny('z', '-', 7, buffer.length()));
+        Assertions.assertEquals(9, buffer.indexOfAny('a', '-', 8, buffer.length()));
+        Assertions.assertEquals(10, buffer.indexOfAny('z', '-', 8, buffer.length()));
+        Assertions.assertEquals(11, buffer.indexOfAny('z', '-', 11, buffer.length()));
+        Assertions.assertEquals(11, buffer.indexOfAny('o', '-', 8, buffer.length()));
+    }
+
     @Test
     public void lastIndexOf() {
         CharBuffer buffer = new CharBuffer("foo-bar-baz");
@@ -112,6 +140,32 @@ public void lastIndexOf() {
         Assertions.assertEquals(-2, buffer.lastIndexOf('a', 4, -2));
     }
 
+    @Test
+    public void lastIndexOfAny() {
+        CharBuffer buffer = new CharBuffer("foo-bar-baz");
+
+        Assertions.assertEquals(0, buffer.lastIndexOfAny('f', 'g'));
+        Assertions.assertEquals(2, buffer.lastIndexOfAny('f', 'o'));
+        Assertions.assertEquals(2, buffer.lastIndexOfAny('o', 'f'));
+        Assertions.assertEquals(2, buffer.lastIndexOfAny('o', 'x'));
+        Assertions.assertEquals(7, buffer.lastIndexOfAny('-', 'o'));
+        Assertions.assertEquals(9, buffer.lastIndexOfAny('-', 'a'));
+        Assertions.assertEquals(10, buffer.lastIndexOfAny('-', 'z'));
+        Assertions.assertEquals(-1, buffer.lastIndexOfAny('x', 'y'));
+
+        Assertions.assertEquals(0, buffer.lastIndexOfAny('f', 'g', 10));
+        Assertions.assertEquals(0, buffer.lastIndexOfAny('f', 'g', 10, -1));
+        Assertions.assertEquals(10, buffer.lastIndexOfAny('f', 'z', 10));
+        Assertions.assertEquals(10, buffer.lastIndexOfAny('f', 'z', 10, -1));
+        Assertions.assertEquals(10, buffer.lastIndexOfAny('a', 'z', 10));
+        Assertions.assertEquals(9, buffer.lastIndexOfAny('a', 'r', 10));
+        Assertions.assertEquals(9, buffer.lastIndexOfAny('a', 'z', 9));
+        Assertions.assertEquals(5, buffer.lastIndexOfAny('a', 'z', 8));
+        Assertions.assertEquals(5, buffer.lastIndexOfAny('a', 'r', 5));
+        Assertions.assertEquals(-1, buffer.lastIndexOfAny('a', 'r', 4));
+        Assertions.assertEquals(-2, buffer.lastIndexOfAny('a', 'r', 4, -2));
+    }
+
     @Test
     public void commonPrefix() {
         Assertions.assertEquals(0, new CharBuffer("foo").commonPrefix(new CharBuffer("bar")));