#23 fixed scheduling

zalando-nakadi · May 23, 2017 · 5ba13f1 · 5ba13f1
1 parent d513e28
commit 5ba13f1
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/...ing-boot-starter/src/main/java/org/zalando/nakadiproducer/EventTransmissionScheduler.java b/...ing-boot-starter/src/main/java/org/zalando/nakadiproducer/EventTransmissionScheduler.java
@@ -0,0 +1,26 @@
+package org.zalando.nakadiproducer;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+import org.zalando.nakadiproducer.transmission.impl.EventTransmitter;
+
+@Component
+public class EventTransmissionScheduler {
+    private final EventTransmitter eventTransmitter;
+    private final boolean scheduledTransmissionEnabled;
+
+    @Autowired
+    public EventTransmissionScheduler(EventTransmitter eventTransmitter, @Value("${nakadi-producer.scheduled-transmission-enabled:true}") boolean scheduledTransmissionEnabled) {
+        this.eventTransmitter = eventTransmitter;
+        this.scheduledTransmissionEnabled = scheduledTransmissionEnabled;
+    }
+
+    @Scheduled(fixedDelayString = "${nakadi-producer.transmission-polling-delay:1000}")
+    protected void sendEventsIfSchedulingEnabled() {
+        if (scheduledTransmissionEnabled) {
+            eventTransmitter.sendEvents();
+        }
+    }
+}
diff --git a/...oot-starter/src/main/java/org/zalando/nakadiproducer/NakadiProducerAutoConfiguration.java b/...oot-starter/src/main/java/org/zalando/nakadiproducer/NakadiProducerAutoConfiguration.java
@@ -29,6 +29,7 @@
 import org.springframework.context.annotation.Configuration;
 import org.springframework.context.annotation.Import;
 import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
+import org.springframework.scheduling.annotation.Scheduled;
 import org.zalando.fahrschein.NakadiClient;
 import org.zalando.nakadiproducer.eventlog.EventLogWriter;
 import org.zalando.nakadiproducer.eventlog.impl.EventLogRepository;
@@ -167,6 +168,7 @@ public EventTransmissionService eventTransmissionService(EventLogRepository even
         return new EventTransmissionService(eventLogRepository, nakadiPublishingClient, objectMapper);
     }
 
+
     @PostConstruct
     public void migrateFlyway() {
         Flyway flyway = new Flyway();