[ISSUE #1310]⚡️Refactor AllocateMessageQueueStrategy (#1311)

mxsm · Nov 25, 2024 · 79a345a · 79a345a
1 parent 6e8aaac
commit 79a345a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rocketmq-client/src/consumer/allocate_message_queue_strategy.rs b/rocketmq-client/src/consumer/allocate_message_queue_strategy.rs
@@ -22,8 +22,8 @@ use crate::Result;
 pub trait AllocateMessageQueueStrategy: Send + Sync {
     fn allocate(
         &self,
-        consumer_group: &str,
-        current_cid: &str,
+        consumer_group: &CheetahString,
+        current_cid: &CheetahString,
         mq_all: &[MessageQueue],
         cid_all: &[CheetahString],
     ) -> Result<Vec<MessageQueue>>;

diff --git a/rocketmq-client/src/consumer/rebalance_strategy/allocate_message_queue_averagely.rs b/rocketmq-client/src/consumer/rebalance_strategy/allocate_message_queue_averagely.rs
@@ -26,8 +26,8 @@ pub struct AllocateMessageQueueAveragely;
 impl AllocateMessageQueueStrategy for AllocateMessageQueueAveragely {
     fn allocate(
         &self,
-        consumer_group: &str,
-        current_cid: &str,
+        consumer_group: &CheetahString,
+        current_cid: &CheetahString,
         mq_all: &[MessageQueue],
         cid_all: &[CheetahString],
     ) -> Result<Vec<MessageQueue>> {