From 166d1f20c1e96f5a570f7d5d0dffac8f30af4379 Mon Sep 17 00:00:00 2001
From: Sodesnei <1452401269@qq.com>
Date: Fri, 8 Oct 2021 20:11:13 +0800
Subject: [PATCH] fix(middle-template): update middle template

---
 plugins/config/middle.gen.go   | 6 ++++--
 plugins/config/middle.template | 6 ++++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/plugins/config/middle.gen.go b/plugins/config/middle.gen.go
index addf8c1..15a055c 100644
--- a/plugins/config/middle.gen.go
+++ b/plugins/config/middle.gen.go
@@ -12,9 +12,11 @@ import (
 
 func GetMiddlewareMap() map[string]plugins.MiddleToolsInterface {
 
+	logMiddle := logMiddle.NewMiddleware()
+	mapKey := mapKey.NewMiddleware()
 	return map[string]plugins.MiddleToolsInterface{
 
-		"logMiddle": logMiddle.NewMiddleware(),
-		"mapKey":    mapKey.NewMiddleware(),
+		logMiddle.Name(): logMiddle,
+		mapKey.Name():    mapKey,
 	}
 }
diff --git a/plugins/config/middle.template b/plugins/config/middle.template
index 10b56e8..d904530 100644
--- a/plugins/config/middle.template
+++ b/plugins/config/middle.template
@@ -12,10 +12,12 @@ import (
 
 
 func GetMiddlewareMap() map[string]plugins.MiddleToolsInterface {
-
+    {%for dir in dirs %}
+        {{dir[0]}}:={{dir[0]}}.NewMiddleware()
+        {%- endfor%}
     return map[string]plugins.MiddleToolsInterface{
 		{%for dir in dirs %}
-        "{{dir[0]}}":{{dir[0]}}.NewMiddleware(),
+        {{dir[0]}}.Name():{{dir[0]}},
         {%- endfor%}
 	}
 }
\ No newline at end of file