Drive JMH benchmarks via Apache Zeppelin.

zeppelin-jmh-interpreter/.gitignore

@@ -0,0 +1,10 @@
+.idea
+.run
+.gradle
+build
+.classpath
+.project
+.settings/
+.DS_Store
+out
+logs

zeppelin-jmh-interpreter/README.md

@@ -0,0 +1,48 @@
+# zeppelin-jmh-interpreter
+JMH interpreter for Apache Zeppelin.
+
+
+## Build
+
+```sh
+./gradlew build
+```
+
+## Deployment
+
+* Update `$ZEPPELIN_HOME/conf/zeppeln-site.xml`
+```xml
+<property>'
+  <name>zeppelin.interpreters</name>
+  <value>...,org.apache.jmh.JMHInterpreter</value>
+</property>
+```
+* Create `$ZEPPELIN_HOME/interpreter/jmh`
+* Copy interpreter jar in `$ZEPPELIN_HOME/interpreter/jmh`
+
+
+## Configuration
+
+TODO
+<table>
+  <tr><th>Parameter</th><th>Default value</th><th>Description</th></tr>
+</table>
+
+## How to use
+
+In Zeppelin, use `%jmh` in a paragraph.
+
+
+Examples:
+```javascript
+%jmh
+
+// Display a table
+db.zipcodes.find({ "city":"CHICAGO", "state": "IL" }).table()
+```
+
+## Examples
+
+
+
+

zeppelin-jmh-interpreter/build.gradle

@@ -0,0 +1,102 @@
+/*
+ * This file was generated by the Gradle 'init' task.
+ */
+
+plugins {
+    id 'java'
+    id 'maven-publish'
+    id 'com.github.johnrengelman.shadow' version '6.1.0'
+    id 'net.nemerosa.versioning' version '2.6.1'
+}
+
+apply plugin: 'com.github.johnrengelman.shadow'
+
+configurations.all {
+    resolutionStrategy {
+        cacheChangingModulesFor 0, 'seconds'
+    }
+}
+
+repositories {
+    mavenLocal()
+    maven {
+        url = uri('http://repository.apache.org/snapshots')
+    }
+
+    maven {
+        url = uri('https://repo.maven.apache.org/maven2/')
+    }
+    maven {
+        url 'https://repo.maven.apache.org/maven2'
+        name 'Maven Central'
+    }
+}
+
+dependencies {
+    implementation ('org.slf4j:slf4j-api:1.7.32')
+    implementation 'org.apache.commons:commons-exec:1.3'
+    implementation 'org.apache.commons:commons-lang3:3.12.0'
+    implementation 'commons-io:commons-io:2.5'
+    implementation 'commons-cli:commons-cli:20040117.000000'
+    implementation 'org.jline:jline-reader:3.20.0'
+    implementation 'org.apache.commons:commons-csv:1.8'
+    implementation 'net.openhft:chronicle-map:3.21ea82' // TODO: remove
+    implementation (group: 'org.apache.solr', name: 'solr-solrj', version: '8.9.0', {
+        exclude group: "io.netty", module: "*"
+        exclude group: "org.slf4j", module: "*"
+        exclude group: "org.eclipse.jetty", module: "*"
+    })
+    implementation ('org.eclipse.jetty:jetty-server:11.0.6'
+    , {
+        exclude group: "org.slf4j", module: "*"
+    })
+    implementation ('org.eclipse.jetty:jetty-servlet:11.0.6'
+    , {
+        exclude group: "org.slf4j", module: "*"
+    })
+    implementation ('org.eclipse.jetty:jetty-http:11.0.6'
+    , {
+        exclude group: "org.slf4j", module: "*"
+    })
+    implementation ('org.eclipse.jetty.http2:http2-common:11.0.6'
+    , {
+        exclude group: "org.slf4j", module: "*"
+    })
+
+   // compileOnly 'javax.servlet:javax.servlet-api:3.1.0'
+
+    testImplementation 'junit:junit:4.13.2'
+    compileOnly 'org.apache.zeppelin:zeppelin-interpreter:0.10.1-SNAPSHOT'
+    testImplementation 'org.apache.zeppelin:zeppelin-interpreter:0.10.1-SNAPSHOT'
+    testImplementation 'org.apache.zeppelin:zeppelin-common:0.10.1-SNAPSHOT'
+    testRuntime  'log4j:log4j:1.2.17'
+}
+
+
+group = 'jmh'
+version = '0.1.0'
+description = 'Zeppelin: JMH interpreter'
+java.targetCompatibility = JavaVersion.VERSION_11
+
+jar {
+    manifest {
+        attributes(
+                'Built-By'       : System.properties['user.name'],
+                'Build-Timestamp': new java.text.SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSSZ").format(new Date()),
+                'Build-Revision' : versioning.info.commit,
+                'Created-By'     : "Gradle ${gradle.gradleVersion}",
+                'Build-Jdk'      : "${System.properties['java.version']} (${System.properties['java.vendor']} ${System.properties['java.vm.version']})",
+                'Build-OS'       : "${System.properties['os.name']} ${System.properties['os.arch']} ${System.properties['os.version']}"
+        )
+    }
+}
+
+shadowJar {
+  outputs.upToDateWhen { false }
+  archiveBaseName = 'jmh'
+  relocate 'org.eclipse.jetty', 'shadow.jetty'
+}
+
+tasks.withType(JavaCompile) {
+    options.encoding = 'UTF-8'
+}

