grpc-bidirectional-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')

class GrpcBidirectionalStream {
    constructor(name, client, newStream) {
        this.grpcStream = new GrpcStream(name, () => {
            const stream = newStream.call(client)
            stream.on('data', (data) => {
                if (log.isDebug() && name && data) {
                    log.debug(`GrpcBidirectionalStream stream data. ${name} on(data): ${data}`)
                }
            })
            stream.on('end', () => {
                if (log.isDebug() && name) {
                    log.debug(`GrpcBidirectionalStream side stream ended. ${name} on(end)`)
                }
            })
            stream.on('error', (e) => {
                if (log.isDebug() && e && name) {
                    log.debug(`GrpcBidirectionalStream stream on error. ${name} on(error): ${JSON.stringify(e)}`)
                }
            })
            stream.on('status', (status) => {
                if (log.isDebug() && status && name) {
                    log.debug(`GrpcBidirectionalStream stream data. ${name} on(status): ${JSON.stringify(status)}`)
                }
            })
            return stream
        })
    }

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

module.exports = GrpcBidirectionalStream