Skip to content

Commit

Permalink
improve query latency, add topN filter (#18)
Browse files Browse the repository at this point in the history
* improve queries execution latency
We swap bool with filter and remove the order->Count aggregation

* add topn counter to query to elasticsearch
Signed-off-by: inge4pres <[email protected]>
  • Loading branch information
inge4pres authored and rockdaboot committed Jul 4, 2022
1 parent 50a42f4 commit abc5b26
Show file tree
Hide file tree
Showing 5 changed files with 55 additions and 60 deletions.
2 changes: 1 addition & 1 deletion src/plugins/profiling/public/app.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ function App({ fetchTopN, fetchElasticFlamechart, fetchPixiFlamechart }: Props)
<EuiSpacer />
<FlameGraphContext.Provider value={pixiFlamegraph}>
<FlameGraphNavigation getter={fetchPixiFlamechart} setter={setPixiFlamegraph} />
<PixiFlamechart projectID={5} />
<PixiFlamechart projectID={'5'} />
</FlameGraphContext.Provider>
</>
),
Expand Down
22 changes: 6 additions & 16 deletions src/plugins/profiling/public/services.ts
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,10 @@ function getFetchQuery(seconds: string): HttpFetchQuery {
return {
index: 'profiling-events',
projectID: 5,
timeFrom: unixTime - parseInt(seconds),
timeFrom: unixTime - parseInt(seconds, 10),
timeTo: unixTime,
// TODO remove hard-coded value for topN items length and expose it through the UI
n: 100,
} as HttpFetchQuery;
}

