[ISSUE #3503]🚀Implement General HA components and complete AutoSwitch HA implementation✨ (#3504)

Author: mxsm

rocketmq-store/src/ha.rs

@@ -20,6 +20,8 @@ pub(crate) mod default_ha_client;
 mod default_ha_connection;
 pub(crate) mod default_ha_service;
 pub(crate) mod flow_monitor;
+pub(crate) mod general_ha_client;
+pub(crate) mod general_ha_connection;
 pub(crate) mod general_ha_service;
 pub(crate) mod ha_client;
 pub(crate) mod ha_connection;

rocketmq-store/src/ha/auto_switch.rs

@@ -14,4 +14,6 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+pub(crate) mod auto_switch_ha_client;
+pub(crate) mod auto_switch_ha_connection;
 pub(crate) mod auto_switch_ha_service;

rocketmq-store/src/ha/auto_switch/auto_switch_ha_client.rs

@@ -0,0 +1,75 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+use crate::ha::ha_client::HAClient;
+use crate::ha::ha_connection_state::HAConnectionState;
+
+pub struct AutoSwitchHAClient;
+
+impl HAClient for AutoSwitchHAClient {
+    async fn start(&self) {
+        todo!()
+    }
+
+    async fn shutdown(&self) {
+        todo!()
+    }
+
+    async fn wakeup(&self) {
+        todo!()
+    }
+
+    async fn update_master_address(&self, new_address: &str) {
+        todo!()
+    }
+
+    async fn update_ha_master_address(&self, new_address: &str) {
+        todo!()
+    }
+
+    fn get_master_address(&self) -> String {
+        todo!()
+    }
+
+    fn get_ha_master_address(&self) -> String {
+        todo!()
+    }
+
+    fn get_last_read_timestamp(&self) -> i64 {
+        todo!()
+    }
+
+    fn get_last_write_timestamp(&self) -> i64 {
+        todo!()
+    }
+
+    fn get_current_state(&self) -> HAConnectionState {
+        todo!()
+    }
+
+    fn change_current_state(&self, ha_connection_state: HAConnectionState) {
+        todo!()
+    }
+
+    async fn close_master(&self) {
+        todo!()
+    }
+
+    fn get_transferred_byte_in_second(&self) -> i64 {
+        todo!()
+    }
+}

rocketmq-store/src/ha/auto_switch/auto_switch_ha_connection.rs

@@ -0,0 +1,17 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+pub struct AutoSwitchHAConnection;

rocketmq-store/src/ha/default_ha_service.rs

@@ -36,6 +36,7 @@ use std::sync::atomic::AtomicI64;
 use std::sync::Arc;
 
 use rocketmq_remoting::protocol::body::ha_runtime_info::HARuntimeInfo;
+use rocketmq_rust::ArcMut;
 use tracing::error;
 
 use crate::ha::ha_client::HAClient;
@@ -61,6 +62,11 @@ impl DefaultHAService {
         // The actual implementation would depend on the specific requirements of the HA service.
         unimplemented!(" notify_transfer_some method is not implemented");
     }
+
+    pub(crate) fn init(&mut self, message_store: ArcMut<LocalFileMessageStore>) -> HAResult<()> {
+        // Initialize the DefaultHAService with the provided message store.
+        Ok(())
+    }
 }
 
 impl HAService for DefaultHAService {

rocketmq-store/src/ha/general_ha_client.rs

@@ -0,0 +1,24 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+use crate::ha::auto_switch::auto_switch_ha_client::AutoSwitchHAClient;
+use crate::ha::default_ha_client::DefaultHAClient;
+
+pub struct GeneralHAClient {
+    default_ha_service: Option<DefaultHAClient>,
+    auto_switch_ha_service: Option<AutoSwitchHAClient>,
+}

rocketmq-store/src/ha/general_ha_connection.rs

@@ -0,0 +1,24 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+use crate::ha::auto_switch::auto_switch_ha_connection::AutoSwitchHAConnection;
+use crate::ha::default_ha_connection::DefaultHAConnection;
+
+pub struct GeneralHAConnection {
+    default_ha_connection: Option<DefaultHAConnection>,
+    auto_switch_ha_connection: Option<AutoSwitchHAConnection>,
+}

rocketmq-store/src/ha/general_ha_service.rs

@@ -49,7 +49,9 @@ impl GeneralHAService {
         {
             self.auto_switch_ha_service = Some(AutoSwitchHAService)
         } else {
-            self.default_ha_service = Some(DefaultHAService::default())
+            let mut default_ha_service = DefaultHAService::default();
+            default_ha_service.init(message_store)?;
+            self.default_ha_service = Some(default_ha_service);
         }
         Ok(())
     }