Lambda関数の品質を自動で守る！僕らが実践した「コンプライアンステスト」の全記録

// 既存のLambda関数（良い例）
func main() {
    log := logger.New("function-name") // ルール通りのログ
    metricsClient := middleware.NewMetricsClient() // ルール通りのメトリクス
    
    // 予期せぬエラーが起きてもプログラムが落ちないようにするおまじない
    defer func() {
        if r := recover(); r != nil {
            log.Error("Panic recovered", r)
        }
    }()
    
    lambda.Start(handler)
}

// 新しい開発者が追加した関数（問題のある例）  
func main() {
    // ログ、メトリクス、エラー回復処理が実装されていない！
    lambda.Start(handler)
}

// pkg/lambda/consistency_test.go
func TestLambdaFunctionConsistency(t *testing.T) {
    // これからチェックする関数の一覧
    lambdaFunctions := []string{
        "../../cmd/register/main.go",
        "../../cmd/login/main.go", 
        // ... 合計11個の関数
    }
    // ... この後、各関数をループでチェックしていく
}

// ...

// 実際のチェック処理
func CheckStructuredLogging(filePath string) bool {
    // Goのソースコードをテキストとして読み込む
    content, err := os.ReadFile(filePath)
    if err != nil {
        return false
    }
    
    codeStr := string(content)
    
    // 【正規表現】ソースコード内に 'logger.New(' という文字列があるかチェック
    loggerNewPattern := `logger\.New\s*\(`
    matched, _ := regexp.MatchString(loggerNewPattern, codeStr)
    if matched {
        return true
    }
    
    // 【正規表現】または、推奨している共通部品 'lambdaCommon.NewCommonHandler' が使われているかチェック
    commonHandlerPattern := `lambdaCommon\.NewCommonHandler`
    matched, _ = regexp.MatchString(commonHandlerPattern, codeStr)
    return matched
}

// ...
func CheckMetricsCollection(filePath string) bool {
    content, err := os.ReadFile(filePath)
    if err != nil {
        return false
    }
    codeStr := string(content)
    
    // パフォーマンス測定に関連するコードが存在するかチェック
    patterns := []string{
        `middleware\.NewMetricsClient\s*\(`, // メトリクス収集部品
        `time\.Now\s*\(\)`,               // 開始時間の記録
        `time\.Since\s*\(`,              // 実行時間の計算
        `lambdaCommon\.NewCommonHandler`, // 共通部品
    }
    
    for _, pattern := range patterns {
        if matched, _ := regexp.MatchString(pattern, codeStr); matched {
            return true
        }
    }
    return false
}

// ...
func CheckPanicRecovery(filePath string) bool {
    content, err := os.ReadFile(filePath)
    if err != nil {
        return false
    }
    codeStr := string(content)
    
    // パニック回復のお決まりパターンが存在するかチェック
    patterns := []string{
        `defer\s+func\s*\(\s*\)\s*{[\s\S]*recover\s*\(\s*\)`,  // deferとrecoverの組み合わせ
        `recover\s*\(\s*\)`,                                    // recover()の呼び出し
        `lambdaCommon\.NewCommonHandler`,                       // 共通部品
    }
    
    for _, pattern := range patterns {
        if matched, _ := regexp.MatchString(pattern, codeStr); matched {
            return true
        }
    }
    return false
}

// ...
func CheckSecureLogging(filePath string) bool {
    content, err := os.ReadFile(filePath)
    if err != nil {
        return false
    }
    codeStr := string(content)
    
    // 危険なログ出力パターンがないかチェック
    dangerousPatterns := []string{
        `log\..*password`,     // 例: log.Printf("password: %s", password)
        `log\..*Password`,
        `log\..*token.*request`, // リクエスト全体をログ出力している
        `fmt\.Printf.*password`,
    }
    
    for _, pattern := range dangerousPatterns {
        if matched, _ := regexp.MatchString(pattern, codeStr); matched {
            return false // 危険なパターンが見つかったら即NG
        }
    }
    
    // 安全なログ出力方法（マスキング機能付きなど）の存在をチェック
    safePatterns := []string{
        `logger\.New\s*\(`,      // 構造化ロガー
        `MaskSensitiveData\s*\(`, // マスキング関数
        `lambdaCommon\.NewCommonHandler`, // 共通部品
    }
    
    for _, pattern := range safePatterns {
        if matched, _ := regexp.MatchString(pattern, codeStr); matched {
            return true
        }
    }
    
    return false
}

=== RUN   TestLambdaFunctionCompliance/post-confirmation
    consistency_test.go:370: Function ../../cmd/post-confirmation/main.go does not meet all compliance requirements:
          ✅ Structured Logging: implemented
          ✅ Metrics Collection: implemented
          ✅ Panic Recovery: implemented
          ✅ CORS Middleware: implemented
          ✅ Request Logging: implemented
          ✅ Security Logging: compliant
          ❌ Performance Metrics: missing
          ❌ Error Handling: missing