refactor

it-at-m · Apr 5, 2024 · 042fc85 · 042fc85
1 parent 18e13ae
commit 042fc85
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 6 deletions.
diff --git a/src/main/java/de/muenchen/dave/domain/mapper/detektor/MessstelleMapper.java b/src/main/java/de/muenchen/dave/domain/mapper/detektor/MessstelleMapper.java
@@ -129,6 +129,7 @@ default void bean2overviewDtoAftermapping(@MappingTarget MessstelleOverviewDTO d
         dto.setStadtbezirk(Stadtbezirk.bezeichnungOf(bean.getStadtbezirkNummer()));
     }
 
+    // TODO Test schreiben
     List<MessstelleAuswertungDTO> bean2auswertungDto(List<Messstelle> bean);
 
     List<MessquerschnittAuswertungDTO> bean2auswertungMqDto(List<Messquerschnitt> bean);

diff --git a/src/main/java/de/muenchen/dave/repositories/elasticsearch/MessstelleIndex.java b/src/main/java/de/muenchen/dave/repositories/elasticsearch/MessstelleIndex.java
@@ -41,6 +41,7 @@ public interface MessstelleIndex extends ElasticsearchRepository<Messstelle, Str
     Page<Messstelle> suggestSearch(String query, Pageable pageable);
 
     List<Messstelle> findAll();
+    List<Messstelle> findAllBySichtbarDatenportalIsTrue();
 
     Optional<Messstelle> findByMstId(String mstId);
 

diff --git a/src/main/java/de/muenchen/dave/services/messstelle/MessstelleIndexService.java b/src/main/java/de/muenchen/dave/services/messstelle/MessstelleIndexService.java
@@ -3,12 +3,13 @@
 import de.muenchen.dave.domain.elasticsearch.detektor.Messstelle;
 import de.muenchen.dave.exceptions.ResourceNotFoundException;
 import de.muenchen.dave.repositories.elasticsearch.MessstelleIndex;
-import java.util.List;
-import java.util.Optional;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 
+import java.util.List;
+import java.util.Optional;
+
 /**
  * Die Klasse {@link MessstelleIndexService} holt alle relevanten Messstellen aus MobidaM und
  * aktualisiert die in Dave gespeichereten Daten.
@@ -38,5 +39,8 @@ public Messstelle findByIdOrThrowException(final String messstelleId) {
     public List<Messstelle> findAllMessstellen() {
         return messstelleIndex.findAll();
     }
+    public List<Messstelle> findAllVisibleMessstellen() {
+        return messstelleIndex.findAllBySichtbarDatenportalIsTrue();
+    }
 
 }
diff --git a/src/main/java/de/muenchen/dave/services/messstelle/MessstelleService.java b/src/main/java/de/muenchen/dave/services/messstelle/MessstelleService.java
@@ -8,13 +8,14 @@
 import de.muenchen.dave.domain.elasticsearch.detektor.Messstelle;
 import de.muenchen.dave.domain.mapper.detektor.MessstelleMapper;
 import de.muenchen.dave.services.CustomSuggestIndexService;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
 /**
  * Die Klasse {@link MessstelleService} holt alle relevanten Messstellen aus MobidaM und
  * aktualisiert die in Dave gespeichereten Daten.
@@ -74,7 +75,7 @@ public boolean isKfzMessstelle(final String messstelleId) {
 
     public List<MessstelleAuswertungDTO> getAllVisibleMessstellenForAuswertung() {
         log.debug("#getAllVisibleMessstellenForAuswertung");
-        final List<Messstelle> messstellen = messstelleIndexService.findAllMessstellen();
+        final List<Messstelle> messstellen = messstelleIndexService.findAllVisibleMessstellen();
         return messstelleMapper.bean2auswertungDto(messstellen);
     }
 }