Add benchmarks for base64 JDK vs apache commons

Author: dkomanov

WORKSPACE

@@ -148,6 +148,15 @@ maven_install(
     repositories = maven_repositories,
 )
 
+maven_install(
+    name = "jwt",
+    artifacts = [
+        "commons-codec:commons-codec:1.15",
+    ],
+    fetch_sources = True,
+    repositories = maven_repositories,
+)
+
 maven_install(
     name = "scala_serialization_maven",
     artifacts = [

src/com/komanov/jwt/base64/BUILD

@@ -0,0 +1,10 @@
+java_library(
+    name = "base64",
+    srcs = ["Base64Helper.java"],
+    visibility = [
+        "//src/com/komanov/jwt/base64:__subpackages__",
+    ],
+    deps = [
+        "@jwt//:commons_codec_commons_codec",
+    ],
+)

src/com/komanov/jwt/base64/Base64Helper.java

@@ -0,0 +1,59 @@
+package com.komanov.jwt.base64;
+
+import static org.apache.commons.codec.binary.BaseNCodec.MIME_CHUNK_SIZE;
+
+public abstract class Base64Helper {
+    private Base64Helper() {
+    }
+
+    public static final class Jdk {
+        private Jdk() {
+        }
+
+        private static final java.util.Base64.Decoder regularDecoder = java.util.Base64.getDecoder();
+        private static final java.util.Base64.Decoder urlSafeDecoder = java.util.Base64.getUrlDecoder();
+
+        private static final java.util.Base64.Encoder regularEncoder = java.util.Base64.getEncoder();
+        private static final java.util.Base64.Encoder urlSafeEncoder = java.util.Base64.getUrlEncoder().withoutPadding();
+
+        public static byte[] encode(byte[] data) {
+            return regularEncoder.encode(data);
+        }
+
+        public static byte[] encodeUrlSafe(byte[] data) {
+            return urlSafeEncoder.encode(data);
+        }
+
+        public static byte[] decode(byte[] data) {
+            return regularDecoder.decode(data);
+        }
+
+        public static byte[] decodeUrlSafe(byte[] data) {
+            return urlSafeDecoder.decode(data);
+        }
+    }
+
+    public static class Commons {
+        private Commons() {
+        }
+
+        private static final org.apache.commons.codec.binary.Base64 regular = new org.apache.commons.codec.binary.Base64(MIME_CHUNK_SIZE, null, false);
+        private static final org.apache.commons.codec.binary.Base64 urlSafe = new org.apache.commons.codec.binary.Base64(MIME_CHUNK_SIZE, null, true);
+
+        public static byte[] encode(byte[] data) {
+            return regular.encode(data);
+        }
+
+        public static byte[] encodeUrlSafe(byte[] data) {
+            return urlSafe.encode(data);
+        }
+
+        public static byte[] decode(byte[] data) {
+            return regular.decode(data);
+        }
+
+        public static byte[] decodeUrlSafe(byte[] data) {
+            return urlSafe.decode(data);
+        }
+    }
+}

src/com/komanov/jwt/base64/jmh/BUILD

@@ -0,0 +1,7 @@
+scala_benchmark_jmh(
+    name = "jmh",
+    srcs = glob(["*.scala"]),
+    deps = [
+        "//src/com/komanov/jwt/base64",
+    ],
+)

src/com/komanov/jwt/base64/jmh/Base64Benchmarks.scala

@@ -0,0 +1,7 @@
+scala_specs2_junit(
+    name = "tests",
+    srcs = glob(["*.scala"]),
+    deps = [
+        "//src/com/komanov/jwt/base64",
+    ],
+)

src/com/komanov/jwt/base64/tests/BUILD

@@ -0,0 +1,56 @@
+package com.komanov.jwt.base64.tests
+
+import com.komanov.jwt.base64.Base64Helper.Jdk
+import com.komanov.jwt.base64.Base64Helper.Commons
+import org.specs2.mutable.SpecificationWithJUnit
+import org.specs2.specification.core.Fragments
+
+import java.nio.charset.StandardCharsets
+import scala.util.Random
+
+class Base64HelperTest extends SpecificationWithJUnit {
+  Fragments.foreach(Seq(
+    "" -> "",
+    "a" -> "YQ==",
+    "abcd" -> "YWJjZA==",
+    "nmmфывыаывпврапопррло" -> "bm1t0YTRi9Cy0YvQsNGL0LLQv9Cy0YDQsNC/0L7Qv9GA0YDQu9C+",
+  )) { case (data, expected) =>
+    s"(jdk) regular encode-decode for input: $data" >> {
+      val encoded = Jdk.encode(data.getBytes(StandardCharsets.UTF_8))
+      str(encoded) mustEqual expected
+      val bytes = Jdk.decode(encoded)
+      str(bytes) mustEqual data
+    }
+
+    s"(commons) regular encode-decode for input: $data" >> {
+      val encoded = Commons.encode(data.getBytes(StandardCharsets.UTF_8))
+      str(encoded) mustEqual expected
+      val bytes = Commons.decode(encoded)
+      str(bytes) mustEqual data
+    }
+  }
+
+  Fragments.foreach(Seq(
+    "" -> "",
+    "a" -> "YQ",
+    "abcd" -> "YWJjZA",
+    "nmmфывыаывпврапопррло" -> "bm1t0YTRi9Cy0YvQsNGL0LLQv9Cy0YDQsNC_0L7Qv9GA0YDQu9C-",
+  )) { case (data, expected) =>
+    s"(jdk) regular encode-decode for input: $data" >> {
+      val encoded = Jdk.encodeUrlSafe(data.getBytes(StandardCharsets.UTF_8))
+      str(encoded) mustEqual expected
+      val bytes = Jdk.decodeUrlSafe(encoded)
+      str(bytes) mustEqual data
+    }
+
+    s"(commons) regular encode-decode for input: $data" >> {
+      val encoded = Commons.encodeUrlSafe(data.getBytes(StandardCharsets.UTF_8))
+      str(encoded) mustEqual expected
+      val bytes = Commons.decodeUrlSafe(encoded)
+      str(bytes) mustEqual data
+    }
+  }
+
+  private def str(bytes: Array[Byte]): String =
+    new String(bytes, StandardCharsets.UTF_8)
+}