mirror of https://github.com/zulip/zulip.git
133 lines
3.3 KiB
TypeScript
133 lines
3.3 KiB
TypeScript
import * as Sentry from "@sentry/browser";
|
|
|
|
import * as blueslip from "./blueslip";
|
|
|
|
export let next_local_id = 0;
|
|
export const messages = new Map<string, MessageState>();
|
|
|
|
export function get_new_local_id(): string {
|
|
next_local_id += 1;
|
|
const local_id = next_local_id;
|
|
return "loc-" + local_id.toString();
|
|
}
|
|
|
|
export class MessageState {
|
|
local_id: string;
|
|
locally_echoed: boolean;
|
|
rendered_changed = false;
|
|
|
|
server_acked = false;
|
|
saw_event = false;
|
|
|
|
txn: Sentry.Transaction | undefined = undefined;
|
|
event_span: Sentry.Span | undefined = undefined;
|
|
|
|
constructor(opts: {local_id: string; locally_echoed: boolean}) {
|
|
this.local_id = opts.local_id;
|
|
this.locally_echoed = opts.locally_echoed;
|
|
}
|
|
|
|
start_send(): Sentry.Transaction {
|
|
this.txn = Sentry.startTransaction({
|
|
op: "function",
|
|
description: "message send",
|
|
name: "message send",
|
|
});
|
|
this.event_span = this.txn.startChild({
|
|
op: "function",
|
|
description: "message send (server event loop)",
|
|
});
|
|
return this.txn;
|
|
}
|
|
|
|
mark_disparity(): void {
|
|
this.rendered_changed = true;
|
|
}
|
|
|
|
report_server_ack(): void {
|
|
this.server_acked = true;
|
|
this.maybe_finish_txn();
|
|
}
|
|
|
|
report_event_received(): void {
|
|
if (!this.event_span) {
|
|
return;
|
|
}
|
|
this.saw_event = true;
|
|
this.event_span.finish();
|
|
this.maybe_finish_txn();
|
|
}
|
|
|
|
maybe_finish_txn(): void {
|
|
if (!this.saw_event || !this.server_acked) {
|
|
return;
|
|
}
|
|
const setTag = (name: string, val: boolean): void => {
|
|
const str_val = val ? "true" : "false";
|
|
this.event_span!.setTag(name, str_val);
|
|
this.txn!.setTag(name, str_val);
|
|
};
|
|
setTag("rendered_changed", this.rendered_changed);
|
|
setTag("locally_echoed", this.locally_echoed);
|
|
this.txn!.finish();
|
|
messages.delete(this.local_id);
|
|
}
|
|
}
|
|
|
|
export function start_tracking_message(opts: {local_id: string; locally_echoed: boolean}): void {
|
|
const local_id = opts.local_id;
|
|
|
|
if (!opts.local_id) {
|
|
blueslip.error("You must supply a local_id");
|
|
return;
|
|
}
|
|
|
|
if (messages.has(local_id)) {
|
|
blueslip.error("We are re-using a local_id");
|
|
return;
|
|
}
|
|
|
|
const state = new MessageState(opts);
|
|
|
|
messages.set(local_id, state);
|
|
}
|
|
|
|
export function get_message_state(local_id: string): MessageState | undefined {
|
|
const state = messages.get(local_id);
|
|
|
|
if (!state) {
|
|
blueslip.warn("Unknown local_id: " + local_id);
|
|
}
|
|
|
|
return state;
|
|
}
|
|
|
|
export function start_send(local_id: string): Sentry.Transaction | undefined {
|
|
const state = get_message_state(local_id);
|
|
if (!state) {
|
|
return undefined;
|
|
}
|
|
|
|
return state.start_send();
|
|
}
|
|
|
|
export function mark_disparity(local_id: string): void {
|
|
const state = get_message_state(local_id);
|
|
if (!state) {
|
|
return;
|
|
}
|
|
state.mark_disparity();
|
|
}
|
|
|
|
export function report_event_received(local_id: string): void {
|
|
if (local_id === undefined) {
|
|
return;
|
|
}
|
|
const state = get_message_state(local_id);
|
|
if (!state) {
|
|
return;
|
|
}
|
|
|
|
state.report_event_received();
|
|
}
|