EvergreenBenchmarkResultWriter.java

/*
 * Copyright 2008-present MongoDB, Inc.
 *
 * Licensed 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 com.mongodb.benchmark.framework;

import org.bson.json.JsonMode;
import org.bson.json.JsonWriter;
import org.bson.json.JsonWriterSettings;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.StringWriter;
import java.nio.charset.StandardCharsets;

public class EvergreenBenchmarkResultWriter implements BenchmarkResultWriter {

    private static final String OUTPUT_FILE_SYSTEM_PROPERTY = "org.mongodb.benchmarks.output";

    private final StringWriter writer;
    private final JsonWriter jsonWriter;

    public EvergreenBenchmarkResultWriter()  {
        writer = new StringWriter();
        jsonWriter = new JsonWriter(writer, JsonWriterSettings.builder().outputMode(JsonMode.RELAXED).indent(true).build());
        jsonWriter.writeStartDocument();
        jsonWriter.writeStartArray("results");
    }

    @Override
    public void write(final BenchmarkResult benchmarkResult) {
        jsonWriter.writeStartDocument();

        jsonWriter.writeStartDocument("info");
        jsonWriter.writeString("test_name", benchmarkResult.getName());

        jsonWriter.writeStartDocument("args");
        jsonWriter.writeInt32("threads", 1);
        jsonWriter.writeEndDocument();
        jsonWriter.writeEndDocument();

        jsonWriter.writeStartArray("metrics");

        jsonWriter.writeStartDocument();
        jsonWriter.writeString("name","ops_per_sec" );
        jsonWriter.writeDouble("value",
                (benchmarkResult.getBytesPerIteration() / 1000000d) /
                        (benchmarkResult.getElapsedTimeNanosAtPercentile(50) / 1000000000d));
        jsonWriter.writeEndDocument();

        jsonWriter.writeEndArray();
        jsonWriter.writeEndDocument();
    }

    @Override
    public void write(final MongocryptBecnhmarkResult result) {
        jsonWriter.writeStartDocument();

        jsonWriter.writeStartDocument("info");
        jsonWriter.writeString("test_name", result.getTestName());

        jsonWriter.writeStartDocument("args");
        jsonWriter.writeInt32("threads", result.getThreadCount());
        jsonWriter.writeEndDocument();
        jsonWriter.writeEndDocument();

        jsonWriter.writeString("created_at", result.getCreatedAt());
        jsonWriter.writeString("completed_at", result.getCompletedAt());
        jsonWriter.writeStartArray("metrics");

        jsonWriter.writeStartDocument();
        jsonWriter.writeString("name", result.getMetricName());
        jsonWriter.writeString("type", result.getMetricType());
        jsonWriter.writeDouble("value", result.getMedianOpsPerSec());
        jsonWriter.writeEndDocument();

        jsonWriter.writeEndArray();
        jsonWriter.writeEndDocument();
    }

    @Override
    public void close() throws IOException {
        jsonWriter.writeEndArray();
        jsonWriter.writeEndDocument();

        try (OutputStreamWriter fileWriter = new OutputStreamWriter(new FileOutputStream(System.getProperty(OUTPUT_FILE_SYSTEM_PROPERTY)),
                StandardCharsets.UTF_8)) {
            fileWriter.write(getJsonResultsArrayFromJsonDocument());
        }
    }

    private String getJsonResultsArrayFromJsonDocument() {
        String jsonDocument = writer.toString();
        int startArrayIndex = jsonDocument.indexOf('[');
        int endArrayIndex = jsonDocument.lastIndexOf(']');
        return writer.toString().substring(startArrayIndex, endArrayIndex + 1) + "\n";
    }
}