Connect Google BigQuery and Amazon S3 NDJSON file 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 AwsS3TargetUploadNdjson {
    async init() {
        // TODO: Create your aws-s3 credential
        // More info at https://yepcode.io/docs/integrations/aws-s3/#credential-configuration
        const awsS3 = yepcode.integration.awsS3("your-aws-s3-credential-name");
        // Transforms the items into a ndjson format
        this.stringifier = ndjson.stringify();
        const targetStream = new PassThrough();
        this.stringifier.pipe(targetStream);
        // TODO: customize the Upload content
        // More info at: https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/interfaces/_aws_sdk_lib_storage.options-1.html
        const upload = new Upload({
            client: awsS3,
            params: {
                Bucket: "your-bucket-name",
                Key: "your-file-name.ndjson",
                Body: targetStream,
            },
        });
        upload.on("httpUploadProgress", (progress) => {
            console.log(`Upload progress`, progress);
        });
        this.uploadPromise = upload.done();
    }

    async consume(item) {
        this.stringifier.write(item);
    }

    async close() {
        try {
            this.stringifier.end();
        } catch (error) {
            console.error(`Error ending stringifier`, error);
        }
        try {
            await this.uploadPromise;
        } catch (error) {
            console.error(`Error ending upload`, error);
        }
    }
}