Expand All @@ -33,11 +35,7 @@ export function getServices(core: CoreStart): Services {
fetchTopN: async (type: string, seconds: string) => {
try {
const query = getFetchQuery(seconds);
const response = await core.http.get(
`${paths.TopN}/${type}`,
{ query }
);
return response;
return await core.http.get(`${paths.TopN}/${type}`, { query });
} catch (e) {
return e;
}
Expand All @@ -46,11 +44,7 @@ export function getServices(core: CoreStart): Services {
fetchElasticFlamechart: async (seconds: string) => {
try {
const query = getFetchQuery(seconds);
const response = await core.http.get(
paths.FlamechartElastic,
{ query }
);
return response;
return await core.http.get(paths.FlamechartElastic, { query });
} catch (e) {
return e;
}
Expand All @@ -59,11 +53,7 @@ export function getServices(core: CoreStart): Services {
fetchPixiFlamechart: async (seconds: string) => {
try {
const query = getFetchQuery(seconds);
const response = await core.http.get(
paths.FlamechartPixi,
{ query }
);
return response;
return await core.http.get(paths.FlamechartPixi, { query });
} catch (e) {
return e;
}
Expand Down
28 changes: 12 additions & 16 deletions src/plugins/profiling/server/routes/mappings.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,10 @@ import { AggregationsAggregationContainer } from '@elastic/elasticsearch/lib/api

interface ProjectTimeQuery {
bool: {
must: Array<
filter: Array<
| {
term: {
ProjectID: {
value: string;
boost: number;
};
ProjectID: string;
};
}
| {
Expand All @@ -40,13 +37,10 @@ export function newProjectTimeQuery(
): ProjectTimeQuery {
return {
bool: {
must: [
filter: [
{
term: {
ProjectID: {
value: projectID,
boost: 1.0,
},
ProjectID: projectID,
},
},
{
Expand All @@ -65,21 +59,23 @@ export function newProjectTimeQuery(
}

export function autoHistogramSumCountOnGroupByField(
searchField: string
searchField: string,
topNItems: number
): AggregationsAggregationContainer {
return {
auto_date_histogram: {
field: '@timestamp',
buckets: 100,
buckets: 50,
},
aggs: {
group_by: {
terms: {
field: searchField,
order: {
Count: 'desc',
},
size: 100,
// We remove the ordering since we will rely directly on the natural
// ordering of Elasticsearch: by default this will be the descending count
// of matched documents. This is not equal to the ordering by sum of Count field,
// but it's a good-enough approximation given the distribution of Count.
size: topNItems,
},
aggs: {
Count: {
Expand Down
24 changes: 15 additions & 9 deletions src/plugins/profiling/server/routes/search_topn.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,13 @@
import { topNElasticSearchQuery } from './search_topn';
import { DataRequestHandlerContext } from '../../../data/server';
import { kibanaResponseFactory } from '../../../../core/server';
import { AggregationsAggregationContainer } from '@elastic/elasticsearch/lib/api/types';
import {
AggregationsAggregationContainer,
AggregationsHistogramAggregate,
AggregationsHistogramBucket,
AggregationsMultiBucketAggregateBase,
AggregationsStringTermsBucket,
} from '@elastic/elasticsearch/lib/api/types';

const anyQuery = 'any::query';
const index = 'test';
Expand All @@ -36,23 +42,23 @@ function mockTopNData() {
histogram: {
buckets: [
{
key_as_string: '1644506880',
key: 1644506880000,
doc_count: 700,
key_as_string: '123',
key: 123000,
doc_count: 10,
group_by: {
buckets: [
{
key: 'vyHke_Kdp2c05tXV7a_Rkg==',
key: '::any::key::',
doc_count: 10,
Count: {
value: 100.0,
},
},
} as AggregationsStringTermsBucket,
],
},
},
} as AggregationsMultiBucketAggregateBase<AggregationsStringTermsBucket>,
} as AggregationsHistogramBucket,
],
},
} as AggregationsHistogramAggregate,
},
},
}),
Expand Down
39 changes: 21 additions & 18 deletions src/plugins/profiling/server/routes/search_topn.ts
Original file line number Diff line number Diff line change
Expand Up @@ -8,19 +8,21 @@
import { schema } from '@kbn/config-schema';
import type { IRouter, KibanaResponseFactory } from 'kibana/server';
import {
AggregationsHistogramAggregate,
AggregationsHistogramBucket,
AggregationsMultiBucketAggregateBase,
AggregationsStringTermsBucket,
} from '@elastic/elasticsearch/lib/api/types';
import type { DataRequestHandlerContext } from '../../../data/server';
import { getRemoteRoutePaths } from '../../common';
import { newProjectTimeQuery, autoHistogramSumCountOnGroupByField } from './mappings';
import { autoHistogramSumCountOnGroupByField, newProjectTimeQuery } from './mappings';

export async function topNElasticSearchQuery(
context: DataRequestHandlerContext,
index: string,
projectID: string,
timeFrom: string,
timeTo: string,
topNItems: number,
searchField: string,
response: KibanaResponseFactory
) {
Expand All @@ -30,23 +32,22 @@ export async function topNElasticSearchQuery(
body: {
query: newProjectTimeQuery(projectID, timeFrom, timeTo),
aggs: {
histogram: autoHistogramSumCountOnGroupByField(searchField),
histogram: autoHistogramSumCountOnGroupByField(searchField, topNItems),
},
},
});

if (searchField === 'StackTraceID') {
const autoDateHistogram = resTopNStackTraces.body.aggregations
?.histogram as AggregationsMultiBucketAggregateBase<AggregationsHistogramBucket>;

const docIDs: string[] = [];
autoDateHistogram.buckets?.forEach((timeInterval: any) => {
timeInterval.group_by.buckets.forEach((stackTraceItem: any) => {
(
resTopNStackTraces.body.aggregations?.histogram as AggregationsHistogramAggregate
).buckets.forEach((timeInterval: AggregationsHistogramBucket) => {
timeInterval.group_by.buckets.forEach((stackTraceItem: AggregationsStringTermsBucket) => {
docIDs.push(stackTraceItem.key);
});
});

const resTraceMetadata = await esClient.mget<any>({
const resTraceMetadata = await esClient.mget({
index: 'profiling-stacktraces',
body: { ids: docIDs },
});
Expand Down Expand Up @@ -76,23 +77,25 @@ export function queryTopNCommon(
path: pathName,
validate: {
query: schema.object({
index: schema.maybe(schema.string()),
projectID: schema.maybe(schema.string()),
timeFrom: schema.maybe(schema.string()),
timeTo: schema.maybe(schema.string()),
index: schema.string(),
projectID: schema.string(),
timeFrom: schema.string(),
timeTo: schema.string(),
n: schema.number(),
}),
},
},
async (context, request, response) => {
const { index, projectID, timeFrom, timeTo } = request.query;
const { index, projectID, timeFrom, timeTo, n } = request.query;

try {
return await topNElasticSearchQuery(
context,
index!,
projectID!,
timeFrom!,
timeTo!,
index,
projectID,
timeFrom,
timeTo,
n,
searchField,
response
);
Expand Down

0 comments on commit abc5b26

Please sign in to comment.