tolelom
cc8368dfba
- /health + /ready 엔드포인트 추가 (DB/Redis 상태 확인) - RequestID 미들웨어 + 구조화 JSON 로깅 - 체인 트랜잭션 per-user rate limit (20 req/min) - DB 커넥션 풀 설정 (MaxOpen 25, MaxIdle 10, MaxLifetime 5m) - Graceful Shutdown 시 Redis/MySQL 연결 정리 - Dockerfile HEALTHCHECK 추가 - CI에 go vet + 빌드 검증 단계 추가 (deploy 전 실행) - 보스 레이드 클라이언트 입장 API (JWT 인증) - Player 프로필 모듈 추가 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"a301_server/pkg/database"
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
const idempotencyTTL = 10 * time.Minute
|
|
const redisTimeout = 5 * time.Second
|
|
|
|
type cachedResponse struct {
|
|
StatusCode int `json:"s"`
|
|
Body json.RawMessage `json:"b"`
|
|
}
|
|
|
|
// Idempotency checks the Idempotency-Key header to prevent duplicate transactions.
|
|
// If the same key is seen again within the TTL, the cached response is returned.
|
|
func Idempotency(c *fiber.Ctx) error {
|
|
key := c.Get("Idempotency-Key")
|
|
if key == "" {
|
|
return c.Next()
|
|
}
|
|
if len(key) > 256 {
|
|
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Idempotency-Key가 너무 깁니다"})
|
|
}
|
|
|
|
// userID가 있으면 키에 포함하여 사용자 간 캐시 충돌 방지
|
|
redisKey := "idempotency:"
|
|
if uid, ok := c.Locals("userID").(uint); ok {
|
|
redisKey += fmt.Sprintf("u%d:", uid)
|
|
}
|
|
redisKey += key
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), redisTimeout)
|
|
defer cancel()
|
|
|
|
// Check if this key was already processed
|
|
cached, err := database.RDB.Get(ctx, redisKey).Bytes()
|
|
if err == nil && len(cached) > 0 {
|
|
var cr cachedResponse
|
|
if json.Unmarshal(cached, &cr) == nil {
|
|
c.Set("Content-Type", "application/json")
|
|
c.Set("X-Idempotent-Replay", "true")
|
|
return c.Status(cr.StatusCode).Send(cr.Body)
|
|
}
|
|
}
|
|
|
|
// Process the request
|
|
if err := c.Next(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Cache successful responses (2xx)
|
|
status := c.Response().StatusCode()
|
|
if status >= 200 && status < 300 {
|
|
cr := cachedResponse{StatusCode: status, Body: c.Response().Body()}
|
|
if data, err := json.Marshal(cr); err == nil {
|
|
writeCtx, writeCancel := context.WithTimeout(context.Background(), redisTimeout)
|
|
defer writeCancel()
|
|
if err := database.RDB.Set(writeCtx, redisKey, data, idempotencyTTL).Err(); err != nil {
|
|
log.Printf("WARNING: idempotency cache write failed (key=%s): %v", key, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|