X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FQueryApplication.java;fp=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FQueryApplication.java;h=813d3b28212acc5b059c2ae4df41b8f45fc535fa;hb=8b1fd41f91be906ac0c691b4df7c528cad9ff583;hp=995c1a13a1b8f4d99103bf7d8ae37bd21d1589ec;hpb=0d1ed8c65f422e5c0cd291c35dba67b1f2205997;p=demos%2Fkafka%2Fwordcount diff --git a/src/main/java/de/juplo/kafka/wordcount/query/QueryApplication.java b/src/main/java/de/juplo/kafka/wordcount/query/QueryApplication.java index 995c1a1..813d3b2 100644 --- a/src/main/java/de/juplo/kafka/wordcount/query/QueryApplication.java +++ b/src/main/java/de/juplo/kafka/wordcount/query/QueryApplication.java @@ -1,14 +1,56 @@ package de.juplo.kafka.wordcount.query; +import com.fasterxml.jackson.databind.ObjectMapper; +import org.apache.kafka.streams.state.HostInfo; import org.springframework.boot.SpringApplication; import org.springframework.boot.autoconfigure.SpringBootApplication; +import org.springframework.boot.autoconfigure.web.ServerProperties; import org.springframework.boot.context.properties.EnableConfigurationProperties; +import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.context.annotation.Bean; + +import java.io.IOException; +import java.net.InetSocketAddress; +import java.net.Socket; @SpringBootApplication @EnableConfigurationProperties(QueryApplicationProperties.class) public class QueryApplication { + @Bean + public QueryStreamProcessor usersStreamProcessor( + ServerProperties serverProperties, + QueryApplicationProperties properties, + ObjectMapper mapper, + ConfigurableApplicationContext context) throws IOException + { + String host; + if (serverProperties.getAddress() == null) + { + HostInfo bootstrapServer = HostInfo.buildFromEndpoint(properties.getBootstrapServer()); + Socket socket = new Socket(); + socket.connect(new InetSocketAddress(bootstrapServer.host(), bootstrapServer.port())); + host = socket.getLocalAddress().getHostAddress(); + } + else + { + host = serverProperties.getAddress().getHostAddress(); + } + + Integer port = serverProperties.getPort() == null ? 8080 : serverProperties.getPort(); + + return new QueryStreamProcessor( + properties.getApplicationId(), + new HostInfo(host, port), + properties.getBootstrapServer(), + properties.getUsersInputTopic(), + properties.getRankingInputTopic(), + mapper, + context); + } + + public static void main(String[] args) { SpringApplication.run(QueryApplication.class, args);