PrebuiltModuleGen: fail prebuilt module generation job when critical modules failed to build

Generating prebuilt modules for critical modules is important because missing them has a bigger
impact on the overall build time. This change adds a report of generated modules when the job
finishes and fails the prebuilt module job if some selected critical modules were missing.

rdar://132094476

Author: nkcsgexi

Sources/SwiftDriver/Jobs/PrebuiltModulesJob.swift

@@ -173,6 +173,7 @@ fileprivate class ModuleCompileDelegate: JobExecutionDelegate {
   var failingCriticalOutputs: Set<VirtualPath>
   let logPath: AbsolutePath?
   let jsonDelegate: JSONOutputDelegate
+  var compiledModules: [String: Int] = [:]
   init(_ jobs: [Job], _ diagnosticsEngine: DiagnosticsEngine, _ verbose: Bool,
               _ logPath: AbsolutePath?, _ jsonDelegate: JSONOutputDelegate) {
     self.diagnosticsEngine = diagnosticsEngine
@@ -201,13 +202,34 @@ fileprivate class ModuleCompileDelegate: JobExecutionDelegate {
     return !failingCriticalOutputs.isEmpty
   }
 
+  public func checkCriticalModulesGenerated() -> Bool {
+    let sortedModules = compiledModules.sorted(by: <)
+    Driver.stdErrQueue.sync {
+      stderrStream.send("===================================================\n")
+      sortedModules.forEach {
+        stderrStream.send("\($0.key): \($0.value)\n")
+      }
+      stderrStream.send("===================================================\n")
+      stderrStream.flush()
+    }
+    let keyModules = ["Swift", "SwiftUI", "Foundation"]
+    return keyModules.allSatisfy { compiledModules.keys.contains($0) }
+  }
+
   public func jobFinished(job: Job, result: ProcessResult, pid: Int) {
     self.jsonDelegate.jobFinished(job, result)
     switch result.exitStatus {
     case .terminated(code: let code):
       if code == 0 {
         printJobInfo(job, false, verbose)
         failingCriticalOutputs.remove(job.outputs[0].file)
+
+        // Keep track of Swift modules that have been already generated.
+        if let seen = compiledModules[job.moduleName] {
+          compiledModules[job.moduleName] = seen + 1
+        } else {
+          compiledModules[job.moduleName] = 1
+        }
       } else {
         failingModules.insert(job.moduleName)
         let result: String = try! result.utf8stderrOutput()
@@ -317,6 +339,9 @@ public class PrebuiltModuleGenerationDelegate: JobExecutionDelegate {
   public var hasCriticalFailure: Bool {
     return compileDelegate.hasCriticalFailure
   }
+  public func checkCriticalModulesGenerated() -> Bool {
+    return compileDelegate.checkCriticalModulesGenerated()
+  }
   public func emitJsonOutput(to path: AbsolutePath) throws {
     let data = try JSONEncoder().encode(self.jsonDelegate)
     if let json = try? JSONSerialization.jsonObject(with: data, options: .mutableContainers),

Sources/swift-build-sdk-interfaces/main.swift

@@ -206,6 +206,9 @@ do {
       if let jsonPath = jsonPath {
         try! delegate.emitJsonOutput(to: jsonPath)
       }
+      if !delegate.checkCriticalModulesGenerated() {
+        exit(1)
+      }
     }
     do {
       try executor.execute(workload: DriverExecutorWorkload.init(jobs, nil, nil, continueBuildingAfterErrors: true),