From 042fc85d0824e20f7b8d82cb51dc440615884470 Mon Sep 17 00:00:00 2001
From: "alexander.kerscher" <alexander.kerscher@muenchen.de>
Date: Fri, 5 Apr 2024 14:06:29 +0200
Subject: [PATCH] refactor

---
 .../dave/domain/mapper/detektor/MessstelleMapper.java    | 1 +
 .../dave/repositories/elasticsearch/MessstelleIndex.java | 1 +
 .../dave/services/messstelle/MessstelleIndexService.java | 8 ++++++--
 .../dave/services/messstelle/MessstelleService.java      | 9 +++++----
 4 files changed, 13 insertions(+), 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
index 43b679dc..c3aa6a3f 100644
--- 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
index 20ffd323..3e86d4e1 100644
--- 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
index 7c6e29ed..3b794386 100644
--- 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
index bfbd0114..1cdd7888 100644
--- 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);
     }
 }