feat: added API gateway

This commit is contained in:
ITQ
2025-12-17 11:00:50 +03:00
parent daa8c24482
commit 340ae43d22
30 changed files with 2838 additions and 3 deletions
+69
View File
@@ -0,0 +1,69 @@
package middleware
import (
"context"
"net/http"
"strings"
"datarush/internal/gw/domain"
"datarush/internal/gw/grpc_client"
)
type contextKey string
const (
UserIDKey contextKey = "user_id"
)
type AuthMiddleware struct {
authClient *grpc_client.AuthClient
}
func NewAuthMiddleware(authClient *grpc_client.AuthClient) *AuthMiddleware {
return &AuthMiddleware{
authClient: authClient,
}
}
func (m *AuthMiddleware) Authenticate(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
authHeader := r.Header.Get("Authorization")
if authHeader == "" {
respondWithError(w, domain.NewUnauthorizedError("missing authorization header"))
return
}
parts := strings.Split(authHeader, " ")
if len(parts) != 2 || parts[0] != "Bearer" {
respondWithError(w, domain.NewUnauthorizedError("invalid authorization header format"))
return
}
token := parts[1]
userID, err := m.authClient.ValidateToken(r.Context(), token)
if err != nil {
respondWithError(w, domain.NewUnauthorizedError("invalid token"))
return
}
ctx := context.WithValue(r.Context(), UserIDKey, userID)
next.ServeHTTP(w, r.WithContext(ctx))
})
}
func GetUserIDFromContext(ctx context.Context) (string, error) {
userID, ok := ctx.Value(UserIDKey).(string)
if !ok || userID == "" {
return "", domain.ErrUnauthorized
}
return userID, nil
}
func respondWithError(w http.ResponseWriter, err *domain.AppError) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(err.StatusCode)
response := domain.NewErrorResponse(err.Err, err.Message)
w.Write([]byte(`{"error":"` + response.Error + `","message":"` + response.Message + `"}`))
}
+21
View File
@@ -0,0 +1,21 @@
package middleware
import (
"net/http"
)
func CORSMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, PATCH, OPTIONS")
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
w.Header().Set("Access-Control-Max-Age", "3600")
if r.Method == http.MethodOptions {
w.WriteHeader(http.StatusNoContent)
return
}
next.ServeHTTP(w, r)
})
}
+47
View File
@@ -0,0 +1,47 @@
package middleware
import (
"log"
"net/http"
"time"
)
type responseWriter struct {
http.ResponseWriter
statusCode int
written int64
}
func (rw *responseWriter) WriteHeader(code int) {
rw.statusCode = code
rw.ResponseWriter.WriteHeader(code)
}
func (rw *responseWriter) Write(b []byte) (int, error) {
n, err := rw.ResponseWriter.Write(b)
rw.written += int64(n)
return n, err
}
func LoggingMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
start := time.Now()
wrapped := &responseWriter{
ResponseWriter: w,
statusCode: http.StatusOK,
}
next.ServeHTTP(wrapped, r)
duration := time.Since(start)
log.Printf(
"%s %s %d %s %s",
r.Method,
r.RequestURI,
wrapped.statusCode,
duration,
r.RemoteAddr,
)
})
}