Properly start db writers

Author: Artem

src/main/java/com/codenameart/rocketmerger/Webhook.java

@@ -3,9 +3,11 @@
 import com.codenameart.rocketmerger.q.DBQueue;
 import com.codenameart.rocketmerger.q.DBWriter;
 import lombok.extern.log4j.Log4j2;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
+import org.springframework.context.ApplicationContext;
 import org.springframework.context.annotation.ComponentScan;
 import org.springframework.http.HttpStatus;
 import org.springframework.stereotype.Controller;
@@ -16,11 +18,13 @@
 import javax.annotation.PostConstruct;
 import java.util.List;
 import java.util.TimeZone;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
 
 /**
  * Created by Artem on 18.12.2017.
  */
-@Log4j2
+@Slf4j
 @Controller
 @EnableAutoConfiguration
 @ComponentScan
@@ -32,6 +36,9 @@ public class Webhook {
     @Autowired
     DBWriter dbWriter;
 
+    @Autowired
+    private ApplicationContext context;
+
     public static void main(String[] args) throws Exception {
         SpringApplication.run(Webhook.class, args);
     }
@@ -40,13 +47,16 @@ public static void main(String[] args) throws Exception {
     @ResponseStatus(HttpStatus.OK)
     void process(@RequestBody List<Message> payload) {
         TimeZone.setDefault(TimeZone.getTimeZone("UTC"));
+        log.debug("Message received");
         payload.stream().map(Message::getMessage).forEach(dbQueue::offer);
     }
 
     @PostConstruct
     public void startDBLoop() {
+        ExecutorService executorService = Executors.newFixedThreadPool(5);
         for (int i = 0; i < 10; i++) {
-            dbWriter.startLoop();
+            DBWriter bean = context.getBean(DBWriter.class);
+            executorService.execute(bean);
         }
     }
 }

src/main/java/com/codenameart/rocketmerger/q/DBQueue.java

@@ -23,4 +23,8 @@ public boolean offer(Pokemon pokemon) {
     public Pokemon poll() {
         return queue.poll();
     }
+
+    public int size() {
+        return queue.size();
+    }
 }

src/main/java/com/codenameart/rocketmerger/q/DBWriter.java

@@ -2,15 +2,16 @@
 
 import com.codenameart.rocketmerger.Pokemon;
 import com.codenameart.rocketmerger.PokemonRepository;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.scheduling.annotation.Async;
 
 import java.util.Date;
 
 /**
  * Created by Artem on 23.12.2017.
  */
-public class DBWriter {
+@Slf4j
+public class DBWriter implements Runnable {
     @Autowired
     DBQueue queue;
 
@@ -20,8 +21,9 @@ public class DBWriter {
     PokemonRepository pokemonRepository;
 
 
-    @Async
-    public void startLoop(){
+    @Override
+    public void run() {
+        log.info("Starting queue DBWriter");
         while (true) {
             if (Thread.currentThread().isInterrupted()) {
                 return;
@@ -36,6 +38,7 @@ public void startLoop(){
                 }
             } else {
                 pokemon.setLast_modified(new Date());
+                log.info("Queue length: " + queue.size());
                 pokemonRepository.save(pokemon);
             }
         }