|
3 | 3 | * SPDX-License-Identifier: Apache-2.0 |
4 | 4 | */ |
5 | 5 |
|
| 6 | +import AsyncLock from 'async-lock' |
6 | 7 | import { globals } from '../../shared' |
7 | 8 | import { telemetry } from '../../shared/telemetry' |
8 | 9 | import { Event, uiEventRecorder } from '../util/eventRecorder' |
| 10 | +import { CWCTelemetryHelper } from '../../codewhispererChat/controllers/chat/telemetryHelper' |
| 11 | +import { TabType } from '../webview/ui/storages/tabsStorage' |
9 | 12 |
|
10 | 13 | export class AmazonQChatMessageDuration { |
| 14 | + private static _asyncLock = new AsyncLock() |
| 15 | + private static getAsyncLock() { |
| 16 | + if (!AmazonQChatMessageDuration._asyncLock) { |
| 17 | + AmazonQChatMessageDuration._asyncLock = new AsyncLock() |
| 18 | + } |
| 19 | + return AmazonQChatMessageDuration._asyncLock |
| 20 | + } |
| 21 | + |
11 | 22 | /** |
12 | 23 | * Record the initial requests in the chat message flow |
13 | 24 | */ |
14 | | - static startChatMessageTelemetry(msg: { traceId: string; startTime: number; trigger?: string }) { |
15 | | - const { traceId, startTime, trigger } = msg |
| 25 | + static startChatMessageTelemetry(msg: { traceId: string; startTime: number; tabID: string; trigger?: string }) { |
| 26 | + const { traceId, startTime, tabID, trigger } = msg |
16 | 27 |
|
17 | | - uiEventRecorder.set(traceId, { |
| 28 | + uiEventRecorder.set(tabID, { |
| 29 | + traceId, |
18 | 30 | events: { |
19 | 31 | chatMessageSent: startTime, |
20 | | - }, |
21 | | - }) |
22 | | - uiEventRecorder.set(traceId, { |
23 | | - events: { |
24 | 32 | editorReceivedMessage: globals.clock.Date.now(), |
25 | 33 | }, |
26 | 34 | }) |
27 | 35 | if (trigger) { |
28 | | - uiEventRecorder.set(traceId, { |
| 36 | + uiEventRecorder.set(tabID, { |
29 | 37 | trigger, |
30 | 38 | }) |
31 | 39 | } |
| 40 | + CWCTelemetryHelper.instance.setDisplayTimeForChunks(tabID, startTime) |
32 | 41 | } |
33 | 42 |
|
34 | 43 | /** |
35 | 44 | * Stop listening to all incoming events and emit what we've found |
36 | 45 | */ |
37 | | - static stopChatMessageTelemetry(msg: { traceId: string }) { |
38 | | - const { traceId } = msg |
| 46 | + static stopChatMessageTelemetry(msg: { tabID: string; time: number; tabType: TabType }) { |
| 47 | + const { tabID, time, tabType } = msg |
39 | 48 |
|
40 | 49 | // We can't figure out what trace this event was associated with |
41 | | - if (!traceId) { |
| 50 | + if (!tabID || tabType !== 'cwc') { |
42 | 51 | return |
43 | 52 | } |
44 | 53 |
|
45 | | - uiEventRecorder.set(traceId, { |
46 | | - events: { |
47 | | - messageDisplayed: globals.clock.Date.now(), |
48 | | - }, |
49 | | - }) |
50 | | - |
51 | | - const metrics = uiEventRecorder.get(traceId) |
| 54 | + // Lock the tab id just in case another event tries to trigger this |
| 55 | + void AmazonQChatMessageDuration.getAsyncLock().acquire(tabID, () => { |
| 56 | + const metrics = uiEventRecorder.get(tabID) |
| 57 | + if (!metrics) { |
| 58 | + return |
| 59 | + } |
52 | 60 |
|
53 | | - // get events sorted by the time they were created |
54 | | - const events = Object.entries(metrics.events) |
55 | | - .map((x) => ({ |
56 | | - event: x[0], |
57 | | - duration: x[1], |
58 | | - })) |
59 | | - .sort((a, b) => { |
60 | | - return a.duration - b.duration |
| 61 | + uiEventRecorder.set(tabID, { |
| 62 | + events: { |
| 63 | + messageDisplayed: time, |
| 64 | + }, |
61 | 65 | }) |
62 | 66 |
|
63 | | - const chatMessageSentTime = events[events.length - 1].duration |
64 | | - // Get the total duration by subtracting when the message was displayed and when the chat message was first sent |
65 | | - const totalDuration = events[events.length - 1].duration - events[0].duration |
| 67 | + const displayTime = metrics.events.messageDisplayed |
| 68 | + const sentTime = metrics.events.chatMessageSent |
| 69 | + if (!displayTime || !sentTime) { |
| 70 | + return |
| 71 | + } |
66 | 72 |
|
67 | | - /** |
68 | | - * Find the time it took to get between two metric events |
69 | | - */ |
70 | | - const timings = new Map<Event, number>() |
71 | | - for (let i = 1; i < events.length; i++) { |
72 | | - const currentEvent = events[i] |
73 | | - const previousEvent = events[i - 1] |
| 73 | + const totalDuration = displayTime - sentTime |
74 | 74 |
|
75 | | - const timeDifference = currentEvent.duration - previousEvent.duration |
| 75 | + function durationFrom(start: Event, end: Event) { |
| 76 | + const startEvent = metrics.events[start] |
| 77 | + const endEvent = metrics.events[end] |
| 78 | + if (!startEvent || !endEvent) { |
| 79 | + return -1 |
| 80 | + } |
| 81 | + return endEvent - startEvent |
| 82 | + } |
76 | 83 |
|
77 | | - timings.set(currentEvent.event as Event, timeDifference) |
78 | | - } |
| 84 | + // TODO: handle onContextCommand round trip time |
| 85 | + if (metrics.trigger !== 'onContextCommand') { |
| 86 | + telemetry.amazonq_chatRoundTrip.emit({ |
| 87 | + amazonqChatMessageSentTime: metrics.events.chatMessageSent ?? -1, |
| 88 | + amazonqEditorReceivedMessageMs: durationFrom('chatMessageSent', 'editorReceivedMessage') ?? -1, |
| 89 | + amazonqFeatureReceivedMessageMs: |
| 90 | + durationFrom('editorReceivedMessage', 'featureReceivedMessage') ?? -1, |
| 91 | + amazonqMessageDisplayedMs: durationFrom('featureReceivedMessage', 'messageDisplayed') ?? -1, |
| 92 | + source: metrics.trigger, |
| 93 | + duration: totalDuration, |
| 94 | + result: 'Succeeded', |
| 95 | + traceId: metrics.traceId, |
| 96 | + }) |
| 97 | + } |
79 | 98 |
|
80 | | - telemetry.amazonq_chatRoundTrip.emit({ |
81 | | - amazonqChatMessageSentTime: chatMessageSentTime, |
82 | | - amazonqEditorReceivedMessageMs: timings.get('editorReceivedMessage') ?? -1, |
83 | | - amazonqFeatureReceivedMessageMs: timings.get('featureReceivedMessage') ?? -1, |
84 | | - amazonqMessageDisplayedMs: timings.get('messageDisplayed') ?? -1, |
85 | | - source: metrics.trigger, |
86 | | - duration: totalDuration, |
87 | | - result: 'Succeeded', |
88 | | - traceId, |
| 99 | + CWCTelemetryHelper.instance.emitAddMessage(tabID, totalDuration, metrics.events.chatMessageSent) |
| 100 | + |
| 101 | + uiEventRecorder.delete(tabID) |
89 | 102 | }) |
| 103 | + } |
| 104 | + |
| 105 | + static updateChatMessageTelemetry(msg: { tabID: string; time: number; tabType: TabType }) { |
| 106 | + const { tabID, time, tabType } = msg |
| 107 | + if (!tabID || tabType !== 'cwc') { |
| 108 | + return |
| 109 | + } |
90 | 110 |
|
91 | | - uiEventRecorder.delete(traceId) |
| 111 | + CWCTelemetryHelper.instance.setDisplayTimeForChunks(tabID, time) |
92 | 112 | } |
93 | 113 | } |
0 commit comments