Use streamx.Transform in createWriteStream/createDiffStream

Author: andrewosh

lib/clients/drive.js

@@ -2,9 +2,8 @@ const grpc = require('@grpc/grpc-js')
 const maybe = require('call-me-maybe')
 const collectStream = require('stream-collector')
 const pumpify = require('pumpify')
-const map = require('through2-map')
 const codecs = require('codecs')
-const { Writable } = require('streamx')
+const { Writable, Transform } = require('streamx')
 
 const { Stat } = require('hyperdrive-schemas')
 const { drive: { services,messages } } = require('../rpc.js')
@@ -233,11 +232,13 @@ class RemoteHyperdrive {
     const call = this._client.createDiffStream(req, toMetadata({ token: this.token }))
     return pumpify.obj(
       call,
-      map.obj(rsp => {
-        return {
-          type: rsp.getType(),
-          name: rsp.getName(),
-          value: fromDiffEntry(rsp.getValue())
+      new Transform({
+        transform (rsp, cb) {
+          return cb(null, {
+            type: rsp.getType(),
+            name: rsp.getName(),
+            value: fromDiffEntry(rsp.getValue())
+          })
         }
       })
     )
@@ -255,7 +256,11 @@ class RemoteHyperdrive {
     const call = this._client.createReadStream(req, toMetadata({ token: this.token }))
     return pumpify(
       call,
-      map.obj(rsp => Buffer.from(rsp.getChunk()))
+      new Transform({
+        transform (rsp, cb) {
+          return cb(null, Buffer.from(rsp.getChunk()))
+        }
+      })
     )
   }