Connect Zyte automatic extraction and Databricks in our serverless environment

class HttpSourceZyteAutomaticExtraction {
    async init() {
        // TODO: Create your http credential with zyte information:
        // More info at https://yepcode.io/docs/integrations/http/#credential-configuration
        // Get your Zyte API key following guide at: https://docs.zyte.com/automatic-extraction-get-started.html
        // baseUrl: https://autoextract.scrapinghub.com/v1
        // user: <your_automatic_extraction_api_key>
        // password: empty
        this.httpClient = yepcode.integration.http(
            "your-http-zyte-credential-name"
        );
    }

    async fetch(publish, done) {
        // TODO: Customize your request checking the API documentation:
        // https://docs.zyte.com/automatic-extraction.html
        const payload = [
            // In this sample we'll use the product list extraction: https://docs.zyte.com/automatic-extraction/product-list.html
            // TODO: Other lists or single object extractions may be done changing this payload:
            //       articles, comments, job posting, real state, reviews
            {
                url: "http://books.toscrape.com/",
                pageType: "productList",
            },
        ];
        const {
            data
        } = await this.httpClient.post("/extract", payload);

        if (data && data[0]) {
            // TODO: If retrieved list is not a productList, this attribute navigations should be changed
            // TODO: You may also keep on retrieving next paginations using information at data[0].productList.paginationNext
            for (const item of data[0].productList.products) {
                await publish(item);
            }
        }
        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();
    }
}