はじめに

AWS SDK for Go v2 を使って CLI ツールなどを開発していると、内部で呼ばれる SDK のロギングもきちんとやってみようという謎のモチベーションがときどき湧いてきますよね。

オプションでログレベルを渡すように設計し、それに応じてログ出力をいい感じに整えたいですが、どうせなら SDK のログも同じフォーマットで出力したいものです。特にデバッグ時には、アプリのログなのか SDK のログなのかが一見してわかるようになっているとトラブルシューティングが捗りそうです。今回はこれを実装してみます。

概要

SDK に対してカスタムロガーを渡さない場合は logging.StandardLogger が呼ばれます。このロガーが簡素な出力を行い、アプリ側で定義したロガーとフォーマットが異なって出力されるためモヤモヤするというのが今回の痒いところになります。このためカスタムロガーを設定することになりますが、どちらから出力されたのかを識別するための情報もほしいところです。

SDK はデフォルトではほとんどログを出力しないため、通常は特に気にする必要はありません。ただし SDK のデバッグログを出力したい場合などは大量のログが出るため「ログの出し方をもっと統一したいなぁ」となりました。これに対する解決策ですが、以下の流れで見通しよく実装できました。

  • 埋め込みを使ってアプリ用と SDK 用でそれぞれ同じ内容のロガーを作る
  • それぞれのロガーにはどちらのログかを判別できる情報 (label) を設定する
  • SDK 用のロガーは logging.Logger インターフェースを実装する
  • それぞれ呼び出す

実装

アプリ用と SDK 用でそれぞれロガーを作る

今回は以下のように log/slog でベーシックなロガーを作ります。アプリ用か SDK 用かを識別するために label を出力するようにしました。label=App のように表示されます。

// newLogger creates a new base logger used by AppLogger and SDKLogger.
func newLogger(w io.Writer, level slog.Leveler, label string) *slog.Logger {
    opts := &slog.HandlerOptions{
        Level: level,
    }
    l := slog.New(slog.NewTextHandler(w, opts))
    l = l.With(slog.String("label", label))
    return l
}

次にこれをアプリ用、SDK 用それぞれのロガーに埋め込みます。アプリ用は log/slog のロガーをラップしているだけです。

// AppLogger is a logger for the application.
type AppLogger struct {
    *slog.Logger
}

// NewAppLogger creates a new logger for the application.
func NewAppLogger(w io.Writer, level slog.Leveler, label string) *AppLogger {
    l := newLogger(w, level, label)
    return &AppLogger{l}
}

SDK 用も同様に、この時点ではラップしているだけです。

// SDKLogger is a logger for the AWS SDK. This implemented the logging.Logger interface.
// See: https://github.com/aws/smithy-go/blob/main/logging/logger.go
type SDKLogger struct {
    *slog.Logger
}

// NewSDKLogger creates a new logger for AWS SDK.
func NewSDKLogger(w io.Writer, level slog.Leveler, label string) *SDKLogger {
    l := newLogger(w, level, label)
    return &SDKLogger{l}
}

logging.Logger インターフェースを実装する

これだけだと SDKLogger をカスタムロガーとして SDK に渡せないため、Logf を実装する必要があります。logging.Classification には WARNDEBUG しかないので、これらを switch 文で振り分けて SDKLogger の各メソッドを単純に実行しています。

// Logf logs a message with formatting.
func (l *SDKLogger) Logf(classification logging.Classification, format string, v ...any) {
    s := fmt.Sprintf(format, v...)
    switch classification {
    case logging.Warn:
        l.Warn(s)
    case logging.Debug:
        l.Debug(s)
    default:
        l.Info(s)
    }
}

呼び出す

あとはアプリの中で適切に呼び出すだけです。ロガーの渡し方については公式ドキュメントを参照してください。例えば以下のような形になるでしょうか。

func main() {
    ctx := context.Background()

    // Writer を定義する
    w := os.Stderr

    // ログレベルを定義する
    level := slog.LevelDebug

    // 各ロガーを定義する
    appLogger := log.NewAppLogger(w, level, "App")
    sdkLogger := log.NewSDKLogger(w, level, "SDK")

    cfg, err := config.LoadDefaultConfig(ctx,
        // SDK Config に SDK 用ロガーを渡す
        config.WithLogger(sdkLogger),

        // SDK Config にログモードを渡す
        // 今回はレスポンスだけ出す
        config.WithClientLogMode(aws.LogResponse),
    )

    // SDK Config からクライアントを作る
    // 今回は例として S3 のクライアントを作る
    client := s3.NewFromConfig(cfg)

    // なんらかの処理を実行してログを出力する
    out, err := client.ListBuckets(ctx, &s3.ListBucketsInput{})
    if err != nil {
        panic(err)
    }
    for _, bucket := range out.Buckets {
        appLogger.Info(aws.ToString(bucket.Name))
    }
}

これを実行してみると、以下のような出力が得られます。今回は標準のテキストハンドラを使っており、メッセージがエスケープされるなどコンソール向けではないですが、統一的なフォーマットでログが表示されていることがわかります。label 属性で App, SDK のどちらが出力したものかも識別できます (見づらいですが、右にスクロールしていただくと label=SDK となっています)

$ go run main.go
time=2025-03-20T19:31:23.506+09:00 level=DEBUG msg="Response\nHTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\nContent-Type: application/xml\r\nDate: Thu, 20 Mar 2025 10:31:24 GMT\r\nServer: AmazonS3\r\nX-Amz-Id-2: XXXX\r\nX-Amz-Request-Id: XXXX\r\n\r\n" label=SDK
time=2025-03-20T19:31:23.507+09:00 level=INFO msg=abc-bucket label=App
time=2025-03-20T19:31:23.507+09:00 level=INFO msg=cdk-hnb659fds-assets-000000000000-ap-northeast-1 label=App
time=2025-03-20T19:31:23.507+09:00 level=INFO msg=cf-templates-111111111111-ap-northeast-1 label=App

汎用化してみた

完全に自分用ですが、github.com/charmbracelet/log というサードパーティー製のロガーをラップする形で、今回紹介した内容をパッケージ化しています。

GitHub - nekrassov01/logwrapper: Small, thin wrapper for logging in Go and AWS SDK
Small, thin wrapper for logging in Go and AWS SDK. Contribute to nekrassov01/logwrapper development by creating an account on GitHub.

これを使うことで以下のような出力を得ることができます (拙作の llcm のログです)

console

おわりに

最後に紹介した自分用のラッパーはミニマム構成で割と気に入っているのですが、今後はやはり Go の標準ライブラリである log/slog で、この文脈でも使える自前のハンドラを書いてみたいと思います。