tasksClient.js

const { CloudTasksClient } = require("@google-cloud/tasks");
const tracerApi = require("@opentelemetry/api");

module.exports = class TasksClient {
    /**
     * Construct variables for this cloud tasks client for use with Google Cloud Tasks.
     * Undefined values may be replaced with defaults
     *
     * @param projectId Google projectId
     * @param keyPath path to service account json
     * @param context trace
     * 
     * Constructs the actual Google cloud tasks client for internal use
     */
    constructor(config) {
        this.projectId = config.projectId;
        this.keyPath = config.keyPath;

        // TRACER CLIENT
        this.context = config.context;

        this.TasksClient = new CloudTasksClient({
            projectId: this.projectId,
            keyFilename: this.keyPath
        });

        // LOGGER CLIENT
        if(config.hasOwnProperty('loggerInstance') && config.loggerInstance !== undefined){
            this.loggerInstance = config.loggerInstance
            this.logger = config.loggerInstance.createLogger({level: process.env.LOGLEVEL ? process.env.LOGLEVEL : 'trace'});
        }
    }

    /**
     * Create Task and send it to queue
     *
     * @param method Google projectId
     * @param url path to service account json
     * @param body hosted queue location
     * @param queue hosted queue location
     *
     */
    async sendTask(config) {

        // determin
        let span;
        if(config.spanName !== null){

            span = this.context.startSpan(config.spanName, {
                parent: tracerApi.getSpan(tracerApi.context.active()) ? tracerApi.getSpan(tracerApi.context.active()) : this.context.startSpan().updateName("sendTask")
            });

            if(this.logger){
                let labelObject = {
                    function: `taskClient.sendTask()`,
                    spanId: span.spanContext.spanId,
                    spanTraceId: span.spanContext.traceId
                };

                let loggerKey = await this.loggerInstance.getLoggerKey(span.spanContext.spanId, {
                    labels: labelObject
                });
                
                this.logger.debug(loggerKey, `KICK CLOUD TASK: ${span.spanContext.spanId}`);
            }

            span.setAttributes(config)

            // creat trace parent
            config.traceparent = '00-' + span.spanContext.traceId + '-' + span.spanContext.spanId + '-0' + span.spanContext.traceFlags
        }

        // set headers on task
        if(config.hasOwnProperty('headers')){
            config.headers['Content-Type'] = 'application/json';
            config.headers['traceparent'] = config.traceparent ? config.traceparent : null;
        }else{
            config.headers = {
                'Content-Type': 'application/json',
                'traceparent': config.traceparent ? config.traceparent : null,
            }
        }

        // configuration for the task
        const request = {
            parent: this.TasksClient.queuePath(this.projectId, config.location, config.queue),
            task: {
                httpRequest: {
                    httpMethod: config.method,
                    url: config.url,
                    headers: config.headers,
                },
                scheduleTime: {
                    seconds: config.scheduleTime ? config.scheduleTime + (Math.round(new Date() / 1000)) : Math.round(new Date() / 1000),
                }
            }
        };
        if(config.method === 'POST' || config.method === 'PUT'){
            request.task.httpRequest.body = Buffer.from(JSON.stringify(config.body)).toString('base64');
        }

        let createdTask = await this.TasksClient.createTask(request);

        if(config.spanName !== null){
            span.end()
        }

        return createdTask;
    }
  
}