serverless-spy
Version:
CDK-based library for writing elegant integration tests on AWS serverless architecture and an additional web console to monitor events in real time.
43 lines (42 loc) • 1.52 kB
JavaScript
import { BatchGetItemCommand as __BatchGetItemCommand } from "@aws-sdk/client-dynamodb";
import { Command as $Command } from "@smithy/smithy-client";
import { DynamoDBDocumentClientCommand } from "../baseCommand/DynamoDBDocumentClientCommand";
import { ALL_VALUES } from "../commands/utils";
export { DynamoDBDocumentClientCommand, $Command };
export class BatchGetCommand extends DynamoDBDocumentClientCommand {
constructor(input) {
super();
this.input = input;
this.inputKeyNodes = {
RequestItems: {
"*": {
Keys: {
"*": ALL_VALUES,
},
},
},
};
this.outputKeyNodes = {
Responses: {
"*": {
"*": ALL_VALUES,
},
},
UnprocessedKeys: {
"*": {
Keys: {
"*": ALL_VALUES,
},
},
},
};
this.clientCommand = new __BatchGetItemCommand(this.input);
this.middlewareStack = this.clientCommand.middlewareStack;
}
resolveMiddleware(clientStack, configuration, options) {
this.addMarshallingMiddleware(configuration);
const stack = clientStack.concat(this.middlewareStack);
const handler = this.clientCommand.resolveMiddleware(stack, configuration, options);
return async () => handler(this.clientCommand);
}
}