UNPKG

@google-cloud/tpu

Version:
917 lines 112 kB
"use strict"; // Copyright 2025 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. // // ** This file is automatically generated by gapic-generator-typescript. ** // ** https://github.com/googleapis/gapic-generator-typescript ** // ** All changes to this file may be overwritten. ** Object.defineProperty(exports, "__esModule", { value: true }); exports.TpuClient = void 0; const jsonProtos = require("../../protos/protos.json"); const google_gax_1 = require("google-gax"); /** * Client JSON configuration object, loaded from * `src/v2alpha1/tpu_client_config.json`. * This file defines retry strategy and timeouts for all API methods in this library. */ const gapicConfig = require("./tpu_client_config.json"); const version = require('../../../package.json').version; /** * Manages TPU nodes and other resources * * TPU API v2alpha1 * @class * @memberof v2alpha1 */ class TpuClient { _terminated = false; _opts; _providedCustomServicePath; _gaxModule; _gaxGrpc; _protos; _defaults; _universeDomain; _servicePath; _log = google_gax_1.loggingUtils.log('tpu'); auth; descriptors = { page: {}, stream: {}, longrunning: {}, batching: {}, }; warn; innerApiCalls; locationsClient; pathTemplates; operationsClient; tpuStub; /** * Construct an instance of TpuClient. * * @param {object} [options] - The configuration object. * The options accepted by the constructor are described in detail * in [this document](https://github.com/googleapis/gax-nodejs/blob/main/client-libraries.md#creating-the-client-instance). * The common options are: * @param {object} [options.credentials] - Credentials object. * @param {string} [options.credentials.client_email] * @param {string} [options.credentials.private_key] * @param {string} [options.email] - Account email address. Required when * using a .pem or .p12 keyFilename. * @param {string} [options.keyFilename] - Full path to the a .json, .pem, or * .p12 key downloaded from the Google Developers Console. If you provide * a path to a JSON file, the projectId option below is not necessary. * NOTE: .pem and .p12 require you to specify options.email as well. * @param {number} [options.port] - The port on which to connect to * the remote host. * @param {string} [options.projectId] - The project ID from the Google * Developer's Console, e.g. 'grape-spaceship-123'. We will also check * the environment variable GCLOUD_PROJECT for your project ID. If your * app is running in an environment which supports * {@link https://cloud.google.com/docs/authentication/application-default-credentials Application Default Credentials}, * your project ID will be detected automatically. * @param {string} [options.apiEndpoint] - The domain name of the * API remote host. * @param {gax.ClientConfig} [options.clientConfig] - Client configuration override. * Follows the structure of {@link gapicConfig}. * @param {boolean} [options.fallback] - Use HTTP/1.1 REST mode. * For more information, please check the * {@link https://github.com/googleapis/gax-nodejs/blob/main/client-libraries.md#http11-rest-api-mode documentation}. * @param {gax} [gaxInstance]: loaded instance of `google-gax`. Useful if you * need to avoid loading the default gRPC version and want to use the fallback * HTTP implementation. Load only fallback version and pass it to the constructor: * ``` * const gax = require('google-gax/build/src/fallback'); // avoids loading google-gax with gRPC * const client = new TpuClient({fallback: true}, gax); * ``` */ constructor(opts, gaxInstance) { // Ensure that options include all the required fields. const staticMembers = this.constructor; if (opts?.universe_domain && opts?.universeDomain && opts?.universe_domain !== opts?.universeDomain) { throw new Error('Please set either universe_domain or universeDomain, but not both.'); } const universeDomainEnvVar = (typeof process === 'object' && typeof process.env === 'object') ? process.env['GOOGLE_CLOUD_UNIVERSE_DOMAIN'] : undefined; this._universeDomain = opts?.universeDomain ?? opts?.universe_domain ?? universeDomainEnvVar ?? 'googleapis.com'; this._servicePath = 'tpu.' + this._universeDomain; const servicePath = opts?.servicePath || opts?.apiEndpoint || this._servicePath; this._providedCustomServicePath = !!(opts?.servicePath || opts?.apiEndpoint); const port = opts?.port || staticMembers.port; const clientConfig = opts?.clientConfig ?? {}; const fallback = opts?.fallback ?? (typeof window !== 'undefined' && typeof window?.fetch === 'function'); opts = Object.assign({ servicePath, port, clientConfig, fallback }, opts); // Request numeric enum values if REST transport is used. opts.numericEnums = true; // If scopes are unset in options and we're connecting to a non-default endpoint, set scopes just in case. if (servicePath !== this._servicePath && !('scopes' in opts)) { opts['scopes'] = staticMembers.scopes; } // Load google-gax module synchronously if needed if (!gaxInstance) { gaxInstance = require('google-gax'); } // Choose either gRPC or proto-over-HTTP implementation of google-gax. this._gaxModule = opts.fallback ? gaxInstance.fallback : gaxInstance; // Create a `gaxGrpc` object, with any grpc-specific options sent to the client. this._gaxGrpc = new this._gaxModule.GrpcClient(opts); // Save options to use in initialize() method. this._opts = opts; // Save the auth object to the client, for use by other methods. this.auth = this._gaxGrpc.auth; // Set useJWTAccessWithScope on the auth object. this.auth.useJWTAccessWithScope = true; // Set defaultServicePath on the auth object. this.auth.defaultServicePath = this._servicePath; // Set the default scopes in auth client if needed. if (servicePath === this._servicePath) { this.auth.defaultScopes = staticMembers.scopes; } this.locationsClient = new this._gaxModule.LocationsClient(this._gaxGrpc, opts); // Determine the client header string. const clientHeader = [ `gax/${this._gaxModule.version}`, `gapic/${version}`, ]; if (typeof process === 'object' && 'versions' in process) { clientHeader.push(`gl-node/${process.versions.node}`); } else { clientHeader.push(`gl-web/${this._gaxModule.version}`); } if (!opts.fallback) { clientHeader.push(`grpc/${this._gaxGrpc.grpcVersion}`); } else { clientHeader.push(`rest/${this._gaxGrpc.grpcVersion}`); } if (opts.libName && opts.libVersion) { clientHeader.push(`${opts.libName}/${opts.libVersion}`); } // Load the applicable protos. this._protos = this._gaxGrpc.loadProtoJSON(jsonProtos); // This API contains "path templates"; forward-slash-separated // identifiers to uniquely identify resources within the API. // Create useful helper objects for these. this.pathTemplates = { acceleratorTypePathTemplate: new this._gaxModule.PathTemplate('projects/{project}/locations/{location}/acceleratorTypes/{accelerator_type}'), locationPathTemplate: new this._gaxModule.PathTemplate('projects/{project}/locations/{location}'), nodePathTemplate: new this._gaxModule.PathTemplate('projects/{project}/locations/{location}/nodes/{node}'), projectPathTemplate: new this._gaxModule.PathTemplate('projects/{project}'), queuedResourcePathTemplate: new this._gaxModule.PathTemplate('projects/{project}/locations/{location}/queuedResources/{queued_resource}'), reservationPathTemplate: new this._gaxModule.PathTemplate('projects/{project}/locations/{location}/reservations/{reservation}'), runtimeVersionPathTemplate: new this._gaxModule.PathTemplate('projects/{project}/locations/{location}/runtimeVersions/{runtime_version}'), }; // Some of the methods on this service return "paged" results, // (e.g. 50 results at a time, with tokens to get subsequent // pages). Denote the keys used for pagination and results. this.descriptors.page = { listNodes: new this._gaxModule.PageDescriptor('pageToken', 'nextPageToken', 'nodes'), listQueuedResources: new this._gaxModule.PageDescriptor('pageToken', 'nextPageToken', 'queuedResources'), listAcceleratorTypes: new this._gaxModule.PageDescriptor('pageToken', 'nextPageToken', 'acceleratorTypes'), listRuntimeVersions: new this._gaxModule.PageDescriptor('pageToken', 'nextPageToken', 'runtimeVersions'), listReservations: new this._gaxModule.PageDescriptor('pageToken', 'nextPageToken', 'reservations') }; const protoFilesRoot = this._gaxModule.protobufFromJSON(jsonProtos); // This API contains "long-running operations", which return a // an Operation object that allows for tracking of the operation, // rather than holding a request open. const lroOptions = { auth: this.auth, grpc: 'grpc' in this._gaxGrpc ? this._gaxGrpc.grpc : undefined }; if (opts.fallback) { lroOptions.protoJson = protoFilesRoot; lroOptions.httpRules = [{ selector: 'google.cloud.location.Locations.GetLocation', get: '/v2alpha1/{name=projects/*/locations/*}', }, { selector: 'google.cloud.location.Locations.ListLocations', get: '/v2alpha1/{name=projects/*}/locations', }, { selector: 'google.longrunning.Operations.CancelOperation', post: '/v2alpha1/{name=projects/*/locations/*/operations/*}:cancel', }, { selector: 'google.longrunning.Operations.DeleteOperation', delete: '/v2alpha1/{name=projects/*/locations/*/operations/*}', }, { selector: 'google.longrunning.Operations.GetOperation', get: '/v2alpha1/{name=projects/*/locations/*/operations/*}', }, { selector: 'google.longrunning.Operations.ListOperations', get: '/v2alpha1/{name=projects/*/locations/*}/operations', }]; } this.operationsClient = this._gaxModule.lro(lroOptions).operationsClient(opts); const createNodeResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.Node'); const createNodeMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const deleteNodeResponse = protoFilesRoot.lookup('.google.protobuf.Empty'); const deleteNodeMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const stopNodeResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.Node'); const stopNodeMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const startNodeResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.Node'); const startNodeMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const updateNodeResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.Node'); const updateNodeMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const performMaintenanceResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.Node'); const performMaintenanceMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const createQueuedResourceResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.QueuedResource'); const createQueuedResourceMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const deleteQueuedResourceResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.QueuedResource'); const deleteQueuedResourceMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const resetQueuedResourceResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.QueuedResource'); const resetQueuedResourceMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const performMaintenanceQueuedResourceResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.QueuedResource'); const performMaintenanceQueuedResourceMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); const simulateMaintenanceEventResponse = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.Node'); const simulateMaintenanceEventMetadata = protoFilesRoot.lookup('.google.cloud.tpu.v2alpha1.OperationMetadata'); this.descriptors.longrunning = { createNode: new this._gaxModule.LongrunningDescriptor(this.operationsClient, createNodeResponse.decode.bind(createNodeResponse), createNodeMetadata.decode.bind(createNodeMetadata)), deleteNode: new this._gaxModule.LongrunningDescriptor(this.operationsClient, deleteNodeResponse.decode.bind(deleteNodeResponse), deleteNodeMetadata.decode.bind(deleteNodeMetadata)), stopNode: new this._gaxModule.LongrunningDescriptor(this.operationsClient, stopNodeResponse.decode.bind(stopNodeResponse), stopNodeMetadata.decode.bind(stopNodeMetadata)), startNode: new this._gaxModule.LongrunningDescriptor(this.operationsClient, startNodeResponse.decode.bind(startNodeResponse), startNodeMetadata.decode.bind(startNodeMetadata)), updateNode: new this._gaxModule.LongrunningDescriptor(this.operationsClient, updateNodeResponse.decode.bind(updateNodeResponse), updateNodeMetadata.decode.bind(updateNodeMetadata)), performMaintenance: new this._gaxModule.LongrunningDescriptor(this.operationsClient, performMaintenanceResponse.decode.bind(performMaintenanceResponse), performMaintenanceMetadata.decode.bind(performMaintenanceMetadata)), createQueuedResource: new this._gaxModule.LongrunningDescriptor(this.operationsClient, createQueuedResourceResponse.decode.bind(createQueuedResourceResponse), createQueuedResourceMetadata.decode.bind(createQueuedResourceMetadata)), deleteQueuedResource: new this._gaxModule.LongrunningDescriptor(this.operationsClient, deleteQueuedResourceResponse.decode.bind(deleteQueuedResourceResponse), deleteQueuedResourceMetadata.decode.bind(deleteQueuedResourceMetadata)), resetQueuedResource: new this._gaxModule.LongrunningDescriptor(this.operationsClient, resetQueuedResourceResponse.decode.bind(resetQueuedResourceResponse), resetQueuedResourceMetadata.decode.bind(resetQueuedResourceMetadata)), performMaintenanceQueuedResource: new this._gaxModule.LongrunningDescriptor(this.operationsClient, performMaintenanceQueuedResourceResponse.decode.bind(performMaintenanceQueuedResourceResponse), performMaintenanceQueuedResourceMetadata.decode.bind(performMaintenanceQueuedResourceMetadata)), simulateMaintenanceEvent: new this._gaxModule.LongrunningDescriptor(this.operationsClient, simulateMaintenanceEventResponse.decode.bind(simulateMaintenanceEventResponse), simulateMaintenanceEventMetadata.decode.bind(simulateMaintenanceEventMetadata)) }; // Put together the default options sent with requests. this._defaults = this._gaxGrpc.constructSettings('google.cloud.tpu.v2alpha1.Tpu', gapicConfig, opts.clientConfig || {}, { 'x-goog-api-client': clientHeader.join(' ') }); // Set up a dictionary of "inner API calls"; the core implementation // of calling the API is handled in `google-gax`, with this code // merely providing the destination and request information. this.innerApiCalls = {}; // Add a warn function to the client constructor so it can be easily tested. this.warn = this._gaxModule.warn; } /** * Initialize the client. * Performs asynchronous operations (such as authentication) and prepares the client. * This function will be called automatically when any class method is called for the * first time, but if you need to initialize it before calling an actual method, * feel free to call initialize() directly. * * You can await on this method if you want to make sure the client is initialized. * * @returns {Promise} A promise that resolves to an authenticated service stub. */ initialize() { // If the client stub promise is already initialized, return immediately. if (this.tpuStub) { return this.tpuStub; } // Put together the "service stub" for // google.cloud.tpu.v2alpha1.Tpu. this.tpuStub = this._gaxGrpc.createStub(this._opts.fallback ? this._protos.lookupService('google.cloud.tpu.v2alpha1.Tpu') : // eslint-disable-next-line @typescript-eslint/no-explicit-any this._protos.google.cloud.tpu.v2alpha1.Tpu, this._opts, this._providedCustomServicePath); // Iterate over each of the methods that the service provides // and create an API call method for each. const tpuStubMethods = ['listNodes', 'getNode', 'createNode', 'deleteNode', 'stopNode', 'startNode', 'updateNode', 'performMaintenance', 'listQueuedResources', 'getQueuedResource', 'createQueuedResource', 'deleteQueuedResource', 'resetQueuedResource', 'performMaintenanceQueuedResource', 'generateServiceIdentity', 'listAcceleratorTypes', 'getAcceleratorType', 'listRuntimeVersions', 'getRuntimeVersion', 'getGuestAttributes', 'listReservations', 'simulateMaintenanceEvent']; for (const methodName of tpuStubMethods) { const callPromise = this.tpuStub.then(stub => (...args) => { if (this._terminated) { return Promise.reject('The client has already been closed.'); } const func = stub[methodName]; return func.apply(stub, args); }, (err) => () => { throw err; }); const descriptor = this.descriptors.page[methodName] || this.descriptors.longrunning[methodName] || undefined; const apiCall = this._gaxModule.createApiCall(callPromise, this._defaults[methodName], descriptor, this._opts.fallback); this.innerApiCalls[methodName] = apiCall; } return this.tpuStub; } /** * The DNS address for this API service. * @deprecated Use the apiEndpoint method of the client instance. * @returns {string} The DNS address for this service. */ static get servicePath() { if (typeof process === 'object' && typeof process.emitWarning === 'function') { process.emitWarning('Static servicePath is deprecated, please use the instance method instead.', 'DeprecationWarning'); } return 'tpu.googleapis.com'; } /** * The DNS address for this API service - same as servicePath. * @deprecated Use the apiEndpoint method of the client instance. * @returns {string} The DNS address for this service. */ static get apiEndpoint() { if (typeof process === 'object' && typeof process.emitWarning === 'function') { process.emitWarning('Static apiEndpoint is deprecated, please use the instance method instead.', 'DeprecationWarning'); } return 'tpu.googleapis.com'; } /** * The DNS address for this API service. * @returns {string} The DNS address for this service. */ get apiEndpoint() { return this._servicePath; } get universeDomain() { return this._universeDomain; } /** * The port for this API service. * @returns {number} The default port for this service. */ static get port() { return 443; } /** * The scopes needed to make gRPC calls for every method defined * in this service. * @returns {string[]} List of default scopes. */ static get scopes() { return [ 'https://www.googleapis.com/auth/cloud-platform' ]; } /** * Return the project ID used by this class. * @returns {Promise} A promise that resolves to string containing the project ID. */ getProjectId(callback) { if (callback) { this.auth.getProjectId(callback); return; } return this.auth.getProjectId(); } getNode(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); this._log.info('getNode request %j', request); const wrappedCallback = callback ? (error, response, options, rawResponse) => { this._log.info('getNode response %j', response); callback(error, response, options, rawResponse); // We verified callback above. } : undefined; return this.innerApiCalls.getNode(request, options, wrappedCallback) ?.then(([response, options, rawResponse]) => { this._log.info('getNode response %j', response); return [response, options, rawResponse]; }).catch((error) => { if (error && 'statusDetails' in error && error.statusDetails instanceof Array) { const protos = this._gaxModule.protobuf.Root.fromJSON(jsonProtos); error.statusDetails = (0, google_gax_1.decodeAnyProtosInArray)(error.statusDetails, protos); } throw error; }); } getQueuedResource(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); this._log.info('getQueuedResource request %j', request); const wrappedCallback = callback ? (error, response, options, rawResponse) => { this._log.info('getQueuedResource response %j', response); callback(error, response, options, rawResponse); // We verified callback above. } : undefined; return this.innerApiCalls.getQueuedResource(request, options, wrappedCallback) ?.then(([response, options, rawResponse]) => { this._log.info('getQueuedResource response %j', response); return [response, options, rawResponse]; }).catch((error) => { if (error && 'statusDetails' in error && error.statusDetails instanceof Array) { const protos = this._gaxModule.protobuf.Root.fromJSON(jsonProtos); error.statusDetails = (0, google_gax_1.decodeAnyProtosInArray)(error.statusDetails, protos); } throw error; }); } generateServiceIdentity(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'parent': request.parent ?? '', }); this.initialize().catch(err => { throw err; }); this._log.info('generateServiceIdentity request %j', request); const wrappedCallback = callback ? (error, response, options, rawResponse) => { this._log.info('generateServiceIdentity response %j', response); callback(error, response, options, rawResponse); // We verified callback above. } : undefined; return this.innerApiCalls.generateServiceIdentity(request, options, wrappedCallback) ?.then(([response, options, rawResponse]) => { this._log.info('generateServiceIdentity response %j', response); return [response, options, rawResponse]; }).catch((error) => { if (error && 'statusDetails' in error && error.statusDetails instanceof Array) { const protos = this._gaxModule.protobuf.Root.fromJSON(jsonProtos); error.statusDetails = (0, google_gax_1.decodeAnyProtosInArray)(error.statusDetails, protos); } throw error; }); } getAcceleratorType(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); this._log.info('getAcceleratorType request %j', request); const wrappedCallback = callback ? (error, response, options, rawResponse) => { this._log.info('getAcceleratorType response %j', response); callback(error, response, options, rawResponse); // We verified callback above. } : undefined; return this.innerApiCalls.getAcceleratorType(request, options, wrappedCallback) ?.then(([response, options, rawResponse]) => { this._log.info('getAcceleratorType response %j', response); return [response, options, rawResponse]; }).catch((error) => { if (error && 'statusDetails' in error && error.statusDetails instanceof Array) { const protos = this._gaxModule.protobuf.Root.fromJSON(jsonProtos); error.statusDetails = (0, google_gax_1.decodeAnyProtosInArray)(error.statusDetails, protos); } throw error; }); } getRuntimeVersion(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); this._log.info('getRuntimeVersion request %j', request); const wrappedCallback = callback ? (error, response, options, rawResponse) => { this._log.info('getRuntimeVersion response %j', response); callback(error, response, options, rawResponse); // We verified callback above. } : undefined; return this.innerApiCalls.getRuntimeVersion(request, options, wrappedCallback) ?.then(([response, options, rawResponse]) => { this._log.info('getRuntimeVersion response %j', response); return [response, options, rawResponse]; }).catch((error) => { if (error && 'statusDetails' in error && error.statusDetails instanceof Array) { const protos = this._gaxModule.protobuf.Root.fromJSON(jsonProtos); error.statusDetails = (0, google_gax_1.decodeAnyProtosInArray)(error.statusDetails, protos); } throw error; }); } getGuestAttributes(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); this._log.info('getGuestAttributes request %j', request); const wrappedCallback = callback ? (error, response, options, rawResponse) => { this._log.info('getGuestAttributes response %j', response); callback(error, response, options, rawResponse); // We verified callback above. } : undefined; return this.innerApiCalls.getGuestAttributes(request, options, wrappedCallback) ?.then(([response, options, rawResponse]) => { this._log.info('getGuestAttributes response %j', response); return [response, options, rawResponse]; }).catch((error) => { if (error && 'statusDetails' in error && error.statusDetails instanceof Array) { const protos = this._gaxModule.protobuf.Root.fromJSON(jsonProtos); error.statusDetails = (0, google_gax_1.decodeAnyProtosInArray)(error.statusDetails, protos); } throw error; }); } createNode(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'parent': request.parent ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('createNode response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('createNode request %j', request); return this.innerApiCalls.createNode(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('createNode response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `createNode()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.create_node.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_CreateNode_async */ async checkCreateNodeProgress(name) { this._log.info('createNode long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.createNode, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } deleteNode(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('deleteNode response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('deleteNode request %j', request); return this.innerApiCalls.deleteNode(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('deleteNode response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `deleteNode()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.delete_node.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_DeleteNode_async */ async checkDeleteNodeProgress(name) { this._log.info('deleteNode long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.deleteNode, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } stopNode(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('stopNode response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('stopNode request %j', request); return this.innerApiCalls.stopNode(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('stopNode response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `stopNode()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.stop_node.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_StopNode_async */ async checkStopNodeProgress(name) { this._log.info('stopNode long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.stopNode, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } startNode(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('startNode response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('startNode request %j', request); return this.innerApiCalls.startNode(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('startNode response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `startNode()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.start_node.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_StartNode_async */ async checkStartNodeProgress(name) { this._log.info('startNode long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.startNode, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } updateNode(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'node.name': request.node.name ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('updateNode response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('updateNode request %j', request); return this.innerApiCalls.updateNode(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('updateNode response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `updateNode()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.update_node.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_UpdateNode_async */ async checkUpdateNodeProgress(name) { this._log.info('updateNode long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.updateNode, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } performMaintenance(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('performMaintenance response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('performMaintenance request %j', request); return this.innerApiCalls.performMaintenance(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('performMaintenance response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `performMaintenance()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.perform_maintenance.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_PerformMaintenance_async */ async checkPerformMaintenanceProgress(name) { this._log.info('performMaintenance long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.performMaintenance, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } createQueuedResource(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'parent': request.parent ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._log.info('createQueuedResource response %j', rawResponse); callback(error, response, rawResponse, _); // We verified callback above. } : undefined; this._log.info('createQueuedResource request %j', request); return this.innerApiCalls.createQueuedResource(request, options, wrappedCallback) ?.then(([response, rawResponse, _]) => { this._log.info('createQueuedResource response %j', rawResponse); return [response, rawResponse, _]; }); } /** * Check the status of the long running operation returned by `createQueuedResource()`. * @param {String} name * The operation name that will be passed. * @returns {Promise} - The promise which resolves to an object. * The decoded operation object has result and metadata field to get information from. * Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#long-running-operations | documentation } * for more details and examples. * @example <caption>include:samples/generated/v2alpha1/tpu.create_queued_resource.js</caption> * region_tag:tpu_v2alpha1_generated_Tpu_CreateQueuedResource_async */ async checkCreateQueuedResourceProgress(name) { this._log.info('createQueuedResource long-running'); const request = new this._gaxModule.operationsProtos.google.longrunning.GetOperationRequest({ name }); const [operation] = await this.operationsClient.getOperation(request); const decodeOperation = new this._gaxModule.Operation(operation, this.descriptors.longrunning.createQueuedResource, this._gaxModule.createDefaultBackoffSettings()); return decodeOperation; } deleteQueuedResource(request, optionsOrCallback, callback) { request = request || {}; let options; if (typeof optionsOrCallback === 'function' && callback === undefined) { callback = optionsOrCallback; options = {}; } else { options = optionsOrCallback; } options = options || {}; options.otherArgs = options.otherArgs || {}; options.otherArgs.headers = options.otherArgs.headers || {}; options.otherArgs.headers['x-goog-request-params'] = this._gaxModule.routingHeader.fromParams({ 'name': request.name ?? '', }); this.initialize().catch(err => { throw err; }); const wrappedCallback = callback ? (error, response, rawResponse, _) => { this._lo