From 3f5cd807642995f4a475d4b9f08b7f8536981c18 Mon Sep 17 00:00:00 2001
From: dockerzhang <dockerzhang@tencent.com>
Date: Thu, 24 Dec 2020 15:56:43 +0800
Subject: [PATCH] add updateRates method for kop to collect publish rate

---
 .../main/java/org/apache/pulsar/broker/service/Producer.java   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pulsar-broker/src/main/java/org/apache/pulsar/broker/service/Producer.java b/pulsar-broker/src/main/java/org/apache/pulsar/broker/service/Producer.java
index c69f9bade52a5..c5966d9f2d666 100644
--- a/pulsar-broker/src/main/java/org/apache/pulsar/broker/service/Producer.java
+++ b/pulsar-broker/src/main/java/org/apache/pulsar/broker/service/Producer.java
@@ -581,7 +581,10 @@ public void updateRates() {
             msgDrop.calculateRate();
             ((NonPersistentPublisherStats) stats).msgDropRate = msgDrop.getRate();
         }
+    }
 
+    public void updateRates(int numOfMessages, long msgSizeInBytes) {
+        msgIn.recordMultipleEvents(numOfMessages, msgSizeInBytes);
     }
 
     public boolean isRemote() {