-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Expand file tree
/
Copy pathBenchMark.java
More file actions
108 lines (90 loc) · 3.95 KB
/
BenchMark.java
File metadata and controls
108 lines (90 loc) · 3.95 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package benchmark;
import graphql.Assert;
import graphql.ExecutionResult;
import graphql.GraphQL;
import graphql.execution.ExecutionStepInfo;
import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;
import graphql.schema.GraphQLSchema;
import graphql.schema.TypeResolver;
import graphql.schema.idl.RuntimeWiring;
import graphql.schema.idl.SchemaGenerator;
import graphql.schema.idl.SchemaParser;
import graphql.schema.idl.TypeDefinitionRegistry;
import org.openjdk.jmh.annotations.Benchmark;
import org.openjdk.jmh.annotations.BenchmarkMode;
import org.openjdk.jmh.annotations.Measurement;
import org.openjdk.jmh.annotations.Mode;
import org.openjdk.jmh.annotations.OutputTimeUnit;
import org.openjdk.jmh.annotations.Warmup;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import static graphql.schema.idl.TypeRuntimeWiring.newTypeWiring;
/**
* See <a href="https://github.com/openjdk/jmh/tree/master/jmh-samples/src/main/java/org/openjdk/jmh/samples/">this link</a> for more samples
* on what you can do with JMH.
* <p>
* You MUST have the JMH plugin for IDEA in place for this to work : <a href="https://github.com/artyushov/idea-jmh-plugin">idea-jmh-plugin</a>
* <p>
* Install it and then just hit "Run" on a certain benchmark method
*/
@Warmup(iterations = 2, time = 5, batchSize = 3)
@Measurement(iterations = 3, time = 10, batchSize = 4)
public class BenchMark {
private static final int NUMBER_OF_FRIENDS = 10 * 100;
private static final GraphQL GRAPHQL = buildGraphQL();
@Benchmark
@BenchmarkMode(Mode.Throughput)
@OutputTimeUnit(TimeUnit.SECONDS)
public ExecutionResult benchMarkSimpleQueriesThroughput() {
return executeQuery();
}
@Benchmark
@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MILLISECONDS)
public ExecutionResult benchMarkSimpleQueriesAvgTime() {
return executeQuery();
}
public static ExecutionResult executeQuery() {
String query = "{ hero { name friends { name friends { name } } } }";
return GRAPHQL.execute(query);
}
private static GraphQL buildGraphQL() {
TypeDefinitionRegistry definitionRegistry = new SchemaParser().parse(BenchmarkUtils.loadResource("starWarsSchema.graphqls"));
DataFetcher<CharacterDTO> heroDataFetcher = environment -> CharacterDTO.mkCharacter(environment, "r2d2", NUMBER_OF_FRIENDS);
TypeResolver typeResolver = env -> env.getSchema().getObjectType("Human");
RuntimeWiring runtimeWiring = RuntimeWiring.newRuntimeWiring()
.type(newTypeWiring("QueryType").dataFetcher("hero", heroDataFetcher))
.type(newTypeWiring("Character").typeResolver(typeResolver))
.build();
GraphQLSchema graphQLSchema = new SchemaGenerator().makeExecutableSchema(definitionRegistry, runtimeWiring);
return GraphQL.newGraphQL(graphQLSchema)
.build();
}
static class CharacterDTO {
private final String name;
private final List<CharacterDTO> friends;
CharacterDTO(String name, List<CharacterDTO> friends) {
this.name = name;
this.friends = friends;
}
public String getName() {
return name;
}
public List<CharacterDTO> getFriends() {
return friends;
}
public static CharacterDTO mkCharacter(DataFetchingEnvironment environment, String name, int friendCount) {
Object sideEffect = environment.getArgument("episode");
Assert.assertNull(sideEffect);
ExecutionStepInfo anotherSideEffect = environment.getExecutionStepInfo();
Assert.assertNotNull(anotherSideEffect);
List<CharacterDTO> friends = new ArrayList<>(friendCount);
for (int i = 0; i < friendCount; i++) {
friends.add(mkCharacter(environment, "friend" + i, 0));
}
return new CharacterDTO(name, friends);
}
}
}