节点中实现重写规则

internal/nodes/http_request.go

@@ -53,6 +53,9 @@ type HTTPRequest struct {
 	origin               *serverconfigs.OriginConfig       // 源站
 	originAddr           string                            // 源站实际地址
 	errors               []string                          // 错误信息
+	rewriteRule          *serverconfigs.HTTPRewriteRule    // 匹配到的重写规则
+	rewriteReplace       string                            // 重写规则的目标
+	rewriteIsExternalURL bool                              // 重写目标是否为外部URL
 }
 
 // 初始化
@@ -106,15 +109,19 @@ func (this *HTTPRequest) Do() {
 
 // 开始调用
 func (this *HTTPRequest) doBegin() {
-	// 重写规则
-	// TODO 需要实现
-
 	// 临时关闭页面
 	if this.web.Shutdown != nil && this.web.Shutdown.IsOn {
 		this.doShutdown()
 		return
 	}
 
+	// 重写规则
+	if this.rewriteRule != nil {
+		if this.doRewrite() {
+			return
+		}
+	}
+
 	// 缓存
 	// TODO
 
@@ -176,7 +183,6 @@ func (this *HTTPRequest) configureWeb(web *serverconfigs.HTTPWebConfig, isTop bo
 	} else {
 		rawPath = this.uri
 	}
-	_ = rawQuery // TODO 暂时不用到这个变量
 
 	// redirect
 	if web.RedirectToHttps != nil && (web.RedirectToHttps.IsPrior || isTop) {
@@ -219,6 +225,61 @@ func (this *HTTPRequest) configureWeb(web *serverconfigs.HTTPWebConfig, isTop bo
 		this.web.Websocket = web.Websocket
 	}
 
+	// 重写规则
+	if len(web.RewriteRefs) > 0 {
+		for index, ref := range web.RewriteRefs {
+			if !ref.IsOn {
+				continue
+			}
+			rewriteRule := web.RewriteRules[index]
+			if !rewriteRule.IsOn {
+				continue
+			}
+			if replace, varMapping, isMatched := rewriteRule.Match(rawPath, this.Format); isMatched {
+				this.addVarMapping(varMapping)
+				this.rewriteRule = rewriteRule
+
+				if rewriteRule.WithQuery {
+					queryIndex := strings.Index(replace, "?")
+					if queryIndex > -1 {
+						if len(rawQuery) > 0 {
+							replace = replace[:queryIndex] + "?" + rawQuery + "&" + replace[queryIndex+1:]
+						}
+					} else {
+						if len(rawQuery) > 0 {
+							replace += "?" + rawQuery
+						}
+					}
+				}
+
+				this.rewriteReplace = replace
+
+				// 如果是外部URL直接返回
+				if rewriteRule.IsExternalURL(replace) {
+					this.rewriteIsExternalURL = true
+					return nil
+				}
+
+				// 如果是内部URL继续解析
+				if replace == this.uri {
+					// URL不变，则停止解析，防止无限循环跳转
+					return nil
+				}
+				this.uri = replace
+
+				// 终止解析的三个条件：
+				//    isBreak = true
+				//    mode = redirect
+				//    replace = external url
+				if rewriteRule.IsBreak || rewriteRule.Mode == serverconfigs.HTTPRewriteModeRedirect {
+					return nil
+				}
+
+				return this.configureWeb(web, isTop, redirects+1)
+			}
+		}
+	}
+
 	// locations
 	if len(web.LocationRefs) > 0 {
 		var resultLocation *serverconfigs.HTTPLocationConfig
@@ -242,6 +303,9 @@ func (this *HTTPRequest) configureWeb(web *serverconfigs.HTTPWebConfig, isTop bo
 			}
 		}
 		if resultLocation != nil {
+			// reset rewrite rule
+			this.rewriteRule = nil
+
 			// Reverse Proxy
 			if resultLocation.ReverseProxyRef != nil && resultLocation.ReverseProxyRef.IsPrior {
 				this.reverseProxyRef = resultLocation.ReverseProxyRef
@@ -250,7 +314,7 @@ func (this *HTTPRequest) configureWeb(web *serverconfigs.HTTPWebConfig, isTop bo
 
 			// Web
 			if resultLocation.Web != nil {
-				err := this.configureWeb(resultLocation.Web, false, redirects)
+				err := this.configureWeb(resultLocation.Web, false, redirects+1)
 				if err != nil {
 					return err
 				}

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
+}

internal/nodes/listener_manager.go

@@ -44,7 +44,13 @@ func (this *ListenerManager) Start(node *nodeconfigs.NodeConfig) error {
 
 	// 所有的新地址
 	groupAddrs := []string{}
-	for _, group := range node.AvailableGroups() {
+	availableServerGroups := node.AvailableGroups()
+
+	if len(availableServerGroups) == 0 {
+		logs.Println("[LISTENER_MANAGER]no available servers to startup")
+	}
+
+	for _, group := range availableServerGroups {
 		addr := group.FullAddr()
 		groupAddrs = append(groupAddrs, addr)
 	}
@@ -59,7 +65,7 @@ func (this *ListenerManager) Start(node *nodeconfigs.NodeConfig) error {
 	}
 
 	// 启动新的或修改老的
-	for _, group := range node.AvailableGroups() {
+	for _, group := range availableServerGroups {
 		addr := group.FullAddr()
 		listener, ok := this.listenersMap[addr]
 		if ok {