chore: check ee license for disk cache (#17749)

* chore: check ee license for disk cache

* chore: add license checks

* revert method renaming

* Tweak comments

* Relaxed

* chore: remove new method from LicenseManager & fmt & clippy

* Uses new License Manager api

Author: dantengsky

src/query/config/src/lib.rs

@@ -49,6 +49,7 @@ pub use inner::CacheConfig;
 pub use inner::CacheStorageTypeConfig as CacheStorageTypeInnerConfig;
 pub use inner::CatalogConfig;
 pub use inner::CatalogHiveConfig;
+pub use inner::DiskCacheConfig as DiskCacheInnerConfig;
 pub use inner::DiskCacheKeyReloadPolicy;
 pub use inner::InnerConfig;
 pub use inner::SpillConfig;

src/query/service/src/global_services.rs

@@ -151,6 +151,7 @@ impl GlobalServices {
             &config.cache,
             &config.query.max_server_memory_usage,
             config.query.tenant_id.tenant_name().to_string(),
+            ee_mode,
         )?;
         TempDirManager::init(&config.spill, config.query.tenant_id.tenant_name())?;
 

src/query/service/src/interpreters/interpreter.rs

@@ -33,6 +33,7 @@ use databend_common_exception::ErrorCode;
 use databend_common_exception::Result;
 use databend_common_exception::ResultExt;
 use databend_common_expression::SendableDataBlockStream;
+use databend_common_license::license_manager::LicenseManagerSwitch;
 use databend_common_pipeline_core::always_callback;
 use databend_common_pipeline_core::processors::PlanProfile;
 use databend_common_pipeline_core::ExecutionInfo;
@@ -42,6 +43,7 @@ use databend_common_sql::PlanExtras;
 use databend_common_sql::Planner;
 use databend_common_storages_system::ProfilesLogElement;
 use databend_common_storages_system::ProfilesLogQueue;
+use databend_storages_common_cache::CacheManager;
 use derive_visitor::DriveMut;
 use derive_visitor::VisitorMut;
 use log::error;
@@ -101,6 +103,10 @@ pub trait Interpreter: Sync + Send {
         ctx.set_status_info("building pipeline");
         ctx.check_aborting().with_context(make_error)?;
 
+        CacheManager::instance().set_allows_disk_cache(
+            LicenseManagerSwitch::instance().is_license_valid(ctx.get_license_key()),
+        );
+
         let mut build_res = match self.execute2().await {
             Ok(build_res) => build_res,
             Err(err) => {

src/query/storages/common/cache/src/lib.rs

@@ -33,6 +33,7 @@ pub use caches::SizedColumnArray;
 pub use manager::CacheManager;
 // Unfortunately, criterion benchmarks need it to be pub
 pub use providers::disk_cache::disk_cache_lru::read_cache_content;
+pub use providers::DiskCacheAccessor;
 pub use providers::DiskCacheBuilder;
 pub use providers::DiskCacheError;
 pub use providers::DiskCacheKey;
@@ -42,7 +43,6 @@ pub use providers::InMemoryLruCache;
 pub use providers::LruDiskCache;
 pub use providers::LruDiskCacheBuilder;
 pub use providers::LruDiskCacheHolder;
-pub use providers::TableDataCache;
 pub use providers::TableDataCacheKey;
 pub use read::CacheKey;
 pub use read::CachedReader;