chore: optimize virtual host routing with radix tree

Replaced linear domain lookup with a radix tree for a faster and more efficient routing system

Author: mirkobrombin

go.mod

@@ -3,7 +3,7 @@ module github.com/mirkobrombin/goup
 go 1.22
 
 require (
-	github.com/mitchellh/mapstructure v1.5.0
+	github.com/armon/go-radix v1.0.0
 	github.com/quic-go/quic-go v0.48.1
 	github.com/rivo/tview v0.0.0-20241103174730-c76f7879f592
 	github.com/sirupsen/logrus v1.9.3

go.sum

@@ -1,3 +1,5 @@
+github.com/armon/go-radix v1.0.0 h1:F4z6KzEeeQIMeLFa97iZU6vupzoecKdU5TX24SNppXI=
+github.com/armon/go-radix v1.0.0/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8=
 github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
 github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
 github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
@@ -30,8 +32,6 @@ github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69
 github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
 github.com/mattn/go-runewidth v0.0.15 h1:UNAjwbU9l54TA3KzvqLGxwWjHmMgBUVhBiTjelZgg3U=
 github.com/mattn/go-runewidth v0.0.15/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
-github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
-github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=
 github.com/onsi/ginkgo/v2 v2.9.5 h1:+6Hr4uxzP4XIUyAkg61dWBw8lb/gc4/X5luuxN/EC+Q=
 github.com/onsi/ginkgo/v2 v2.9.5/go.mod h1:tvAoo1QUJwNEU2ITftXTpR7R1RbCzoZUOs3RonqW57k=
 github.com/onsi/gomega v1.27.6 h1:ENqfyGeS5AX/rlXDd/ETokDz93u0YufY1Pgxuy/PvWE=

internal/server/server.go

@@ -7,6 +7,7 @@ import (
 	"strings"
 	"sync"
 
+	"github.com/armon/go-radix"
 	"github.com/mirkobrombin/goup/internal/config"
 	"github.com/mirkobrombin/goup/internal/logger"
 	"github.com/mirkobrombin/goup/internal/plugin"
@@ -131,7 +132,7 @@ func startVirtualHostServer(port int, configs []config.SiteConfig, mwManager *mi
 	identifier := fmt.Sprintf("port_%d", port)
 	logger := loggers[identifier]
 
-	domainHandlers := make(map[string]http.Handler)
+	radixTree := radix.New()
 
 	for _, conf := range configs {
 		// We do not want to start a server if the root directory does not exist
@@ -148,19 +149,18 @@ func startVirtualHostServer(port int, configs []config.SiteConfig, mwManager *mi
 			continue
 		}
 
-		domainHandlers[conf.Domain] = handler
+		radixTree.Insert(conf.Domain, handler)
 	}
 
 	// Main handler that routes requests based on the Host header
 	mainHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		host := r.Host
-
 		if colonIndex := strings.Index(host, ":"); colonIndex != -1 {
 			host = host[:colonIndex]
 		}
 
-		if handler, ok := domainHandlers[host]; ok {
-			handler.ServeHTTP(w, r)
+		if handler, found := radixTree.Get(host); found {
+			handler.(http.Handler).ServeHTTP(w, r)
 		} else {
 			http.NotFound(w, r)
 		}