fix

ydb-platform · Apr 26, 2024 · db4f171 · db4f171
1 parent 363d83f
commit db4f171
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ydb/library/yql/tools/dqrun/dqrun.cpp b/ydb/library/yql/tools/dqrun/dqrun.cpp
@@ -643,7 +643,6 @@ int RunMain(int argc, const char* argv[])
         qStorage = MakeFileQStorage(qStorageDir);
         qContext = TQContext(qStorage->MakeReader(opId));
     } else if (res.Has("capture")) {
-        Y_ENSURE(opId);
         qStorage = MakeFileQStorage(qStorageDir);
         qContext = TQContext(qStorage->MakeWriter(opId));
     }
@@ -997,7 +996,10 @@ int RunMain(int argc, const char* argv[])
         program->SetQueryName(progFile);
     }
 
-    program->SetQContext(qContext);
+    if (qStorage) {
+        program->SetQContext(qContext);
+    }
+
     if (paramsFile) {
         TString parameters = TFileInput(paramsFile).ReadAll();
         program->SetParametersYson(parameters);