Make NIX_SHOW_STATS work with new-style commands

2021-02-05 13:35:31 +01:00 · 2021-02-05 13:35:31 +01:00 · 7331da99ab
parent e64cf8e0a3
commit 7331da99ab
2 changed files with 8 additions and 0 deletions
--- a/src/libcmd/command.hh
+++ b/src/libcmd/command.hh
@ -48,6 +48,8 @@ struct EvalCommand : virtual StoreCommand, MixEvalArgs
    ref<EvalState> getEvalState();

    std::shared_ptr<EvalState> evalState;
+
+    ~EvalCommand();
 };

 struct MixFlakeOptions : virtual Args, EvalCommand
--- a/src/libcmd/installables.cc
+++ b/src/libcmd/installables.cc
@ -280,6 +280,12 @@ ref<EvalState> EvalCommand::getEvalState()
    return ref<EvalState>(evalState);
 }

+EvalCommand::~EvalCommand()
+{
+    if (evalState)
+        evalState->printStats();
+}
+
 void completeFlakeRef(ref<Store> store, std::string_view prefix)
 {
    if (prefix == "")