コンテンツにスキップ

ミドルウェア

どのリクエストも、普段は意識しないミドルウェアのスタックを通ります。panic 回復用の ハンドラを自分で設置していなくても、フレームワークが panic を捕捉します。しかし、 レスポンスヘッダーが足りない、ログに request ID がない、ヘルスプローブが 503 を返す、 といった問題を調べるには、ソケットからハンドラまでの間で何が実行されているかを知る 必要があります。以下では、組み込まれている処理、その実行順、設定との対応、独自の ミドルウェアを挿入できる位置を説明します。

すべてのフレームは1本の番号線の上にいて、外側から内側へ昇順に並びます。フレームワーク 自身のフレームは10の倍数を占めます — BASIC の行番号と同じで、理由も同じです。隙間は 独自ミドルウェアを挿入するためのものです。各番号には公開された定数(pw.SlotRequestIDpw.SlotAccessLog など)が あるので、位置は裸の整数ではなく名前からの相対で書けます。

スロット フレーム 役割 スイッチ
リクエスト追跡 graceful shutdown のため処理中リクエストを数える 常時
10 OpenTelemetry リクエストのルートスパンを開く トレースの出力先があるときだけ
20 リソース注入 ロガー・データベース・設定をコンテキストへ 常時
30 リクエスト ID 全ログ行が携える ID を検証または発行 middleware.request_id
40 アクセスログ 1リクエスト1行、所要時間つき middleware.access_log
50 recover panic を交渉済みのエラーレスポンスへ変換 middleware.recovery
52 レスポンスポリシー CSP・HSTS とクロスオリジンの印を、何かが書き込む前に security.headers.enabled, security.cors.enabled
70 リクエストタイムアウト リクエスト全体を時間で縛る middleware.request_timeout
80 ボディ上限 リクエストボディの読み取り量に蓋をする server.max_request_body
90 公開アセット 動的処理の前に静的ツリーを返す server.public.enabled
100 プローブ health と readiness を、認証より上で server.health, server.readiness
110–150 拡張 ストレージ・セッション・認証・CSRF・ガード 拡張ごと
160 API ドキュメント OpenAPI ドキュメントと UI を、ガードの下で server.openapi, server.apidoc
あなたのハンドラ pw.Run に渡した mux

1つだけ 10 の倍数に乗っていない枠があります。レスポンスポリシーが 52 にいるのは、 そこで何を設定するかではなく、何を書き込むかの都合です。Access-Control-Allow-Origin の付いていないクロスオリジンレスポンスは、ブラウザが誰にも渡しません。ステータスごと 渡しません。だから印を付ける枠は、リクエストを拒否しうる枠すべてより先に走る必要が あります — 上の表に出てこない 55 のプロセス全体レートリミットも含めて。60 はその下でした。 つまりクロスオリジンの呼び出し元には、429 が読めない形で届いていた。移した副産物として、 その 429 にセキュリティヘッダも乗るようになりました。クロスオリジンリクエストを参照。

リクエスト追跡だけは線の外、最外周にいます。shutdown の計数は番号つきの全ステップを 観測しなければならないからです。ハンドラは定義上の最内端。そのあいだは、番号だけが 順序を決めます。

並び順はアルファベット順でも歴史的経緯でもありません。1つ1つの位置が論証です。 リクエスト ID がアクセスログの外側にあるのは、ログの行が ID を載せられるように。 アクセスログが recover の外側にあるのは、panic したリクエストも所要時間と 500 つきで 記録に残るように。プローブが拡張チェーンより上にあるのは、セッションストアが落ちていても liveness チェックは成功するように — 依存先の障害は readiness を落とすべきであって、 再起動ループに化けるべきではありません。そして OpenAPI ドキュメントはガードのに います。API 表面全体の地図には、そこに描かれたルートと同じ保護がかかるべきだからです。

圧縮はこれらの隣で設定しますが(middleware.compression)、チェーンのフレームでは ありません。適用されるのはレスポンスを書く場所です。いつ有効にすべきかは 圧縮のガイドにあります。

110 から 150 の区間もハードコードされていません。import された機能がそこに自分を登録し、 チェーンは同じ番号で組み上がります。

