grpc-client-side-stream.js

/**
 * Pinpoint Node.js Agent
 * Copyright 2021-present NAVER Corp.
 * Apache License v2.0
 */

'use strict'

const log = require('../utils/logger')
const GrpcStream = require('./grpc-stream')

const DEFAULT_CLIENT_REQUEST_TIMEOUT = 5 * 60 * 1000

class GrpcClientSideStream {
    constructor(name, client, newStream) {
        this.grpcStream = new GrpcStream(name, () => {
            this.deadline = Date.now()

            const self = this
            const stream = newStream.call(client, (err, response) => {
                if (response) {
                    if (log.isDebug()) {
                        log.debug(`GrpcClientSideStream Server side stream data. ${name} on(data): ${response}`)
                    }
                } else if (err) {
                    if (log.isDebug()) {
                        log.debug(`GrpcClientSideStream callback err: ${err}`)
                    }
                }

                this.grpcStream.endWithStream(stream)
                
                if (self.callback && typeof self.callback === 'function') {
                    self.callback(err, response)
                }
            })
            stream.on('status', (status) => {
                if (log.isDebug() && status) {
                    log.debug(`GrpcClientSideStream Server side stream data. ${name} on(status): ${JSON.stringify(status)}`)
                }
            })
            return stream
        })
    }

    write(data) {
        this.grpcStream.write(data)

        const now = Date.now()
        if (now - this.deadline > DEFAULT_CLIENT_REQUEST_TIMEOUT) {
            this.grpcStream.end()
        }
    }
}

module.exports = GrpcClientSideStream