🏗️ microservice fabric splitting

common/middleware/authorization_middleware.go

@@ -0,0 +1,32 @@
+package middleware
+
+import (
+	"github.com/zeromicro/go-zero/core/logx"
+	"net/http"
+	"schisandra-album-cloud-microservices/common/errors"
+	"schisandra-album-cloud-microservices/common/xhttp"
+	"strconv"
+	"time"
+)
+
+func AuthorizationMiddleware(w http.ResponseWriter, r *http.Request) {
+	expireAtStr := r.Header.Get("X-Expire-At")
+	if expireAtStr == "" {
+		xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New(http.StatusForbidden, "unauthorized"))
+		return
+	}
+	expireAtInt, err := strconv.ParseInt(expireAtStr, 10, 64)
+	if err != nil {
+		logx.Errorf("Failed to parse X-Expire-At: %v", err)
+		xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New(http.StatusForbidden, "unauthorized"))
+		return
+	}
+	expireAt := time.Unix(expireAtInt, 0)
+	currentTime := time.Now()
+
+	remainingTime := expireAt.Sub(currentTime)
+	if remainingTime < time.Minute*5 {
+		xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New(http.StatusUnauthorized, "token expired"))
+		return
+	}
+}