スロット 定数 登録するもの
110 pw.SlotStorage ストレージクライアントを開くセッションバックエンド
120 pw.SlotSession フレームワークのセッション解決
130 pw.SlotAuthentication plugin/auth
140 pw.SlotCSRF security.csrf が有効にする CSRF 検査
150 pw.SlotGuard 認証ガード

番号には意図があります。150 のガードは、120 で解決されたセッションと 130 で確定した 認証を必ず観測できる。機能を提供するパッケージ — 再利用するコンポーネントパッケージでも、 自分のアプリ内のパッケージでも — は init から登録します。

package audit
import (
"context"
"net/http"
"github.com/shibukawa/popcornweb/pw"
)
func init() {
pw.RegisterExtension(pw.Extension{
Name: "audit",
Slot: pw.SlotGuard + 1, // 認証の後、ガードの後
Setup: func(ctx context.Context) (pw.Middleware, error) {
return func(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// セッションと認証はこの上で解決済み。
next.ServeHTTP(w, r)
})
}, nil
},
})
}

Setup は起動時に一度、設定のパースとデータベースの起動の後に走り、ハンドラが見るのと 同じリソースを受け取ります。設定を間違えた拡張は最初のリクエストではなく起動を失敗させる、 というわけです。nil のミドルウェアを返せば何も設置されません。無効化された機能は そうやって抜けます。

pw.RegisterMiddleware はスロット・名前・素の func(http.Handler) http.Handler を 受け取り、pw.Runpw.Middlewares が組むチェーンのその位置に収まります。呼ぶのは main から、全パッケージの init の後、チェーンが組まれる前。pw.RegisterSessionStore と同じタイミングで、理由も同じです。チェーンは一度だけ組まれるので、後から登録しても どこにも入りません。

小さなミドルウェアの一番おいしい使い道は、リクエストごとの事実を一度だけ導出して、 下の全員に読ませることです。session.RequestScope は まさにこのための配置で、代表例はリクエスト時刻です。書き込みのたびに time.Now() を 呼ぶハンドラは、タイムスタンプをリクエスト内に撒き散らします。1回のフォーム送信で 更新した3行が、ハンドラの処理時間ぶんずつずれた3つの updated_at を持つことになる。 かわりに、瞬間を一度だけ捕まえます。型から登録までのプログラム全体はこうなります。

cmd/myapp/main.go
package main
import (
"context"
"log"
"net/http"
"time"
"github.com/shibukawa/popcornweb/pw"
"github.com/shibukawa/popcornweb/session"
"myapp/handlers"
)
type RequestTime struct {
At time.Time `json:"at"`
}
func withRequestTime(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if handle, ok := session.Value[RequestTime](r.Context()); ok {
handle.Set(RequestTime{At: time.Now()})
}
next.ServeHTTP(w, r)
})
}
func main() {
// このミドルウェアはセッション状態に書き込むので、
// 120 のセッション解決より下に置く。
pw.RegisterSessionStore[RequestTime]("request_time", session.RequestScope)
pw.RegisterMiddleware(pw.SlotSession+5, "request_time", withRequestTime)
if err := pw.Run(context.Background(), handlers.Handlers()); err != nil {
log.Fatal(err)
}
}

以後、そのリクエストの書き込みは session.Load[RequestTime]updated_at に使い、 1回の送信は1つの瞬間を刻みます。同じ形は「ちょうど1リクエストのあいだ真である事実」 なら何にでも使えます。bearer トークンが解決するスコープ集合。リクエスト冒頭で取った フィーチャーフラグのスナップショット — 処理の途中でフラグの切り替えが見えてしまわない ように。

この例が開いたままにしている判断は番号だけで、それはミドルウェアが何を観測したいかで 選びます。20 より上ではコンテキストにリソースが無い。50 より下なら panic は recover が 受け止める。120 より下ならセッションは解決済み。150 より後ろにはガードが通した リクエストしか来ない。リクエスト時刻が pw.SlotSession+5 にいるのは、120 より上には 存在しないセッション状態へ書き込むからです。ヘッダを読むだけのミドルウェアなら ずっと上でいい — たとえば pw.SlotAccessLog-5 なら、30 のリクエスト ID は発行済みで、 40 のアクセスログがこの先を計時してくれます。同じ番号の2つは登録順に走るので、 順序に依存しない組なら相乗りで構いません。

