WIP: Autorest generated code and some changes

sdk/search/azure-search-documents/pom.xml

@@ -107,4 +107,22 @@
       <scope>test</scope>
     </dependency>
   </dependencies>
+  <!-- plugins -->
+  <build>
+    <plugins>
+      <plugin>
+        <groupId>org.revapi</groupId>
+        <artifactId>revapi-maven-plugin</artifactId>
+        <configuration>
+          <oldArtifacts>
+            <artifact>com.azure:azure-search-documents:11.7.10</artifact>
+          </oldArtifacts>
+          <newArtifacts>
+            <artifact>com.azure:azure-search-documents:11.8.0-beta.9</artifact>
+          </newArtifacts>
+          <failBuildOnProblemsFound>true</failBuildOnProblemsFound>
+        </configuration>
+      </plugin>
+    </plugins>
+  </build>
 </project>

sdk/search/azure-search-documents/search/src/main/java/com/azure/search/KnowledgeRetrievals.java

@@ -0,0 +1,161 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+// Code generated by Microsoft (R) AutoRest Code Generator.
+// Changes may cause incorrect behavior and will be lost if the code is regenerated.
+
+package com.azure.search;
+
+import com.azure.core.annotation.BodyParam;
+import com.azure.core.annotation.ExpectedResponses;
+import com.azure.core.annotation.HeaderParam;
+import com.azure.core.annotation.Host;
+import com.azure.core.annotation.HostParam;
+import com.azure.core.annotation.Post;
+import com.azure.core.annotation.QueryParam;
+import com.azure.core.annotation.ReturnType;
+import com.azure.core.annotation.ServiceInterface;
+import com.azure.core.annotation.ServiceMethod;
+import com.azure.core.annotation.UnexpectedResponseExceptionType;
+import com.azure.core.http.rest.Response;
+import com.azure.core.http.rest.RestProxy;
+import com.azure.core.util.Context;
+import com.azure.core.util.FluxUtil;
+import com.azure.search.models.ErrorResponseException;
+import com.azure.search.models.KnowledgeBaseRetrievalRequest;
+import com.azure.search.models.KnowledgeBaseRetrievalResponse;
+import com.azure.search.models.RequestOptions;
+import java.util.UUID;
+import reactor.core.publisher.Mono;
+
+/**
+ * An instance of this class provides access to all the operations defined in KnowledgeRetrievals.
+ */
+public final class KnowledgeRetrievals {
+    /**
+     * The proxy service used to perform REST calls.
+     */
+    private final KnowledgeRetrievalsService service;
+
+    /**
+     * The service client containing this operation class.
+     */
+    private final SearchClient client;
+
+    /**
+     * Initializes an instance of KnowledgeRetrievals.
+     * 
+     * @param client the instance of the service client containing this operation class.
+     */
+    KnowledgeRetrievals(SearchClient client) {
+        this.service = RestProxy.create(KnowledgeRetrievalsService.class, client.getHttpPipeline(),
+            client.getSerializerAdapter());
+        this.client = client;
+    }
+
+    /**
+     * The interface defining all the services for SearchClientKnowledgeRetrievals to be used by the proxy service to
+     * perform REST calls.
+     */
+    @Host("{endpoint}/knowledgebases('{knowledgeBaseName}')")
+    @ServiceInterface(name = "SearchClientKnowledgeRetrievals")
+    public interface KnowledgeRetrievalsService {
+        @Post("/retrieve")
+        @ExpectedResponses({ 200, 206 })
+        @UnexpectedResponseExceptionType(ErrorResponseException.class)
+        Mono<Response<KnowledgeBaseRetrievalResponse>> retrieve(@HostParam("endpoint") String endpoint,
+            @HostParam("knowledgeBaseName") String knowledgeBaseName,
+            @HeaderParam("x-ms-client-request-id") UUID xMsClientRequestId,
+            @QueryParam("api-version") String apiVersion,
+            @HeaderParam("x-ms-query-source-authorization") String xMsQuerySourceAuthorization,
+            @BodyParam("application/json") KnowledgeBaseRetrievalRequest retrievalRequest,
+            @HeaderParam("Accept") String accept, Context context);
+    }
+
+    /**
+     * KnowledgeBase retrieves relevant data from backing stores.
+     * 
+     * @param retrievalRequest The retrieval request to process.
+     * @param xMsQuerySourceAuthorization Token identifying the user for which the query is being executed. This token
+     * is used to enforce security restrictions on documents.
+     * @param requestOptions Parameter group.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws ErrorResponseException thrown if the request is rejected by server.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return the output contract for the retrieval response along with {@link Response} on successful completion of
+     * {@link Mono}.
+     */
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    public Mono<Response<KnowledgeBaseRetrievalResponse>> retrieveWithResponseAsync(
+        KnowledgeBaseRetrievalRequest retrievalRequest, String xMsQuerySourceAuthorization,
+        RequestOptions requestOptions) {
+        return FluxUtil.withContext(context -> retrieveWithResponseAsync(retrievalRequest, xMsQuerySourceAuthorization,
+            requestOptions, context));
+    }
+
+    /**
+     * KnowledgeBase retrieves relevant data from backing stores.
+     * 
+     * @param retrievalRequest The retrieval request to process.
+     * @param xMsQuerySourceAuthorization Token identifying the user for which the query is being executed. This token
+     * is used to enforce security restrictions on documents.
+     * @param requestOptions Parameter group.
+     * @param context The context to associate with this operation.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws ErrorResponseException thrown if the request is rejected by server.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return the output contract for the retrieval response along with {@link Response} on successful completion of
+     * {@link Mono}.
+     */
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    public Mono<Response<KnowledgeBaseRetrievalResponse>> retrieveWithResponseAsync(
+        KnowledgeBaseRetrievalRequest retrievalRequest, String xMsQuerySourceAuthorization,
+        RequestOptions requestOptions, Context context) {
+        final String accept = "application/json";
+        UUID xMsClientRequestIdInternal = null;
+        if (requestOptions != null) {
+            xMsClientRequestIdInternal = requestOptions.getXMsClientRequestId();
+        }
+        UUID xMsClientRequestId = xMsClientRequestIdInternal;
+        return service.retrieve(this.client.getEndpoint(), this.client.getKnowledgeBaseName(), xMsClientRequestId,
+            this.client.getApiVersion(), xMsQuerySourceAuthorization, retrievalRequest, accept, context);
+    }
+
+    /**
+     * KnowledgeBase retrieves relevant data from backing stores.
+     * 
+     * @param retrievalRequest The retrieval request to process.
+     * @param xMsQuerySourceAuthorization Token identifying the user for which the query is being executed. This token
+     * is used to enforce security restrictions on documents.
+     * @param requestOptions Parameter group.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws ErrorResponseException thrown if the request is rejected by server.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return the output contract for the retrieval response on successful completion of {@link Mono}.
+     */
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    public Mono<KnowledgeBaseRetrievalResponse> retrieveAsync(KnowledgeBaseRetrievalRequest retrievalRequest,
+        String xMsQuerySourceAuthorization, RequestOptions requestOptions) {
+        return retrieveWithResponseAsync(retrievalRequest, xMsQuerySourceAuthorization, requestOptions)
+            .flatMap(res -> Mono.justOrEmpty(res.getValue()));
+    }
+
+    /**
+     * KnowledgeBase retrieves relevant data from backing stores.
+     * 
+     * @param retrievalRequest The retrieval request to process.
+     * @param xMsQuerySourceAuthorization Token identifying the user for which the query is being executed. This token
+     * is used to enforce security restrictions on documents.
+     * @param requestOptions Parameter group.
+     * @param context The context to associate with this operation.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws ErrorResponseException thrown if the request is rejected by server.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return the output contract for the retrieval response on successful completion of {@link Mono}.
+     */
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    public Mono<KnowledgeBaseRetrievalResponse> retrieveAsync(KnowledgeBaseRetrievalRequest retrievalRequest,
+        String xMsQuerySourceAuthorization, RequestOptions requestOptions, Context context) {
+        return retrieveWithResponseAsync(retrievalRequest, xMsQuerySourceAuthorization, requestOptions, context)
+            .flatMap(res -> Mono.justOrEmpty(res.getValue()));
+    }
+}