87 lines
2.8 KiB
JavaScript
87 lines
2.8 KiB
JavaScript
import { createRequiredError, defineDriver } from "./utils/index.mjs";
|
|
import { CosmosClient } from "@azure/cosmos";
|
|
import { DefaultAzureCredential } from "@azure/identity";
|
|
const DRIVER_NAME = "azure-cosmos";
|
|
export default defineDriver((opts) => {
|
|
let client;
|
|
const getCosmosClient = async () => {
|
|
if (client) {
|
|
return client;
|
|
}
|
|
if (!opts.endpoint) {
|
|
throw createRequiredError(DRIVER_NAME, "endpoint");
|
|
}
|
|
if (opts.accountKey) {
|
|
const cosmosClient = new CosmosClient({
|
|
endpoint: opts.endpoint,
|
|
key: opts.accountKey
|
|
});
|
|
const { database } = await cosmosClient.databases.createIfNotExists({
|
|
id: opts.databaseName || "unstorage"
|
|
});
|
|
const { container } = await database.containers.createIfNotExists({
|
|
id: opts.containerName || "unstorage"
|
|
});
|
|
client = container;
|
|
} else {
|
|
const credential = new DefaultAzureCredential();
|
|
const cosmosClient = new CosmosClient({
|
|
endpoint: opts.endpoint,
|
|
aadCredentials: credential
|
|
});
|
|
const { database } = await cosmosClient.databases.createIfNotExists({
|
|
id: opts.databaseName || "unstorage"
|
|
});
|
|
const { container } = await database.containers.createIfNotExists({
|
|
id: opts.containerName || "unstorage"
|
|
});
|
|
client = container;
|
|
}
|
|
return client;
|
|
};
|
|
return {
|
|
name: DRIVER_NAME,
|
|
options: opts,
|
|
getInstance: getCosmosClient,
|
|
async hasItem(key) {
|
|
const item = await (await getCosmosClient()).item(key).read();
|
|
return item.resource ? true : false;
|
|
},
|
|
async getItem(key) {
|
|
const item = await (await getCosmosClient()).item(key).read();
|
|
return item.resource ? item.resource.value : null;
|
|
},
|
|
async setItem(key, value) {
|
|
const modified = /* @__PURE__ */ new Date();
|
|
await (await getCosmosClient()).items.upsert(
|
|
{ id: key, value, modified },
|
|
{ consistencyLevel: "Session" }
|
|
);
|
|
},
|
|
async removeItem(key) {
|
|
await (await getCosmosClient()).item(key).delete({ consistencyLevel: "Session" });
|
|
},
|
|
async getKeys() {
|
|
const iterator = (await getCosmosClient()).items.query(
|
|
`SELECT { id } from c`
|
|
);
|
|
return (await iterator.fetchAll()).resources.map((item) => item.id);
|
|
},
|
|
async getMeta(key) {
|
|
const item = await (await getCosmosClient()).item(key).read();
|
|
return {
|
|
mtime: item.resource?.modified ? new Date(item.resource.modified) : void 0
|
|
};
|
|
},
|
|
async clear() {
|
|
const iterator = (await getCosmosClient()).items.query(
|
|
`SELECT { id } from c`
|
|
);
|
|
const items = (await iterator.fetchAll()).resources;
|
|
for (const item of items) {
|
|
await (await getCosmosClient()).item(item.id).delete({ consistencyLevel: "Session" });
|
|
}
|
|
}
|
|
};
|
|
});
|