get-caller-identity.go

package main

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"

	"github.com/aws/aws-sdk-go-v2/config"
	"github.com/aws/aws-sdk-go-v2/service/sts"
)

type loggingRoundTripper struct {
	logger *log.Logger
	next   http.RoundTripper
}

func (lrt *loggingRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
	// Send the request
	resp, err := lrt.next.RoundTrip(req)
	if err != nil {
		return nil, err
	}

	// Copy the response body for logging
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}

	// Log the response body
	lrt.logger.Printf("Response Body: %s\n", string(body))

	// IMPORTANT: We need to replace the response body because it's been read
	resp.Body = io.NopCloser(bytes.NewReader(body))

	return resp, nil
}

func main() {

	// Initialize logger
	logger := log.New(os.Stdout, "", log.LstdFlags)

	// Load the Shared AWS Configuration (~/.aws/config)
	cfg, err := config.LoadDefaultConfig(context.TODO(),
		config.WithHTTPClient(&http.Client{
			Transport: &loggingRoundTripper{
				logger: logger,
				next:   http.DefaultTransport,
			},
		}),
	)
	if err != nil {
		log.Fatalf("unable to load SDK config, %v", err)
	}

	// Create an Amazon STS service client
	svc := sts.NewFromConfig(cfg)

	// Call the GetCallerIdentity operation
	result, err := svc.GetCallerIdentity(context.TODO(), &sts.GetCallerIdentityInput{})
	if err != nil {
		log.Fatalf("unable to get caller identity, %v", err)
	}

	fmt.Printf("%+v\n", result)
}