1use std::collections::hash_map::Entry;
2use std::collections::{HashMap, HashSet, VecDeque};
3
4use super::{RedDBRuntime, RuntimeQueryResult, RuntimeResultCacheEntry};
5
6const RESULT_CACHE_BACKEND_KEY: &str = "runtime.result_cache.backend";
7const RESULT_CACHE_DEFAULT_BACKEND: &str = "legacy";
8const RESULT_CACHE_BLOB_NAMESPACE: &str = "runtime.result_cache";
9const RESULT_CACHE_EMBEDDED_COLLECTION: &str = "red_internal_result_cache";
10const RESULT_CACHE_TTL_SECS: u64 = 30;
11const RESULT_CACHE_MAX_ENTRIES: usize = 1000;
12const RESULT_CACHE_ENABLED_KEY: &str = "runtime.result_cache.enabled";
13const RESULT_CACHE_TTL_KEY: &str = "runtime.result_cache.ttl_seconds";
14const RESULT_CACHE_CAPACITY_KEY: &str = "runtime.result_cache.capacity_entries";
15const RESULT_CACHE_PAYLOAD_MAGIC: &[u8; 8] = b"RDRC0001";
16
17#[derive(Clone, Copy, Debug, PartialEq, Eq)]
18enum RuntimeResultCacheBackend {
19 Legacy,
20 BlobCache,
21 Shadow,
22}
23
24fn trim_result_cache(
25 map: &mut HashMap<String, RuntimeResultCacheEntry>,
26 order: &mut VecDeque<String>,
27 max_entries: usize,
28) -> u64 {
29 let mut evicted = 0u64;
30 while map.len() > max_entries {
31 if let Some(oldest) = order.pop_front() {
32 if map.remove(&oldest).is_some() {
33 evicted += 1;
34 }
35 } else {
36 break;
37 }
38 }
39 evicted
40}
41
42fn result_cache_fingerprint(result: &RuntimeQueryResult) -> String {
43 format!(
44 "{:?}|{}|{}|{}|{}|{:?}",
45 result.result,
46 result.query,
47 result.statement,
48 result.engine,
49 result.affected_rows,
50 result.statement_type
51 )
52}
53
54fn mode_to_byte(mode: reddb_rql::modes::QueryMode) -> u8 {
55 match mode {
56 reddb_rql::modes::QueryMode::Sql => 0,
57 reddb_rql::modes::QueryMode::Gremlin => 1,
58 reddb_rql::modes::QueryMode::Cypher => 2,
59 reddb_rql::modes::QueryMode::Sparql => 3,
60 reddb_rql::modes::QueryMode::Path => 4,
61 reddb_rql::modes::QueryMode::Natural => 5,
62 reddb_rql::modes::QueryMode::Unknown => 255,
63 }
64}
65
66fn mode_from_byte(byte: u8) -> Option<reddb_rql::modes::QueryMode> {
67 match byte {
68 0 => Some(reddb_rql::modes::QueryMode::Sql),
69 1 => Some(reddb_rql::modes::QueryMode::Gremlin),
70 2 => Some(reddb_rql::modes::QueryMode::Cypher),
71 3 => Some(reddb_rql::modes::QueryMode::Sparql),
72 4 => Some(reddb_rql::modes::QueryMode::Path),
73 5 => Some(reddb_rql::modes::QueryMode::Natural),
74 255 => Some(reddb_rql::modes::QueryMode::Unknown),
75 _ => None,
76 }
77}
78
79fn result_cache_static_str(value: &str) -> Option<&'static str> {
80 match value {
81 "select" => Some("select"),
82 "materialized-graph" => Some("materialized-graph"),
83 "runtime-red-schema" => Some("runtime-red-schema"),
84 "runtime-fdw" => Some("runtime-fdw"),
85 "runtime-table-rls" => Some("runtime-table-rls"),
86 "runtime-table" => Some("runtime-table"),
87 "runtime-join-rls" => Some("runtime-join-rls"),
88 "runtime-join" => Some("runtime-join"),
89 "runtime-vector" => Some("runtime-vector"),
90 "runtime-hybrid" => Some("runtime-hybrid"),
91 "runtime-secret" => Some("runtime-secret"),
92 "runtime-config" => Some("runtime-config"),
93 "runtime-tenant" => Some("runtime-tenant"),
94 "runtime-explain" => Some("runtime-explain"),
95 "runtime-tree" => Some("runtime-tree"),
96 "runtime-kv" => Some("runtime-kv"),
97 "runtime-queue" => Some("runtime-queue"),
98 _ => None,
99 }
100}
101
102fn write_u32(out: &mut Vec<u8>, value: usize) -> Option<()> {
103 let value = u32::try_from(value).ok()?;
104 out.extend_from_slice(&value.to_le_bytes());
105 Some(())
106}
107
108fn write_string(out: &mut Vec<u8>, value: &str) -> Option<()> {
109 write_u32(out, value.len())?;
110 out.extend_from_slice(value.as_bytes());
111 Some(())
112}
113
114fn write_bytes(out: &mut Vec<u8>, value: &[u8]) -> Option<()> {
115 write_u32(out, value.len())?;
116 out.extend_from_slice(value);
117 Some(())
118}
119
120fn read_u8(input: &mut &[u8]) -> Option<u8> {
121 let (&value, rest) = input.split_first()?;
122 *input = rest;
123 Some(value)
124}
125
126fn read_u32(input: &mut &[u8]) -> Option<usize> {
127 if input.len() < 4 {
128 return None;
129 }
130 let value = u32::from_le_bytes(input[..4].try_into().ok()?) as usize;
131 *input = &input[4..];
132 Some(value)
133}
134
135fn read_u64(input: &mut &[u8]) -> Option<u64> {
136 if input.len() < 8 {
137 return None;
138 }
139 let value = u64::from_le_bytes(input[..8].try_into().ok()?);
140 *input = &input[8..];
141 Some(value)
142}
143
144fn read_string(input: &mut &[u8]) -> Option<String> {
145 let len = read_u32(input)?;
146 if input.len() < len {
147 return None;
148 }
149 let value = String::from_utf8(input[..len].to_vec()).ok()?;
150 *input = &input[len..];
151 Some(value)
152}
153
154fn read_bytes<'a>(input: &mut &'a [u8]) -> Option<&'a [u8]> {
155 let len = read_u32(input)?;
156 if input.len() < len {
157 return None;
158 }
159 let value = &input[..len];
160 *input = &input[len..];
161 Some(value)
162}
163
164fn encode_result_cache_payload(entry: &RuntimeResultCacheEntry) -> Option<Vec<u8>> {
165 let result = &entry.result;
166 if result.result.stats.vector.is_some()
168 || result.result.pre_serialized_json.is_some()
169 || result_cache_static_str(result.statement).is_none()
170 || result_cache_static_str(result.engine).is_none()
171 || result_cache_static_str(result.statement_type).is_none()
172 || result.result.records.iter().any(|record| {
173 !record.nodes.is_empty()
174 || !record.edges.is_empty()
175 || !record.paths.is_empty()
176 || !record.vector_results.is_empty()
177 })
178 {
179 return None;
180 }
181
182 let mut out = Vec::new();
183 out.extend_from_slice(RESULT_CACHE_PAYLOAD_MAGIC);
184 write_string(&mut out, &result.query)?;
185 out.push(mode_to_byte(result.mode));
186 write_string(&mut out, result.statement)?;
187 write_string(&mut out, result.engine)?;
188 out.extend_from_slice(&result.affected_rows.to_le_bytes());
189 write_string(&mut out, result.statement_type)?;
190
191 write_u32(&mut out, result.result.columns.len())?;
192 for column in &result.result.columns {
193 write_string(&mut out, column)?;
194 }
195 out.extend_from_slice(&result.result.stats.nodes_scanned.to_le_bytes());
196 out.extend_from_slice(&result.result.stats.edges_scanned.to_le_bytes());
197 out.extend_from_slice(&result.result.stats.rows_scanned.to_le_bytes());
198 out.extend_from_slice(&result.result.stats.segments_total.to_le_bytes());
199 out.extend_from_slice(&result.result.stats.segments_scanned.to_le_bytes());
200 out.extend_from_slice(&result.result.stats.segments_pruned.to_le_bytes());
201 out.extend_from_slice(&result.result.stats.exec_time_us.to_le_bytes());
202
203 write_u32(&mut out, result.result.records.len())?;
204 for record in &result.result.records {
205 let fields = record.iter_fields().collect::<Vec<_>>();
206 write_u32(&mut out, fields.len())?;
207 for (name, value) in fields {
208 write_string(&mut out, name)?;
209 let mut encoded = Vec::new();
210 reddb_types::value_codec::encode(value, &mut encoded);
211 write_bytes(&mut out, &encoded)?;
212 }
213 }
214
215 write_u32(&mut out, entry.scopes.len())?;
216 for scope in &entry.scopes {
217 write_string(&mut out, scope)?;
218 }
219 Some(out)
220}
221
222fn decode_result_cache_payload(mut input: &[u8]) -> Option<(RuntimeQueryResult, HashSet<String>)> {
223 if input.len() < RESULT_CACHE_PAYLOAD_MAGIC.len()
224 || &input[..RESULT_CACHE_PAYLOAD_MAGIC.len()] != RESULT_CACHE_PAYLOAD_MAGIC
225 {
226 return None;
227 }
228 input = &input[RESULT_CACHE_PAYLOAD_MAGIC.len()..];
229
230 let query = read_string(&mut input)?;
231 let mode = mode_from_byte(read_u8(&mut input)?)?;
232 let statement = result_cache_static_str(&read_string(&mut input)?)?;
233 let engine = result_cache_static_str(&read_string(&mut input)?)?;
234 let affected_rows = read_u64(&mut input)?;
235 let statement_type = result_cache_static_str(&read_string(&mut input)?)?;
236
237 let mut columns = Vec::new();
238 for _ in 0..read_u32(&mut input)? {
239 columns.push(read_string(&mut input)?);
240 }
241 let stats = crate::storage::query::unified::QueryStats {
242 nodes_scanned: read_u64(&mut input)?,
243 edges_scanned: read_u64(&mut input)?,
244 rows_scanned: read_u64(&mut input)?,
245 segments_total: read_u64(&mut input)?,
246 segments_scanned: read_u64(&mut input)?,
247 segments_pruned: read_u64(&mut input)?,
248 exec_time_us: read_u64(&mut input)?,
249 vector: None,
250 };
251
252 let mut records = Vec::new();
253 for _ in 0..read_u32(&mut input)? {
254 let mut record = crate::storage::query::unified::UnifiedRecord::new();
255 for _ in 0..read_u32(&mut input)? {
256 let name = read_string(&mut input)?;
257 let bytes = read_bytes(&mut input)?;
258 let (value, used) = reddb_types::value_codec::decode(bytes).ok()?;
259 if used != bytes.len() {
260 return None;
261 }
262 record.set_owned(name, value);
263 }
264 records.push(record);
265 }
266
267 let mut scopes = HashSet::new();
268 for _ in 0..read_u32(&mut input)? {
269 scopes.insert(read_string(&mut input)?);
270 }
271 if !input.is_empty() {
272 return None;
273 }
274
275 Some((
276 RuntimeQueryResult {
277 query,
278 mode,
279 statement,
280 engine,
281 result: crate::storage::query::unified::UnifiedResult {
282 columns,
283 records,
284 stats,
285 pre_serialized_json: None,
286 },
287 affected_rows,
288 statement_type,
289 bookmark: None,
290 notice: None,
291 },
292 scopes,
293 ))
294}
295
296impl RedDBRuntime {
297 fn result_cache_backend(&self) -> RuntimeResultCacheBackend {
298 match self
299 .config_string(RESULT_CACHE_BACKEND_KEY, RESULT_CACHE_DEFAULT_BACKEND)
300 .as_str()
301 {
302 "blob_cache" => RuntimeResultCacheBackend::BlobCache,
303 "shadow" => RuntimeResultCacheBackend::Shadow,
304 _ => RuntimeResultCacheBackend::Legacy,
305 }
306 }
307
308 fn result_cache_enabled(&self) -> bool {
309 self.config_bool(RESULT_CACHE_ENABLED_KEY, true)
310 }
311
312 fn result_cache_ttl_secs(&self) -> u64 {
313 self.config_u64(RESULT_CACHE_TTL_KEY, RESULT_CACHE_TTL_SECS)
314 }
315
316 fn result_cache_capacity(&self) -> usize {
317 self.config_u64(RESULT_CACHE_CAPACITY_KEY, RESULT_CACHE_MAX_ENTRIES as u64) as usize
318 }
319
320 pub fn result_cache_metrics(&self) -> (u64, u64, u64) {
321 use std::sync::atomic::Ordering::Relaxed;
322 (
323 self.inner.result_cache_hits.load(Relaxed),
324 self.inner.result_cache_misses.load(Relaxed),
325 self.inner.result_cache_evictions.load(Relaxed),
326 )
327 }
328
329 fn record_result_cache_evictions(&self, evicted: u64) {
330 if evicted > 0 {
331 self.inner
332 .result_cache_evictions
333 .fetch_add(evicted, std::sync::atomic::Ordering::Relaxed);
334 }
335 }
336
337 pub(super) fn get_result_cache_entry(&self, key: &str) -> Option<RuntimeQueryResult> {
338 if !self.result_cache_enabled() {
339 return None;
340 }
341 let hit = self.get_result_cache_entry_inner(key);
342 let counter = if hit.is_some() {
343 &self.inner.result_cache_hits
344 } else {
345 &self.inner.result_cache_misses
346 };
347 counter.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
348 hit
349 }
350
351 fn get_result_cache_entry_inner(&self, key: &str) -> Option<RuntimeQueryResult> {
352 match self.result_cache_backend() {
353 RuntimeResultCacheBackend::Legacy => self.get_legacy_result_cache_entry(key),
354 RuntimeResultCacheBackend::BlobCache => self.get_blob_result_cache_entry(key),
355 RuntimeResultCacheBackend::Shadow => {
356 let legacy = self.get_legacy_result_cache_entry(key);
357 let blob = self.get_blob_result_cache_entry(key);
358 if let (Some(ref legacy), Some(ref blob)) = (&legacy, &blob) {
359 if result_cache_fingerprint(legacy) != result_cache_fingerprint(blob) {
360 self.inner
361 .result_cache_shadow_divergences
362 .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
363 tracing::warn!(
364 key,
365 metric = crate::runtime::METRIC_CACHE_SHADOW_DIVERGENCE_TOTAL,
366 "result cache shadow backend diverged from legacy"
367 );
368 }
369 }
370 legacy
371 }
372 }
373 }
374
375 fn get_legacy_result_cache_entry(&self, key: &str) -> Option<RuntimeQueryResult> {
376 let ttl = self.result_cache_ttl_secs();
377 let cache = self.inner.result_cache.read();
378 cache.0.get(key).and_then(|entry| {
379 if entry.cached_at.elapsed().as_secs() < ttl {
380 Some(entry.result.clone())
381 } else {
382 None
383 }
384 })
385 }
386
387 fn get_blob_result_cache_entry(&self, key: &str) -> Option<RuntimeQueryResult> {
388 if self.inner.embedded_single_file {
389 if let Some(bytes) = self.get_embedded_result_cache_payload(key) {
390 let policy = crate::storage::cache::BlobCachePolicy::default()
391 .ttl_ms(self.result_cache_ttl_secs() * 1000)
392 .priority(200);
393 let put = crate::storage::cache::BlobCachePut::new(bytes).with_policy(policy);
394 if self
395 .inner
396 .result_blob_cache
397 .put(RESULT_CACHE_BLOB_NAMESPACE, key, put)
398 .is_err()
399 {
400 return None;
401 }
402 }
403 }
404 self.get_blob_result_cache_entry_from_blob_cache(key)
405 }
406
407 fn get_blob_result_cache_entry_from_blob_cache(&self, key: &str) -> Option<RuntimeQueryResult> {
408 let hit = self
409 .inner
410 .result_blob_cache
411 .get(RESULT_CACHE_BLOB_NAMESPACE, key)?;
412 {
413 let cache = self.inner.result_blob_entries.read();
414 if let Some(entry) = cache.0.get(key) {
415 return Some(entry.result.clone());
416 }
417 }
418
419 let (result, scopes) = decode_result_cache_payload(hit.value())?;
420 let mut cache = self.inner.result_blob_entries.write();
421 let (ref mut map, ref mut order) = *cache;
422 let new_entry = RuntimeResultCacheEntry {
423 result: result.clone(),
424 cached_at: std::time::Instant::now(),
425 scopes,
426 };
427 match map.entry(key.to_string()) {
429 Entry::Occupied(mut slot) => {
430 slot.insert(new_entry);
431 }
432 Entry::Vacant(slot) => {
433 order.push_back(slot.key().clone());
434 slot.insert(new_entry);
435 }
436 }
437 let evicted = trim_result_cache(map, order, self.result_cache_capacity());
438 drop(cache);
439 self.record_result_cache_evictions(evicted);
440 Some(result)
441 }
442
443 fn get_embedded_result_cache_payload(&self, key: &str) -> Option<Vec<u8>> {
444 let manager = self
445 .inner
446 .db
447 .store()
448 .get_collection(RESULT_CACHE_EMBEDDED_COLLECTION)?;
449 let mut latest: Option<(u64, Vec<u8>)> = None;
450 manager.for_each_entity(|entity| {
451 let Some(row) = entity.data.as_row() else {
452 return true;
453 };
454 let namespace_matches = row.get_field("namespace").and_then(|value| match value {
455 reddb_types::Value::Text(value) => Some(value.as_ref()),
456 _ => None,
457 }) == Some(RESULT_CACHE_BLOB_NAMESPACE);
458 let key_matches = row.get_field("key").and_then(|value| match value {
459 reddb_types::Value::Text(value) => Some(value.as_ref()),
460 _ => None,
461 }) == Some(key);
462 if namespace_matches && key_matches {
463 if let Some(reddb_types::Value::Blob(payload)) = row.get_field("payload") {
464 let id = entity.id.raw();
465 if latest
466 .as_ref()
467 .is_none_or(|(latest_id, _)| id >= *latest_id)
468 {
469 latest = Some((id, payload.clone()));
470 }
471 }
472 }
473 true
474 });
475 latest.map(|(_, payload)| payload)
476 }
477
478 pub(super) fn put_result_cache_entry(&self, key: &str, entry: RuntimeResultCacheEntry) {
479 if !self.result_cache_enabled() {
480 return;
481 }
482 match self.result_cache_backend() {
483 RuntimeResultCacheBackend::Legacy => self.put_legacy_result_cache_entry(key, entry),
484 RuntimeResultCacheBackend::BlobCache => self.put_blob_result_cache_entry(key, entry),
485 RuntimeResultCacheBackend::Shadow => {
486 self.put_legacy_result_cache_entry(key, entry.clone());
487 self.put_blob_result_cache_entry(key, entry);
488 }
489 }
490 }
491
492 fn put_legacy_result_cache_entry(&self, key: &str, entry: RuntimeResultCacheEntry) {
493 let capacity = self.result_cache_capacity();
494 let mut cache = self.inner.result_cache.write();
495 let (ref mut map, ref mut order) = *cache;
496 match map.entry(key.to_string()) {
498 Entry::Occupied(mut slot) => {
499 slot.insert(entry);
500 }
501 Entry::Vacant(slot) => {
502 order.push_back(slot.key().clone());
503 slot.insert(entry);
504 }
505 }
506 let evicted = trim_result_cache(map, order, capacity);
507 drop(cache);
508 self.record_result_cache_evictions(evicted);
509 }
510
511 fn put_blob_result_cache_entry(&self, key: &str, entry: RuntimeResultCacheEntry) {
512 let policy = crate::storage::cache::BlobCachePolicy::default()
513 .ttl_ms(self.result_cache_ttl_secs() * 1000)
514 .priority(200);
515 let dependencies = entry.scopes.iter().cloned().collect::<Vec<_>>();
516 let bytes = encode_result_cache_payload(&entry)
517 .unwrap_or_else(|| result_cache_fingerprint(&entry.result).into_bytes());
518 if self.inner.embedded_single_file {
519 self.put_embedded_result_cache_payload(key, &bytes, &dependencies);
520 }
521 let put = crate::storage::cache::BlobCachePut::new(bytes)
522 .with_dependencies(dependencies)
523 .with_policy(policy);
524 if self
525 .inner
526 .result_blob_cache
527 .put(RESULT_CACHE_BLOB_NAMESPACE, key, put)
528 .is_err()
529 {
530 return;
531 }
532
533 let capacity = self.result_cache_capacity();
534 let mut cache = self.inner.result_blob_entries.write();
535 let (ref mut map, ref mut order) = *cache;
536 match map.entry(key.to_string()) {
538 Entry::Occupied(mut slot) => {
539 slot.insert(entry);
540 }
541 Entry::Vacant(slot) => {
542 order.push_back(slot.key().clone());
543 slot.insert(entry);
544 }
545 }
546 let evicted = trim_result_cache(map, order, capacity);
547 drop(cache);
548 self.record_result_cache_evictions(evicted);
549 }
550
551 fn put_embedded_result_cache_payload(&self, key: &str, bytes: &[u8], scopes: &[String]) {
552 let store = self.inner.db.store();
553 let _ = store.get_or_create_collection(RESULT_CACHE_EMBEDDED_COLLECTION);
554 let entity = crate::storage::UnifiedEntity::new(
555 crate::storage::EntityId::new(0),
556 crate::storage::EntityKind::TableRow {
557 table: std::sync::Arc::from(RESULT_CACHE_EMBEDDED_COLLECTION),
558 row_id: 0,
559 },
560 crate::storage::EntityData::Row(crate::storage::RowData {
561 columns: Vec::new(),
562 named: Some(HashMap::from([
563 (
564 "namespace".to_string(),
565 reddb_types::Value::text(RESULT_CACHE_BLOB_NAMESPACE),
566 ),
567 ("key".to_string(), reddb_types::Value::text(key.to_string())),
568 (
569 "payload".to_string(),
570 reddb_types::Value::Blob(bytes.to_vec()),
571 ),
572 (
573 "scopes".to_string(),
574 reddb_types::Value::text(scopes.join("\n")),
575 ),
576 ])),
577 schema: None,
578 }),
579 );
580 let _ = store.insert_auto(RESULT_CACHE_EMBEDDED_COLLECTION, entity);
581 }
582
583 fn invalidate_embedded_result_cache(&self) {
584 let Some(manager) = self
585 .inner
586 .db
587 .store()
588 .get_collection(RESULT_CACHE_EMBEDDED_COLLECTION)
589 else {
590 return;
591 };
592 let ids = manager
593 .query_all(|_| true)
594 .into_iter()
595 .map(|entity| entity.id)
596 .collect::<Vec<_>>();
597 if !ids.is_empty() {
598 let _ = self
599 .inner
600 .db
601 .store()
602 .delete_batch(RESULT_CACHE_EMBEDDED_COLLECTION, &ids);
603 }
604 }
605
606 fn invalidate_embedded_result_cache_for_scope(&self, scope: &str) {
607 let Some(manager) = self
608 .inner
609 .db
610 .store()
611 .get_collection(RESULT_CACHE_EMBEDDED_COLLECTION)
612 else {
613 return;
614 };
615 let ids = manager
616 .query_all(|entity| {
617 entity
618 .data
619 .as_row()
620 .and_then(|row| row.get_field("scopes"))
621 .and_then(|value| match value {
622 reddb_types::Value::Text(value) => Some(value.as_ref()),
623 _ => None,
624 })
625 .is_some_and(|scopes| scopes.lines().any(|entry| entry == scope))
626 })
627 .into_iter()
628 .map(|entity| entity.id)
629 .collect::<Vec<_>>();
630 if !ids.is_empty() {
631 let _ = self
632 .inner
633 .db
634 .store()
635 .delete_batch(RESULT_CACHE_EMBEDDED_COLLECTION, &ids);
636 }
637 }
638
639 pub fn result_cache_shadow_divergences(&self) -> u64 {
640 self.inner
641 .result_cache_shadow_divergences
642 .load(std::sync::atomic::Ordering::Relaxed)
643 }
644
645 pub fn invalidate_result_cache(&self) {
646 self.invalidate_result_cache_process_only();
647 if self.inner.embedded_single_file {
648 self.invalidate_embedded_result_cache();
649 }
650 }
651
652 pub(crate) fn invalidate_result_cache_process_only(&self) {
653 let mut cache = self.inner.result_cache.write();
654 cache.0.clear();
655 cache.1.clear();
656 let mut blob_entries = self.inner.result_blob_entries.write();
657 blob_entries.0.clear();
658 blob_entries.1.clear();
659 self.inner
660 .result_blob_cache
661 .invalidate_namespace(RESULT_CACHE_BLOB_NAMESPACE);
662 let mut ask_entries = self.inner.ask_answer_cache_entries.write();
663 ask_entries.0.clear();
664 ask_entries.1.clear();
665 self.inner
666 .result_blob_cache
667 .invalidate_namespace(super::ASK_ANSWER_CACHE_NAMESPACE);
668 }
669
670 pub(crate) fn invalidate_result_cache_for_table(&self, table: &str) {
671 let legacy_has_match = {
672 let cache = self.inner.result_cache.read();
673 let (ref map, _) = *cache;
674 !map.is_empty() && map.values().any(|entry| entry.scopes.contains(table))
675 };
676 let blob_has_match = {
677 let cache = self.inner.result_blob_entries.read();
678 let (ref map, _) = *cache;
679 !map.is_empty() && map.values().any(|entry| entry.scopes.contains(table))
680 };
681 if legacy_has_match {
682 let mut cache = self.inner.result_cache.write();
683 let (ref mut map, ref mut order) = *cache;
684 map.retain(|_, entry| !entry.scopes.contains(table));
685 order.retain(|key| map.contains_key(key));
686 }
687
688 if matches!(
689 self.result_cache_backend(),
690 RuntimeResultCacheBackend::BlobCache | RuntimeResultCacheBackend::Shadow
691 ) {
692 let mut blob_entries = self.inner.result_blob_entries.write();
693 let (ref mut blob_map, ref mut blob_order) = *blob_entries;
694 blob_map.clear();
695 blob_order.clear();
696 self.inner
697 .result_blob_cache
698 .invalidate_namespace(RESULT_CACHE_BLOB_NAMESPACE);
699 if self.inner.embedded_single_file {
700 self.invalidate_embedded_result_cache();
701 }
702 } else if blob_has_match {
703 let mut blob_entries = self.inner.result_blob_entries.write();
704 let (ref mut blob_map, ref mut blob_order) = *blob_entries;
705 blob_map.retain(|_, entry| !entry.scopes.contains(table));
706 blob_order.retain(|key| blob_map.contains_key(key));
707 if self.inner.embedded_single_file {
708 self.invalidate_embedded_result_cache_for_scope(table);
709 }
710 }
711 let mut ask_entries = self.inner.ask_answer_cache_entries.write();
712 ask_entries.0.clear();
713 ask_entries.1.clear();
714 self.inner
715 .result_blob_cache
716 .invalidate_namespace(super::ASK_ANSWER_CACHE_NAMESPACE);
717 }
718}