Spaces:
Build error
Build error
File size: 5,155 Bytes
84d2a97 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 |
use segment::types::{Filter, PointIdType};
use super::vector_ops;
use crate::operations::payload_ops::PayloadOps;
use crate::operations::{point_ops, CollectionUpdateOperations};
/// Structure to define what part of the shard are affected by the operation
pub enum OperationEffectArea {
Empty,
Points(Vec<PointIdType>),
Filter(Filter),
}
/// Estimate how many points will be affected by the operation
pub enum PointsOperationEffect {
/// No points affected
Empty,
/// Some points are affected
Some(Vec<PointIdType>),
/// Too many to enumerate, so we just say that it is a lot
Many,
}
pub trait EstimateOperationEffectArea {
fn estimate_effect_area(&self) -> OperationEffectArea;
}
impl EstimateOperationEffectArea for CollectionUpdateOperations {
fn estimate_effect_area(&self) -> OperationEffectArea {
match self {
CollectionUpdateOperations::PointOperation(point_operation) => {
point_operation.estimate_effect_area()
}
CollectionUpdateOperations::VectorOperation(vector_operation) => {
vector_operation.estimate_effect_area()
}
CollectionUpdateOperations::PayloadOperation(payload_operation) => {
payload_operation.estimate_effect_area()
}
CollectionUpdateOperations::FieldIndexOperation(_) => OperationEffectArea::Empty,
}
}
}
impl EstimateOperationEffectArea for point_ops::PointOperations {
fn estimate_effect_area(&self) -> OperationEffectArea {
match self {
point_ops::PointOperations::UpsertPoints(insert_operations) => {
insert_operations.estimate_effect_area()
}
point_ops::PointOperations::DeletePoints { ids } => {
OperationEffectArea::Points(ids.clone())
}
point_ops::PointOperations::DeletePointsByFilter(filter) => {
OperationEffectArea::Filter(filter.clone())
}
point_ops::PointOperations::SyncPoints(sync_op) => {
debug_assert!(
false,
"SyncPoints operation should not be used during transfer"
);
OperationEffectArea::Points(sync_op.points.iter().map(|x| x.id).collect())
}
}
}
}
impl EstimateOperationEffectArea for vector_ops::VectorOperations {
fn estimate_effect_area(&self) -> OperationEffectArea {
match self {
vector_ops::VectorOperations::UpdateVectors(update_operation) => {
let ids = update_operation.points.iter().map(|p| p.id).collect();
OperationEffectArea::Points(ids)
}
vector_ops::VectorOperations::DeleteVectors(ids, _) => {
OperationEffectArea::Points(ids.points.clone())
}
vector_ops::VectorOperations::DeleteVectorsByFilter(filter, _) => {
OperationEffectArea::Filter(filter.clone())
}
}
}
}
impl EstimateOperationEffectArea for point_ops::PointInsertOperationsInternal {
fn estimate_effect_area(&self) -> OperationEffectArea {
match self {
point_ops::PointInsertOperationsInternal::PointsBatch(batch) => {
OperationEffectArea::Points(batch.ids.clone())
}
point_ops::PointInsertOperationsInternal::PointsList(list) => {
OperationEffectArea::Points(list.iter().map(|x| x.id).collect())
}
}
}
}
impl EstimateOperationEffectArea for PayloadOps {
fn estimate_effect_area(&self) -> OperationEffectArea {
match self {
PayloadOps::SetPayload(set_payload) => {
if let Some(points) = &set_payload.points {
OperationEffectArea::Points(points.clone())
} else if let Some(filter) = &set_payload.filter {
OperationEffectArea::Filter(filter.clone())
} else {
OperationEffectArea::Empty
}
}
PayloadOps::DeletePayload(delete_payload) => {
if let Some(points) = &delete_payload.points {
OperationEffectArea::Points(points.clone())
} else if let Some(filter) = &delete_payload.filter {
OperationEffectArea::Filter(filter.clone())
} else {
OperationEffectArea::Empty
}
}
PayloadOps::ClearPayload { points } => OperationEffectArea::Points(points.clone()),
PayloadOps::ClearPayloadByFilter(filter) => OperationEffectArea::Filter(filter.clone()),
PayloadOps::OverwritePayload(set_payload) => {
if let Some(points) = &set_payload.points {
OperationEffectArea::Points(points.clone())
} else if let Some(filter) = &set_payload.filter {
OperationEffectArea::Filter(filter.clone())
} else {
OperationEffectArea::Empty
}
}
}
}
}
|