Skip to content

Commit d89cd84

Browse files
committed
Fixed issues #1
1 parent ce55390 commit d89cd84

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ydb/core/fq/libs/compute/ydb/result_writer_actor.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ class TResultSetWriterActor : public TBaseComputeActor<TResultSetWriterActor> {
115115

116116
if (!splittedResultSets.Success) {
117117
LOG_E("ResultSetId: " << ResultSetId << " Can't split script result: " << splittedResultSets.Issues.ToOneLineString());
118-
Send(Parent, new TEvYdbCompute::TEvResultSetWriterResponse(ResultSetId, splittedResultSets.Issues, NYdb::EStatus::BAD_REQUEST));
118+
Send(Parent, new TEvYdbCompute::TEvResultSetWriterResponse(ResultSetId, splittedResultSets.Issues, NYdb::EStatus::INTERNAL_ERROR));
119119
FailedAndPassAway();
120120
return;
121121
}
@@ -132,7 +132,7 @@ class TResultSetWriterActor : public TBaseComputeActor<TResultSetWriterActor> {
132132

133133
if (WriterInflight.empty()) {
134134
SendReplyAndPassAway();
135-
} else if (FetchToken && WriterInflight.size() < MAX_WRITER_INFLIGHT) {
135+
} else if (FetchToken && (WriterInflight.size() + ResultChunks.size()) < 2 * MAX_WRITER_INFLIGHT) {
136136
SendFetchScriptResultRequest();
137137
}
138138
}
@@ -168,7 +168,7 @@ class TResultSetWriterActor : public TBaseComputeActor<TResultSetWriterActor> {
168168
writeResultCounters->Ok->Inc();
169169
LOG_I("ResultSetId: " << ResultSetId << " Cookie: " << cookie << " Result successfully written for offset " << meta.Offset);
170170
if (FetchToken) {
171-
if (FetchToken != LastProcessedToken && WriterInflight.size() < MAX_WRITER_INFLIGHT) {
171+
if (FetchToken != LastProcessedToken && (WriterInflight.size() + ResultChunks.size()) < 2 * MAX_WRITER_INFLIGHT) {
172172
SendFetchScriptResultRequest();
173173
}
174174
} else if (WriterInflight.empty()) {

0 commit comments

Comments
 (0)