From ea1cd5ff3029315c1f89b98b820ceeebfba0e4df Mon Sep 17 00:00:00 2001
From: Aleksandr Nogikh <nogikh@google.com>
Date: Wed, 21 Aug 2024 14:28:06 +0200
Subject: pkg/build: introduce the build jobs parameter

This parameter defines the number of cores dedicated to the kernel build
process. By default, it's equal to the number of available CPUs.
---
 pkg/build/build.go | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'pkg/build/build.go')

diff --git a/pkg/build/build.go b/pkg/build/build.go
index 4d6e3bce3..2274ceb8e 100644
--- a/pkg/build/build.go
+++ b/pkg/build/build.go
@@ -12,6 +12,7 @@ import (
 	"os"
 	"path/filepath"
 	"regexp"
+	"runtime"
 	"strings"
 	"time"
 
@@ -37,6 +38,7 @@ type Params struct {
 	SysctlFile   string
 	Config       []byte
 	Tracer       debugtracer.DebugTracer
+	BuildJobs    int // If 0, all CPUs will be used.
 	Build        json.RawMessage
 }
 
@@ -71,6 +73,9 @@ func Image(params Params) (details ImageDetails, err error) {
 	if params.Tracer == nil {
 		params.Tracer = &debugtracer.NullTracer{}
 	}
+	if params.BuildJobs == 0 {
+		params.BuildJobs = runtime.NumCPU()
+	}
 	var builder builder
 	builder, err = getBuilder(params.TargetOS, params.TargetArch, params.VMType)
 	if err != nil {
-- 
cgit mrf-deployment