Skip to content

Commit df2a7ba

Browse files
authored
Merge pull request #127 from adinovri/feat_batchGetEnrichActivities
feat: add capability in batch client to get enrichment activities to use enrichment flags
2 parents e57d6c0 + dad14c7 commit df2a7ba

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

src/main/java/io/getstream/client/BatchClient.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import io.getstream.core.exceptions.StreamException;
99
import io.getstream.core.http.Token;
1010
import io.getstream.core.models.*;
11+
import io.getstream.core.options.EnrichmentFlags;
1112
import io.getstream.core.utils.DefaultOptions;
1213
import java.util.List;
1314
import java8.util.J8Arrays;
@@ -91,8 +92,13 @@ public CompletableFuture<List<EnrichedActivity>> getEnrichedActivitiesByID(
9192

9293
public CompletableFuture<List<EnrichedActivity>> getEnrichedActivitiesByID(String... activityIDs)
9394
throws StreamException {
95+
return getEnrichedActivitiesByID(DefaultOptions.DEFAULT_ENRICHMENT_FLAGS, activityIDs);
96+
}
97+
98+
public CompletableFuture<List<EnrichedActivity>> getEnrichedActivitiesByID(
99+
EnrichmentFlags flags, String... activityIDs) throws StreamException {
94100
final Token token = buildActivityToken(secret, TokenAction.READ);
95-
return batch.getEnrichedActivitiesByID(token, activityIDs);
101+
return batch.getEnrichedActivitiesByID(token, flags, activityIDs);
96102
}
97103

98104
public CompletableFuture<List<Activity>> getActivitiesByForeignID(

src/main/java/io/getstream/core/StreamBatch.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import io.getstream.core.http.Token;
1515
import io.getstream.core.models.*;
1616
import io.getstream.core.options.CustomQueryParameter;
17+
import io.getstream.core.options.EnrichmentFlags;
1718
import io.getstream.core.options.RequestOption;
1819
import java.io.IOException;
1920
import java.net.MalformedURLException;
@@ -151,7 +152,7 @@ public CompletableFuture<List<Activity>> getActivitiesByID(Token token, String..
151152
}
152153

153154
public CompletableFuture<List<EnrichedActivity>> getEnrichedActivitiesByID(
154-
Token token, String... activityIDs) throws StreamException {
155+
Token token, EnrichmentFlags flags, String... activityIDs) throws StreamException {
155156
checkNotNull(activityIDs, "No activities to get");
156157
checkArgument(activityIDs.length > 0, "No activities to get");
157158

@@ -163,6 +164,7 @@ public CompletableFuture<List<EnrichedActivity>> getEnrichedActivitiesByID(
163164
url,
164165
key,
165166
token,
167+
flags,
166168
new CustomQueryParameter("ids", Joiner.on(",").join(activityIDs))))
167169
.thenApply(
168170
response -> {

0 commit comments

Comments
 (0)