istream/Pause: implement _FillBucketList()

CM4all · Aug 12, 2023 · 28b065b · 28b065b
1 parent e8ae38d
commit 28b065b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/istream/PauseIstream.cxx b/src/istream/PauseIstream.cxx
@@ -6,6 +6,7 @@
 #include "ForwardIstream.hxx"
 #include "UnusedPtr.hxx"
 #include "New.hxx"
+#include "Bucket.hxx"
 #include "event/DeferEvent.hxx"
 
 class PauseIstream final : public ForwardIstream {
@@ -59,6 +60,18 @@ class PauseIstream final : public ForwardIstream {
 		}
 	}
 
+	void _FillBucketList(IstreamBucketList &list) override {
+		if (resumed) {
+			defer_read.Cancel();
+			ForwardIstream::_FillBucketList(list);
+		} else {
+			list.SetMore();
+
+			/* we'll try again after Resume() gets called */
+			want_read = true;
+		}
+	}
+
 	int _AsFd() noexcept override {
 		return resumed
 			? ForwardIstream::_AsFd()