2 files changed, 19 insertions, 17 deletions
diff --git a/include/classes/BigPipe/BigPipe.php b/include/classes/BigPipe/BigPipe.php
index 9bb9fcb..d84083e 100644
--- a/include/classes/BigPipe/BigPipe.php
+++ b/include/classes/BigPipe/BigPipe.php
@@ -40,21 +40,6 @@ class BigPipe {
 	}
 
 	#===============================================================================
-	# Prints a single pagelet response
-	#===============================================================================
-	private static function singleResponse(Pagelet $Pagelet) {
-		$pageletJSON = $Pagelet->getStructure();
-
-		$pageletHTML = removeLineBreaksAndTabs($Pagelet->getHTML());
-		$pageletHTML = str_replace('--', '&#45;&#45;', $pageletHTML);
-
-		$pageletJSON = json_encode($pageletJSON);
-
-		echo "<code hidden id=\"_{$Pagelet->getID()}\"><!-- {$pageletHTML} --></code>\n";
-		echo "<script>BigPipe.onPageletArrive({$pageletJSON}, document.getElementById(\"_{$Pagelet->getID()}\"));</script>\n\n";
-	}
-
-	#===============================================================================
 	# Sends output buffer so far as possible towards user
 	#===============================================================================
 	public static function flushOutputBuffer() {
@@ -80,8 +65,7 @@ class BigPipe {
 
 			if(self::enabled()) {
 				foreach($pagelets as $Pagelet) {
-					self::singleResponse($Pagelet);
-					self::flushOutputBuffer();
+					$Pagelet->flush();
 				}
 			}
 
diff --git a/include/classes/BigPipe/Pagelet.php b/include/classes/BigPipe/Pagelet.php
index ae465ab..101ec34 100644
--- a/include/classes/BigPipe/Pagelet.php
+++ b/include/classes/BigPipe/Pagelet.php
@@ -148,6 +148,24 @@ class Pagelet extends Item {
 	}
 
 	#===============================================================================
+	# Flush pagelet immediately
+	#===============================================================================
+	public function flush() {
+		if(BigPipe::enabled()) {
+			$pageletHTML = removeLineBreaksAndTabs($this->getHTML());
+			$pageletHTML = str_replace('--', '&#45;&#45;', $pageletHTML);
+
+			$pageletJSON = json_encode($this->getStructure());
+
+			echo "<code hidden id=\"_{$this->getID()}\"><!-- {$pageletHTML} --></code>\n";
+			echo "<script>BigPipe.onPageletArrive({$pageletJSON}, document.getElementById(\"_{$this->getID()}\"));</script>\n\n";
+
+			BigPipe::dequeue($this);
+			BigPipe::flushOutputBuffer();
+		}
+	}
+
+	#===============================================================================
 	# Magic method: __toString()
 	#===============================================================================
 	public function __toString() {