diff --git a/src/main/java/io/getstream/client/BatchClient.java b/src/main/java/io/getstream/client/BatchClient.java index 62a6def0..e9cd30be 100644 --- a/src/main/java/io/getstream/client/BatchClient.java +++ b/src/main/java/io/getstream/client/BatchClient.java @@ -8,6 +8,7 @@ import io.getstream.core.exceptions.StreamException; import io.getstream.core.http.Token; import io.getstream.core.models.*; +import io.getstream.core.options.EnrichmentFlags; import io.getstream.core.utils.DefaultOptions; import java.util.List; import java8.util.J8Arrays; @@ -91,8 +92,13 @@ public CompletableFuture> getEnrichedActivitiesByID( public CompletableFuture> getEnrichedActivitiesByID(String... activityIDs) throws StreamException { + return getEnrichedActivitiesByID(DefaultOptions.DEFAULT_ENRICHMENT_FLAGS, activityIDs); + } + + public CompletableFuture> getEnrichedActivitiesByID( + EnrichmentFlags flags, String... activityIDs) throws StreamException { final Token token = buildActivityToken(secret, TokenAction.READ); - return batch.getEnrichedActivitiesByID(token, activityIDs); + return batch.getEnrichedActivitiesByID(token, flags, activityIDs); } public CompletableFuture> getActivitiesByForeignID( diff --git a/src/main/java/io/getstream/core/StreamBatch.java b/src/main/java/io/getstream/core/StreamBatch.java index 1abf5535..e171b572 100644 --- a/src/main/java/io/getstream/core/StreamBatch.java +++ b/src/main/java/io/getstream/core/StreamBatch.java @@ -14,6 +14,7 @@ import io.getstream.core.http.Token; import io.getstream.core.models.*; import io.getstream.core.options.CustomQueryParameter; +import io.getstream.core.options.EnrichmentFlags; import io.getstream.core.options.RequestOption; import java.io.IOException; import java.net.MalformedURLException; @@ -151,7 +152,7 @@ public CompletableFuture> getActivitiesByID(Token token, String.. } public CompletableFuture> getEnrichedActivitiesByID( - Token token, String... activityIDs) throws StreamException { + Token token, EnrichmentFlags flags, String... activityIDs) throws StreamException { checkNotNull(activityIDs, "No activities to get"); checkArgument(activityIDs.length > 0, "No activities to get"); @@ -163,6 +164,7 @@ public CompletableFuture> getEnrichedActivitiesByID( url, key, token, + flags, new CustomQueryParameter("ids", Joiner.on(",").join(activityIDs)))) .thenApply( response -> {