1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use std::net::TcpStream;

use serde::Serialize;
use serde_json::{Map, Value};

use crate::actor::{Actor, ActorMessageStatus, ActorRegistry};
use crate::StreamId;

#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct TimelineMemoryReply {
    js_object_size: u64,
    js_string_size: u64,
    js_other_size: u64,
    dom_size: u64,
    style_size: u64,
    other_size: u64,
    total_size: u64,
    js_milliseconds: f64,
    #[serde(rename = "nonJSMilliseconds")]
    non_js_milliseconds: f64,
}

pub struct MemoryActor {
    pub name: String,
}

impl Actor for MemoryActor {
    fn name(&self) -> String {
        self.name.clone()
    }

    fn handle_message(
        &self,
        _registry: &ActorRegistry,
        _msg_type: &str,
        _msg: &Map<String, Value>,
        _stream: &mut TcpStream,
        _id: StreamId,
    ) -> Result<ActorMessageStatus, ()> {
        Ok(ActorMessageStatus::Ignored)
    }
}

impl MemoryActor {
    /// return name of actor
    pub fn create(registry: &ActorRegistry) -> String {
        let actor_name = registry.new_name("memory");
        let actor = MemoryActor {
            name: actor_name.clone(),
        };

        registry.register_later(Box::new(actor));
        actor_name
    }

    pub fn measure(&self) -> TimelineMemoryReply {
        TimelineMemoryReply {
            js_object_size: 1,
            js_string_size: 1,
            js_other_size: 1,
            dom_size: 1,
            style_size: 1,
            other_size: 1,
            total_size: 1,
            js_milliseconds: 1.1,
            non_js_milliseconds: 1.1,
        }
    }
}