-
Notifications
You must be signed in to change notification settings - Fork 41
/
QueryProfileExplainAnalyzeAggregation.java
88 lines (75 loc) · 3.52 KB
/
QueryProfileExplainAnalyzeAggregation.java
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
/*
* Copyright 2024 Google LLC
*
* 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.example.datastore.queryprofile;
// [START datastore_query_explain_analyze_aggregation]
import static com.google.cloud.datastore.aggregation.Aggregation.count;
import com.google.cloud.datastore.AggregationQuery;
import com.google.cloud.datastore.AggregationResult;
import com.google.cloud.datastore.AggregationResults;
import com.google.cloud.datastore.Datastore;
import com.google.cloud.datastore.DatastoreOptions;
import com.google.cloud.datastore.Entity;
import com.google.cloud.datastore.Query;
import com.google.cloud.datastore.StructuredQuery;
import com.google.cloud.datastore.models.ExecutionStats;
import com.google.cloud.datastore.models.ExplainMetrics;
import com.google.cloud.datastore.models.ExplainOptions;
import com.google.cloud.datastore.models.PlanSummary;
import com.google.common.collect.Iterables;
import java.util.List;
import java.util.Map;
public class QueryProfileExplainAnalyzeAggregation {
public static void invoke() throws Exception {
// Instantiates a client
Datastore datastore = DatastoreOptions.getDefaultInstance().getService();
// Build the query
StructuredQuery<Entity> query = Query.newEntityQueryBuilder().setKind("Task").build();
// Build the aggregation query
AggregationQuery aggregationQuery =
Query.newAggregationQueryBuilder()
.over(query)
.addAggregation(count().as("countVal"))
.build();
// Set the explain options with analyze = true to get back the query stats, plan info, and
// aggregation results
AggregationResults results =
datastore.runAggregation(
aggregationQuery, ExplainOptions.newBuilder().setAnalyze(true).build());
if (!results.getExplainMetrics().isPresent()) {
throw new Exception("No explain metrics returned");
}
ExplainMetrics explainMetrics = results.getExplainMetrics().get();
if (!explainMetrics.getExecutionStats().isPresent()) {
throw new Exception("No execution stats returned");
}
ExecutionStats executionStats = explainMetrics.getExecutionStats().get();
Map<String, Object> debugStats = executionStats.getDebugStats();
System.out.println("----- Debug Stats -----");
debugStats.forEach((key, val) -> System.out.println(key + ": " + val));
System.out.println("----------");
long resultsReturned = executionStats.getResultsReturned();
System.out.println("Results returned: " + resultsReturned);
// Get the plan summary
PlanSummary planSummary = explainMetrics.getPlanSummary();
List<Map<String, Object>> indexesUsed = planSummary.getIndexesUsed();
System.out.println("----- Indexes Used -----");
indexesUsed.forEach(map -> map.forEach((key, val) -> System.out.println(key + ": " + val)));
System.out.println("----- Aggregation Results -----");
AggregationResult result = Iterables.getOnlyElement(results);
System.out.println("Count: " + result.getLong("countVal"));
}
}
// [END datastore_query_explain_analyze_aggregation]