presidium 0.15.23 → 0.15.29

package/DynamoStream.js

@@ -144,17 +144,19 @@ DynamoStream.prototype.getRecords = async function* getRecords(
     */
 
   }).promise().then(get('ShardIterator'))
+
   let records = await this.client.getRecords({
     ShardIterator: startingShardIterator,
     Limit: this.getRecordsLimit
   }).promise()
-
   if (records.Records.length > 0) {
     yield* records.Records.map(assign({
       table: always(this.table),
       shardId: always(Shard.ShardId),
     }))
   }
+  await new Promise(resolve => setTimeout(resolve, this.getRecordsInterval))
+
   while (!this.closed && records.NextShardIterator != null) {
     records = await this.client.getRecords({
       ShardIterator: records.NextShardIterator,

package/KinesisStream.js

@@ -116,7 +116,6 @@ KinesisStream.prototype.delete = function deleteStream() {
  * ```
  */
 KinesisStream.prototype.putRecord = async function putRecord(data, options = {}) {
-  await this.ready
   return this.kinesis.client.putRecord({
     StreamName: this.name,
     Data: data,
@@ -142,14 +141,19 @@ KinesisStream.prototype.listShards = async function* listShards() {
     StreamName: this.name,
     MaxResults: this.listShardsLimit,
   }).promise()
-  yield* shards.Shards
+  if (shards.Shards.length > 0) {
+    yield* shards.Shards
+  }
+
   while (!this.closed && shards.NextToken != null) {
     shards = await this.kinesis.client.listShards({
       StreamName: this.name,
       MaxResults: this.listShardsLimit,
       NextToken: shards.NextToken,
     })
-    yield* shards.Shards
+    if (shards.Shards.length > 0) {
+      yield* shards.Shards
+    }
   }
 }
 
@@ -163,19 +167,25 @@ KinesisStream.prototype.getRecords = async function* getRecords(Shard) {
       Timestamp: this.shardIteratorTimestamp,
     },
   }).promise().then(get('ShardIterator'))
+
   let records = await this.kinesis.client.getRecords({
     ShardIterator: startingShardIterator,
     Limit: this.getRecordsLimit,
   }).promise()
+  if (records.Records.length > 0) {
+    yield* records.Records
+  }
   await new Promise(resolve => setTimeout(resolve, this.getRecordsInterval))
-  yield* records.Records
+
   while (!this.closed && records.NextShardIterator != null) {
     records = await this.kinesis.client.getRecords({
       ShardIterator: records.NextShardIterator,
       Limit: this.getRecordsLimit,
     }).promise()
+    if (records.Records.length > 0) {
+      yield* records.Records
+    }
     await new Promise(resolve => setTimeout(resolve, this.getRecordsInterval))
-    yield* records.Records
   }
 }
 
@@ -186,11 +196,13 @@ KinesisStream.prototype[Symbol.asyncIterator] = async function* generateRecords(
   let muxAsyncIterator = Mux.race([
     ...shards.map(Shard => this.getRecords(Shard)),
     (async function* UpdateShardsGenerator() {
-      while (true) {
+      while (!this.closed) {
         await new Promise(resolve => {
           setTimeout(resolve, this.shardUpdatePeriod)
         })
-        yield SymbolUpdateShards
+        if (!this.closed) {
+          yield SymbolUpdateShards
+        }
       }
     }).call(this),
   ])

package/KinesisStream.test.js

@@ -65,6 +65,8 @@ const test = new Test('KinesisStream', KinesisStream)
   // wait a second for shard update
   await new Promise(resolve => setTimeout(thunkify(resolve, 'hey'), 1000))
   myStream.close()
+  // ensure shards don't update after close
+  await new Promise(resolve => setTimeout(resolve, 1000))
 })
 
 .after(async function() {

package/TranscribeStream.js

@@ -177,6 +177,10 @@ TranscribeStream.prototype.sendAudioChunk = function (chunk) {
   this.websocket.send(bytes)
 }
 
+TranscribeStream.prototype.close = function () {
+  this.websocket.close()
+}
+
 /**
  * @name marshalHeaders
  *

package/TranscribeStream.test.js

@@ -53,7 +53,7 @@ const test = new Test('TranscribeStream', async function () {
       wav.fromMuLaw()
       testTranscribeStream.sendAudioChunk(Buffer.from(wav.data.samples))
     } else if (event.event == 'stop') {
-      testTranscribeStream.websocket.close()
+      testTranscribeStream.close()
     }
   })
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "presidium",
-  "version": "0.15.23",
+  "version": "0.15.29",
   "description": "A library for creating web services",
   "author": "Richard Tong",
   "license": "MIT",