Merge pull request #317 from qmwu2000/master

apollo-client/pom.xml

@@ -65,5 +65,13 @@
 			<scope>test</scope>
 		</dependency>
 		<!-- end of test -->
+		<!-- dal-jdbc -->
+		<dependency>
+			<groupId>org.unidal.framework</groupId>
+			<artifactId>dal-jdbc</artifactId>
+			<version>2.4.0</version>
+			<scope>provided</scope>
+		</dependency>
+		<!-- end of dal-jdbc -->
 	</dependencies>
 </project>

apollo-client/src/main/java/com/ctrip/framework/apollo/ds/ApolloDataSourceProvider.java

+package com.ctrip.framework.apollo.ds;
+
+import org.codehaus.plexus.logging.LogEnabled;
+import org.codehaus.plexus.logging.Logger;
+import org.unidal.dal.jdbc.datasource.DataSourceProvider;
+import org.unidal.dal.jdbc.datasource.model.entity.DataSourcesDef;
+import org.unidal.dal.jdbc.datasource.model.transform.DefaultSaxParser;
+import org.unidal.lookup.annotation.Named;
+
+import com.ctrip.framework.apollo.ConfigFile;
+import com.ctrip.framework.apollo.ConfigService;
+import com.ctrip.framework.apollo.core.enums.ConfigFileFormat;
+import com.ctrip.framework.foundation.Foundation;
+
+/**
+ * Data source provider based on Apollo configuration service.
+ * <p>
+ *
+ * Use following component definition to replace default
+ * <code>DataSourceProvider</code>:
+ * <p>
+ * <code><pre>
+ *   public List<Component> defineComponents() {
+ *      List<Component> all = new ArrayList<>();
+ * 
+ *      all.add(A(ApolloDataSourceProvider.class));
+ * 
+ *      return all;
+ *   }
+ * </pre></code>
+ * 
+ * <b>WARNING:</b> all defined <code>DataSourceProvider</code> components will
+ * be taken affect. DO NOT define unused <code>DataSourceProvider</code>
+ * component.
+ */
+@Named(type = DataSourceProvider.class, value = "apollo")
+public class ApolloDataSourceProvider implements DataSourceProvider, LogEnabled {
+   private Logger m_logger;
+
+   private DataSourcesDef m_def;
+
+   @Override
+   public DataSourcesDef defineDatasources() {
+      if (m_def == null) {
+         ConfigFile file = ConfigService.getConfigFile("datasources", ConfigFileFormat.XML);
+         String appId = Foundation.app().getAppId();
+         String envType = Foundation.server().getEnvType();
+
+         if (file.hasContent()) {
+            String content = file.getContent();
+
+            m_logger.info(String.format("Found datasources.xml from Apollo(env=%s, app.id=%s)!", envType, appId));
+
+            try {
+               m_def = DefaultSaxParser.parse(content);
+            } catch (Exception e) {
+               throw new IllegalStateException(String.format("Error when parsing datasources.xml from Apollo(env=%s, app.id=%s)!", envType, appId), e);
+            }
+         } else {
+            m_logger.warn(String.format("Can't get datasources.xml from Apollo(env=%s, app.id=%s)!", envType, appId));
+            m_def = new DataSourcesDef();
+         }
+      }
+
+      return m_def;
+   }
+
+   @Override
+   public void enableLogging(Logger logger) {
+      m_logger = logger;
+   }
+}