forked from graphql-java/graphql-java
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathBenchMark.java
More file actions
110 lines (92 loc) · 3.92 KB
/
BenchMark.java
File metadata and controls
110 lines (92 loc) · 3.92 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
109
110
package benchmark;
import graphql.GraphQL;
import graphql.execution.ExecutionStepInfo;
import graphql.execution.instrumentation.tracing.TracingInstrumentation;
import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;
import graphql.schema.GraphQLSchema;
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.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import static graphql.schema.idl.TypeRuntimeWiring.newTypeWiring;
/**
* See https://github.com/openjdk/jmh/tree/master/jmh-samples/src/main/java/org/openjdk/jmh/samples/ 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 : https://github.com/artyushov/idea-jmh-plugin
* <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;
static GraphQL graphQL = buildGraphQL();
@Benchmark
@BenchmarkMode(Mode.Throughput)
@OutputTimeUnit(TimeUnit.SECONDS)
public void benchMarkSimpleQueriesThroughput() {
executeQuery();
}
@Benchmark
@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MILLISECONDS)
public void benchMarkSimpleQueriesAvgTime() {
executeQuery();
}
public static void executeQuery() {
String query = "{ hero { name friends { name friends { name } } } }";
graphQL.execute(query);
}
private static GraphQL buildGraphQL() {
InputStream sdl = BenchMark.class.getClassLoader().getResourceAsStream("starWarsSchema.graphqls");
TypeDefinitionRegistry definitionRegistry = new SchemaParser().parse(new InputStreamReader(sdl));
DataFetcher heroDataFetcher = environment -> CharacterDTO.mkCharacter(environment, "r2d2", NUMBER_OF_FRIENDS);
RuntimeWiring runtimeWiring = RuntimeWiring.newRuntimeWiring()
.type(
newTypeWiring("QueryType").dataFetcher("hero", heroDataFetcher))
.type(newTypeWiring("Character").typeResolver(
env -> env.getSchema().getObjectType("Human")
))
.build();
GraphQLSchema graphQLSchema = new SchemaGenerator().makeExecutableSchema(definitionRegistry, runtimeWiring);
return GraphQL.newGraphQL(graphQLSchema)
.instrumentation(new TracingInstrumentation())
.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");
ExecutionStepInfo anotherSideEffect = environment.getExecutionStepInfo();
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);
}
}
}