Connect AWS Redshift and Dremio in our serverless environment

class AwsRedshiftSourceSelect {
    async init() {
        // TODO: Create your aws-redshift credential
        // More info at https://yepcode.io/docs/integrations/aws-redshift/#credential-configuration
        this.awsRedshift = yepcode.integration.awsRedshift(
            "your-aws-redshift-credential-name"
        );

        // TODO: Customize the command to be executed
        // More info at https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-redshift-data/interfaces/executestatementcommandinput.html
        const {
            Id
        } = await this.awsRedshift.send(
            new ExecuteStatementCommand({
                ClusterIdentifier: "your-cluster-identifier",
                Database: "your-database-name",
                DbUser: "your-database-user",
                Sql: "SELECT * FROM your-table-name;",
            })
        );
        this.statementId = Id;
        await this._awaitStatementToFinish();
    }

    async fetch(publish, done) {
        let nextToken;
        // if result is returned paginated, iterate to publish entire statement result
        do {
            const result = await this.awsRedshift.send(
                new GetStatementResultCommand({
                    Id: this.statementId,
                    NextToken: nextToken,
                })
            );
            nextToken = result.NextToken;
            for (const record of result.Records) {
                // Each record come as a list of record cells
                // More info at https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-redshift-data/interfaces/getstatementresultcommandoutput.html

                // TODO: Customize record columns mapping to the item that will be published
                const item = {
                    id: record.id,
                    ...record
                };

                await publish(item);
            }
        } while (nextToken);
        done();
    }

    async close() {}

    _awaitStatementToFinish() {
        return new Promise((resolve, reject) => {
            const describeStatementCommand = new DescribeStatementCommand({
                Id: this.statementId,
            });

            // Configure interval to check statement status
            const intervalId = setInterval(async () => {
                const result = await this.awsRedshift.send(describeStatementCommand);
                if (result.Status === "FINISHED") {
                    clearInterval(intervalId);
                    resolve();
                    return;
                }

                if (result.Status === "FAILED" || result.Status === "ABORTED") {
                    clearInterval(intervalId);
                    reject();
                    return;
                }
            }, 1000);
        });
    }
}

class HttpTargetDremioInsert {
    async init() {
        // TODO: Create your http credential with dremio information:
        // More info at https://yepcode.io/docs/integrations/http/#credential-configuration
        // Official docs: https://docs.dremio.com/current/reference/api/
        this.dremio = await yepcode.integration.http(
            "your-dremio-credential-name"
        );
    }
    async consume(item) {
        const projectId = "<your-dremio-project-id>";
        // TODO: Map your item to row and customize INSERT statement
        // More info at https://docs.dremio.com/cloud/reference/sql/commands/insert/
        const query = `INSERT INTO your_source_path.your_table VALUES (${item.id}, ${item.name})`;

        // Use the SQL API to submit queries. The response contains the ID for the job associated with the SQL query
        // Official docs: https://docs.dremio.com/cloud/reference/api/sql
        const {
            data: job
        } = await this.dremio.post(`${projectId}/sql`, {
            sql: query,
        });

        // Use the Job API to check the job status. Jobs final status are COMPLETED, FAILED and CANCELED
        // Official docs: https://docs.dremio.com/cloud/reference/api/job/
        const finalJobStates = ["COMPLETED", "FAILED", "CANCELED"];
        let jobInfoResponse = null;
        do {
            jobInfoResponse = await this.dremio.get(
                `${this.projectId}/job/${job.id}`
            );
        } while (!finalJobStates.includes(jobInfoResponse.data.jobState));

        const {
            data: jobInfo
        } = jobInfoResponse;

        if (jobInfo.jobState === "COMPLETED") {
            console.log("Success!!");
        } else if (jobInfo.jobState === "FAILED") {
            console.log(`Uups! Something was wrong: ${jobInfo.errorMessage}`);
        } else {
            console.log(
                `The job's been cancelled: ${jobInfo.cancellationReason}`
            );
        }
    }

    async close() {}
}