@ServiceMethod(returns = ReturnType.COLLECTION)
public SearchPagedFlux search(String searchText, SearchOptions searchOptions) {
    SearchRequest request = createSearchRequest(searchText, searchOptions);
    // The firstPageResponse shared among all functional calls below.
    // Do not initial new instance directly in func call.
    final SearchFirstPageResponseWrapper firstPageResponse = new SearchFirstPageResponseWrapper();
    Function<String, Mono<SearchPagedResponse>> func = continuationToken -> withContext(
        context -> search(request, continuationToken, firstPageResponse, context));
    return new SearchPagedFlux(() -> func.apply(null), func);
}

private Mono<SearchPagedResponse> search(SearchRequest request, String continuationToken,
    SearchFirstPageResponseWrapper firstPageResponseWrapper, Context context) {
    if (continuationToken == null && firstPageResponseWrapper.getFirstPageResponse() != null) {
        return Mono.just(firstPageResponseWrapper.getFirstPageResponse());  // First subscriber response always returned here
    }
    SearchRequest requestToUse = (continuationToken == null)
        ? request
        : SearchContinuationToken.deserializeToken(serviceVersion.getVersion(), continuationToken);

    return restClient.getDocuments()
        .searchPostWithResponseAsync(requestToUse, null, context)
        .onErrorMap(MappingUtils::exceptionMapper)
        .map(response -> {
            SearchDocumentsResult result = response.getValue();

            SearchPagedResponse page
                = new SearchPagedResponse(new SimpleResponse<>(response, getSearchResults(result, serializer)),
                    createContinuationToken(result, serviceVersion), result.getFacets(), result.getCount(),
                    result.getCoverage(), result.getAnswers(), result.getSemanticPartialResponseReason(),
                    result.getSemanticPartialResponseType());
            if (continuationToken == null) {
                firstPageResponseWrapper.setFirstPageResponse(page); // First subscriber response set here
            }
            return page;
        });
}