This is an automated email from the ASF dual-hosted git repository.
duanzhengqiang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git
The following commit(s) were added to refs/heads/master by this push:
new f18228a9db6 Add ReadwriteSplittingContextManagerLifecycleListener to
load qualified data source state (#34240)
f18228a9db6 is described below
commit f18228a9db6c29c57c9ccdde34531c305f90b2f7
Author: Haoran Meng <[email protected]>
AuthorDate: Fri Jan 3 19:05:37 2025 +0800
Add ReadwriteSplittingContextManagerLifecycleListener to load qualified
data source state (#34240)
* Add ReadwriteSplittingContextManagerLifecycleListener to load qualified
data source state
* Fix checkstyle
---
...teSplittingContextManagerLifecycleListener.java | 57 ++++++++++++++++++++++
...anager.listener.ContextManagerLifecycleListener | 18 +++++++
2 files changed, 75 insertions(+)
diff --git
a/features/readwrite-splitting/core/src/main/java/org/apache/shardingsphere/readwritesplitting/listener/ReadwriteSplittingContextManagerLifecycleListener.java
b/features/readwrite-splitting/core/src/main/java/org/apache/shardingsphere/readwritesplitting/listener/ReadwriteSplittingContextManagerLifecycleListener.java
new file mode 100644
index 00000000000..6d6d97cd452
--- /dev/null
+++
b/features/readwrite-splitting/core/src/main/java/org/apache/shardingsphere/readwritesplitting/listener/ReadwriteSplittingContextManagerLifecycleListener.java
@@ -0,0 +1,57 @@
+/*
+ * 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.
+ */
+
+package org.apache.shardingsphere.readwritesplitting.listener;
+
+import org.apache.shardingsphere.infra.metadata.ShardingSphereMetaData;
+import
org.apache.shardingsphere.infra.metadata.database.schema.QualifiedDataSource;
+import
org.apache.shardingsphere.infra.rule.attribute.datasource.StaticDataSourceRuleAttribute;
+import
org.apache.shardingsphere.infra.state.datasource.qualified.QualifiedDataSourceState;
+import org.apache.shardingsphere.mode.manager.ContextManager;
+import
org.apache.shardingsphere.mode.manager.listener.ContextManagerLifecycleListener;
+
+import java.util.Map;
+
+/**
+ * Readwrite splitting context manager lifecycle listener.
+ */
+public class ReadwriteSplittingContextManagerLifecycleListener implements
ContextManagerLifecycleListener {
+
+ @Override
+ public void onInitialized(final ContextManager contextManager) {
+ if
(contextManager.getComputeNodeInstanceContext().getModeConfiguration().isCluster())
{
+ updateQualifiedDataSourceState(contextManager);
+ }
+ }
+
+ @Override
+ public void onDestroyed(final ContextManager contextManager) {
+ }
+
+ private void updateQualifiedDataSourceState(final ContextManager
contextManager) {
+ Map<String, QualifiedDataSourceState> qualifiedDataSourceStateMap =
contextManager.getPersistServiceFacade().getQualifiedDataSourceStatePersistService().load();
+ qualifiedDataSourceStateMap.forEach((key, value) ->
updateQualifiedDataSourceState(contextManager.getMetaDataContexts().getMetaData(),
new QualifiedDataSource(key), value));
+ }
+
+ private void updateQualifiedDataSourceState(final ShardingSphereMetaData
metaData, final QualifiedDataSource qualifiedDataSource, final
QualifiedDataSourceState state) {
+ metaData.getAllDatabases().forEach(each -> {
+ if (each.getName().equals(qualifiedDataSource.getDatabaseName())) {
+
each.getRuleMetaData().getAttributes(StaticDataSourceRuleAttribute.class).forEach(attribute
-> attribute.updateStatus(qualifiedDataSource, state.getState()));
+ }
+ });
+ }
+}
diff --git
a/features/readwrite-splitting/core/src/main/resources/META-INF/services/org.apache.shardingsphere.mode.manager.listener.ContextManagerLifecycleListener
b/features/readwrite-splitting/core/src/main/resources/META-INF/services/org.apache.shardingsphere.mode.manager.listener.ContextManagerLifecycleListener
new file mode 100644
index 00000000000..3820234565e
--- /dev/null
+++
b/features/readwrite-splitting/core/src/main/resources/META-INF/services/org.apache.shardingsphere.mode.manager.listener.ContextManagerLifecycleListener
@@ -0,0 +1,18 @@
+#
+# 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.
+#
+
+org.apache.shardingsphere.readwritesplitting.listener.ReadwriteSplittingContextManagerLifecycleListener
