diff --git a/src/main/java/com/threerings/presents/peer/server/PeerManager.java b/src/main/java/com/threerings/presents/peer/server/PeerManager.java index 3d9acb06f..51fab580b 100644 --- a/src/main/java/com/threerings/presents/peer/server/PeerManager.java +++ b/src/main/java/com/threerings/presents/peer/server/PeerManager.java @@ -70,6 +70,8 @@ import com.threerings.presents.dobj.EntryUpdatedEvent; import com.threerings.presents.dobj.ObjectAccessException; import com.threerings.presents.dobj.SetListener; import com.threerings.presents.dobj.Subscriber; +import com.threerings.presents.net.DownstreamMessage; +import com.threerings.presents.net.Message; import com.threerings.presents.peer.client.PeerService; import com.threerings.presents.peer.data.ClientInfo; import com.threerings.presents.peer.data.NodeObject; @@ -213,6 +215,14 @@ public abstract class PeerManager /** The total number of messages sent to all of our peers. */ public long peerMessagesOut; + public void notePeerMessageReceived (Message msg) { + peerMessagesIn.incrementAndGet(); + } + + public void notePeerMessageSent (DownstreamMessage msg) { + peerMessagesOut++; + } + @Override public Stats clone () { try { Stats cstats = (Stats)super.clone(); diff --git a/src/main/java/com/threerings/presents/peer/server/PeerSession.java b/src/main/java/com/threerings/presents/peer/server/PeerSession.java index ca6c7f7d2..61c08945f 100644 --- a/src/main/java/com/threerings/presents/peer/server/PeerSession.java +++ b/src/main/java/com/threerings/presents/peer/server/PeerSession.java @@ -61,7 +61,7 @@ public class PeerSession extends PresentsSession public void handleMessage (Message message) { super.handleMessage(message); - _stats.peerMessagesIn.incrementAndGet(); + _stats.notePeerMessageReceived(message); } @Override // from PresentsSession @@ -121,7 +121,7 @@ public class PeerSession extends PresentsSession if (!super.postMessage(msg, expect)) { return false; } - _stats.peerMessagesOut++; + _stats.notePeerMessageSent(msg); return true; }