インターセプター入門

var RequestLogger = Interceptor("RequestLogger", func() {
    Description("受信リクエストとそのタイミングをログに記録します")
    
    // リザルトからメソッドのステータスを読み取りたい
    ReadResult(func() {
        Attribute("status", Int, "返却されたステータスコード") // HTTPではなくビジネスロジックのステータスコード
    })
    
    // タイミング情報をリザルトに追加する
    WriteResult(func() {
        Attribute("processedAt", String, "リクエストが処理された時刻")
        Attribute("duration", Int, "処理時間（ミリ秒）")
    })
})

var _ = Service("calculator", func() {
    // サービス内のすべてのメソッドに適用
    ServerInterceptor(RequestLogger)
    
    Method("add", func() {
        // メソッド固有のインターセプター
        ServerInterceptor(ValidateNumbers)
        
        Payload(func() {
            Attribute("a", Int)
            Attribute("b", Int)
        })
        Result(Int)
    })
})

func (i *ServerInterceptors) RequestLogger(ctx context.Context, info *RequestLoggerInfo, next goa.Endpoint) (any, error) {
    start := time.Now()
    
    // 次のインターセプターまたは最終エンドポイントを呼び出す
    res, err := next(ctx, info.RawPayload())
    if err != nil {
        return nil, err
    }
    
    // 型安全なインターフェースを通じてリザルトにアクセス
    r := info.Result(res)
    
    // タイミング情報を追加
    r.SetProcessedAt(time.Now().Format(time.RFC3339))
    r.SetDuration(int(time.Since(start).Milliseconds()))
    
    return res, nil
}

// ServerInterceptorsはすべてのサーバーサイドインターセプターのインターフェースを定義
type ServerInterceptors interface {
    RequestLogger(ctx context.Context, info *RequestLoggerInfo, next goa.Endpoint) (any, error)
    // ... 他のインターセプター ...
}

// 情報構造体はインターセプションに関するメタデータを提供
type RequestLoggerInfo struct {
    service    string
    method     string
    callType   goa.InterceptorCallType
    rawPayload any
}

// リザルトにアクセスするための型安全なインターフェース
type RequestLoggerResult interface {
    Status() int
    SetProcessedAt(string)
    SetDuration(int)
}

type interceptors struct {
    logger *log.Logger
}

func NewInterceptors(logger *log.Logger) *interceptors {
    return &interceptors{logger: logger}
}

func (i *interceptors) RequestLogger(ctx context.Context, info *RequestLoggerInfo, next goa.Endpoint) (any, error) {
    // 前の例からの実装
    start := time.Now()
    res, err := next(ctx, info.RawPayload())
    if err != nil {
        return nil, err
    }
    
    r := info.Result(res)
    r.SetProcessedAt(time.Now().Format(time.RFC3339))
    r.SetDuration(int(time.Since(start).Milliseconds()))
    
    return res, nil
}

func main() {
    // サービス実装を作成
    svc := NewService()
    
    // インターセプターを作成
    interceptors := NewInterceptors(log.Default())
    
    // インターセプター付きでエンドポイントを作成
    endpoints := NewEndpoints(svc, interceptors)
    
    // ... 通常通り続行 ...
}