[ISSUE #1277]⚡️Optimize name server DefaultRequestProcessor#wipe_write_perm_of_broker_by_lock

Author: mxsm

rocketmq-namesrv/src/processor/default_request_processor.rs

@@ -320,7 +320,7 @@ impl DefaultRequestProcessor {
             .expect("decode WipeWritePermOfBrokerRequestHeader failed");
         let wipe_topic_cnt = self
             .route_info_manager
-            .wipe_write_perm_of_broker_by_lock(request_header.broker_name.as_str());
+            .wipe_write_perm_of_broker_by_lock(&request_header.broker_name);
         RemotingCommand::create_response_command()
             .set_command_custom_header(WipeWritePermOfBrokerResponseHeader::new(wipe_topic_cnt))
     }
@@ -331,7 +331,7 @@ impl DefaultRequestProcessor {
             .expect("decode AddWritePermOfBrokerRequestHeader failed");
         let add_topic_cnt = self
             .route_info_manager
-            .add_write_perm_of_broker_by_lock(request_header.broker_name.as_str());
+            .add_write_perm_of_broker_by_lock(&request_header.broker_name);
         RemotingCommand::create_response_command()
             .set_command_custom_header(AddWritePermOfBrokerResponseHeader::new(add_topic_cnt))
     }

rocketmq-namesrv/src/route/route_info_manager.rs

@@ -675,22 +675,35 @@ impl RouteInfoManager {
         Some(group_member)
     }
 
-    pub(crate) fn wipe_write_perm_of_broker_by_lock(&mut self, broker_name: &str) -> i32 {
-        self.operate_write_perm_of_broker(broker_name, RequestCode::WipeWritePermOfBroker)
+    #[inline]
+    pub(crate) fn wipe_write_perm_of_broker_by_lock(&self, broker_name: &CheetahString) -> i32 {
+        let lock = self.lock.write();
+        let cnt =
+            self.operate_write_perm_of_broker(broker_name, RequestCode::WipeWritePermOfBroker);
+        drop(lock);
+        cnt
     }
 
-    pub(crate) fn add_write_perm_of_broker_by_lock(&mut self, broker_name: &str) -> i32 {
-        self.operate_write_perm_of_broker(broker_name, RequestCode::AddWritePermOfBroker)
+    #[inline]
+    pub(crate) fn add_write_perm_of_broker_by_lock(&self, broker_name: &CheetahString) -> i32 {
+        let lock = self.lock.write();
+        let cnt = self.operate_write_perm_of_broker(broker_name, RequestCode::AddWritePermOfBroker);
+        drop(lock);
+        cnt
     }
 
     fn operate_write_perm_of_broker(
-        &mut self,
-        broker_name: &str,
+        &self,
+        broker_name: &CheetahString,
         request_code: RequestCode,
     ) -> i32 {
         let mut topic_cnt = 0;
-        for (_topic, qd_map) in self.topic_queue_table.iter_mut() {
-            let qd = qd_map.get_mut(broker_name).unwrap();
+        for (_topic, qd_map) in self.topic_queue_table.mut_from_ref().iter_mut() {
+            let qd = qd_map.get_mut(broker_name);
+            if qd.is_none() {
+                continue;
+            }
+            let qd = qd.unwrap();
             let mut perm = qd.perm;
             match request_code {
                 RequestCode::WipeWritePermOfBroker => {