@@ -193,12 +193,13 @@ class TLocalTaskRunner: public ITaskRunner {
193
193
194
194
/* ______________________________________________________________________________________________*/
195
195
196
- class TAbstractFactory : public IProxyFactory {
196
+ class TLocalFactory : public IProxyFactory {
197
197
public:
198
- TAbstractFactory (const NKikimr::NMiniKQL::IFunctionRegistry* functionRegistry,
198
+ TLocalFactory (const NKikimr::NMiniKQL::IFunctionRegistry* functionRegistry,
199
199
NKikimr::NMiniKQL::TComputationNodeFactory compFactory,
200
200
TTaskTransformFactory taskTransformFactory,
201
- std::shared_ptr<NKikimr::NMiniKQL::TComputationPatternLRUCache> patternCache)
201
+ std::shared_ptr<NKikimr::NMiniKQL::TComputationPatternLRUCache> patternCache,
202
+ bool terminateOnError)
202
203
: DeterministicMode(!!GetEnv(" YQL_DETERMINISTIC_MODE" ))
203
204
, RandomProvider(
204
205
DeterministicMode
@@ -210,6 +211,7 @@ class TAbstractFactory: public IProxyFactory {
210
211
: CreateDefaultTimeProvider())
211
212
, FunctionRegistry(functionRegistry)
212
213
, TaskTransformFactory(std::move(taskTransformFactory))
214
+ , TerminateOnError(terminateOnError)
213
215
{
214
216
ExecutionContext.FuncRegistry = FunctionRegistry;
215
217
ExecutionContext.ComputationFactory = compFactory;
@@ -218,29 +220,6 @@ class TAbstractFactory: public IProxyFactory {
218
220
ExecutionContext.PatternCache = patternCache;
219
221
}
220
222
221
- protected:
222
- bool DeterministicMode;
223
- TIntrusivePtr<IRandomProvider> RandomProvider;
224
- TIntrusivePtr<ITimeProvider> TimeProvider;
225
- const NKikimr::NMiniKQL::IFunctionRegistry* FunctionRegistry;
226
- TTaskTransformFactory TaskTransformFactory;
227
-
228
- NDq::TDqTaskRunnerContext ExecutionContext;
229
- };
230
-
231
- /* ______________________________________________________________________________________________*/
232
-
233
- class TLocalFactory : public TAbstractFactory {
234
- public:
235
- TLocalFactory (const NKikimr::NMiniKQL::IFunctionRegistry* functionRegistry,
236
- NKikimr::NMiniKQL::TComputationNodeFactory compFactory,
237
- TTaskTransformFactory taskTransformFactory,
238
- std::shared_ptr<NKikimr::NMiniKQL::TComputationPatternLRUCache> patternCache,
239
- bool terminateOnError)
240
- : TAbstractFactory(functionRegistry, compFactory, taskTransformFactory, patternCache)
241
- , TerminateOnError(terminateOnError)
242
- { }
243
-
244
223
ITaskRunner::TPtr GetOld (const TDqTaskSettings& task, const TString& traceId) override {
245
224
return new TLocalTaskRunner (task, Get (task, NDqProto::DQ_STATS_MODE_BASIC, traceId));
246
225
}
@@ -287,6 +266,12 @@ class TLocalFactory: public TAbstractFactory {
287
266
}
288
267
289
268
private:
269
+ bool DeterministicMode;
270
+ TIntrusivePtr<IRandomProvider> RandomProvider;
271
+ TIntrusivePtr<ITimeProvider> TimeProvider;
272
+ const NKikimr::NMiniKQL::IFunctionRegistry* FunctionRegistry;
273
+ TTaskTransformFactory TaskTransformFactory;
274
+ NDq::TDqTaskRunnerContext ExecutionContext;
290
275
const bool TerminateOnError;
291
276
};
292
277
0 commit comments