Connect Google BigTable and Snowflake in our serverless environment

class GoogleBigtableSourceGetRows {
    async init() {
        // TODO: Create your google-bigtable credential
        // More info at https://yepcode.io/docs/integrations/google-bigtable/#credential-configuration
        const googleBigTable = yepcode.integration.googleBigTable(
            "your-google-bigquery-bigtable-name"
        );
        // TODO: Customize your instance and table names
        this.table = await googleBigTable
            .instance("your-instance-name")
            .table("your-table-name");
    }

    async fetch(publish, done) {
        // You can customize get options. eg: get matching keys
        // Look at: https://cloud.google.com/nodejs/docs/reference/bigtable/latest/bigtable/getrowsoptions
        const getRowsOptions = {};
        const [rows] = await this.table.getRows(getRowsOptions);

        for (const rowResponse of rows) {
            const [row] = await rowResponse.get();
            const item = this._mapRowToItem(row);
            await publish(item);
        }
        done();
    }

    _mapRowToItem(row) {
        const columnsAndLastValue = Object.entries(row.data).map(
            ([columnName, columnContent]) => [
                columnName,
                this._getLastColumnValue(columnContent),
            ]
        );
        return Object.fromEntries(columnsAndLastValue);
    }

    _getLastColumnValue(columnContent) {
        // The second [0] is to access the last value
        // You can access previous values by accessing the next
        // elements in the list
        return columnContent[0][0].value;
    }

    async close() {}
}

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

    async consume(item) {
        await new Promise((resolve, reject) => {
            this.snowflake.execute({
                // TODO: Customize your SQL INSERT statement
                sqlText: "insert into YOUR_TABLE(ID, NAME) values(?, ?)",
                // TODO: Map item to any of the ? in insert sentence
                binds: [item.id, item.name],
                complete: (err, stmt, rows) => {
                    if (err) {
                        reject(err);
                        return;
                    }
                    resolve([stmt, rows]);
                },
            });
        });
    }

    async close() {
        await this.snowflake.destroy();
    }
}