/* * Copyright 2022 Google LLC * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * https://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.google.cloud.discoveryengine.v1beta.stub; import static com.google.cloud.discoveryengine.v1beta.SchemaServiceClient.ListSchemasPagedResponse; import com.google.api.HttpRule; import com.google.api.core.BetaApi; import com.google.api.core.InternalApi; import com.google.api.gax.core.BackgroundResource; import com.google.api.gax.core.BackgroundResourceAggregation; import com.google.api.gax.httpjson.ApiMethodDescriptor; import com.google.api.gax.httpjson.HttpJsonCallSettings; import com.google.api.gax.httpjson.HttpJsonOperationSnapshot; import com.google.api.gax.httpjson.HttpJsonStubCallableFactory; import com.google.api.gax.httpjson.ProtoMessageRequestFormatter; import com.google.api.gax.httpjson.ProtoMessageResponseParser; import com.google.api.gax.httpjson.ProtoRestSerializer; import com.google.api.gax.httpjson.longrunning.stub.HttpJsonOperationsStub; import com.google.api.gax.rpc.ClientContext; import com.google.api.gax.rpc.OperationCallable; import com.google.api.gax.rpc.UnaryCallable; import com.google.cloud.discoveryengine.v1beta.CreateSchemaMetadata; import com.google.cloud.discoveryengine.v1beta.CreateSchemaRequest; import com.google.cloud.discoveryengine.v1beta.DeleteSchemaMetadata; import com.google.cloud.discoveryengine.v1beta.DeleteSchemaRequest; import com.google.cloud.discoveryengine.v1beta.GetSchemaRequest; import com.google.cloud.discoveryengine.v1beta.ListSchemasRequest; import com.google.cloud.discoveryengine.v1beta.ListSchemasResponse; import com.google.cloud.discoveryengine.v1beta.Schema; import com.google.cloud.discoveryengine.v1beta.UpdateSchemaMetadata; import com.google.cloud.discoveryengine.v1beta.UpdateSchemaRequest; import com.google.common.collect.ImmutableMap; import com.google.longrunning.Operation; import com.google.protobuf.Empty; import com.google.protobuf.TypeRegistry; import java.io.IOException; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.concurrent.TimeUnit; import javax.annotation.Generated; // AUTO-GENERATED DOCUMENTATION AND CLASS. /** * REST stub implementation for the SchemaService service API. * *

This class is for advanced usage and reflects the underlying API directly. */ @BetaApi @Generated("by gapic-generator-java") public class HttpJsonSchemaServiceStub extends SchemaServiceStub { private static final TypeRegistry typeRegistry = TypeRegistry.newBuilder() .add(Empty.getDescriptor()) .add(CreateSchemaMetadata.getDescriptor()) .add(DeleteSchemaMetadata.getDescriptor()) .add(Schema.getDescriptor()) .add(UpdateSchemaMetadata.getDescriptor()) .build(); private static final ApiMethodDescriptor getSchemaMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1beta.SchemaService/GetSchema") .setHttpMethod("GET") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1beta/{name=projects/*/locations/*/dataStores/*/schemas/*}", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "name", request.getName()); return fields; }) .setAdditionalPaths( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/schemas/*}") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor(request -> null) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(Schema.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .build(); private static final ApiMethodDescriptor listSchemasMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1beta.SchemaService/ListSchemas") .setHttpMethod("GET") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1beta/{parent=projects/*/locations/*/dataStores/*}/schemas", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "parent", request.getParent()); return fields; }) .setAdditionalPaths( "/v1beta/{parent=projects/*/locations/*/collections/*/dataStores/*}/schemas") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "pageSize", request.getPageSize()); serializer.putQueryParam(fields, "pageToken", request.getPageToken()); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor(request -> null) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(ListSchemasResponse.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .build(); private static final ApiMethodDescriptor createSchemaMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1beta.SchemaService/CreateSchema") .setHttpMethod("POST") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1beta/{parent=projects/*/locations/*/dataStores/*}/schemas", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "parent", request.getParent()); return fields; }) .setAdditionalPaths( "/v1beta/{parent=projects/*/locations/*/collections/*/dataStores/*}/schemas") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "schemaId", request.getSchemaId()); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor( request -> ProtoRestSerializer.create() .toBody("schema", request.getSchema(), true)) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(Operation.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .setOperationSnapshotFactory( (CreateSchemaRequest request, Operation response) -> HttpJsonOperationSnapshot.create(response)) .build(); private static final ApiMethodDescriptor updateSchemaMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1beta.SchemaService/UpdateSchema") .setHttpMethod("PATCH") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1beta/{schema.name=projects/*/locations/*/dataStores/*/schemas/*}", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam( fields, "schema.name", request.getSchema().getName()); return fields; }) .setAdditionalPaths( "/v1beta/{schema.name=projects/*/locations/*/collections/*/dataStores/*/schemas/*}") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam( fields, "allowMissing", request.getAllowMissing()); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor( request -> ProtoRestSerializer.create() .toBody("schema", request.getSchema(), true)) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(Operation.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .setOperationSnapshotFactory( (UpdateSchemaRequest request, Operation response) -> HttpJsonOperationSnapshot.create(response)) .build(); private static final ApiMethodDescriptor deleteSchemaMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1beta.SchemaService/DeleteSchema") .setHttpMethod("DELETE") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1beta/{name=projects/*/locations/*/dataStores/*/schemas/*}", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "name", request.getName()); return fields; }) .setAdditionalPaths( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/schemas/*}") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor(request -> null) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(Operation.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .setOperationSnapshotFactory( (DeleteSchemaRequest request, Operation response) -> HttpJsonOperationSnapshot.create(response)) .build(); private final UnaryCallable getSchemaCallable; private final UnaryCallable listSchemasCallable; private final UnaryCallable listSchemasPagedCallable; private final UnaryCallable createSchemaCallable; private final OperationCallable createSchemaOperationCallable; private final UnaryCallable updateSchemaCallable; private final OperationCallable updateSchemaOperationCallable; private final UnaryCallable deleteSchemaCallable; private final OperationCallable deleteSchemaOperationCallable; private final BackgroundResource backgroundResources; private final HttpJsonOperationsStub httpJsonOperationsStub; private final HttpJsonStubCallableFactory callableFactory; public static final HttpJsonSchemaServiceStub create(SchemaServiceStubSettings settings) throws IOException { return new HttpJsonSchemaServiceStub(settings, ClientContext.create(settings)); } public static final HttpJsonSchemaServiceStub create(ClientContext clientContext) throws IOException { return new HttpJsonSchemaServiceStub( SchemaServiceStubSettings.newHttpJsonBuilder().build(), clientContext); } public static final HttpJsonSchemaServiceStub create( ClientContext clientContext, HttpJsonStubCallableFactory callableFactory) throws IOException { return new HttpJsonSchemaServiceStub( SchemaServiceStubSettings.newHttpJsonBuilder().build(), clientContext, callableFactory); } /** * Constructs an instance of HttpJsonSchemaServiceStub, using the given settings. This is * protected so that it is easy to make a subclass, but otherwise, the static factory methods * should be preferred. */ protected HttpJsonSchemaServiceStub( SchemaServiceStubSettings settings, ClientContext clientContext) throws IOException { this(settings, clientContext, new HttpJsonSchemaServiceCallableFactory()); } /** * Constructs an instance of HttpJsonSchemaServiceStub, using the given settings. This is * protected so that it is easy to make a subclass, but otherwise, the static factory methods * should be preferred. */ protected HttpJsonSchemaServiceStub( SchemaServiceStubSettings settings, ClientContext clientContext, HttpJsonStubCallableFactory callableFactory) throws IOException { this.callableFactory = callableFactory; this.httpJsonOperationsStub = HttpJsonOperationsStub.create( clientContext, callableFactory, typeRegistry, ImmutableMap.builder() .put( "google.longrunning.Operations.GetOperation", HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/branches/*/operations/*}") .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/models/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/dataStores/*/branches/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/dataStores/*/models/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/dataStores/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1beta/{name=projects/*/locations/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1beta/{name=projects/*/operations/*}") .build()) .build()) .put( "google.longrunning.Operations.ListOperations", HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/branches/*}/operations") .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*/models/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*/dataStores/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/collections/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/dataStores/*/branches/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/dataStores/*/models/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1beta/{name=projects/*/locations/*/dataStores/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1beta/{name=projects/*/locations/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1beta/{name=projects/*}/operations") .build()) .build()) .build()); HttpJsonCallSettings getSchemaTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(getSchemaMethodDescriptor) .setTypeRegistry(typeRegistry) .build(); HttpJsonCallSettings listSchemasTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(listSchemasMethodDescriptor) .setTypeRegistry(typeRegistry) .build(); HttpJsonCallSettings createSchemaTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(createSchemaMethodDescriptor) .setTypeRegistry(typeRegistry) .build(); HttpJsonCallSettings updateSchemaTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(updateSchemaMethodDescriptor) .setTypeRegistry(typeRegistry) .build(); HttpJsonCallSettings deleteSchemaTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(deleteSchemaMethodDescriptor) .setTypeRegistry(typeRegistry) .build(); this.getSchemaCallable = callableFactory.createUnaryCallable( getSchemaTransportSettings, settings.getSchemaSettings(), clientContext); this.listSchemasCallable = callableFactory.createUnaryCallable( listSchemasTransportSettings, settings.listSchemasSettings(), clientContext); this.listSchemasPagedCallable = callableFactory.createPagedCallable( listSchemasTransportSettings, settings.listSchemasSettings(), clientContext); this.createSchemaCallable = callableFactory.createUnaryCallable( createSchemaTransportSettings, settings.createSchemaSettings(), clientContext); this.createSchemaOperationCallable = callableFactory.createOperationCallable( createSchemaTransportSettings, settings.createSchemaOperationSettings(), clientContext, httpJsonOperationsStub); this.updateSchemaCallable = callableFactory.createUnaryCallable( updateSchemaTransportSettings, settings.updateSchemaSettings(), clientContext); this.updateSchemaOperationCallable = callableFactory.createOperationCallable( updateSchemaTransportSettings, settings.updateSchemaOperationSettings(), clientContext, httpJsonOperationsStub); this.deleteSchemaCallable = callableFactory.createUnaryCallable( deleteSchemaTransportSettings, settings.deleteSchemaSettings(), clientContext); this.deleteSchemaOperationCallable = callableFactory.createOperationCallable( deleteSchemaTransportSettings, settings.deleteSchemaOperationSettings(), clientContext, httpJsonOperationsStub); this.backgroundResources = new BackgroundResourceAggregation(clientContext.getBackgroundResources()); } @InternalApi public static List getMethodDescriptors() { List methodDescriptors = new ArrayList<>(); methodDescriptors.add(getSchemaMethodDescriptor); methodDescriptors.add(listSchemasMethodDescriptor); methodDescriptors.add(createSchemaMethodDescriptor); methodDescriptors.add(updateSchemaMethodDescriptor); methodDescriptors.add(deleteSchemaMethodDescriptor); return methodDescriptors; } public HttpJsonOperationsStub getHttpJsonOperationsStub() { return httpJsonOperationsStub; } @Override public UnaryCallable getSchemaCallable() { return getSchemaCallable; } @Override public UnaryCallable listSchemasCallable() { return listSchemasCallable; } @Override public UnaryCallable listSchemasPagedCallable() { return listSchemasPagedCallable; } @Override public UnaryCallable createSchemaCallable() { return createSchemaCallable; } @Override public OperationCallable createSchemaOperationCallable() { return createSchemaOperationCallable; } @Override public UnaryCallable updateSchemaCallable() { return updateSchemaCallable; } @Override public OperationCallable updateSchemaOperationCallable() { return updateSchemaOperationCallable; } @Override public UnaryCallable deleteSchemaCallable() { return deleteSchemaCallable; } @Override public OperationCallable deleteSchemaOperationCallable() { return deleteSchemaOperationCallable; } @Override public final void close() { try { backgroundResources.close(); } catch (RuntimeException e) { throw e; } catch (Exception e) { throw new IllegalStateException("Failed to close resource", e); } } @Override public void shutdown() { backgroundResources.shutdown(); } @Override public boolean isShutdown() { return backgroundResources.isShutdown(); } @Override public boolean isTerminated() { return backgroundResources.isTerminated(); } @Override public void shutdownNow() { backgroundResources.shutdownNow(); } @Override public boolean awaitTermination(long duration, TimeUnit unit) throws InterruptedException { return backgroundResources.awaitTermination(duration, unit); } }