common_meta/cache/table/
schema.rs1use std::sync::Arc;
16
17use futures_util::future::BoxFuture;
18use moka::future::Cache;
19use snafu::OptionExt;
20
21use crate::cache::{CacheContainer, Initializer};
22use crate::error::ValueNotExistSnafu;
23use crate::instruction::CacheIdent;
24use crate::key::schema_name::{SchemaManager, SchemaName, SchemaNameKey, SchemaNameValue};
25use crate::kv_backend::KvBackendRef;
26
27pub type SchemaCache = CacheContainer<SchemaName, Arc<SchemaNameValue>, CacheIdent>;
28pub type SchemaCacheRef = Arc<SchemaCache>;
29
30pub fn new_schema_cache(
32 name: String,
33 cache: Cache<SchemaName, Arc<SchemaNameValue>>,
34 kv_backend: KvBackendRef,
35) -> SchemaCache {
36 let schema_manager = SchemaManager::new(kv_backend.clone());
37 let init = init_factory(schema_manager);
38
39 CacheContainer::new(name, cache, Box::new(invalidator), init, filter)
40}
41
42fn init_factory(schema_manager: SchemaManager) -> Initializer<SchemaName, Arc<SchemaNameValue>> {
43 Arc::new(move |schema_name| {
44 let manager = schema_manager.clone();
45 Box::pin(async move {
46 let schema_value = manager
47 .get(SchemaNameKey {
48 catalog: &schema_name.catalog_name,
49 schema: &schema_name.schema_name,
50 })
51 .await?
52 .context(ValueNotExistSnafu)?
53 .into_inner();
54 Ok(Some(Arc::new(schema_value)))
55 })
56 })
57}
58
59fn invalidator<'a>(
60 cache: &'a Cache<SchemaName, Arc<SchemaNameValue>>,
61 ident: &'a CacheIdent,
62) -> BoxFuture<'a, crate::error::Result<()>> {
63 Box::pin(async move {
64 if let CacheIdent::SchemaName(schema_name) = ident {
65 cache.invalidate(schema_name).await
66 }
67 Ok(())
68 })
69}
70
71fn filter(ident: &CacheIdent) -> bool {
72 matches!(ident, CacheIdent::SchemaName(_))
73}