2e9dcf49a8
- Singleton CosmosClient with env var config (COSMOS_ENDPOINT, COSMOS_KEY, COSMOS_DATABASE) - Simple getContainer() for services - Container registry with registerContainers(), getRegisteredContainer(), initializeAllContainers() for dashboards - ContainerConfig type with partitionKeyPath and optional defaultTtl - _resetClient() and _resetRegistry() for test isolation - Peer dep: @azure/cosmos >=4.0.0
72 lines
2.0 KiB
TypeScript
72 lines
2.0 KiB
TypeScript
/**
|
|
* Container registry for dashboards that need partition key validation
|
|
* and createIfNotExists support.
|
|
*/
|
|
|
|
import { Container, PartitionKeyDefinition } from "@azure/cosmos";
|
|
import { getDatabase, getCosmosClient } from "./client.js";
|
|
import type { ContainerConfig } from "./types.js";
|
|
|
|
const _registry: Map<string, ContainerConfig> = new Map();
|
|
const _containerCache: Map<string, Container> = new Map();
|
|
|
|
/**
|
|
* Register containers with their partition key configuration.
|
|
* Call once at app startup before any getRegisteredContainer() calls.
|
|
*/
|
|
export function registerContainers(
|
|
definitions: Record<string, ContainerConfig>,
|
|
): void {
|
|
for (const [name, config] of Object.entries(definitions)) {
|
|
_registry.set(name, config);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get a container that was previously registered.
|
|
* Throws if the container name is unknown.
|
|
*/
|
|
export function getRegisteredContainer(name: string): Container {
|
|
if (!_registry.has(name)) {
|
|
throw new Error(
|
|
`Unknown container '${name}'. Valid: ${[..._registry.keys()].join(", ")}`,
|
|
);
|
|
}
|
|
|
|
let container = _containerCache.get(name);
|
|
if (!container) {
|
|
container = getDatabase().container(name);
|
|
_containerCache.set(name, container);
|
|
}
|
|
return container;
|
|
}
|
|
|
|
/**
|
|
* Create all registered containers if they don't exist.
|
|
* Call from a seed script or on first deploy.
|
|
*/
|
|
export async function initializeAllContainers(): Promise<void> {
|
|
const client = getCosmosClient();
|
|
const dbId = process.env.COSMOS_DATABASE || "lysnrai";
|
|
const { database } = await client.databases.createIfNotExists({ id: dbId });
|
|
|
|
for (const [name, config] of _registry.entries()) {
|
|
await database.containers.createIfNotExists({
|
|
id: name,
|
|
partitionKey: {
|
|
paths: [config.partitionKeyPath],
|
|
} as PartitionKeyDefinition,
|
|
...(config.defaultTtl != null && { defaultTtl: config.defaultTtl }),
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Reset the registry (useful for testing).
|
|
* @internal
|
|
*/
|
|
export function _resetRegistry(): void {
|
|
_registry.clear();
|
|
_containerCache.clear();
|
|
}
|