节点中实现重写规则

internal/nodes/http_request_rewrite.go

@@ -0,0 +1,41 @@
+package nodes
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"net/http"
+)
+
+// 调用Rewrite
+func (this *HTTPRequest) doRewrite() (shouldShop bool) {
+	if this.rewriteRule == nil {
+		return
+	}
+
+	// 代理
+	if this.rewriteRule.Mode == serverconfigs.HTTPRewriteModeProxy {
+		// 外部URL
+		if this.rewriteIsExternalURL {
+			host := this.Host
+			if len(this.rewriteRule.ProxyHost) > 0 {
+				host = this.rewriteRule.ProxyHost
+			}
+			this.doURL(this.RawReq.Method, this.rewriteReplace, host, 0)
+			return true
+		}
+
+		// 内部URL继续
+		return false
+	}
+
+	// 跳转
+	if this.rewriteRule.Mode == serverconfigs.HTTPRewriteModeRedirect {
+		if this.rewriteRule.RedirectStatus > 0 {
+			http.Redirect(this.writer, this.RawReq, this.rewriteReplace, this.rewriteRule.RedirectStatus)
+		} else {
+			http.Redirect(this.writer, this.RawReq, this.rewriteReplace, http.StatusTemporaryRedirect)
+		}
+		return true
+	}
+
+	return true
+}