core.Interceptor

コントローラーの実行前後に、認証、ログ出力、CORS、トランザクション処理を追加するインターフェースです。Spine v0.6.0では、事前処理、成功後の処理、レスポンス前の処理、最終的なクリーンアップをそれぞれ実装します。

インターフェース

import "github.com/NARUBROWN/spine/core"

type Interceptor interface {
    PreHandle(ctx ExecutionContext, meta HandlerMeta) error
    PostHandle(ctx ExecutionContext, meta HandlerMeta)
    BeforeResponse(ctx ExecutionContext, meta HandlerMeta, executionErr error) error
    AfterCompletion(ctx ExecutionContext, meta HandlerMeta, err error)
}

メソッド	呼び出しのタイミング	戻り値
`PreHandle`	コントローラーの実行前。登録順	`nil`の場合は続行、エラーの場合は中断
`PostHandle`	コントローラー、レスポンスの準備、実行後フックが成功した後。登録の逆順	なし
`BeforeResponse`	実際のレスポンスの書き込み前。`PreHandle`が成功したインターセプターを逆順に実行	エラーがある場合は最終的な実行エラーに含める
`AfterCompletion`	レスポンス処理後。成功と失敗にかかわらず登録の逆順	なし

ライフサイクルと実装例

次の例のauth、transactionFrom、トランザクションの実装はアプリで提供します。core.Interceptorとして登録するには、4つのメソッドのシグネチャをすべて実装する必要があります。

PreHandle

コントローラーの呼び出し前に登録順で実行します。nilを返すと次の段階に進み、通常のエラーを返すと失敗します。core.ErrAbortPipelineを返すと、コントローラーを呼び出さずに正常に中断します。

グローバルインターセプターはルーティング前に実行されるため、空のHandlerMeta{}を受け取ります。ルートインターセプターはルーティングと入力値の解釈後に実行され、対象ハンドラーの情報を受け取ります。

func (i *AuthInterceptor) PreHandle(ctx core.ExecutionContext, meta core.HandlerMeta) error {
    token := ctx.Header("Authorization")
    if token == "" {
        return httperr.Unauthorized("인증이 필요합니다")
    }
    user, err := i.auth.Validate(token)
    if err != nil {
        return httperr.Unauthorized("유효하지 않은 토큰입니다")
    }
    ctx.Set("auth.user", user)
    return nil
}

PostHandle

コントローラーの戻り値をレスポンスとして準備し、実行後フックまで成功した後に逆順で実行します。エラーを返せないため、成功ログや観測用の処理に使用します。

func (i *LoggingInterceptor) PostHandle(ctx core.ExecutionContext, meta core.HandlerMeta) {
    log.Printf("[RES] %s %s prepared", ctx.Method(), ctx.Path())
}

BeforeResponse

レスポンスを実際のトランスポート層に書き込む前に、逆順で実行します。executionErrには、事前処理、ルーティング、入力値の解釈、コントローラー、レスポンスの準備、実行後フックで、この時点までに発生したエラーが渡されます。先に実行したBeforeResponseが返したエラーもまとめて、次のインターセプターに渡します。

リクエストスコープのトランザクションは、この段階でコミットまたはロールバックしてください。v0.6.0はレスポンスのシリアライズと、Cookieおよびステータスコードの検証を先に行うため、レスポンスの準備に失敗した場合にロールバックできます。

func (i *TxInterceptor) BeforeResponse(
    ctx core.ExecutionContext,
    meta core.HandlerMeta,
    executionErr error,
) error {
    tx, ok := transactionFrom(ctx)
    if !ok {
        return nil
    }
    if executionErr != nil {
        return tx.Rollback()
    }
    return tx.Commit()
}

AfterCompletion

成功と失敗にかかわらず、レスポンス処理後に逆順で呼び出します。最終的なerrを受け取り、ログ出力、メトリクスの収集、リソースの解放に使用します。レスポンスの送信結果も、この段階で確認できます。トランザクションをコミットするかどうかは、前の段階ですでに決定している必要があります。

func (i *LoggingInterceptor) AfterCompletion(
    ctx core.ExecutionContext,
    meta core.HandlerMeta,
    err error,
) {
    if err != nil {
        log.Printf("[ERR] %s %s: %v", ctx.Method(), ctx.Path(), err)
    }
}

