@@ -27,7 +27,7 @@ class DemoSslEchoHandler extends WebSocketUriHandler
27
27
/**
28
28
* @var TcpStream[][]
29
29
*/
30
- protected $ streams ;
30
+ protected $ streams = array () ;
31
31
protected $ server ;
32
32
33
33
/**
@@ -48,18 +48,21 @@ protected function getStream(IWebSocketConnection $user, $id)
48
48
*/
49
49
protected function getStreamsByUser (IWebSocketConnection $ user )
50
50
{
51
- return $ this ->streams [$ user ->getId ()];
51
+ return array_key_exists ( $ user -> getId (), $ this ->streams ) ? $ this -> streams [$ user ->getId ()] : [ ];
52
52
}
53
53
54
54
protected function removeStream (IWebSocketConnection $ user , TcpStream $ stream )
55
55
{
56
56
unset($ this ->streams [$ user ->getId ()][$ stream ->getId ()]);
57
57
}
58
58
59
+ protected function addStream (IWebSocketConnection $ user , TcpStream $ stream ){
60
+ $ this ->streams [$ user ->getId ()][$ stream ->getId ()] = $ stream ;
61
+ }
62
+
59
63
public function __construct (\Devristo \Phpws \Server \SocketServer $ server , $ logger )
60
64
{
61
65
parent ::__construct ($ logger );
62
- $ this ->streams = new DefaultDict (array ());
63
66
$ this ->socketServer = $ server ;
64
67
}
65
68
@@ -132,7 +135,7 @@ protected function requestConnect(IWebSocketConnection $user, $message)
132
135
$ uriHandler ->removeStream ($ user , $ stream );
133
136
});
134
137
135
- $ this ->streams [ $ user -> getId ()][ $ stream -> getId ()] = $ stream ;
138
+ $ this ->addStream ( $ user , $ stream) ;
136
139
137
140
$ user ->sendString (json_encode (array (
138
141
'connection ' => $ stream ->getId (),
0 commit comments