From 79fc84f3db738d0ddf67305393c8c9120f93c407 Mon Sep 17 00:00:00 2001 From: yukun Date: Tue, 16 Nov 2021 17:13:10 +0800 Subject: [PATCH] Optimize updatePageInfo in rocksmq (#11711) Signed-off-by: fishpenguin --- .../util/rocksmq/server/rocksmq/rocksmq_impl.go | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go b/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go index a04254a8a9..d125eabcf9 100644 --- a/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go +++ b/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go @@ -519,23 +519,14 @@ func (rmq *rocksmq) updatePageInfo(topicName string, msgIDs []UniqueID, msgSizes if err != nil { return err } - - // Update message size to 0 - err = rmq.kv.Save(msgSizeKey, strconv.FormatInt(0, 10)) - if err != nil { - return err - } curMsgSize = 0 } else { curMsgSize += msgSize - // Update message size to current message size - err := rmq.kv.Save(msgSizeKey, strconv.FormatInt(curMsgSize, 10)) - if err != nil { - return err - } } } - return nil + // Update message size to current message size + err = rmq.kv.Save(msgSizeKey, strconv.FormatInt(curMsgSize, 10)) + return err } // Consume steps: