Connect Google BigQuery and PostgreSQL in our serverless environment

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

    async fetch(publish, done) {
        // TODO: Customize your query
        const query = `SELECT *
      FROM \`your_db.your_schema.your_table_name\`
      LIMIT 10`;

        // for all options, see https://cloud.google.com/bigquery/docs/reference/rest/v2/jobs/query
        const options = {
            query: query,
            // Location must match that of the dataset(s) referenced in the query.
            location: "US",
        };

        // run the query as a job
        const [job] = await this.googleBigQuery.createQueryJob(options);
        console.log(`Job ${job.id} started.`);

        // wait for the query to finish
        const [rows] = await job.getQueryResults();
        for (const row of rows) {
            await publish(row);
        }
        done();
    }

    async close() {}
}

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

    async consume(item) {
        // TODO: Customize the SQL INSERT statement and map your item to row
        await this.postgresPool.query(
            "INSERT INTO yourTable(id, name) VALUES($1, $2)",
            [item.id, item.name]
        );
    }

    async close() {
        await this.postgresPool.end();
    }
}