Generate compose file for rabbitmq streams

Signed-off-by: Eddú Meléndez <eddu.melendez@gmail.com>

Author: eddumelendez

start-site/src/main/java/io/spring/start/site/extension/dependency/springamqp/SpringAmqpProjectGenerationConfiguration.java

@@ -16,9 +16,14 @@
 
 package io.spring.start.site.extension.dependency.springamqp;
 
+import java.util.function.Consumer;
+
 import io.spring.initializr.generator.buildsystem.Build;
 import io.spring.initializr.generator.condition.ConditionalOnPlatformVersion;
 import io.spring.initializr.generator.condition.ConditionalOnRequestedDependency;
+import io.spring.initializr.generator.container.docker.compose.ComposeConfig;
+import io.spring.initializr.generator.container.docker.compose.ComposeService;
+import io.spring.initializr.generator.container.docker.compose.ComposeServiceConfig;
 import io.spring.initializr.generator.project.ProjectGenerationConfiguration;
 import io.spring.start.site.container.ComposeFileCustomizer;
 import io.spring.start.site.container.DockerServiceResolver;
@@ -68,18 +73,36 @@ ServiceConnectionsCustomizer rabbitServiceConnectionsCustomizer(Build build,
 		@ConditionalOnRequestedDependency("docker-compose")
 		ComposeFileCustomizer rabbitComposeFileCustomizer(Build build, DockerServiceResolver serviceResolver) {
 			return (composeFile) -> {
-				if (isAmqpEnabled(build)) {
-					serviceResolver.doWith("rabbit", (service) -> composeFile.services()
-						.add("rabbitmq",
-								service.andThen((builder) -> builder.environment("RABBITMQ_DEFAULT_USER", "myuser")
-									.environment("RABBITMQ_DEFAULT_PASS", "secret")
-									.ports(5672))));
-				}
+				serviceResolver.doWith("rabbit", (service) -> {
+					if (isAmqpEnabled(build)) {
+						Consumer<ComposeService.Builder> composeService = (builder) -> builder
+							.environment("RABBITMQ_DEFAULT_USER", "myuser")
+							.environment("RABBITMQ_DEFAULT_PASS", "secret")
+							.ports(5672);
+						if (isAmqpStreamsEnabled(build)) {
+							composeFile.configs()
+								.add("plugins", ComposeConfig.Builder.forContent("[rabbitmq_stream].").build());
+							composeFile.services()
+								.add("rabbitmq",
+										service.andThen(composeService)
+											.andThen((builder) -> builder.ports(5552)
+												.config(ComposeServiceConfig.ofLong("plugins",
+														"/etc/rabbitmq/enabled_plugins"))));
+						}
+						else {
+							composeFile.services().add("rabbitmq", service.andThen(composeService));
+						}
+					}
+				});
 			};
 		}
 
 		private boolean isAmqpEnabled(Build build) {
-			return build.dependencies().has("amqp") || build.dependencies().has("amqp-streams");
+			return build.dependencies().has("amqp");
+		}
+
+		private boolean isAmqpStreamsEnabled(Build build) {
+			return build.dependencies().has("amqp-streams");
 		}
 
 	}

start-site/src/test/java/io/spring/start/site/extension/dependency/springamqp/SpringAmqpProjectGenerationConfigurationTests.java

@@ -84,7 +84,7 @@ void springAmqpStreamsWithoutDockerCompose() {
 	@Test
 	void springAmqpStreamsWithDockerCompose() {
 		ProjectRequest request = createProjectRequest("docker-compose", "amqp-streams");
-		assertThat(composeFile(request)).hasSameContentAs(new ClassPathResource("compose/rabbitmq.yaml"));
+		assertThat(composeFile(request)).hasSameContentAs(new ClassPathResource("compose/rabbitmq-streams.yaml"));
 	}
 
 }

start-site/src/test/resources/compose/rabbitmq-streams.yaml

@@ -0,0 +1,16 @@
+services:
+  rabbitmq:
+    image: 'rabbitmq:latest'
+    environment:
+      - 'RABBITMQ_DEFAULT_PASS=secret'
+      - 'RABBITMQ_DEFAULT_USER=myuser'
+    ports:
+      - '5552'
+      - '5672'
+    configs:
+      - source: "plugins"
+        target: "/etc/rabbitmq/enabled_plugins"
+
+configs:
+  plugins:
+    content: "[rabbitmq_stream]."