package benchmark;

import com.google.common.base.Charsets;

import com.google.common.io.Resources;

import graphql.language.Document;

import graphql.normalized.NormalizedQueryTree;

import graphql.normalized.NormalizedQueryTreeFactory;

import graphql.parser.Parser;

import graphql.schema.GraphQLSchema;

import graphql.schema.idl.SchemaGenerator;

import org.openjdk.jmh.annotations.Benchmark;

import org.openjdk.jmh.annotations.BenchmarkMode;

import org.openjdk.jmh.annotations.Fork;

import org.openjdk.jmh.annotations.Measurement;

import org.openjdk.jmh.annotations.Mode;

import org.openjdk.jmh.annotations.OutputTimeUnit;

import org.openjdk.jmh.annotations.Scope;

import org.openjdk.jmh.annotations.Setup;

import org.openjdk.jmh.annotations.State;

import org.openjdk.jmh.annotations.Threads;

import org.openjdk.jmh.annotations.Warmup;

import java.io.IOException;

import java.net.URL;

import java.util.Collections;

import java.util.concurrent.ExecutionException;

import java.util.concurrent.TimeUnit;

import static com.google.common.io.Resources.getResource;

@State(Scope.Benchmark)

@BenchmarkMode(Mode.Throughput)

@Warmup(iterations = 2)

@Measurement(iterations = 2, timeUnit = TimeUnit.NANOSECONDS)

public class NQBenchmark2 {

@State(Scope.Benchmark)

public static class MyState {

GraphQLSchema schema;

Document document;

@Setup

public void setup() {

try {

String schemaString = readFromClasspath("large-schema-2.graphqls");

schema = SchemaGenerator.createdMockedSchema(schemaString);

String query = readFromClasspath("large-schema-2-query.graphql");

document = Parser.parse(query);

} catch (Exception e) {

System.out.println(e);

throw new RuntimeException(e);

}

}

private String readFromClasspath(String file) throws IOException {

URL url = getResource(file);

return Resources.toString(url, Charsets.UTF_8);

}

}

@Benchmark

@Warmup(iterations = 2)

@Measurement(iterations = 100, time = 10)

@Threads(1)

@Fork(3)

@BenchmarkMode(Mode.AverageTime)

@OutputTimeUnit(TimeUnit.MILLISECONDS)

public NormalizedQueryTree benchMarkAvgTime(MyState myState) throws ExecutionException, InterruptedException {

NormalizedQueryTree normalizedQuery = NormalizedQueryTreeFactory.createNormalizedQuery(myState.schema, myState.document, null, Collections.emptyMap());

System.out.println("fields size:" + normalizedQuery.getFieldToNormalizedField().size());

return normalizedQuery;

}

}

query operation {...Fragment1 ...Fragment2 ...Fragment3} fragment Fragment3 on Object27 {field91 {field29 field30 field27 field1} field57 {field58 field59 field1}} fragment Fragment2 on Object27 {field91 {field29 field30 field27 field1} field57 {field58 field59 field1}} fragment Fragment1 on Object27 {field91 {field27 field1} field57 {field58 field59 field1}}