Allow scoping ContentGridLinkCollector to specific entities

contentgrid-spring-data-rest/src/main/java/com/contentgrid/spring/data/rest/links/AggregateLinkCollector.java

@@ -7,7 +7,7 @@
 @RequiredArgsConstructor
 class AggregateLinkCollector implements LinkCollector {
     private final LinkCollector delegate;
-    private final Iterable<ContentGridLinkCollector> collectors;
+    private final Iterable<ContentGridLinkCollector<?>> collectors;
 
     @Override
     public Links getLinksFor(Object object) {
@@ -18,7 +18,10 @@ public Links getLinksFor(Object object) {
     public Links getLinksFor(Object object, Links existing) {
         existing = delegate.getLinksFor(object, existing);
         for (var collector : collectors) {
-            existing = collector.getLinksFor(object, existing);
+            final Links finalExisting = existing;
+            existing = LambdaSafe.callback(ContentGridLinkCollector.class, collector, object, finalExisting)
+                    .invokeAnd(c -> c.getLinksFor(object, finalExisting))
+                    .get(finalExisting);
         }
         return existing;
     }
@@ -27,7 +30,10 @@ public Links getLinksFor(Object object, Links existing) {
     public Links getLinksForNested(Object object, Links existing) {
         existing = delegate.getLinksForNested(object, existing);
         for (var collector : collectors) {
-            existing = collector.getLinksForNested(object, existing);
+            final Links finalExisting = existing;
+            existing = LambdaSafe.callback(ContentGridLinkCollector.class, collector, object, finalExisting)
+                    .invokeAnd(c -> c.getLinksForNested(object, finalExisting))
+                    .get(finalExisting);
         }
         return existing;
     }

contentgrid-spring-data-rest/src/main/java/com/contentgrid/spring/data/rest/links/ContentGridLinkCollector.java

@@ -2,9 +2,11 @@
 
 import org.springframework.hateoas.Links;
 
-public interface ContentGridLinkCollector {
+public interface ContentGridLinkCollector<T> {
 
-    Links getLinksFor(Object object, Links existing);
+    Links getLinksFor(T object, Links existing);
 
-    Links getLinksForNested(Object object, Links existing);
+    default Links getLinksForNested(T object, Links existing) {
+        return existing;
+    }
 }

contentgrid-spring-data-rest/src/main/java/com/contentgrid/spring/data/rest/links/ContentGridSpringContentRestLinksConfiguration.java

@@ -15,7 +15,7 @@
 @Import(ContentGridSpringDataLinksConfiguration.class)
 public class ContentGridSpringContentRestLinksConfiguration {
     @Bean
-    ContentGridLinkCollector contentGridSpringContentLinkCollector(
+    ContentGridLinkCollector<?> contentGridSpringContentLinkCollector(
             PersistentEntities entities, Stores stores, MappingContext mappingContext,
             RestConfiguration restConfiguration, ContentPropertyToRequestMappingContext requestMappingContext,
             ContentPropertyToLinkrelMappingContext linkrelMappingContext, MessageResolver resolver

contentgrid-spring-data-rest/src/main/java/com/contentgrid/spring/data/rest/links/ContentGridSpringDataLinksConfiguration.java

@@ -29,7 +29,7 @@ public class ContentGridSpringDataLinksConfiguration {
 
     @Bean
     RepositoryRestConfigurer contentGridLinkCollectorConfigurer(
-            ObjectProvider<ContentGridLinkCollector> collectors
+            ObjectProvider<ContentGridLinkCollector<?>> collectors
     ) {
         return new RepositoryRestConfigurer() {
             @Override
@@ -40,7 +40,8 @@ public LinkCollector customizeLinkCollector(LinkCollector collector) {
     }
 
     @Bean
-    ContentGridLinkCollector contentGridRelationLinkCollector(PersistentEntities entities, Associations associations, SelfLinkProvider selfLinkProvider, MessageResolver resolver) {
+    ContentGridLinkCollector<?> contentGridRelationLinkCollector(PersistentEntities entities, Associations associations,
+            SelfLinkProvider selfLinkProvider, MessageResolver resolver) {
         return new SpringDataAssociationLinkCollector(entities, associations, selfLinkProvider, resolver);
     }