feat: Allows to search subscriptions by api key

gravitee-management-api-model/src/main/java/io/gravitee/management/model/subscription/SubscriptionQuery.java

@@ -35,6 +35,8 @@ public class SubscriptionQuery {
 
     private Collection<String> applications;
 
+    private String apiKey;
+
     private long from, to;
 
     public Collection<String> getApis() {
@@ -81,6 +83,14 @@ public void setApplication(String application) {
         this.applications = Collections.singleton(application);
     }
 
+    public String getApiKey() {
+        return apiKey;
+    }
+
+    public void setApiKey(String apiKey) {
+        this.apiKey = apiKey;
+    }
+
     public long getFrom() {
         return from;
     }
@@ -107,11 +117,12 @@ public boolean equals(Object o) {
                 Objects.equals(apis, that.apis) &&
                 Objects.equals(plans, that.plans) &&
                 Objects.equals(statuses, that.statuses) &&
-                Objects.equals(applications, that.applications);
+                Objects.equals(applications, that.applications) &&
+                Objects.equals(apiKey, that.apiKey);
     }
 
     @Override
     public int hashCode() {
-        return Objects.hash(apis, plans, statuses, applications, from, to);
+        return Objects.hash(apis, plans, statuses, applications, apiKey, from, to);
     }
 }

gravitee-management-api-rest/src/main/java/io/gravitee/management/rest/resource/ApiSubscriptionsResource.java

@@ -207,6 +207,9 @@ private static class SubscriptionParam {
         @ApiModelProperty(dataType = "string", allowableValues = "accepted, pending, rejected, closed", value = "Subscription status")
         private ListSubscriptionStatusParam status;
 
+        @QueryParam("api_key")
+        private String apiKey;
+
         public String getApi() {
             return api;
         }
@@ -239,6 +242,14 @@ public void setStatus(ListSubscriptionStatusParam status) {
             this.status = status;
         }
 
+        public String getApiKey() {
+            return apiKey;
+        }
+
+        public void setApiKey(String apiKey) {
+            this.apiKey = apiKey;
+        }
+
         private SubscriptionQuery toQuery() {
             SubscriptionQuery query = new SubscriptionQuery();
 
@@ -256,6 +267,8 @@ private SubscriptionQuery toQuery() {
                 query.setApplications(applications.getValue());
             }
 
+            query.setApiKey(this.apiKey);
+
             return query;
         }
     }

gravitee-management-api-rest/src/main/java/io/gravitee/management/rest/resource/ApplicationSubscriptionsResource.java

@@ -297,6 +297,9 @@ private static class SubscriptionParam {
         @ApiModelProperty(dataType = "string", allowableValues = "accepted, pending, rejected, closed", value = "Subscription status")
         private ListSubscriptionStatusParam status;
 
+        @QueryParam("api_key")
+        private String apiKey;
+
         public ListStringParam getPlans() {
             return plans;
         }
@@ -329,6 +332,14 @@ public void setStatus(ListSubscriptionStatusParam status) {
             this.status = status;
         }
 
+        public String getApiKey() {
+            return apiKey;
+        }
+
+        public void setApiKey(String apiKey) {
+            this.apiKey = apiKey;
+        }
+
         private SubscriptionQuery toQuery() {
             SubscriptionQuery query = new SubscriptionQuery();
 
@@ -346,6 +357,9 @@ private SubscriptionQuery toQuery() {
                 query.setStatuses(status.getStatus());
             }
 
+            if (apiKey != null) {
+                query.setApiKey(apiKey);
+            }
             return query;
         }
     }

gravitee-management-api-service/src/main/java/io/gravitee/management/service/impl/SubscriptionServiceImpl.java

@@ -45,11 +45,14 @@
 import java.util.*;
 import java.util.function.Predicate;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 import static io.gravitee.repository.management.model.Audit.AuditProperties.API;
 import static io.gravitee.repository.management.model.Audit.AuditProperties.APPLICATION;
 import static io.gravitee.repository.management.model.Subscription.AuditEvent.*;
 import static java.lang.System.lineSeparator;
+import static java.util.Collections.emptyList;
+import static java.util.stream.Collectors.toList;
 
 /**
  * @author David BRASSELY (david.brassely at graviteesource.com)
@@ -119,7 +122,7 @@ public Collection<SubscriptionEntity> findByApplicationAndPlan(String applicatio
             query.setApplication(application);
         } else if (isAuthenticated()) {
             Set<ApplicationListItem> applications = applicationService.findByUser(getAuthenticatedUsername());
-            query.setApplications(applications.stream().map(ApplicationListItem::getId).collect(Collectors.toList()));
+            query.setApplications(applications.stream().map(ApplicationListItem::getId).collect(toList()));
         }
 
         return search(query);
@@ -702,10 +705,15 @@ public Collection<SubscriptionEntity> search(SubscriptionQuery query) {
                         .collect(Collectors.toSet()));
             }
 
-            List<SubscriptionEntity> subscriptions = subscriptionRepository.search(builder.build())
-                    .stream().map(this::convert).collect(Collectors.toList());
-            return subscriptions;
-
+            Stream<SubscriptionEntity> subscriptionsStream =
+                    subscriptionRepository.search(builder.build()).stream().map(this::convert);
+            if (query.getApiKey() != null && !query.getApiKey().isEmpty()) {
+                subscriptionsStream = subscriptionsStream.filter(subscriptionEntity -> {
+                    final Set<ApiKeyEntity> apiKeys = apiKeyService.findBySubscription(subscriptionEntity.getId());
+                    return apiKeys.stream().anyMatch(apiKeyEntity -> apiKeyEntity.getKey().equals(query.getApiKey()));
+                });
+            }
+            return subscriptionsStream.collect(toList());
         } catch (TechnicalException ex) {
             logger.error("An error occurs while trying to search for subscriptions: {}", query, ex);
             throw new TechnicalManagementException(
@@ -739,11 +747,20 @@ public Page<SubscriptionEntity> search(SubscriptionQuery query, Pageable pageabl
                                     .pageSize(pageable.getPageSize())
                                     .build());
 
-            List<SubscriptionEntity> content = pageSubscription.getContent()
-                    .stream().map(this::convert).collect(Collectors.toList());
+            Stream<SubscriptionEntity> subscriptionsStream = pageSubscription.getContent().stream().map(this::convert);
 
-            return new Page<>(content, pageSubscription.getPageNumber() + 1,
-                    (int) pageSubscription.getPageElements(), pageSubscription.getTotalElements());
+            if (query.getApiKey() != null && !query.getApiKey().isEmpty()) {
+                subscriptionsStream = subscriptionsStream.filter(subscriptionEntity -> {
+                    final Set<ApiKeyEntity> apiKeys = apiKeyService.findBySubscription(subscriptionEntity.getId());
+                    return apiKeys.stream().anyMatch(apiKeyEntity -> apiKeyEntity.getKey().equals(query.getApiKey()));
+                });
+                final Optional<SubscriptionEntity> sub = subscriptionsStream.findAny();
+                List<SubscriptionEntity> subscriptionEntities = sub.map(Collections::singletonList).orElse(emptyList());
+                return new Page<>(subscriptionEntities, 1, subscriptionEntities.size(), subscriptionEntities.size());
+            } else {
+                return new Page<>(subscriptionsStream.collect(toList()), pageSubscription.getPageNumber() + 1,
+                        (int) pageSubscription.getPageElements(), pageSubscription.getTotalElements());
+            }
         } catch (TechnicalException ex) {
             logger.error("An error occurs while trying to search for pageable subscriptions: {}", query, ex);
             throw new TechnicalManagementException(