Time distribution across script event categories.

1 files changed, 5 insertions, 4 deletions

diff --git a/components/script/dom/websocket.rs b/components/script/dom/websocket.rs
index f92d484c4e5..25b8d832fe4 100644
--- a/components/script/dom/websocket.rs
+++ b/components/script/dom/websocket.rs
@@ -22,6 +22,7 @@ use dom::closeevent::CloseEvent;
 use dom::event::{Event, EventBubbles, EventCancelable};
 use dom::eventtarget::{EventTarget, EventTargetTypeId};
 use dom::messageevent::MessageEvent;
+use script_task::ScriptTaskEventCategory::WebSocketEvent;
 use script_task::{Runnable, CommonScriptMsg};
 
 use net_traits::hosts::replace_hosts;
@@ -182,7 +183,7 @@ impl WebSocket {
                     let task = box CloseTask {
                         addr: address,
                     };
-                    sender.send(CommonScriptMsg::RunnableMsg(task)).unwrap();
+                    sender.send(CommonScriptMsg::RunnableMsg(WebSocketEvent, task)).unwrap();
                     return;
                 }
             };
@@ -192,7 +193,7 @@ impl WebSocket {
                 addr: address.clone(),
                 sender: ws_sender.clone(),
             };
-            sender.send(CommonScriptMsg::RunnableMsg(open_task)).unwrap();
+            sender.send(CommonScriptMsg::RunnableMsg(WebSocketEvent, open_task)).unwrap();
 
             for message in receiver.incoming_messages() {
                 let message = match message {
@@ -208,7 +209,7 @@ impl WebSocket {
                         let task = box CloseTask {
                             addr: address,
                         };
-                        sender.send(CommonScriptMsg::RunnableMsg(task)).unwrap();
+                        sender.send(CommonScriptMsg::RunnableMsg(WebSocketEvent, task)).unwrap();
                         break;
                     },
                     Err(_) => break,
@@ -217,7 +218,7 @@ impl WebSocket {
                     address: address.clone(),
                     message: message,
                 };
-                sender.send(CommonScriptMsg::RunnableMsg(message_task)).unwrap();
+                sender.send(CommonScriptMsg::RunnableMsg(WebSocketEvent, message_task)).unwrap();
             }
         });