Polish Spring Cloud Bus

spring-cloud-starter-alibaba/spring-cloud-starter-bus-rocketmq/pom.xml

@@ -0,0 +1,33 @@
+<?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">
+    <parent>
+        <groupId>org.springframework.cloud</groupId>
+        <artifactId>spring-cloud-starter-alibaba</artifactId>
+        <version>0.2.1.BUILD-SNAPSHOT</version>
+        <relativePath>../pom.xml</relativePath>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>org.springframework.cloud</groupId>
+    <artifactId>spring-cloud-starter-bus-rocketmq</artifactId>
+    <name>Spring Cloud Alibaba Bus RocketMQ</name>
+
+    <dependencies>
+
+        <!-- Spring Cloud Stream RocketMQ -->
+        <dependency>
+            <groupId>org.springframework.cloud</groupId>
+            <artifactId>spring-cloud-stream-binder-rocketmq</artifactId>
+        </dependency>
+
+        <!-- Spring Cloud Bus -->
+        <dependency>
+            <groupId>org.springframework.cloud</groupId>
+            <artifactId>spring-cloud-bus</artifactId>
+        </dependency>
+
+    </dependencies>
+
+</project>

spring-cloud-starter-alibaba/spring-cloud-starter-bus-rocketmq/src/main/java/org/springframework/cloud/bus/rocketmq/env/RocketMQBusEnvironmentPostProcessor.java

@@ -0,0 +1,106 @@
+/*
+ * 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.springframework.cloud.bus.rocketmq.env;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.env.EnvironmentPostProcessor;
+import org.springframework.cloud.bus.BusEnvironmentPostProcessor;
+import org.springframework.core.Ordered;
+import org.springframework.core.env.ConfigurableEnvironment;
+import org.springframework.core.env.MapPropertySource;
+import org.springframework.core.env.MutablePropertySources;
+import org.springframework.core.env.PropertySource;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.springframework.cloud.bus.SpringCloudBusClient.INPUT;
+
+/**
+ * The lowest precedence {@link EnvironmentPostProcessor} configures default RocketMQ Bus Properties that will be
+ * appended into {@link SpringApplication#defaultProperties}
+ *
+ * @author <a href="mailto:mercyblitz@gmail.com">Mercy</a>
+ * @see BusEnvironmentPostProcessor
+ * @since 0.2.1
+ */
+public class RocketMQBusEnvironmentPostProcessor implements EnvironmentPostProcessor, Ordered {
+
+    /**
+     * The name of {@link PropertySource} of {@link SpringApplication#defaultProperties}
+     */
+    private static final String PROPERTY_SOURCE_NAME = "defaultProperties";
+
+    @Override
+    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
+
+        addDefaultPropertySource(environment);
+
+    }
+
+    private void addDefaultPropertySource(ConfigurableEnvironment environment) {
+
+        Map<String, Object> map = new HashMap<String, Object>();
+
+        configureDefaultProperties(map);
+
+        addOrReplace(environment.getPropertySources(), map);
+    }
+
+    private void configureDefaultProperties(Map<String, Object> source) {
+        // Required Properties
+        String groupBindingPropertyName = createBindingPropertyName(INPUT, "group");
+        source.put(groupBindingPropertyName, "rocketmq-bus-group");
+    }
+
+    private String createBindingPropertyName(String channel, String propertyName) {
+        return "spring.cloud.stream.bindings." + channel + "." + propertyName;
+    }
+
+    /**
+     * Copy from {@link BusEnvironmentPostProcessor#addOrReplace(MutablePropertySources, Map)}
+     *
+     * @param propertySources {@link MutablePropertySources}
+     * @param map             Default RocketMQ Bus Properties
+     */
+    private void addOrReplace(MutablePropertySources propertySources,
+                              Map<String, Object> map) {
+        MapPropertySource target = null;
+        if (propertySources.contains(PROPERTY_SOURCE_NAME)) {
+            PropertySource<?> source = propertySources.get(PROPERTY_SOURCE_NAME);
+            if (source instanceof MapPropertySource) {
+                target = (MapPropertySource) source;
+                for (String key : map.keySet()) {
+                    if (!target.containsProperty(key)) {
+                        target.getSource().put(key, map.get(key));
+                    }
+                }
+            }
+        }
+        if (target == null) {
+            target = new MapPropertySource(PROPERTY_SOURCE_NAME, map);
+        }
+        if (!propertySources.contains(PROPERTY_SOURCE_NAME)) {
+            propertySources.addLast(target);
+        }
+    }
+
+    @Override
+    public int getOrder() {
+        return LOWEST_PRECEDENCE;
+    }
+}

spring-cloud-starter-alibaba/spring-cloud-starter-bus-rocketmq/src/main/resources/META-INF/spring.factories

@@ -0,0 +1,3 @@
+# EnvironmentPostProcessor
+org.springframework.boot.env.EnvironmentPostProcessor=\
+org.springframework.cloud.bus.rocketmq.env.RocketMQBusEnvironmentPostProcessor