updating/creating markets

src/main/java/io/dietz/ed/companion/component/ZeroMqHandler.java

@@ -1,6 +1,7 @@
 package io.dietz.ed.companion.component;
 
 import java.io.IOException;
+import java.time.LocalDateTime;
 import java.util.zip.DataFormatException;
 import java.util.zip.Inflater;
 
@@ -13,6 +14,9 @@ import org.zeromq.ZContext;
 import org.zeromq.ZMQ;
 
 import io.dietz.ed.companion.domain.MarketWithRoot;
+import io.dietz.ed.companion.models.eddb.Market;
+import io.dietz.ed.companion.repositories.MarketRepository;
+import io.dietz.ed.companion.repositories.StationRepository;
 
 public class ZeroMqHandler implements Runnable {
 
@@ -28,6 +32,12 @@ public class ZeroMqHandler implements Runnable {
     @Autowired
     private ObjectMapper objectMapper;
 
+    @Autowired
+    private MarketRepository marketRepository;
+
+    @Autowired
+    private StationRepository stationRepository;
+
     public void msg(String msg) {
         System.out.println(msg);
     }
@@ -66,9 +76,24 @@ public class ZeroMqHandler implements Runnable {
                             
                             // outputString contains a json message
                             if (outputString.contains(filterSchema)) {
-                                msg(outputString);
-                                var test = objectMapper.readValue(outputString, MarketWithRoot.class);
-                                msg(test.getMarket().toString());
+                                //msg(outputString);
+
+                                MarketWithRoot marketContainer = objectMapper.readValue(outputString, MarketWithRoot.class);
+
+                                msg(marketContainer.getMarket().toString());
+
+                                var test = marketRepository.findById(marketContainer.getMarket().marketId).orElseGet(() -> {
+                                    msg("Market created!");
+                                    Market market = new Market(marketContainer.getMarket().marketId);
+                                    marketRepository.saveAndFlush(market);
+                                    return market;
+                                });
+
+                                if(test.getStation() != null) {
+                                    test.getStation().setUpdatedAt(LocalDateTime.now());
+                                    stationRepository.saveAndFlush(test.getStation());
+                                    msg(test.getStation().getName() + " updated!");
+                                }
                             }
 
                         } catch (DataFormatException | IOException e) {