Don't close channel which not yet started

ls-milkyway · Jan 9, 2022 · c6fab36 · c6fab36
1 parent c320997
commit c6fab36
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/batch_recognizer.cc b/src/batch_recognizer.cc
@@ -123,13 +123,14 @@ BatchRecognizer::~BatchRecognizer() {
 
 void BatchRecognizer::FinishStream(uint64_t id)
 {
-    Vector<BaseFloat> wave;
-    SubVector<BaseFloat> chunk(wave.Data(), 0);
-    dynamic_batcher_->Push(id, false, true, chunk);
-    streams_.erase(id);
+    if (streams_.find(id) != streams_.end()) {
+       Vector<BaseFloat> wave;
+       SubVector<BaseFloat> chunk(wave.Data(), 0);
+       dynamic_batcher_->Push(id, false, true, chunk);
+       streams_.erase(id);
+    }
 }
 
-
 void BatchRecognizer::PushLattice(uint64_t id, CompactLattice &clat, BaseFloat offset)
 {
     fst::ScaleLattice(fst::GraphLatticeScale(0.9), &clat);