WAF动作支持有效范围

internal/waf/action_base.go

@@ -2,5 +2,22 @@
 
 package waf
 
+import (
+	"net/http"
+)
+
 type BaseAction struct {
 }
+
+// CloseConn 关闭连接
+func (this *BaseAction) CloseConn(writer http.ResponseWriter) error {
+	// 断开连接
+	hijack, ok := writer.(http.Hijacker)
+	if ok {
+		conn, _, err := hijack.Hijack()
+		if err == nil {
+			return conn.Close()
+		}
+	}
+	return nil
+}