登録を拒否する位置が2つあります。100 と 160、プローブと API ドキュメントです。これらは ミドルウェアではなくハンドラで、同じ位置を誰かと分け合えません。panic は移動先の基準に なる定数の名前を挙げます。

線の外に残る継ぎ目は1つ。pw.Middlewares の戻り値を包む位置で、フレームワークが 答えてしまうリクエスト — プローブ込み — まで観測したい稀なミドルウェアのためのものです。

handler, err := pw.Middlewares(mux)
if err != nil {
log.Fatal(err)
}
err = http.ListenAndServe(":8080", myOutermost(handler))

この位置はスタックが提供するものを全部手放します。リクエスト ID も recover も、 コンテキストのリソースも無い。生のリクエストを観測すること自体が目的のときだけ選んで ください。それ以外は、線の上の番号が意図を語り、チェーンがそれを守ります。

番号線は同じものです。pwfast.RegisterMiddleware はスロット・名前・ func(fasthttp.RequestHandler) fasthttp.RequestHandler を受け取り、スロット定数も同じ 定数で、pwfast.Runpwfast.Startpwfast.Middlewares が組むチェーンの同じ位置に 収まります。拒否も3つとも同じです。nil、名前の重複、100 と 160 の固定フレーム。

違うのは包み方だけで、これはそうならざるを得ませんでした。ミドルウェアは自分より下の 全部を包むので、1つをアダプタで挟むと、その下のチェーンごと反対側のトランスポートの ハンドラ型に引きずり込まれます。ルート単位の逃げ道が避けているコストが、チェーン全体に 一度でかかる。登録のほうには違える理由が無かった。だから main.gomain_fasthttp.go の差は、包み方だけです。

cmd/myapp/main_fasthttp.go
//go:build fasthttp
package main
import (
"context"
"log"
"time"
"github.com/shibukawa/popcornweb/pwfast"
"github.com/shibukawa/popcornweb/pwsession"
"github.com/shibukawa/popcornweb/session"
"github.com/shibukawa/tinygodriver/fasthttp"
"myapp/handlers"
)
type RequestTime struct {
At time.Time `json:"at"`
}
func withRequestTime(next fasthttp.RequestHandler) fasthttp.RequestHandler {
return func(r *fasthttp.RequestCtx) {
if handle, ok := session.Value[RequestTime](r); ok {
handle.Set(RequestTime{At: time.Now()})
}
next(r)
}
}
func main() {
// セッションストアの登録はトランスポートに依らないので pwfast ではなく
// pwsession。pw.RegisterSessionStore はこれの re-export。
pwsession.RegisterStore[RequestTime]("request_time", session.RequestScope)
pwfast.RegisterMiddleware(pwfast.SlotSession+5, "request_time", withRequestTime)
mux := pwfast.NewServeMux()
handlers.RegisterRoutes(pwfast.Routes(mux))
if err := pwfast.Run(context.Background(), mux.Handler); err != nil {
log.Fatal(err)
}
}

このトランスポートではリクエスト値そのものがコンテキストです。だから net/http 版が r.Context() を渡すところで session.Valuer を受け取る。ミドルウェア本体の移植は、 ほぼこの置き換えで終わります。包み方でもヘッダの読み書きでもない部分は トランスポートに依らないので、判断する側を素の関数に切り出して両方から呼べば、 ビルドごとに書くのは自分のリクエストに手を入れる数行だけになります。

登録の隣に継ぎ目が2つあります。import した機能 — 認証プラグイン、ストレージ連携 — は ここには登録しません。pwfast.RuntimeOptions.Extra でフレームを渡し、それを pwfast.Run の呼び出しでアプリが名指しする。パッケージを import しただけで このチェーンに何かが加わることはない、というわけです。もう1つはフレームワークが 設置する全部の外側で、それは pwfast.Start です。組み上がったチェーンと、起動が 開いたものを閉じる shutdown を返すので、包んでから自分で serve できます。

serverless ターゲットに追加の作業はありません。pw build --target はエントリポイントの RunStart へ書き換えますが、その呼び出しに達する時点で main は登録を済ませて いるので、フレームは関数ハンドラが serve するチェーンに入っています。 サーバーレスを参照。