Merge pull request #606 from takazamy/master

update bytesWritten, bytesRead on doneConnect
illuspas · Oct 14, 2023 · 641e970 · 641e970 · twilson90 · Mar 31, 2024
2 parents a13c3b6 + 2778f6f
commit 641e970
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/node_flv_session.js b/src/node_flv_session.js
@@ -91,6 +91,8 @@ class NodeFlvSession {
       }
       Logger.log(`[${this.TAG} play] Close stream. id=${this.id} streamPath=${this.playStreamPath}`);
       Logger.log(`[${this.TAG} disconnect] id=${this.id}`);
+      this.connectCmdObj.bytesWritten = this.res.socket.bytesWritten;
+      this.connectCmdObj.bytesRead = this.res.socket.bytesRead;
       context.nodeEvent.emit('doneConnect', this.id, this.connectCmdObj);
       this.res.end();
       context.idlePlayers.delete(this.id);

diff --git a/src/node_rtmp_session.js b/src/node_rtmp_session.js
@@ -217,6 +217,9 @@ class NodeRtmpSession {
       }
 
       Logger.log(`[rtmp disconnect] id=${this.id}`);
+
+      this.connectCmdObj.bytesWritten = this.socket.bytesWritten;
+      this.connectCmdObj.bytesRead = this.socket.bytesRead;
       context.nodeEvent.emit('doneConnect', this.id, this.connectCmdObj);
 
       context.sessions.delete(this.id);