Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -292,15 +292,19 @@ for (String phoneNumber: result.getPhoneNumbers()) {
```

### Purchase Search
<!-- embedme ./src/samples/java/com/azure/communication/administration/ReadmeSamples.java#L385-L391 -->
<!-- embedme ./src/samples/java/com/azure/communication/administration/ReadmeSamples.java#L385-L395 -->
```java
Duration duration = Duration.ofSeconds(1);
String phoneNumberSearchId = "SEARCH_ID_TO_PURCHASE";
PhoneNumberClient phoneNumberClient = createPhoneNumberClient();

SyncPoller<Void, Void> res =
SyncPoller<PhoneNumberSearch, PhoneNumberSearch> res =
phoneNumberClient.beginPurchaseSearch(phoneNumberSearchId, duration);
res.waitForCompletion();
PhoneNumberSearch result = res.getFinalResult();

System.out.println("Search Id: " + result.getSearchId());
System.out.println("Purchase status: " + result.getStatus());
```

## Contributing
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -847,48 +847,53 @@ Mono<PhoneNumberSearch>> createSearchFetchResultOperation() {
*/

@ServiceMethod(returns = ReturnType.COLLECTION)
public PollerFlux<Void, Void> beginPurchaseSearch(String searchId, Duration pollInterval) {
public PollerFlux<PhoneNumberSearch,
PhoneNumberSearch> beginPurchaseSearch(String searchId, Duration pollInterval) {
Objects.requireNonNull(searchId, "'searchId' can not be null.");
Objects.requireNonNull(pollInterval, "'pollInterval' can not be null.");

if (pollInterval == null) {
pollInterval = Duration.ofSeconds(5);
Comment thread
jbeauregardb marked this conversation as resolved.
Outdated
}

return new PollerFlux<Void, Void>(pollInterval,
return new PollerFlux<PhoneNumberSearch, PhoneNumberSearch>(pollInterval,
purchaseSearchActivationOperation(searchId),
purchaseSearchPollOperation(searchId),
(activationResponse, pollingContext) -> Mono.error(new RuntimeException("Cancellation is not supported")),
purchaseSearchFetchResultOperation());
}

private Function<PollingContext<Void>, Mono<Void>> purchaseSearchActivationOperation(String searchId) {

private Function<PollingContext<PhoneNumberSearch>,
Mono<PhoneNumberSearch>> purchaseSearchActivationOperation(String searchId) {
Comment thread
jbeauregardb marked this conversation as resolved.
Outdated
return (pollingContext) -> {
Mono<Void> response = purchaseSearch(searchId);
return response;
Mono<PhoneNumberSearch> res = purchaseSearch(searchId).flatMap(agh -> {
Comment thread
jbeauregardb marked this conversation as resolved.
Outdated
return getSearchById(searchId);
});
return res;
};
}

private Function<PollingContext<Void>, Mono<PollResponse<Void>>>
private Function<PollingContext<PhoneNumberSearch>, Mono<PollResponse<PhoneNumberSearch>>>
purchaseSearchPollOperation(String searchId) {
return (pollingContext) -> getSearchById(searchId)
.flatMap(getSearchResponse -> {
SearchStatus statusResponse = getSearchResponse.getStatus();
if (statusResponse.equals(SearchStatus.RESERVED)
|| statusResponse.equals(SearchStatus.EXPIRED)
if (statusResponse.equals(SearchStatus.EXPIRED)
|| statusResponse.equals(SearchStatus.SUCCESS)) {
return Mono.just(new PollResponse<>(
LongRunningOperationStatus.SUCCESSFULLY_COMPLETED, null));
LongRunningOperationStatus.SUCCESSFULLY_COMPLETED, getSearchResponse));
}
if (statusResponse.equals(SearchStatus.ERROR)) {
return Mono.just(new PollResponse<>(
LongRunningOperationStatus.FAILED, null));
LongRunningOperationStatus.FAILED, getSearchResponse));
}
return Mono.just(new PollResponse<>(LongRunningOperationStatus.IN_PROGRESS, null));
return Mono.just(new PollResponse<>(LongRunningOperationStatus.IN_PROGRESS, getSearchResponse));
});
}

private Function<PollingContext<Void>,
Mono<Void>> purchaseSearchFetchResultOperation() {
private Function<PollingContext<PhoneNumberSearch>,
Mono<PhoneNumberSearch>> purchaseSearchFetchResultOperation() {
return pollingContext -> {
return Mono.empty();
return Mono.just(pollingContext.getLatestResponse().getValue());
};

}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -550,7 +550,7 @@ public SyncPoller<PhoneNumberSearch, PhoneNumberSearch> beginCreateSearch(
* @return A {@link SyncPoller} object with the search result
*/
@ServiceMethod(returns = ReturnType.COLLECTION)
public SyncPoller<Void, Void> beginPurchaseSearch(
public SyncPoller<PhoneNumberSearch, PhoneNumberSearch> beginPurchaseSearch(
String searchId, Duration pollInterval) {
return phoneNumberAsyncClient.beginPurchaseSearch(searchId, pollInterval).getSyncPoller();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -386,8 +386,12 @@ public void beginPurchaseSearch() {
String phoneNumberSearchId = "SEARCH_ID_TO_PURCHASE";
PhoneNumberClient phoneNumberClient = createPhoneNumberClient();

SyncPoller<Void, Void> res =
SyncPoller<PhoneNumberSearch, PhoneNumberSearch> res =
phoneNumberClient.beginPurchaseSearch(phoneNumberSearchId, duration);
res.waitForCompletion();
PhoneNumberSearch result = res.getFinalResult();

System.out.println("Search Id: " + result.getSearchId());
System.out.println("Purchase status: " + result.getStatus());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -531,15 +531,15 @@ public void beginCreateSearch(HttpClient httpClient) {
@ParameterizedTest
@MethodSource("com.azure.core.test.TestBase#getHttpClients")
public void beginPurchaseSearch(HttpClient httpClient) {
Duration pollInterval = Duration.ofSeconds(5);
PollerFlux<Void, Void> poller =
this.getClient(httpClient).beginPurchaseSearch(SEARCH_ID, pollInterval);
poller.takeUntil(apr -> apr.getStatus() == LongRunningOperationStatus.SUCCESSFULLY_COMPLETED);
Mono<PhoneNumberSearch> mono = this.getClient(httpClient).getSearchById(SEARCH_ID);
StepVerifier.create(mono)
.assertNext(item -> {
assertEquals(SearchStatus.SUCCESS, item.getStatus());
});
Duration pollInterval = Duration.ofSeconds(1);
PhoneNumberAsyncClient client = this.getClient(httpClient);
PollerFlux<PhoneNumberSearch, PhoneNumberSearch> poller =
client.beginPurchaseSearch(SEARCH_ID, pollInterval);
AsyncPollResponse<PhoneNumberSearch, PhoneNumberSearch> asyncRes =
poller.takeUntil(apr -> apr.getStatus() == LongRunningOperationStatus.SUCCESSFULLY_COMPLETED)
.blockLast();
PhoneNumberSearch testResult = asyncRes.getValue();
assertEquals(SearchStatus.SUCCESS, testResult.getStatus());
}

private PhoneNumberAsyncClient getClient(HttpClient httpClient) {
Expand Down
Loading