Connect Azure Blob NDJSON file and Databricks in our serverless environment

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

    async fetch(publish, done) {
        // TODO: customize the container name and blob name to download
        // // More info at https://docs.microsoft.com/en-us/javascript/api/@azure/storage-blob/blobserviceclient?view=azure-node-latest#@azure-storage-blob-blobserviceclient-getcontainerclient
        const containerClient = this.azureBlob.getContainerClient("container-name");
        const blobClient = containerClient.getBlobClient("blob-name");
        const result = await blobClient.download();

        // result.readableStreamBody is a stream of object content

        result.readableStreamBody
            .pipe(ndjson.parse())
            .on("data", async (line) => {
                await publish(line);
            })
            .on("end", 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();
    }
}