Skip to content

Commit 5d09964

Browse files
committed
Remove usage of parallelStream in favor of stream
1 parent a67f585 commit 5d09964

File tree

4 files changed

+11
-18
lines changed

4 files changed

+11
-18
lines changed

gradle.properties

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
version=0.9.2-SNAPSHOT
1+
version=0.9.3-SNAPSHOT
22
ossrh.username=tecton-team

src/main/java/ai/tecton/client/request/GetFeaturesBatchRequest.java

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -198,8 +198,7 @@ public GetFeaturesBatchRequest(
198198
// For batch requests, partition the requestDataList into n sublists of size
199199
// microBatchSize and create GetFeaturesMicroBatchRequest for each
200200
this.requestList =
201-
ListUtils.partition(requestDataList, microBatchSize)
202-
.parallelStream()
201+
ListUtils.partition(requestDataList, microBatchSize).stream()
203202
.map(
204203
requestData ->
205204
new GetFeaturesMicroBatchRequest(
@@ -212,8 +211,7 @@ public GetFeaturesBatchRequest(
212211
} else {
213212
// For microBatchSize=1, create a List of individual GetFeaturesRequest objects
214213
this.requestList =
215-
requestDataList
216-
.parallelStream()
214+
requestDataList.stream()
217215
.map(
218216
requestData ->
219217
new GetFeaturesRequest(
@@ -400,7 +398,7 @@ private static void validateParameters(
400398
if (requestDataList == null || requestDataList.isEmpty()) {
401399
throw new InvalidRequestParameterException(TectonErrorMessage.INVALID_REQUEST_DATA_LIST);
402400
}
403-
requestDataList.parallelStream().forEach(AbstractGetFeaturesRequest::validateRequestParameters);
401+
requestDataList.forEach(AbstractGetFeaturesRequest::validateRequestParameters);
404402
if (microBatchSize > RequestConstants.MAX_MICRO_BATCH_SIZE || microBatchSize < 1) {
405403
throw new InvalidRequestParameterException(
406404
String.format(

src/main/java/ai/tecton/client/response/GetFeaturesBatchResponse.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,16 +49,14 @@ public GetFeaturesBatchResponse(
4949

5050
// Serialize list of HttpResponse into list of GetFeaturesMicroBatchResponse
5151
List<GetFeaturesMicroBatchResponse> microBatchResponses =
52-
httpResponseList
53-
.parallelStream()
52+
httpResponseList.stream()
5453
.map(httpResponse -> parseSingleHttpResponse(httpResponse, microBatchSize))
5554
.collect(Collectors.toList());
5655

5756
// Concatenate list of GetFeaturesResponse objects from each microbatch into a single list
5857
// Maintain ordering
5958
this.batchResponseList =
60-
microBatchResponses
61-
.parallelStream()
59+
microBatchResponses.stream()
6260
.map(microBatch -> microBatch.microBatchResponseList)
6361
.flatMap(List::stream)
6462
.collect(Collectors.toList());

src/main/java/ai/tecton/client/transport/TectonHttpClient.java

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,7 @@ public List<HttpResponse> performParallelRequests(
114114
// Map request body to OkHttp Request
115115
// ordering of requests is maintained
116116
List<Request> requestList =
117-
requestBodyList
118-
.parallelStream()
117+
requestBodyList.stream()
119118
.map(
120119
requestBody ->
121120
new HttpRequest(url.url().toString(), endpoint, method, apiKey, requestBody))
@@ -153,12 +152,10 @@ public void onResponse(Call call, Response response) {
153152
};
154153

155154
// Enqueue all calls
156-
requestList
157-
.parallelStream()
158-
.forEach(
159-
request -> {
160-
client.newCall(request).enqueue(callback);
161-
});
155+
requestList.forEach(
156+
request -> {
157+
client.newCall(request).enqueue(callback);
158+
});
162159

163160
// Wait until A) all calls have completed or B) specified timeout has elapsed
164161
try {

0 commit comments

Comments
 (0)