milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: Fix performance issue and use after free bug () 

cherry pick some fixes in https://github.com/milvus-io/milvus/pull/39249

Signed-off-by: sunby <sunbingyi1992@gmail.com>
pull/39390/head
Bingyi Sun 2025-01-17 11:51:03 +08:00 committed by GitHub
parent 47e410feb1
commit 56cb1683eb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 10 deletions
internal/core/src
exec/expression
segcore

19
internal/core/src/exec/expression/Expr.h
View File

@ -684,10 +684,13 @@ class SegmentExpr : public Expr {
if constexpr (std::is_same_v<T, std::string_view> ||
std::is_same_v<T, Json>) {
if (segment_->type() == SegmentType::Sealed) {
valid_data = segment_
->get_batch_views<T>(
field_id_, i, data_pos, size)
.second.data();
auto batch_views = segment_->get_batch_views<T>(
field_id_, i, data_pos, size);
valid_data = batch_views.second.data();
ApplyValidData(valid_data,
res + processed_size,
valid_res + processed_size,
size);
}
} else {
auto chunk = segment_->chunk_data<T>(field_id_, i);
@ -695,11 +698,11 @@ class SegmentExpr : public Expr {
if (valid_data != nullptr) {
valid_data += data_pos;
}
ApplyValidData(valid_data,
res + processed_size,
valid_res + processed_size,
size);
}
ApplyValidData(valid_data,
res + processed_size,
valid_res + processed_size,
size);
}
processed_size += size;

4
internal/core/src/segcore/SegmentChunkReader.cpp
View File

@ -138,8 +138,8 @@ SegmentChunkReader::GetChunkDataAccessor<std::string>(
current_chunk_size =
segment_->chunk_size(field_id, current_chunk_id);
}
auto chunk_data = chunk_info.first;
auto chunk_valid_data = chunk_info.second;
auto& chunk_data = chunk_info.first;
auto& chunk_valid_data = chunk_info.second;
if (current_chunk_pos < chunk_valid_data.size() &&
!chunk_valid_data[current_chunk_pos]) {
current_chunk_pos++;