Connect Databricks and Supabase 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 SupabaseTargetInsert {
    async init() {
        // TODO: Create your supabase credential
        // More info at https://yepcode.io/docs/integrations/supabase/#credential-configuration
        this.supabase = yepcode.integration.supabase(
            "your-supabase-credential-name"
        );
    }

    async consume(item) {
        // TODO: Customize table and columns
        // You may check the full API documentation at:
        // https://supabase.com/docs/reference/javascript/insert
        const {
            error
        } = await this.supabase
            .from("your-table")
            .insert({
                id: item.id,
                name: item.name
            });

        if (error) {
            throw error;
        }
    }

    async close() {}
}