feat(repository): introduce search by layername parameter

terrestris · Oct 9, 2024 · c0b157f · c0b157f
1 parent e94264c
commit c0b157f
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/shogun-lib/src/main/java/de/terrestris/shogun/lib/repository/LayerRepository.java b/shogun-lib/src/main/java/de/terrestris/shogun/lib/repository/LayerRepository.java
@@ -17,8 +17,26 @@
 package de.terrestris.shogun.lib.repository;
 
 import de.terrestris.shogun.lib.model.Layer;
+import java.util.List;
+import java.util.Optional;
 import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
 import org.springframework.stereotype.Repository;
 
 @Repository
-public interface LayerRepository extends BaseCrudRepository<Layer, Long>, JpaSpecificationExecutor<Layer> { }
+public interface LayerRepository extends BaseCrudRepository<Layer, Long>, JpaSpecificationExecutor<Layer> {
+
+    @Query(
+        value = "SELECT * FROM shogun.layers WHERE source_config ->> 'layerNames' = :layerName LIMIT 1",
+        nativeQuery = true
+    )
+    Optional<Layer> findFirstByLayerName(@Param("layerName") String layerName);
+
+    @Query(
+        value = "SELECT * FROM shogun.layers WHERE source_config ->> 'layerNames' = :layerName",
+        nativeQuery = true
+    )
+    List<Layer> findByLayerName(@Param("layerName") String layerName);
+
+}