Connect Notion pages and Databricks in our serverless environment

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

    async fetch(publish, done) {
        // You may want to check the official docs:
        // https://developers.notion.com/reference/retrieve-a-page

        // TODO: Customize your page id
        const pageId = "your-notion-page-id";
        const response = await this.notion.pages.retrieve({
            page_id: pageId
        });
        await publish(response);
        done();
    }

    async close() {}
}

class DatabricksTargetInsert {
    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 consume(item) {
        // TODO: Customize your sql query
        // More info at: https://github.com/databricks/databricks-sql-nodejs/blob/main/tests/e2e/data_types.test.js#L151
        const operation = await this.session.executeStatement(
            `INSERT INTO yourTable VALUES(${item.id}, ${item.name})`
        );
        const result = await operation.fetchAll();
        await operation.close();
        return result;
    }

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