From 286ee22d183026ad990d9f33c7a02d3809fa55b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mr=C2=B7x?= <as2252258@163.com>
Date: Wed, 9 Sep 2020 10:36:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 HttpServer/IInterface/After.php | 12 ++++++++++++
 HttpServer/Route/Middleware.php |  7 ++++++-
 HttpServer/Route/Node.php       |  4 +---
 3 files changed, 19 insertions(+), 4 deletions(-)
 create mode 100644 HttpServer/IInterface/After.php

diff --git a/HttpServer/IInterface/After.php b/HttpServer/IInterface/After.php
new file mode 100644
index 00000000..808295bd
--- /dev/null
+++ b/HttpServer/IInterface/After.php
@@ -0,0 +1,12 @@
+<?php
+
+
+namespace HttpServer\IInterface;
+
+
+interface After
+{
+
+	public function onHandler($params);
+
+}
diff --git a/HttpServer/Route/Middleware.php b/HttpServer/Route/Middleware.php
index 6d136e65..af841050 100644
--- a/HttpServer/Route/Middleware.php
+++ b/HttpServer/Route/Middleware.php
@@ -11,6 +11,7 @@ namespace HttpServer\Route;
 use Closure;
 use Exception;
 use HttpServer\Route\Dispatch\Dispatch;
+use Swoole\Coroutine;
 
 /**
  * Class Middleware
@@ -50,7 +51,11 @@ class Middleware
 	public function getGenerate($node)
 	{
 		$last = function ($passable) use ($node) {
-			return Dispatch::create($node->handler, $passable)->dispatch();
+			$responseData = Dispatch::create($node->handler, $passable)->dispatch();
+			response()->send($responseData, 200);
+			Coroutine::create(function () use ($responseData, $node) {
+				$node->afterDispatch($responseData);
+			});
 		};
 		$data = Reduce::reduce($last, $this->annotation($node));
 		return $node->callback = $data;
diff --git a/HttpServer/Route/Node.php b/HttpServer/Route/Node.php
index 417110b4..c5639111 100644
--- a/HttpServer/Route/Node.php
+++ b/HttpServer/Route/Node.php
@@ -100,9 +100,7 @@ class Node extends Application
 	 */
 	public function afterDispatch($response)
 	{
-		return Reduce::reduce(function () use ($response) {
-			return $response;
-		}, $this->_after);
+		return call_user_func($this->_after, $response);
 	}