ECSで稼働するGoアプリケーションに PGOを導入する現実的な戦略を考える

package main

import (
	"context"
	"errors"
	"log"
	"net/http"
	_ "net/http/pprof" // importするだけでDefaultServeMuxにハンドラが登録される
	"os/signal"
	"syscall"
	"time"
)

func main() {
	// シグナルハンドリングの設定 (SIGINT, SIGTERM)
	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
	defer stop()

	// 1. PGOプロファイル取得用の裏口 (Port 6060)
	// DefaultServeMux (pprofが登録済み) を使用
	// ALB等からはルーティングせず、Security Groupで外部からのアクセスを完全に遮断する
	// ECS Execを用いてlocalhost経由でのみアクセスさせる
	go func() {
		log.Println("Pprof admin server listening on :6060")
		if err := http.ListenAndServe(":6060", nil); err != nil {
			log.Printf("pprof server error: %v", err)
		}
	}()

	// 2. メインのAPIサーバ (Port 8080)
	mux := http.NewServeMux()
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Hello, World!"))
	})

	srv := &http.Server{
		Addr:    ":8080",
		Handler: mux,
	}

	// サーバー起動 (非同期)
	go func() {
		log.Println("Main API server listening on :8080")
		if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			log.Fatalf("listen: %s\n", err)
		}
	}()

	// シグナル待ち
	<-ctx.Done()
	log.Println("shutting down gracefully...")

	// グレースフルシャットダウン (5秒の猶予)
	shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := srv.Shutdown(shutdownCtx); err != nil {
		log.Fatal("Server forced to shutdown: ", err)
	}

	log.Println("Server exiting")
}

resource "aws_ecs_task_definition" "main" {
  family                   = "my-sample-task"
  requires_compatibilities = ["FARGATE"]
  network_mode             = "awsvpc"
  cpu                      = 256
  memory                   = 512
  execution_role_arn       = aws_iam_role.ecs_task_execution_role.arn
  task_role_arn            = aws_iam_role.ecs_task_role.arn

  container_definitions = jsonencode([
    {
      name      = "my-sample"
      image     = "my-sample-image:latest"
      essential = true
      portMappings = [
        {
          # メインのAPI用
          containerPort = 8080
          hostPort      = 8080
          protocol      = "tcp"
        },
        {
          # PGOプロファイル収集用
          containerPort = 6060
          hostPort      = 6060
          protocol      = "tcp"
        }
      ]
      # ... その他の設定
    }
  ])
}

aws ecs execute-command \
    --cluster my-cluster \
    --task <TaskID> \
    --container sample-app \
    --interactive \
    --command "curl -o /tmp/default.pgo http://localhost:6060/debug/pprof/profile?seconds=30"

resource "aws_ecs_service" "main" {
  name            = "my-sample-service"
  cluster         = aws_ecs_cluster.main.id
  task_definition = aws_ecs_task_definition.main.arn
  desired_count   = 1
  launch_type     = "FARGATE"

  # ECS Execを有効化する
  enable_execute_command = true

  network_configuration {
    subnets         = var.private_subnets
    security_groups = [aws_security_group.ecs_task.id]
  }

  load_balancer {
    target_group_arn = aws_lb_target_group.main.arn
    container_name   = "my-app"
    container_port   = 8080
  }
}

# 取得した全ファイルを引数に渡して、1つの default.pgo にまとめる
go tool pprof -proto service_users.prof service_search.prof service_reports.prof > default.pgo

package main

import (
	"log"
	"os"
	"runtime/pprof"
	// ... aws sdk imports
)

func main() {
	// --- PGO計測開始 ---

	f, err := os.Create("cpu.prof")
	if err != nil {
		log.Printf("failed to create profile: %v", err)
	} else {
		if err := pprof.StartCPUProfile(f); err != nil {
			log.Printf("failed to start profile: %v", err)
			f.Close()
		} else {
			// 終了時に確実に停止 & S3アップロード
			defer func() {
				pprof.StopCPUProfile()
				f.Close()
				uploadProfileToS3("cpu.prof") // S3アップロード関数（実装省略）
			}()
		}
	}
	// ------------------

	log.Println("バッチ処理開始...")
	// ... 実際のビジネスロジック ...
}