Skip to content

Commit 19dee3e

Browse files
committed
refactor: Don't nest complex types
Signed-off-by: Anush008 <[email protected]>
1 parent 32d94f7 commit 19dee3e

File tree

1 file changed

+26
-55
lines changed

1 file changed

+26
-55
lines changed

src/ops/storages/qdrant.rs

Lines changed: 26 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -128,11 +128,11 @@ fn values_to_payload(
128128
BasicValue::Float32(v) => (*v as f64).into(),
129129
BasicValue::Float64(v) => (*v).into(),
130130
BasicValue::Range(v) => json!({ "start": v.start, "end": v.end }),
131-
BasicValue::Uuid(v) => json!({ "uuid": v.to_string() }),
132-
BasicValue::Date(v) => json!({ "date": v.to_string() }),
133-
BasicValue::LocalDateTime(v) => json!({ "local_datetime": v.to_string() }),
134-
BasicValue::Time(v) => json!({ "time": v.to_string() }),
135-
BasicValue::OffsetDateTime(v) => json!({ "offset_datetime": v.to_string() }),
131+
BasicValue::Uuid(v) => v.to_string().into(),
132+
BasicValue::Date(v) => v.to_string().into(),
133+
BasicValue::LocalDateTime(v) => v.to_string().into(),
134+
BasicValue::Time(v) => v.to_string().into(),
135+
BasicValue::OffsetDateTime(v) => v.to_string().into(),
136136
BasicValue::Json(v) => (**v).clone(),
137137
BasicValue::Vector(v) => {
138138
let vector = convert_to_vector(v.to_vec());
@@ -223,60 +223,31 @@ fn into_value(point: &ScoredPoint, schema: &FieldSchema) -> Result<Value> {
223223
}
224224
}),
225225

226-
BasicValueType::Uuid => point.payload.get(field_name).and_then(|v| {
227-
v.as_struct().and_then(|s| {
228-
s.fields
229-
.get("uuid")?
230-
.as_str()?
231-
.parse()
232-
.ok()
233-
.map(BasicValue::Uuid)
234-
})
235-
}),
226+
BasicValueType::Uuid => point
227+
.payload
228+
.get(field_name)
229+
.and_then(|v| v.as_str()?.parse().ok().map(BasicValue::Uuid)),
236230

237-
BasicValueType::Date => point.payload.get(field_name).and_then(|v| {
238-
v.as_struct().and_then(|s| {
239-
s.fields
240-
.get("date")?
241-
.as_str()?
242-
.parse()
243-
.ok()
244-
.map(BasicValue::Date)
245-
})
246-
}),
231+
BasicValueType::Date => point
232+
.payload
233+
.get(field_name)
234+
.and_then(|v| v.as_str()?.parse().ok().map(BasicValue::Date)),
247235

248-
BasicValueType::Time => point.payload.get(field_name).and_then(|v| {
249-
v.as_struct().and_then(|s| {
250-
s.fields
251-
.get("time")?
252-
.as_str()?
253-
.parse()
254-
.ok()
255-
.map(BasicValue::Time)
256-
})
257-
}),
236+
BasicValueType::Time => point
237+
.payload
238+
.get(field_name)
239+
.and_then(|v| v.as_str()?.parse().ok().map(BasicValue::Time)),
258240

259-
BasicValueType::LocalDateTime => point.payload.get(field_name).and_then(|v| {
260-
v.as_struct().and_then(|s| {
261-
s.fields
262-
.get("local_datetime")?
263-
.as_str()?
264-
.parse()
265-
.ok()
266-
.map(BasicValue::LocalDateTime)
267-
})
268-
}),
241+
BasicValueType::LocalDateTime => point
242+
.payload
243+
.get(field_name)
244+
.and_then(|v| v.as_str()?.parse().ok().map(BasicValue::LocalDateTime)),
245+
246+
BasicValueType::OffsetDateTime => point
247+
.payload
248+
.get(field_name)
249+
.and_then(|v| v.as_str()?.parse().ok().map(BasicValue::OffsetDateTime)),
269250

270-
BasicValueType::OffsetDateTime => point.payload.get(field_name).and_then(|v| {
271-
v.as_struct().and_then(|s| {
272-
s.fields
273-
.get("offset_datetime")?
274-
.as_str()?
275-
.parse()
276-
.ok()
277-
.map(BasicValue::OffsetDateTime)
278-
})
279-
}),
280251
BasicValueType::Range => point.payload.get(field_name).and_then(|v| {
281252
v.as_struct().and_then(|s| {
282253
let start = s.fields.get("start").and_then(|f| f.as_integer());

0 commit comments

Comments
 (0)