zeppelin-jmh-interpreter/buildSrc/src/main/java/org/apache/lucene/gradle/WrapperDownloader.java

@@ -0,0 +1,129 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.lucene.gradle;
+
+import java.io.IOException;
+import java.net.URL;
+import java.nio.channels.Channels;
+import java.nio.channels.FileChannel;
+import java.nio.channels.ReadableByteChannel;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.EnumSet;
+import java.util.Locale;
+
+import static java.nio.file.StandardCopyOption.REPLACE_EXISTING;
+import static java.nio.file.StandardOpenOption.APPEND;
+
+/**
+ * Standalone class that can be used to download a gradle-wrapper.jar
+ * <p>
+ * Has no dependencies outside of standard java libraries
+ */
+public class WrapperDownloader {
+  public static void main(String[] args) {
+    if (args.length != 1) {
+      System.err.println("Usage: java WrapperDownloader.java <destination>");
+      System.exit(1);
+    }
+
+    try {
+      new WrapperDownloader().run(Paths.get(args[0]));
+    } catch (Exception e) {
+      System.err.println("ERROR: " + e.getMessage());
+      System.exit(1);
+    }
+  }
+
+  public void run(Path destination) throws IOException, NoSuchAlgorithmException {
+    Path checksumPath = destination.resolveSibling(destination.getFileName().toString() + ".sha256");
+    if (!Files.exists(checksumPath)) {
+      throw new IOException("Checksum file not found: " + checksumPath);
+    }
+    String expectedChecksum = Files.readString(checksumPath, StandardCharsets.UTF_8).trim();
+
+    Path versionPath = destination.resolveSibling(destination.getFileName().toString() + ".version");
+    if (!Files.exists(versionPath)) {
+      throw new IOException("Wrapper version file not found: " + versionPath);
+    }
+    String wrapperVersion = Files.readString(versionPath, StandardCharsets.UTF_8).trim();
+
+    MessageDigest digest = MessageDigest.getInstance("SHA-256");
+
+    if (Files.exists(destination)) {
+      if (checksum(digest, destination).equalsIgnoreCase(expectedChecksum)) {
+        // File exists, checksum matches, good to go!
+        return;
+      } else {
+        System.err.println("Checksum mismatch, will attempt to re-download gradle-wrapper.jar");
+        System.out.println(destination);
+        Files.delete(destination);
+      }
+    }
+
+    URL url = new URL("https://github.com/gradle/gradle/raw/v" + wrapperVersion + "/gradle/wrapper/gradle-wrapper.jar");
+    System.err.println("Downloading gradle-wrapper.jar from " + url);
+
+    // As of v6.0.1 the wrapper is approximately 60K
+    // Can increase this if gradle wrapper ever goes beyond 500K, but keep a safety check
+    final int maxSize = 512 * 1024;
+
+    // Zero-copy save the jar to a temp file
+    Path temp = Files.createTempFile(destination.getParent(), ".gradle-wrapper", ".tmp");
+    try {
+      try (ReadableByteChannel in = Channels.newChannel(url.openStream());
+           FileChannel out = FileChannel.open(temp, EnumSet.of(APPEND))) {
+        out.transferFrom(in, 0, maxSize);
+      } catch (IOException e) {
+        throw new IOException("Could not download gradle-wrapper.jar (" + e.getMessage() + ").");
+      }
+
+      String checksum = checksum(digest, temp);
+      if (!checksum.equalsIgnoreCase(expectedChecksum)) {
+        throw new IOException(String.format(Locale.ROOT,
+                "Checksum mismatch on downloaded gradle-wrapper.jar (was: %s, expected: %s).",
+                checksum,
+                expectedChecksum));
+      }
+
+      Files.move(temp, destination, REPLACE_EXISTING);
+      temp = null;
+    } finally {
+      if (temp != null) {
+        Files.deleteIfExists(temp);
+      }
+    }
+  }
+
+  private String checksum(MessageDigest messageDigest, Path path) throws IOException {
+    try {
+      char[] hex = "0123456789abcdef".toCharArray();
+      byte[] digest = messageDigest.digest(Files.readAllBytes(path));
+      StringBuilder sb = new StringBuilder();
+      for (byte b : digest) {
+        sb.append(hex[(b >> 4) & 0xf]).append(hex[b & 0xf]);
+      }
+      return sb.toString();
+    } catch (IOException e) {
+      throw new IOException("Could not compute digest of file: " + path + " (" + e.getMessage() + ")");
+    }
+  }
+}

zeppelin-jmh-interpreter/gradle/wrapper/gradle-wrapper.jar.sha256

@@ -0,0 +1 @@
+e996d452d2645e70c01c11143ca2d3742734a28da2bf61f25c82bdc288c9e637

zeppelin-jmh-interpreter/gradle/wrapper/gradle-wrapper.jar.version

@@ -0,0 +1 @@
+6.8.3

zeppelin-jmh-interpreter/gradle/wrapper/gradle-wrapper.properties

@@ -0,0 +1,5 @@
+distributionBase=GRADLE_USER_HOME
+distributionPath=wrapper/dists
+distributionUrl=https\://services.gradle.org/distributions/gradle-6.8.3-bin.zip
+zipStoreBase=GRADLE_USER_HOME
+zipStorePath=wrapper/dists