integrate Zookeeper as configuration center with ZookeeperDynamicConf…

…igManager
sofastack · Jul 21, 2024 · 86c0c52 · 86c0c52
1 parent d163820
commit 86c0c52
Show file tree

Hide file tree

Showing 7 changed files with 301 additions and 0 deletions.
diff --git a/all/pom.xml b/all/pom.xml
@@ -310,6 +310,11 @@
             <artifactId>sofa-rpc-config-apollo</artifactId>
             <version>${project.version}</version>
         </dependency>
+        <dependency>
+            <groupId>com.alipay.sofa</groupId>
+            <artifactId>sofa-rpc-config-zk</artifactId>
+            <version>${project.version}</version>
+        </dependency>
         <dependency>
             <groupId>com.alipay.sofa</groupId>
             <artifactId>bolt</artifactId>
@@ -565,6 +570,7 @@
                                     <include>com.alipay.sofa:sofa-rpc-tracer-opentracing-resteasy</include>
                                     <include>com.alipay.sofa:sofa-rpc-tracer-opentracing-triple</include>
                                     <include>com.alipay.sofa:sofa-rpc-config-apollo</include>
+                                    <include>com.alipay.sofa:sofa-rpc-config-zk</include>
                                     <include>com.alipay.sofa:sofa-rpc-doc-swagger</include>
                                     <!-- TODO -->
                                 </includes>

