1
1
Fork 0
mirror of https://github.com/NixOS/nix.git synced 2025-11-09 03:56:01 +01:00

Combine DerivationBuilder::{prepareBuild,startBuilder}

After many other cleanups, it turns out there is no reason for these to
be separate methods. We can combine them to simplify things.
This commit is contained in:
John Ericson 2025-09-03 17:58:50 -04:00
parent 14c206f05a
commit 2acb9559d5
3 changed files with 33 additions and 41 deletions

View file

@ -689,6 +689,8 @@ Goal::Co DerivationBuildingGoal::tryToBuild()
#else #else
assert(!hook); assert(!hook);
Descriptor builderOut;
// Will continue here while waiting for a build user below // Will continue here while waiting for a build user below
while (true) { while (true) {
@ -781,7 +783,17 @@ Goal::Co DerivationBuildingGoal::tryToBuild()
}); });
} }
if (!builder->prepareBuild()) { std::optional<Descriptor> builderOutOpt;
try {
/* Okay, we have to build. */
builderOutOpt = builder->startBuild();
} catch (BuildError & e) {
builder.reset();
outputLocks.unlock();
worker.permanentFailure = true;
co_return doneFailure(std::move(e)); // InputRejected
}
if (!builderOutOpt) {
if (!actLock) if (!actLock)
actLock = std::make_unique<Activity>( actLock = std::make_unique<Activity>(
*logger, *logger,
@ -790,26 +802,15 @@ Goal::Co DerivationBuildingGoal::tryToBuild()
fmt("waiting for a free build user ID for '%s'", Magenta(worker.store.printStorePath(drvPath)))); fmt("waiting for a free build user ID for '%s'", Magenta(worker.store.printStorePath(drvPath))));
co_await waitForAWhile(); co_await waitForAWhile();
continue; continue;
} } else {
builderOut = *std::move(builderOutOpt);
};
break; break;
} }
actLock.reset(); actLock.reset();
Descriptor builderOut;
try {
/* Okay, we have to build. */
builderOut = builder->startBuilder();
} catch (BuildError & e) {
builder.reset();
outputLocks.unlock();
worker.permanentFailure = true;
co_return doneFailure(std::move(e)); // InputRejected
}
worker.childStarted(shared_from_this(), {builderOut}, true, true); worker.childStarted(shared_from_this(), {builderOut}, true, true);
started(); started();

View file

@ -147,17 +147,15 @@ struct DerivationBuilder : RestrictionContext
* locks as needed). After this is run, the builder should be * locks as needed). After this is run, the builder should be
* started. * started.
* *
* @returns true if successful, false if we could not acquire a build * @returns logging pipe if successful, `std::nullopt` if we could
* user. In that case, the caller must wait and then try again. * not acquire a build user. In that case, the caller must wait and
*/ * then try again.
virtual bool prepareBuild() = 0;
/**
* Start building a derivation.
* *
* @return logging pipe * @note "success" just means that we were able to set up the environment
* and start the build. The builder could have immediately exited with
* failure, and that would still be considered a successful start.
*/ */
virtual Descriptor startBuilder() = 0; virtual std::optional<Descriptor> startBuild() = 0;
/** /**
* Tear down build environment after the builder exits (either on * Tear down build environment after the builder exits (either on

View file

@ -214,9 +214,7 @@ protected:
public: public:
bool prepareBuild() override; std::optional<Descriptor> startBuild() override;
Descriptor startBuilder() override;
SingleDrvOutputs unprepareBuild() override; SingleDrvOutputs unprepareBuild() override;
@ -470,19 +468,6 @@ bool DerivationBuilderImpl::killChild()
return ret; return ret;
} }
bool DerivationBuilderImpl::prepareBuild()
{
if (useBuildUsers()) {
if (!buildUser)
buildUser = getBuildUser();
if (!buildUser)
return false;
}
return true;
}
SingleDrvOutputs DerivationBuilderImpl::unprepareBuild() SingleDrvOutputs DerivationBuilderImpl::unprepareBuild()
{ {
/* Since we got an EOF on the logger pipe, the builder is presumed /* Since we got an EOF on the logger pipe, the builder is presumed
@ -679,8 +664,16 @@ static bool checkNotWorldWritable(std::filesystem::path path)
return true; return true;
} }
Descriptor DerivationBuilderImpl::startBuilder() std::optional<Descriptor> DerivationBuilderImpl::startBuild()
{ {
if (useBuildUsers()) {
if (!buildUser)
buildUser = getBuildUser();
if (!buildUser)
return std::nullopt;
}
/* Make sure that no other processes are executing under the /* Make sure that no other processes are executing under the
sandbox uids. This must be done before any chownToBuilder() sandbox uids. This must be done before any chownToBuilder()
calls. */ calls. */