mirror of
https://github.com/denoland/deno
synced 2024-10-05 23:59:24 +00:00
f3c0f0565b
This commit adds an ability to "ref" or "unref" pending ops. Up to this point Deno had a notion of "async ops" and "unref async ops"; the former keep event loop alive, while the latter do not block event loop from finishing. It was not possible to change between op types after dispatching, one had to decide which type to use before dispatch. Instead of storing ops in two separate "FuturesUnordered" collections, now ops are stored in a single collection, with supplemental "HashSet" storing ids of promises that were "unrefed". Two APIs were added to "Deno.core": "Deno.core.refOp(promiseId)" which allows to mark promise id to be "refed" and keep event loop alive (the default behavior) "Deno.core.unrefOp(promiseId)" which allows to mark promise id as "unrefed" which won't block event loop from exiting
90 lines
2.7 KiB
Rust
90 lines
2.7 KiB
Rust
// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.
|
|
use crate::serde::Serialize;
|
|
use crate::OpId;
|
|
use std::cell::RefCell;
|
|
use std::cell::RefMut;
|
|
|
|
// TODO(@AaronO): split into AggregateMetrics & PerOpMetrics
|
|
#[derive(Clone, Default, Debug, Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct OpMetrics {
|
|
pub ops_dispatched: u64,
|
|
pub ops_dispatched_sync: u64,
|
|
pub ops_dispatched_async: u64,
|
|
// TODO(bartlomieju): this field is never updated
|
|
pub ops_dispatched_async_unref: u64,
|
|
pub ops_completed: u64,
|
|
pub ops_completed_sync: u64,
|
|
pub ops_completed_async: u64,
|
|
// TODO(bartlomieju): this field is never updated
|
|
pub ops_completed_async_unref: u64,
|
|
pub bytes_sent_control: u64,
|
|
pub bytes_sent_data: u64,
|
|
pub bytes_received: u64,
|
|
}
|
|
|
|
// TODO(@AaronO): track errors
|
|
#[derive(Default, Debug)]
|
|
pub struct OpsTracker {
|
|
pub ops: RefCell<Vec<OpMetrics>>,
|
|
}
|
|
|
|
impl OpsTracker {
|
|
pub fn per_op(&self) -> Vec<OpMetrics> {
|
|
self.ops.borrow().clone()
|
|
}
|
|
|
|
pub fn aggregate(&self) -> OpMetrics {
|
|
let mut sum = OpMetrics::default();
|
|
|
|
for metrics in self.ops.borrow().iter() {
|
|
sum.ops_dispatched += metrics.ops_dispatched;
|
|
sum.ops_dispatched_sync += metrics.ops_dispatched_sync;
|
|
sum.ops_dispatched_async += metrics.ops_dispatched_async;
|
|
sum.ops_dispatched_async_unref += metrics.ops_dispatched_async_unref;
|
|
sum.ops_completed += metrics.ops_completed;
|
|
sum.ops_completed_sync += metrics.ops_completed_sync;
|
|
sum.ops_completed_async += metrics.ops_completed_async;
|
|
sum.ops_completed_async_unref += metrics.ops_completed_async_unref;
|
|
sum.bytes_sent_control += metrics.bytes_sent_control;
|
|
sum.bytes_sent_data += metrics.bytes_sent_data;
|
|
sum.bytes_received += metrics.bytes_received;
|
|
}
|
|
|
|
sum
|
|
}
|
|
|
|
fn ensure_capacity(&self, op_id: OpId) {
|
|
let ops = &mut self.ops.borrow_mut();
|
|
if op_id >= ops.len() {
|
|
let delta_len = 1 + op_id - ops.len();
|
|
ops.extend(vec![OpMetrics::default(); delta_len])
|
|
}
|
|
}
|
|
|
|
fn metrics_mut(&self, id: OpId) -> RefMut<OpMetrics> {
|
|
self.ensure_capacity(id);
|
|
RefMut::map(self.ops.borrow_mut(), |ops| ops.get_mut(id).unwrap())
|
|
}
|
|
|
|
pub fn track_sync(&self, id: OpId) {
|
|
let metrics = &mut self.metrics_mut(id);
|
|
metrics.ops_dispatched += 1;
|
|
metrics.ops_completed += 1;
|
|
metrics.ops_dispatched_sync += 1;
|
|
metrics.ops_completed_sync += 1;
|
|
}
|
|
|
|
pub fn track_async(&self, id: OpId) {
|
|
let metrics = &mut self.metrics_mut(id);
|
|
metrics.ops_dispatched += 1;
|
|
metrics.ops_dispatched_async += 1;
|
|
}
|
|
|
|
pub fn track_async_completed(&self, id: OpId) {
|
|
let metrics = &mut self.metrics_mut(id);
|
|
metrics.ops_completed += 1;
|
|
metrics.ops_completed_async += 1;
|
|
}
|
|
}
|