74 lines
1.9 KiB
TypeScript
74 lines
1.9 KiB
TypeScript
|
import { KV, KvKey } from "../../plug-api/types.ts";
|
||
|
|
import { KvPrimitives, KvQueryOptions } from "./kv_primitives.ts";
|
||
|
|
import {
|
||
|
|
createClient,
|
||
|
|
DynamoDBClient,
|
||
|
|
} from "https://denopkg.com/chiefbiiko/dynamodb@master/mod.ts";
|
||
|
|
|
||
|
|
export type AwsOptions = {
|
||
|
|
accessKey: string;
|
||
|
|
secretKey: string;
|
||
|
|
region: string;
|
||
|
|
};
|
||
|
|
|
||
|
|
const keySeparator = "\0";
|
||
|
|
|
||
|
|
const batchReadSize = 100;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Start of an implementation, to be continued at some point
|
||
|
|
*/
|
||
|
|
|
||
|
|
export class DynamoDBKvPrimitives implements KvPrimitives {
|
||
|
|
client: DynamoDBClient;
|
||
|
|
partitionKey: string;
|
||
|
|
tableName: string;
|
||
|
|
|
||
|
|
constructor(tableName: string, partitionKey: string, options: AwsOptions) {
|
||
|
|
this.tableName = tableName;
|
||
|
|
this.partitionKey = partitionKey;
|
||
|
|
this.client = createClient({
|
||
|
|
credentials: {
|
||
|
|
accessKeyId: options.accessKey,
|
||
|
|
secretAccessKey: options.secretKey,
|
||
|
|
},
|
||
|
|
region: options.region,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
batchGet(keys: KvKey[]): Promise<any[]> {
|
||
|
|
const allResults: any[] = [];
|
||
|
|
const promises: Promise<any>[] = [];
|
||
|
|
for (let i = 0; i < keys.length; i += batchReadSize) {
|
||
|
|
const batch = keys.slice(i, i + batchReadSize);
|
||
|
|
promises.push(
|
||
|
|
this.client.batchGetItem(
|
||
|
|
{
|
||
|
|
RequestItems: {
|
||
|
|
[this.tableName]: {
|
||
|
|
Keys: batch.map((key) => ({
|
||
|
|
pk: this.partitionKey,
|
||
|
|
sk: key.join(keySeparator),
|
||
|
|
})),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
throw new Error("Method not implemented.");
|
||
|
|
}
|
||
|
|
batchSet(entries: KV[]): Promise<void> {
|
||
|
|
throw new Error("Method not implemented.");
|
||
|
|
}
|
||
|
|
batchDelete(keys: KvKey[]): Promise<void> {
|
||
|
|
throw new Error("Method not implemented.");
|
||
|
|
}
|
||
|
|
query(options: KvQueryOptions): AsyncIterableIterator<KV> {
|
||
|
|
throw new Error("Method not implemented.");
|
||
|
|
}
|
||
|
|
close(): void {
|
||
|
|
throw new Error("Method not implemented.");
|
||
|
|
}
|
||
|
|
}
|