blob: d30b458b3858fdf6ebe780609e58d3d5865fac82 [file] [log] [blame]
// Copyright (C) 2019 The Android Open Source Project
//
// 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.
import {assertTrue} from '../base/logging';
import {Time, time} from '../base/time';
import {Args, ArgValue} from '../common/arg_types';
import {ChromeSliceSelection, getLegacySelection} from '../common/state';
import {
CounterDetails,
globals,
SliceDetails,
ThreadStateDetails,
} from '../frontend/globals';
import {
publishCounterDetails,
publishSliceDetails,
publishThreadStateDetails,
} from '../frontend/publish';
import {Engine} from '../trace_processor/engine';
import {
durationFromSql,
LONG,
NUM,
NUM_NULL,
STR,
STR_NULL,
timeFromSql,
} from '../trace_processor/query_result';
import {SLICE_TRACK_KIND} from '../tracks/chrome_slices';
import {Controller} from './controller';
export interface SelectionControllerArgs {
engine: Engine;
}
interface ThreadDetails {
tid: number;
threadName?: string;
}
interface ProcessDetails {
pid?: number;
processName?: string;
uid?: number;
packageName?: string;
versionCode?: number;
}
// This class queries the TP for the details on a specific slice that has
// been clicked.
export class SelectionController extends Controller<'main'> {
private lastSelectedId?: number | string;
private lastSelectedKind?: string;
constructor(private args: SelectionControllerArgs) {
super('main');
}
run() {
const selection = getLegacySelection(globals.state);
if (!selection || selection.kind === 'AREA') return;
const selectWithId = [
'SLICE',
'COUNTER',
'CHROME_SLICE',
'HEAP_PROFILE',
'THREAD_STATE',
];
if (
!selectWithId.includes(selection.kind) ||
(selectWithId.includes(selection.kind) &&
selection.id === this.lastSelectedId &&
selection.kind === this.lastSelectedKind)
) {
return;
}
const selectedId = selection.id;
const selectedKind = selection.kind;
this.lastSelectedId = selectedId;
this.lastSelectedKind = selectedKind;
if (selectedId === undefined) return;
if (selection.kind === 'COUNTER') {
this.counterDetails(
selection.leftTs,
selection.rightTs,
selection.id,
).then((results) => {
if (
results !== undefined &&
selection.kind === selectedKind &&
selection.id === selectedId
) {
publishCounterDetails(results);
}
});
} else if (selection.kind === 'SLICE') {
this.sliceDetails(selectedId as number);
} else if (selection.kind === 'THREAD_STATE') {
this.threadStateDetails(selection.id);
} else if (selection.kind === 'CHROME_SLICE') {
this.chromeSliceDetails(selection);
}
}
async chromeSliceDetails(selection: ChromeSliceSelection) {
const selectedId = selection.id;
const table = selection.table;
let leafTable: string;
let promisedArgs: Promise<Args>;
// TODO(b/155483804): This is a hack to ensure annotation slices are
// selectable for now. We should tidy this up when improving this class.
if (table === 'annotation') {
leafTable = 'annotation_slice';
promisedArgs = Promise.resolve(new Map());
} else {
const result = await this.args.engine.query(`
SELECT
type as leafTable,
arg_set_id as argSetId
FROM slice WHERE id = ${selectedId}`);
if (result.numRows() === 0) {
return;
}
const row = result.firstRow({
leafTable: STR,
argSetId: NUM,
});
leafTable = row.leafTable;
const argSetId = row.argSetId;
promisedArgs = this.getArgs(argSetId);
}
const promisedDetails = this.args.engine.query(`
SELECT *, ABS_TIME_STR(ts) as absTime FROM ${leafTable} WHERE id = ${selectedId};
`);
const [details, args] = await Promise.all([promisedDetails, promisedArgs]);
if (details.numRows() <= 0) return;
const rowIter = details.iter({});
assertTrue(rowIter.valid());
// A few columns are hard coded as part of the SliceDetails interface.
// Long term these should be handled generically as args but for now
// handle them specially:
let ts = undefined;
let absTime = undefined;
let dur = undefined;
let name = undefined;
let category = undefined;
let threadDur = undefined;
let threadTs = undefined;
let trackId = undefined;
// We select all columns from the leafTable to ensure that we include
// additional fields from the child tables (like `thread_dur` from
// `thread_slice` or `frame_number` from `frame_slice`).
// However, this also includes some basic columns (especially from `slice`)
// that are not interesting (i.e. `arg_set_id`, which has already been used
// to resolve and show the arguments) and should not be shown to the user.
const ignoredColumns = [
'type',
'depth',
'parent_id',
'stack_id',
'parent_stack_id',
'arg_set_id',
'thread_instruction_count',
'thread_instruction_delta',
];
for (const k of details.columns()) {
const v = rowIter.get(k);
switch (k) {
case 'id':
break;
case 'ts':
ts = timeFromSql(v);
break;
case 'thread_ts':
threadTs = timeFromSql(v);
break;
case 'absTime':
/* eslint-disable @typescript-eslint/strict-boolean-expressions */
if (v) absTime = `${v}`;
/* eslint-enable */
break;
case 'name':
name = `${v}`;
break;
case 'dur':
dur = durationFromSql(v);
break;
case 'thread_dur':
threadDur = durationFromSql(v);
break;
case 'category':
case 'cat':
category = `${v}`;
break;
case 'track_id':
trackId = Number(v);
break;
default:
if (!ignoredColumns.includes(k)) args.set(k, `${v}`);
}
}
const selected: SliceDetails = {
id: selectedId,
ts,
threadTs,
absTime,
dur,
threadDur,
name,
category,
args,
};
if (trackId !== undefined) {
const columnInfo = (
await this.args.engine.query(`
WITH
leafTrackTable AS (SELECT type FROM track WHERE id = ${trackId}),
cols AS (
SELECT name
FROM pragma_table_info((SELECT type FROM leafTrackTable))
)
SELECT
type as leafTrackTable,
'upid' in cols AS hasUpid,
'utid' in cols AS hasUtid
FROM leafTrackTable
`)
).firstRow({hasUpid: NUM, hasUtid: NUM, leafTrackTable: STR});
const hasUpid = columnInfo.hasUpid !== 0;
const hasUtid = columnInfo.hasUtid !== 0;
if (hasUtid) {
const utid = (
await this.args.engine.query(`
SELECT utid
FROM ${columnInfo.leafTrackTable}
WHERE id = ${trackId};
`)
).firstRow({
utid: NUM,
}).utid;
Object.assign(selected, await this.computeThreadDetails(utid));
} else if (hasUpid) {
const upid = (
await this.args.engine.query(`
SELECT upid
FROM ${columnInfo.leafTrackTable}
WHERE id = ${trackId};
`)
).firstRow({
upid: NUM,
}).upid;
Object.assign(selected, await this.computeProcessDetails(upid));
}
}
// Check selection is still the same on completion of query.
if (selection === getLegacySelection(globals.state)) {
publishSliceDetails(selected);
}
}
async getArgs(argId: number): Promise<Args> {
const args = new Map<string, ArgValue>();
const query = `
select
key AS name,
display_value AS value
FROM args
WHERE arg_set_id = ${argId}
`;
const result = await this.args.engine.query(query);
const it = result.iter({
name: STR,
value: STR_NULL,
});
for (; it.valid(); it.next()) {
const name = it.name;
const value = it.value || 'NULL';
if (name === 'destination slice id' && !isNaN(Number(value))) {
const destTrackId = await this.getDestTrackId(value);
args.set('Destination Slice', {
kind: 'SLICE',
trackId: destTrackId,
sliceId: Number(value),
rawValue: value,
});
} else {
args.set(name, value);
}
}
return args;
}
async getDestTrackId(sliceId: string): Promise<string> {
const trackIdQuery = `select track_id as trackId from slice
where slice_id = ${sliceId}`;
const result = await this.args.engine.query(trackIdQuery);
const trackId = result.firstRow({trackId: NUM}).trackId;
// TODO(hjd): If we had a consistent mapping from TP track_id
// UI track id for slice tracks this would be unnecessary.
let trackKey = '';
for (const track of Object.values(globals.state.tracks)) {
const trackInfo = globals.trackManager.resolveTrackInfo(track.uri);
if (trackInfo?.kind === SLICE_TRACK_KIND) {
const trackIds = trackInfo?.trackIds;
if (trackIds && trackIds.length > 0 && trackIds[0] === trackId) {
trackKey = track.key;
break;
}
}
}
return trackKey;
}
// TODO(altimin): We currently rely on the ThreadStateDetails for supporting
// marking the area (the rest goes is handled by ThreadStateTab
// directly. Refactor it to be plugin-friendly and remove this.
async threadStateDetails(id: number) {
const query = `
SELECT
ts,
thread_state.dur as dur
from thread_state
where thread_state.id = ${id}
`;
const result = await this.args.engine.query(query);
const selection = getLegacySelection(globals.state);
if (result.numRows() > 0 && selection) {
const row = result.firstRow({
ts: LONG,
dur: LONG,
});
const selected: ThreadStateDetails = {
ts: Time.fromRaw(row.ts),
dur: row.dur,
};
publishThreadStateDetails(selected);
}
}
async sliceDetails(id: number) {
const sqlQuery = `SELECT
sched.ts,
sched.dur,
sched.priority,
sched.end_state as endState,
sched.utid,
sched.cpu,
thread_state.id as threadStateId
FROM sched left join thread_state using(ts, utid, cpu)
WHERE sched.id = ${id}`;
const result = await this.args.engine.query(sqlQuery);
// Check selection is still the same on completion of query.
const selection = getLegacySelection(globals.state);
if (result.numRows() > 0 && selection) {
const row = result.firstRow({
ts: LONG,
dur: LONG,
priority: NUM,
endState: STR_NULL,
utid: NUM,
cpu: NUM,
threadStateId: NUM_NULL,
});
const ts = Time.fromRaw(row.ts);
const dur = row.dur;
const priority = row.priority;
const endState = row.endState;
const utid = row.utid;
const cpu = row.cpu;
// eslint-disable-next-line @typescript-eslint/strict-boolean-expressions
const threadStateId = row.threadStateId || undefined;
const selected: SliceDetails = {
ts,
dur,
priority,
endState,
cpu,
id,
utid,
threadStateId,
};
Object.assign(selected, await this.computeThreadDetails(utid));
this.schedulingDetails(ts, utid)
.then((wakeResult) => {
Object.assign(selected, wakeResult);
})
.finally(() => {
publishSliceDetails(selected);
});
}
}
async counterDetails(
ts: time,
rightTs: time,
id: number,
): Promise<CounterDetails> {
const counter = await this.args.engine.query(
`SELECT value, track_id as trackId FROM counter WHERE id = ${id}`,
);
const row = counter.iter({
value: NUM,
trackId: NUM,
});
const value = row.value;
const trackId = row.trackId;
// Finding previous value. If there isn't previous one, it will return 0 for
// ts and value.
const previous = await this.args.engine.query(`SELECT
MAX(ts),
IFNULL(value, 0) as value
FROM counter WHERE ts < ${ts} and track_id = ${trackId}`);
const previousValue = previous.firstRow({value: NUM}).value;
const endTs = rightTs !== -1n ? rightTs : globals.state.traceTime.end;
const delta = value - previousValue;
const duration = endTs - ts;
const trackKey = globals.trackManager.trackKeyByTrackId.get(trackId);
const name = trackKey ? globals.state.tracks[trackKey].name : undefined;
return {startTime: ts, value, delta, duration, name};
}
async schedulingDetails(ts: time, utid: number) {
// Find the ts of the first wakeup before the current slice.
const wakeResult = await this.args.engine.query(`
select ts, waker_utid as wakerUtid
from thread_state
where utid = ${utid} and ts < ${ts} and state = 'R'
order by ts desc
limit 1
`);
if (wakeResult.numRows() === 0) {
return undefined;
}
const wakeFirstRow = wakeResult.firstRow({ts: LONG, wakerUtid: NUM_NULL});
const wakeupTs = wakeFirstRow.ts;
const wakerUtid = wakeFirstRow.wakerUtid;
if (wakerUtid === null) {
return undefined;
}
// Find the previous sched slice for the current utid.
const prevSchedResult = await this.args.engine.query(`
select ts
from sched
where utid = ${utid} and ts < ${ts}
order by ts desc
limit 1
`);
// If this is the first sched slice for this utid or if the wakeup found
// was after the previous slice then we know the wakeup was for this slice.
if (
prevSchedResult.numRows() !== 0 &&
wakeupTs < prevSchedResult.firstRow({ts: LONG}).ts
) {
return undefined;
}
// Find the sched slice with the utid of the waker running when the
// sched wakeup occurred. This is the waker.
const wakerResult = await this.args.engine.query(`
select cpu
from sched
where
utid = ${wakerUtid} and
ts < ${wakeupTs} and
ts + dur >= ${wakeupTs};
`);
if (wakerResult.numRows() === 0) {
return undefined;
}
const wakerRow = wakerResult.firstRow({cpu: NUM});
return {wakeupTs, wakerUtid, wakerCpu: wakerRow.cpu};
}
async computeThreadDetails(
utid: number,
): Promise<ThreadDetails & ProcessDetails> {
const threadInfo = (
await this.args.engine.query(`
SELECT tid, name, upid
FROM thread
WHERE utid = ${utid};
`)
).firstRow({tid: NUM, name: STR_NULL, upid: NUM_NULL});
const threadDetails = {
tid: threadInfo.tid,
threadName: threadInfo.name || undefined,
};
// eslint-disable-next-line @typescript-eslint/strict-boolean-expressions
if (threadInfo.upid) {
return Object.assign(
{},
threadDetails,
await this.computeProcessDetails(threadInfo.upid),
);
}
return threadDetails;
}
async computeProcessDetails(upid: number): Promise<ProcessDetails> {
const details: ProcessDetails = {};
const processResult = (
await this.args.engine.query(`
SELECT pid, name, uid FROM process WHERE upid = ${upid};
`)
).firstRow({pid: NUM, name: STR_NULL, uid: NUM_NULL});
details.pid = processResult.pid;
details.processName = processResult.name || undefined;
if (processResult.uid === null) {
return details;
}
details.uid = processResult.uid;
const packageResult = await this.args.engine.query(`
SELECT
package_name as packageName,
version_code as versionCode
FROM package_list WHERE uid = ${details.uid};
`);
// The package_list table is not populated in some traces so we need to
// check if the result has returned any rows.
if (packageResult.numRows() > 0) {
const packageDetails = packageResult.firstRow({
packageName: STR,
versionCode: NUM,
});
details.packageName = packageDetails.packageName;
details.versionCode = packageDetails.versionCode;
}
return details;
}
}