Connect Databricks and AWS Redshift in our serverless environment

class DatabricksSourceSelect {
    async init() {
        // TODO: Create your databricks credential
        // More info at https://yepcode.io/docs/integrations/databricks/#credential-configuration
        this.databricks = yepcode.integration.databricks(
            "your-databricks-credential-name"
        );
        this.session = await this.databricks.openSession();
    }

    async fetch(publish, done) {
        // TODO: Customize your sql query
        // More info at: https://github.com/databricks/databricks-sql-nodejs/blob/main/tests/e2e/batched_fetch.test.js#L21
        const queryOperation = await this.session.executeStatement(
            `SELECT * FROM your_table_name`, {
                runAsync: true
            }
        );
        const results = await queryOperation.fetchAll();
        await queryOperation.close();

        for (const result of results) {
            await publish(result);
        }
        done();
    }

    async close() {
        await this.session.close();
        await this.databricks.close();
    }
}

class AwsRedshiftTargetInsert {
    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"
        );
    }

    async consume(item) {
        // TODO: Customize your sql query
        const sqlStatement = `INSERT INTO your-table-name VALUES (${item.someProp}, '${item.someStringProp}')`;

        // 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
        this.awsRedshift.send(
            new ExecuteStatementCommand({
                ClusterIdentifier: "your-cluster-identifier",
                Database: "your-database",
                DbUser: "your-db-user",
                Sql: sqlStatement,
            })
        );
    }

    async close() {}
}