File tree Expand file tree Collapse file tree 2 files changed +10
-2
lines changed
yql/essentials/minikql/comp_nodes Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -985,7 +985,7 @@ class TSqueezeSetFlowWrapper : public TStatefulFlowCodegeneratorNode<TSqueezeSet
985
985
986
986
NUdf::TUnboxedValuePod DoCalculate (NUdf::TUnboxedValue& state, TComputationContext& ctx) const {
987
987
if (state.IsFinish ()) {
988
- return state. Release () ;
988
+ return state;
989
989
} else if (state.IsInvalid ()) {
990
990
MakeState (ctx, state);
991
991
}
@@ -1162,7 +1162,7 @@ class TSqueezeSetWideWrapper : public TStatefulFlowCodegeneratorNode<TSqueezeSet
1162
1162
1163
1163
NUdf::TUnboxedValuePod DoCalculate (NUdf::TUnboxedValue& state, TComputationContext& ctx) const {
1164
1164
if (state.IsFinish ()) {
1165
- return state. Release () ;
1165
+ return state;
1166
1166
} else if (state.IsInvalid ()) {
1167
1167
MakeState (ctx, state);
1168
1168
}
Original file line number Diff line number Diff line change @@ -147,6 +147,10 @@ Y_UNIT_TEST_SUITE(TMiniKQLToDictTest) {
147
147
148
148
status = res.Fetch (v);
149
149
UNIT_ASSERT_VALUES_EQUAL (NUdf::EFetchStatus::Finish, status);
150
+ // XXX: Check whether the internal state is not released
151
+ // and the sentinel is still set (see more info in YQL-19866).
152
+ status = res.Fetch (v);
153
+ UNIT_ASSERT_VALUES_EQUAL (NUdf::EFetchStatus::Finish, status);
150
154
};
151
155
152
156
for (auto stream : {true , false }) {
@@ -201,6 +205,10 @@ Y_UNIT_TEST_SUITE(TMiniKQLToDictTest) {
201
205
202
206
status = res.Fetch (v);
203
207
UNIT_ASSERT_VALUES_EQUAL (NUdf::EFetchStatus::Finish, status);
208
+ // XXX: Check whether the internal state is not released
209
+ // and the sentinel is still set (see more info in YQL-19866).
210
+ status = res.Fetch (v);
211
+ UNIT_ASSERT_VALUES_EQUAL (NUdf::EFetchStatus::Finish, status);
204
212
};
205
213
206
214
for (auto hashed : {true , false }) {
You can’t perform that action at this time.
0 commit comments