登録とスコープ

app.Interceptorはapp.InterceptorFor(boot.InterceptorAll, ...)と同じです。HTTPリクエストとWebSocketメッセージの両方に適用し、イベントコンシューマーには適用しません。HTTPレスポンスを扱うCORSのように、特定のトランスポートに依存する場合はスコープを指定してください。

スコープ	適用対象
`boot.InterceptorHTTP`	HTTPリクエスト
`boot.InterceptorWebSocket`	WebSocketメッセージとオプションの接続前処理
`boot.InterceptorAll`	両方のスコープ

app.InterceptorFor(boot.InterceptorAll, auth)
app.InterceptorFor(boot.InterceptorHTTP, corsInterceptor)
app.InterceptorFor(boot.InterceptorWebSocket, wsRateLimiter)

app.Route("GET", "/admin/users/:id", (*AdminController).GetUser,
    route.WithInterceptors((*AuthInterceptor)(nil)),
)

(*AuthInterceptor)(nil)のような型付きnilポインターを登録すると、コンテナからその型のインスタンスを取得して使用します。同じ型のnilポインターは、コンテナの共有インスタンスを使用します。

グローバルインターセプターに同じポインターを複数のスコープで登録すると、スコープをまとめて1回実行します。同じ型でも、異なるポインターはそれぞれ実行します。値型も、登録した項目ごとにそれぞれ実行します。

WebSocket接続前の処理

接続認証が必要なインターセプターは、core.Interceptorに加えて次のオプションのインターフェースも実装します。

type WebSocketHandshakeInterceptor interface {
    PreHandshake(ctx WebSocketHandshakeContext, meta HandlerMeta) error
}

接続枠を先に予約した後、HTTPアップグレード前にPreHandshakeを実行します。そのため、認証待ちの接続リクエストもMaxConnectionsに含めます。PreHandleは接続ごとではなく、WebSocketメッセージごとに引き続き実行します。リクエストのヘッダー、クエリ、Cookieなどは、WebSocketHandshakeContextで読み取ります。

func (i *AuthInterceptor) PreHandshake(
    ctx core.WebSocketHandshakeContext,
    meta core.HandlerMeta,
) error {
    token := ctx.Header("Authorization")
    if token == "" {
        return httperr.Unauthorized("missing credentials")
    }
    return i.verify(token)
}

実行順序

正常なHTTP処理の流れは次のとおりです。

text

Global.PreHandle → Router → ArgumentResolver → Route.PreHandle
→ コントローラー → レスポンスの準備 → 実行後フック
→ Route.PostHandle → Global.PostHandle
→ Route.BeforeResponse → Global.BeforeResponse
→ 実際のレスポンスの書き込み
→ Route.AfterCompletion → Global.AfterCompletion

逆順で実行する各段階は、ルートインターセプターから始まり、グローバルインターセプターに続きます。ルーティング完了後の呼び出しでは、解決済みのHandlerMetaを受け取ります。

制約とエラー

BeforeResponseは、PreHandleがnilを返したインターセプターだけを呼び出します。エラーやcore.ErrAbortPipelineを返したインターセプターは対象に含めません。
AfterCompletionは、グローバルインターセプターと、ルーティングで特定したルートインターセプターに対して実行します。そのインターセプターのPreHandleが実行されていなくても呼び出される場合があるため、リソースの存在を確認してから解放してください。
ルートインターセプターの認証は、入力値の解釈後に実行します。入力値の解釈前に認証する必要がある場合は、グローバルインターセプターを使用してください。
core.ErrAbortPipelineで中断する場合は、インターセプターが必要なレスポンスを準備する必要があります。コントローラーと、それ以降の成功時の処理段階は実行しません。
DBコミット後に、実際のレスポンス送信が失敗する場合があります。BeforeResponseは、DBとネットワーク送信を原子的にまとめるものではありません。

具体的な早期終了とエラーの伝播はv0.6.0のパイプライン実装、接続前処理の順序はWebSocketランタイムの実装を参照してください。

core.Interceptor ​

インターフェース ​

ライフサイクルと実装例 ​

PreHandle ​

PostHandle ​

BeforeResponse ​

AfterCompletion ​

登録とスコープ ​

WebSocket接続前の処理 ​

実行順序 ​

制約とエラー ​

関連情報 ​