HTTPミドルウェア

func ExampleMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 前処理
        // 例：ロギング、メトリクス、トレーシング

        next.ServeHTTP(w, r)

        // 後処理
        // 例：レスポンスのロギング、クリーンアップ
    })
}

// ベースとなるHTTPハンドラーを作成
handler := mux

// 標準的なミドルウェアチェーンを追加
handler = debug.HTTP()(handler)                    // デバッグログ制御
handler = otelhttp.NewHandler(handler, "service")  // OpenTelemetry計装
handler = log.HTTP(ctx)(handler)                   // リクエストログ
handler = goahttpmiddleware.RequestID()(handler)   // リクエストID生成
handler = goahttpmiddleware.PopulateRequestContext()(handler)  // Goaコンテキスト設定

// パスフィルタリング付きのロギングミドルウェア
handler = log.HTTP(ctx, 
    log.WithPathFilter(regexp.MustCompile(`^/(healthz|metrics)$`)))(handler)

// OpenTelemetryトレーシングミドルウェア
handler = otelhttp.NewHandler(handler, "service-name",
    otelhttp.WithMessageEvents(otelhttp.ReadEvents, otelhttp.WriteEvents))

func ContextEnrichmentMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // リクエストスコープの値を追加
        ctx := r.Context()
        ctx = context.WithValue(ctx, "request.start", time.Now())
        ctx = context.WithValue(ctx, "request.id", r.Header.Get("X-Request-ID"))
        
        // 強化されたコンテキストで続行
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}

func SecurityHeadersMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // セキュリティヘッダーを設定
        w.Header().Set("X-Frame-Options", "DENY")
        w.Header().Set("X-Content-Type-Options", "nosniff")
        w.Header().Set("X-XSS-Protection", "1; mode=block")
        
        next.ServeHTTP(w, r)
    })
}

func OptimizedMiddleware(next http.Handler) http.Handler {
    // 高コストなオブジェクトを事前コンパイル
    pathRegex := regexp.MustCompile(`^/api/v\d+/`)
    
    // 頻繁にアロケーションされるオブジェクトにsync.Poolを使用
    bufPool := sync.Pool{
        New: func() interface{} {
            return new(bytes.Buffer)
        },
    }
    
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // マッチしないパスの場合はミドルウェアをスキップ
        if !pathRegex.MatchString(r.URL.Path) {
            next.ServeHTTP(w, r)
            return
        }
        
        // プールされたリソースを使用
        buf := bufPool.Get().(*bytes.Buffer)
        buf.Reset()
        defer bufPool.Put(buf)
        
        next.ServeHTTP(w, r)
    })
}

func ErrorHandlingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                // リクエストコンテキストとともにエラーをログ
                log.Printf("パニックを回復: %v", err)
                
                // 500レスポンスを返す
                http.Error(w, "内部サーバーエラー", http.StatusInternalServerError)
            }
        }()
        
        next.ServeHTTP(w, r)
    })
}

func main() {
    // 1. ベースとなるマルチプレクサを作成
    mux := goahttp.NewMuxer()
    
    // 2. Goaサーバーを作成してマウント
    server := genhttp.New(endpoints, mux, decoder, encoder, eh, eh)
    genhttp.Mount(mux, server)
    
    // 3. ミドルウェアスタックを追加
    handler := mux
    handler = debug.HTTP()(handler)                // デバッグログ
    handler = otelhttp.NewHandler(handler, "svc")  // トレーシング
    handler = log.HTTP(ctx)(handler)               // リクエストログ
    handler = goahttpmiddleware.RequestID()(handler) // リクエストID
    
    // 4. タイムアウト付きのHTTPサーバーを作成
    httpServer := &http.Server{
        Addr:              ":8080",
        Handler:           handler,
        ReadHeaderTimeout: 10 * time.Second,
        WriteTimeout:      30 * time.Second,
        IdleTimeout:       120 * time.Second,
    }
}

func TestMiddleware(t *testing.T) {
    // テストハンドラーを作成
    handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusOK)
    })
    
    // ミドルウェアを追加
    handler = YourMiddleware(handler)
    
    // テストリクエストを作成
    req := httptest.NewRequest("GET", "/test", nil)
    rec := httptest.NewRecorder()
    
    // テスト実行
    handler.ServeHTTP(rec, req)
    
    // 結果を検証
    if rec.Code != http.StatusOK {
        t.Errorf("ステータス %d を取得、期待値は %d", rec.Code, http.StatusOK)
    }
}