Support VirtualHost

cmd/s3-balance/main.go

@@ -91,6 +91,7 @@ func main() {
 		metricsService,
 		cfg.S3API.ProxyMode,
 		cfg.S3API.AuthRequired,
+		cfg.S3API.VirtualHost,
 	)
 
 	// 注册配置热更新回调

internal/api/s3_handler.go

@@ -20,6 +20,7 @@ type S3Handler struct {
 	metrics       *metrics.Metrics
 	proxyMode     bool
 	authRequired  bool
+	virtualHost   bool
 }
 
 // NewS3Handler 创建新的S3兼容API处理器
@@ -33,6 +34,7 @@ func NewS3Handler(
 	metrics *metrics.Metrics,
 	proxyMode bool,
 	authRequired bool,
+	virtualHost bool,
 ) *S3Handler {
 	return &S3Handler{
 		bucketManager: bucketManager,
@@ -44,6 +46,7 @@ func NewS3Handler(
 		metrics:       metrics,
 		proxyMode:     proxyMode,
 		authRequired:  authRequired,
+		virtualHost:   virtualHost,
 	}
 }
 
@@ -73,5 +76,6 @@ func (h *S3Handler) RegisterS3Routes(router *mux.Router) {
 	router.HandleFunc("/{bucket}/{key:.*}", h.handleObjectOperations).Methods("GET", "HEAD", "PUT", "DELETE")
 
 	// 添加认证中间件
+	router.Use(h.virtualHostMiddleware)
 	router.Use(h.authMiddleware)
 }

internal/api/utils.go

@@ -4,9 +4,7 @@ import (
 	"encoding/xml"
 	"fmt"
 	"net/http"
-	"net/url"
 	"strconv"
-	"strings"
 	"time"
 
 	"github.com/DullJZ/s3-balance/internal/storage"
@@ -66,29 +64,3 @@ func (h *S3Handler) setObjectHeaders(w http.ResponseWriter, obj *storage.Object)
 		w.Header().Set("Content-Type", "application/octet-stream")
 	}
 }
-
-
-// 辅助函数：解析S3路径
-func parseS3Path(requestPath string) (bucket string, key string) {
-	requestPath = strings.TrimPrefix(requestPath, "/")
-	parts := strings.SplitN(requestPath, "/", 2)
-
-	if len(parts) > 0 {
-		bucket = parts[0]
-	}
-	if len(parts) > 1 {
-		key = parts[1]
-	}
-
-	return bucket, key
-}
-
-// 辅助函数：URL编码/解码
-func urlEncodePath(p string) string {
-	return strings.ReplaceAll(url.QueryEscape(p), "+", "%20")
-}
-
-func urlDecodePath(p string) string {
-	decoded, _ := url.QueryUnescape(p)
-	return decoded
-}

internal/api/virtual_host_middleware.go

@@ -0,0 +1,72 @@
+package api
+
+import (
+	"net"
+	"net/http"
+	"strings"
+)
+
+// virtualHostMiddleware 支持根据 Host 头推断存储桶名称
+func (h *S3Handler) virtualHostMiddleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if !h.virtualHost {
+			next.ServeHTTP(w, r)
+			return
+		}
+
+		bucketName := h.bucketFromHost(r.Host)
+		if bucketName == "" {
+			next.ServeHTTP(w, r)
+			return
+		}
+
+		// 若路由中已包含桶名称则无需改写
+		if strings.HasPrefix(r.URL.Path, "/"+bucketName) {
+			next.ServeHTTP(w, r)
+			return
+		}
+
+		// 确保桶存在
+		if _, ok := h.bucketManager.GetBucket(bucketName); !ok {
+			next.ServeHTTP(w, r)
+			return
+		}
+
+		newPath := "/" + bucketName
+		if r.URL.Path != "/" {
+			newPath += r.URL.Path
+		}
+
+		clone := r.Clone(r.Context())
+		clone.URL.Path = newPath
+		clone.RequestURI = newPath
+
+		next.ServeHTTP(w, clone)
+	})
+}
+
+func (h *S3Handler) bucketFromHost(host string) string {
+	if host == "" {
+		return ""
+	}
+
+	cleanHost := host
+	if strings.Contains(host, ":") {
+		hostname, _, err := net.SplitHostPort(host)
+		if err == nil {
+			cleanHost = hostname
+		}
+	}
+
+	parts := strings.Split(cleanHost, ".")
+	if len(parts) == 0 {
+		return ""
+	}
+
+	candidate := parts[0]
+	if candidate == "" {
+		return ""
+	}
+
+	return candidate
+}