diff --git a/config/config-zk/pom.xml b/config/config-zk/pom.xml
@@ -0,0 +1,112 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>com.alipay.sofa</groupId>
+        <artifactId>sofa-rpc-config</artifactId>
+        <version>${revision}</version>
+    </parent>
+
+    <artifactId>sofa-rpc-config-zk</artifactId>
+
+    <dependencies>
+        <dependency>
+            <groupId>com.alipay.sofa</groupId>
+            <artifactId>sofa-rpc-log-common-tools</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.alipay.sofa</groupId>
+            <artifactId>sofa-rpc-log</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.alipay.sofa</groupId>
+            <artifactId>sofa-rpc-api</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.apache.curator</groupId>
+            <artifactId>curator-framework</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.zookeeper</groupId>
+            <artifactId>zookeeper</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.slf4j</groupId>
+            <artifactId>slf4j-log4j12</artifactId>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <scope>test</scope>
+        </dependency>
+
+    </dependencies>
+
+    <build>
+        <sourceDirectory>src/main/java</sourceDirectory>
+        <resources>
+            <resource>
+                <directory>src/main/resources</directory>
+                <filtering>false</filtering>
+                <includes>
+                    <include>**/**</include>
+                </includes>
+            </resource>
+        </resources>
+        <testSourceDirectory>src/test/java</testSourceDirectory>
+        <testResources>
+            <testResource>
+                <directory>src/test/resources</directory>
+                <filtering>false</filtering>
+                <includes>
+                    <include>**/**</include>
+                </includes>
+            </testResource>
+        </testResources>
+
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>${maven.compiler.source}</source>
+                    <target>${maven.compiler.source}</target>
+                    <encoding>${project.build.sourceEncoding}</encoding>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-install-plugin</artifactId>
+                <configuration>
+                    <skip>${module.install.skip}</skip>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-deploy-plugin</artifactId>
+                <configuration>
+                    <skip>${module.deploy.skip}</skip>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-surefire-plugin</artifactId>
+                <configuration>
+                    <skipTests>${skipTests}</skipTests>
+                    <includes>
+                        <!-- 这里需要根据自己的需要指定要跑的单元测试 -->
+                        <include>**/*Test.java</include>
+                    </includes>
+                    <!-- 如无特殊需求，将forkMode设置为once -->
+                    <forkMode>once</forkMode>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+</project>
diff --git a/...config-zk/src/main/java/com/alipay/sofa/rpc/dynamic/zk/ZookeeperDynamicConfigManager.java b/...config-zk/src/main/java/com/alipay/sofa/rpc/dynamic/zk/ZookeeperDynamicConfigManager.java
@@ -0,0 +1,112 @@
+/*
+ * 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 com.alipay.sofa.rpc.dynamic.zk;
+
+import com.alipay.sofa.rpc.auth.AuthRuleGroup;
+import com.alipay.sofa.rpc.common.utils.StringUtils;
+import com.alipay.sofa.rpc.dynamic.DynamicConfigKeyHelper;
+import com.alipay.sofa.rpc.dynamic.DynamicConfigManager;
+import com.alipay.sofa.rpc.dynamic.DynamicHelper;
+import com.alipay.sofa.rpc.ext.Extension;
+import org.apache.curator.framework.CuratorFramework;
+import org.apache.curator.framework.CuratorFrameworkFactory;
+import org.apache.curator.retry.ExponentialBackoffRetry;
+
+/**
+ * @author Narziss
+ * @version ZookeeperDynamicConfigManager.java, v 0.1 2024年07月20日 09:23 Narziss
+ */
+
+@Extension(value = "zookeeper", override = true)
+public class ZookeeperDynamicConfigManager extends DynamicConfigManager {
+
+    private CuratorFramework zkClient;
+    private static final String ADDRESS = "127.0.0.1:2181";
+    private static final String DEFAULT_NAMESPACE = "sofa-rpc";
+    private static final String CONFIG_NODE = "/config";
+    private static final String Default_GROUP = "/application";
+
+    protected ZookeeperDynamicConfigManager(String appName) {
+        super(appName);
+        zkClient = CuratorFrameworkFactory.builder()
+                .connectString(ADDRESS)
+                .retryPolicy(new ExponentialBackoffRetry(1000, 3))
+                .namespace(DEFAULT_NAMESPACE)
+                .build();
+        zkClient.start();
+    }
+
+    @Override
+    public void initServiceConfiguration(String service) {
+        //TODO not now
+    }
+
+    @Override
+    public String getProviderServiceProperty(String service, String key) {
+        try {
+            String path = getPrefixPath() + DynamicConfigKeyHelper.buildProviderServiceProKey(service, key);
+            byte[] bytes = zkClient.getData().forPath(path);
+            return bytes != null ? new String(bytes) : DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        } catch (Exception e) {
+            return DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        }
+    }
+
+    @Override
+    public String getConsumerServiceProperty(String service, String key) {
+        try {
+            String path = getPrefixPath() + DynamicConfigKeyHelper.buildConsumerServiceProKey(service, key);
+            byte[] bytes = zkClient.getData().forPath(path);
+            return bytes != null ? new String(bytes) : DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        } catch (Exception e) {
+            return DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        }
+    }
+
+    @Override
+    public String getProviderMethodProperty(String service, String method, String key) {
+        try {
+            String path = getPrefixPath() + DynamicConfigKeyHelper.buildProviderMethodProKey(service, method, key);
+            byte[] bytes = zkClient.getData().forPath(path);
+            return bytes != null ? new String(bytes) : DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        } catch (Exception e) {
+            return DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        }
+    }
+
+    @Override
+    public String getConsumerMethodProperty(String service, String method, String key) {
+        try {
+            String path = getPrefixPath() + DynamicConfigKeyHelper.buildConsumerMethodProKey(service, method, key);
+            byte[] bytes = zkClient.getData().forPath(path);
+            return bytes != null ? new String(bytes) : DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        } catch (Exception e) {
+            return DynamicHelper.DEFAULT_DYNAMIC_VALUE;
+        }
+
+    }
+
+    @Override
+    public AuthRuleGroup getServiceAuthRule(String service) {
+        //TODO 暂不支持
+        return null;
+    }
+
+    public String getPrefixPath() {
+        return CONFIG_NODE+Default_GROUP+StringUtils.CONTEXT_SEP;
+    }
+}
diff --git a/...ain/resources/META-INF/services/sofa-rpc/com.alipay.sofa.rpc.dynamic.DynamicConfigManager b/...ain/resources/META-INF/services/sofa-rpc/com.alipay.sofa.rpc.dynamic.DynamicConfigManager
@@ -0,0 +1 @@
+zookeeper=com.alipay.sofa.rpc.dynamic.zk.ZookeeperDynamicConfigManager
diff --git a/...ig-zk/src/test/java/com/alipay/sofa/rpc/dynamic/zk/ZookeeperDynamicConfigManagerTest.java b/...ig-zk/src/test/java/com/alipay/sofa/rpc/dynamic/zk/ZookeeperDynamicConfigManagerTest.java
@@ -0,0 +1,53 @@
+/*
+ * 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 com.alipay.sofa.rpc.dynamic.zk;
+
+import com.alipay.sofa.rpc.dynamic.DynamicHelper;
+import com.alipay.sofa.rpc.log.Logger;
+import com.alipay.sofa.rpc.log.LoggerFactory;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class ZookeeperDynamicConfigManagerTest {
+
+    private final static Logger        logger                     = LoggerFactory
+                                                                      .getLogger(ZookeeperDynamicConfigManager.class);
+
+    private ZookeeperDynamicConfigManager zookeeperDynamicConfigManager = new ZookeeperDynamicConfigManager("test");
+
+    @Test
+    public void getProviderServiceProperty() {
+        String result = zookeeperDynamicConfigManager.getProviderServiceProperty("serviceName", "timeout");
+        Assert.assertEquals(DynamicHelper.DEFAULT_DYNAMIC_VALUE, result);
+    }
+
+    @Test
+    public void getConsumerServiceProperty() {
+    }
+
+    @Test
+    public void getProviderMethodProperty() {
+    }
+
+    @Test
+    public void getConsumerMethodProperty() {
+    }
+
+    @Test
+    public void getServiceAuthRule() {
+    }
+}
diff --git a/config/config-zk/src/test/resources/log4j.xml b/config/config-zk/src/test/resources/log4j.xml
@@ -0,0 +1,16 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE log4j:configuration SYSTEM "log4j.dtd">
+<log4j:configuration xmlns:log4j="http://jakarta.apache.org/log4j/" debug="false">
+
+    <appender name="CONSOLE" class="org.apache.log4j.ConsoleAppender">
+        <layout class="org.apache.log4j.PatternLayout">
+            <param name="ConversionPattern" value="%d %t %5p [%c:%M:%L] - %m%n"/>
+        </layout>
+    </appender>
+
+    <root>
+        <level value="INFO"/>
+        <appender-ref ref="CONSOLE"/>
+    </root>
+
+</log4j:configuration>
diff --git a/config/pom.xml b/config/pom.xml
@@ -15,6 +15,7 @@
 
     <modules>
         <module>config-apollo</module>
+        <module>config-zk</module>
     </modules>
 
     <dependencyManagement>