istream/Suspend: support buckets

CM4all · Aug 16, 2023 · bb83799 · bb83799
1 parent 113e5a1
commit bb83799
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/src/istream/SuspendIstream.cxx b/src/istream/SuspendIstream.cxx
@@ -6,6 +6,7 @@
 #include "ForwardIstream.hxx"
 #include "UnusedPtr.hxx"
 #include "New.hxx"
+#include "Bucket.hxx"
 #include "event/FineTimerEvent.hxx"
 
 class SuspendIstream final : public ForwardIstream {
@@ -41,6 +42,13 @@ class SuspendIstream final : public ForwardIstream {
 			Schedule();
 	}
 
+	void _FillBucketList(IstreamBucketList &list) override {
+		if (ready)
+			ForwardIstream::_FillBucketList(list);
+		else
+			list.SetMore();
+	}
+
 	int _AsFd() noexcept override {
 		return ready ? ForwardIstream::_AsFd() : -1;
 	}
@@ -53,7 +61,18 @@ class SuspendIstream final : public ForwardIstream {
 
 	void OnTimer() noexcept {
 		ready = true;
-		input.Read();
+
+		switch (InvokeReady()) {
+		case IstreamReadyResult::OK:
+			break;
+
+		case IstreamReadyResult::FALLBACK:
+			input.Read();
+			break;
+
+		case IstreamReadyResult::CLOSED:
+			break;
+		}
 	}
 };