Connect Google BigQuery and Segment 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 SegmentTargetSendTrackEvent {
    async init() {
        // TODO: Create your segment credential
        // More info at https://yepcode.io/docs/integrations/segment/#credential-configuration
        this.segment = yepcode.integration.segment("your-segment-credential-name");
    }

    async consume(item) {
        // TODO: Customize the track event with your item info
        // More info at: https://segment.com/docs/connections/sources/catalog/libraries/server/node/#track
        this.segment.track({
            userId: item.userId,
            event: item.event,
            properties: item.properties,
        });
    }

    async close() {
        await this.segment.flush();
    }
}