@azure/cosmos
Version:
Microsoft Azure Cosmos DB Service Node.js SDK for NOSQL API
187 lines • 8.49 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.StoredProcedure = void 0;
const index_js_1 = require("../../common/index.js");
const extractPartitionKey_js_1 = require("../../extractPartitionKey.js");
const index_js_2 = require("../../request/index.js");
const ClientUtils_js_1 = require("../ClientUtils.js");
const StoredProcedureResponse_js_1 = require("./StoredProcedureResponse.js");
const diagnostics_js_1 = require("../../utils/diagnostics.js");
/**
* Operations for reading, replacing, deleting, or executing a specific, existing stored procedure by id.
*
* For operations to create, read all, or query Stored Procedures,
*/
class StoredProcedure {
container;
id;
clientContext;
/**
* Returns a reference URL to the resource. Used for linking in Permissions.
*/
get url() {
return (0, index_js_1.createStoredProcedureUri)(this.container.database.id, this.container.id, this.id);
}
/**
* Creates a new instance of {@link StoredProcedure} linked to the parent {@link Container}.
* @param container - The parent {@link Container}.
* @param id - The id of the given {@link StoredProcedure}.
* @hidden
*/
constructor(container, id, clientContext) {
this.container = container;
this.id = id;
this.clientContext = clientContext;
}
/**
* Read the {@link StoredProcedureDefinition} for the given {@link StoredProcedure}.
* @example
* ```ts snippet:StoredProcedureRead
* import { CosmosClient } from "@azure/cosmos";
*
* const endpoint = "https://your-account.documents.azure.com";
* const key = "<database account masterkey>";
* const client = new CosmosClient({ endpoint, key });
* const { database } = await client.databases.createIfNotExists({ id: "Test Database" });
* const { container } = await database.containers.createIfNotExists({ id: "Test Container" });
*
* const { resource: sproc } = await container.scripts.storedProcedure("<sproc-id>").read();
* ```
*/
async read(options) {
return (0, diagnostics_js_1.withDiagnostics)(async (diagnosticNode) => {
const path = (0, index_js_1.getPathFromLink)(this.url);
const id = (0, index_js_1.getIdFromLink)(this.url);
const response = await this.clientContext.read({
path,
resourceType: index_js_1.ResourceType.sproc,
resourceId: id,
options,
diagnosticNode,
});
return new StoredProcedureResponse_js_1.StoredProcedureResponse(response.result, response.headers, response.code, this, (0, diagnostics_js_1.getEmptyCosmosDiagnostics)());
}, this.clientContext);
}
/**
* Replace the given {@link StoredProcedure} with the specified {@link StoredProcedureDefinition}.
* @param body - The specified {@link StoredProcedureDefinition} to replace the existing definition.
* @example
* ```ts snippet:StoredProcedureReplace
* import { CosmosClient, StoredProcedureDefinition } from "@azure/cosmos";
*
* const endpoint = "https://your-account.documents.azure.com";
* const key = "<database account masterkey>";
* const client = new CosmosClient({ endpoint, key });
* const { database } = await client.databases.createIfNotExists({ id: "Test Database" });
* const { container } = await database.containers.createIfNotExists({ id: "Test Container" });
*
* const sprocDefinition: StoredProcedureDefinition = {
* id: "sample sproc",
* body: "function () { const x = 10; }",
* };
*
* const { resource: sproc } = await container.scripts.storedProcedures.create(sprocDefinition);
*
* sproc.body = function () {
* const x = 20;
* console.log(x);
* };
* const { resource: replacedSproc } = await container.scripts
* .storedProcedure(sproc.id)
* .replace(sproc);
* ```
*/
async replace(body, options) {
return (0, diagnostics_js_1.withDiagnostics)(async (diagnosticNode) => {
if (body.body) {
body.body = body.body.toString();
}
const err = {};
if (!(0, index_js_1.isResourceValid)(body, err)) {
throw err;
}
const path = (0, index_js_1.getPathFromLink)(this.url);
const id = (0, index_js_1.getIdFromLink)(this.url);
const response = await this.clientContext.replace({
body,
path,
resourceType: index_js_1.ResourceType.sproc,
resourceId: id,
options,
diagnosticNode,
});
return new StoredProcedureResponse_js_1.StoredProcedureResponse(response.result, response.headers, response.code, this, (0, diagnostics_js_1.getEmptyCosmosDiagnostics)());
}, this.clientContext);
}
/**
* Delete the given {@link StoredProcedure}.
* @example
* ```ts snippet:StoredProcedureDelete
* import { CosmosClient } from "@azure/cosmos";
*
* const endpoint = "https://your-account.documents.azure.com";
* const key = "<database account masterkey>";
* const client = new CosmosClient({ endpoint, key });
* const { database } = await client.databases.createIfNotExists({ id: "Test Database" });
* const { container } = await database.containers.createIfNotExists({ id: "Test Container" });
*
* await container.scripts.storedProcedure("<sproc-id>").delete();
* ```
*/
async delete(options) {
return (0, diagnostics_js_1.withDiagnostics)(async (diagnosticNode) => {
const path = (0, index_js_1.getPathFromLink)(this.url);
const id = (0, index_js_1.getIdFromLink)(this.url);
const response = await this.clientContext.delete({
path,
resourceType: index_js_1.ResourceType.sproc,
resourceId: id,
options,
diagnosticNode,
});
return new StoredProcedureResponse_js_1.StoredProcedureResponse(response.result, response.headers, response.code, this, (0, diagnostics_js_1.getEmptyCosmosDiagnostics)());
}, this.clientContext);
}
/**
* Execute the given {@link StoredProcedure}.
*
* The specified type, T, is not enforced by the client.
* Be sure to validate the response from the stored procedure matches the type, T, you provide.
*
* @param partitionKey - The partition key to use when executing the stored procedure
* @param params - Array of parameters to pass as arguments to the given {@link StoredProcedure}.
* @param options - Additional options, such as the partition key to invoke the {@link StoredProcedure} on.
* * @example
* ```ts snippet:StoredProcedureExecute
* import { CosmosClient } from "@azure/cosmos";
*
* const endpoint = "https://your-account.documents.azure.com";
* const key = "<database account masterkey>";
* const client = new CosmosClient({ endpoint, key });
* const { database } = await client.databases.createIfNotExists({ id: "Test Database" });
* const { container } = await database.containers.createIfNotExists({ id: "Test Container" });
*
* const { resource: result } = await container.scripts
* .storedProcedure("<sproc-id>")
* .execute(undefined);
* ```
*/
async execute(partitionKey, params, options) {
return (0, diagnostics_js_1.withDiagnostics)(async (diagnosticNode) => {
if (partitionKey === undefined) {
const partitionKeyResponse = await (0, ClientUtils_js_1.readPartitionKeyDefinition)(diagnosticNode, this.container);
partitionKey = (0, extractPartitionKey_js_1.undefinedPartitionKey)(partitionKeyResponse);
}
const response = await this.clientContext.execute({
sprocLink: this.url,
params,
options,
partitionKey,
diagnosticNode,
});
return new index_js_2.ResourceResponse(response.result, response.headers, response.code, (0, diagnostics_js_1.getEmptyCosmosDiagnostics)());
}, this.clientContext);
}
}
exports.StoredProcedure = StoredProcedure;
//# sourceMappingURL=StoredProcedure.js.map