修改method枚举

karldoenitz · Oct 13, 2023 · 3d29a3f · 3d29a3f
1 parent 4953f83
commit 3d29a3f
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 26 deletions.
diff --git a/TigoWeb/enums.go b/TigoWeb/enums.go
@@ -1,23 +1,27 @@
 package TigoWeb
 
-// MethodMapping http请求方式的一个映射
-var MethodMapping = map[string]string{
-	"get":     "Get",
-	"head":    "Head",
-	"post":    "Post",
-	"put":     "Put",
-	"delete":  "Delete",
-	"connect": "Connect",
-	"options": "Options",
-	"trace":   "Trace",
-	"GET":     "Get",
-	"HEAD":    "Head",
-	"POST":    "Post",
-	"PUT":     "Put",
-	"DELETE":  "Delete",
-	"CONNECT": "Connect",
-	"OPTIONS": "Options",
-	"TRACE":   "Trace",
-}
-
 const Version = "1.6.5"
+
+// MethodEnum 根据http请求方式获取对应的函数名
+//   - httpMethod: http请求方式
+func MethodEnum(httpMethod string) string {
+	var MethodMapping = map[string]string{
+		"get":     "Get",
+		"head":    "Head",
+		"post":    "Post",
+		"put":     "Put",
+		"delete":  "Delete",
+		"connect": "Connect",
+		"options": "Options",
+		"trace":   "Trace",
+		"GET":     "Get",
+		"HEAD":    "Head",
+		"POST":    "Post",
+		"PUT":     "Put",
+		"DELETE":  "Delete",
+		"CONNECT": "Connect",
+		"OPTIONS": "Options",
+		"TRACE":   "Trace",
+	}
+	return MethodMapping[httpMethod]
+}
diff --git a/TigoWeb/urlpattern.go b/TigoWeb/urlpattern.go
@@ -22,11 +22,11 @@ type UrlPatternHandle struct {
 }
 
 // Handle 封装HTTP请求的中间件，主要有以下功能：
-//  - 1、根据反射找到挂载的handler；
-//  - 2、调用handler的InitHandler方法；
-//  - 3、进行HTTP请求预处理，包括判断请求方式是否合法等；
-//  - 4、调用handler中的功能方法；
-//  - 5、进行HTTP请求结束处理。
+//   - 1、根据反射找到挂载的handler；
+//   - 2、调用handler的InitHandler方法；
+//   - 3、进行HTTP请求预处理，包括判断请求方式是否合法等；
+//   - 4、调用handler中的功能方法；
+//   - 5、进行HTTP请求结束处理。
 func (urlPatternMidWare UrlPatternHandle) Handle(responseWriter http.ResponseWriter, request *http.Request) {
 	handlerType := reflect.TypeOf(urlPatternMidWare.Handler)
 	if handlerType.Kind() == reflect.Ptr {
@@ -41,7 +41,7 @@ func (urlPatternMidWare UrlPatternHandle) Handle(responseWriter http.ResponseWri
 	// 调用BeforeRequest方法
 	VoidFuncCall(handler, FnBeforeRequest)
 	// 根据http请求方式调用相关方法
-	VoidFuncCall(handler, MethodMapping[request.Method])
+	VoidFuncCall(handler, MethodEnum(request.Method))
 	// 调用TeardownRequest方法
 	VoidFuncCall(handler, FnTeardownRequest)
 }