Connect Xata and AWS DynamoDB in our serverless environment

class HttpSourceXataQuery {
    async init() {
        // TODO: Create team variables. More info at https://yepcode.io/docs/processes/team-variables
        // TODO: Add your http credential with xata information:
        // More info at https://www.npmjs.com/package/axios
        // baseUrl: Your Xata.io base url https://${your-workspace}.${your-zone}.xata.sh/db'
        // HTTP Headers: { "Authorization": "Bearer <your-xata-api-key>" }
        this.httpClient = axios.create({
            baseURL: `https://${yepcode.env.XATA_WORKSPACE}.${yepcode.env.XATA_ZONE}.xata.sh/db`,
            headers: {
                Authorization: `Bearer ${yepcode.env.XATA_API_KEY}`,
            },
        });
    }

    async fetch(publish, done) {
        // TODO: Set your table endpoint parameters
        // You may check API documentation for more details
        // https://xata.io/docs/overview
        const database = "<your-database>";
        const branch = "<your-branch>";
        const table = "<your-table>";
        const pagination = {
            page: {
                size: 15
            }
        };

        await this.httpClient
            .post(`/${database}:${branch}/tables/${table}/query`, pagination)
            .then(({
                data
            }) => {
                for (const row of data.records) {
                    publish(row);
                }
            })
            .catch((error) => {
                console.error(
                    `An error was thrown retrieving data`,
                    error.response.data
                );
                throw error;
            });

        done();
    }

    async close() {}
}

class AwsDynamodbTargetPut {
    async init() {
        // More info at https://www.npmjs.com/package/@aws-sdk/client-dynamodb
        // TODO: Create team variables. More info at https://yepcode.io/docs/processes/team-variables
        this.awsDynamodb = new DynamoDBClient({
            region: yepcode.env.AWS_REGION,
            credentials: {
                accessKeyId: yepcode.env.AWS_ACCESS_KEY_ID,
                secretAccessKey: yepcode.env.AWS_SECRET_ACCESS_KEY,
            },
        });
    }

    async consume(item) {
        // TODO: Customize the command to be executed
        // More info at: https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-dynamodb/interfaces/putitemcommandinput.html
        const putItemCommand = new PutItemCommand({
            TableName: "table-name",
            Item: {
                "table-primary-key-name": {
                    S: item.id
                },
                email: {
                    S: item.email
                },
                age: {
                    N: item.age
                },
                // etc
            },
        });

        await this.awsDynamodb.send(putItemCommand);
    }

    async close() {}
}