Avoid a fatal error in PartManager.

This commit is contained in:
levlam 2023-03-08 20:34:23 +03:00
parent 5cd6b2d853
commit 4b00266a5b
2 changed files with 2 additions and 1 deletions

View File

@ -170,6 +170,7 @@ void FileLoadManager::cancel(QueryId query_id) {
}
on_error_impl(it->second, Status::Error(-1, "Canceled"));
}
void FileLoadManager::update_local_file_location(QueryId query_id, const LocalFileLocation &local) {
if (stop_flag_) {
return;

View File

@ -311,7 +311,7 @@ Result<Part> PartsManager::start_part() {
}
Status PartsManager::set_known_prefix(int64 size, bool is_ready) {
if (!known_prefix_flag_ || size < static_cast<size_t>(known_prefix_size_)) {
if (!known_prefix_flag_ || size < known_prefix_size_ || (size == 0 && !part_status_.empty())) {
CHECK(is_upload_);
return Status::Error("FILE_UPLOAD_RESTART");
}