Coverage Report

Created: 2026-05-28 19:50

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/runtime/exec_env.h
Line
Count
Source
1
// Licensed to the Apache Software Foundation (ASF) under one
2
// or more contributor license agreements.  See the NOTICE file
3
// distributed with this work for additional information
4
// regarding copyright ownership.  The ASF licenses this file
5
// to you under the Apache License, Version 2.0 (the
6
// "License"); you may not use this file except in compliance
7
// with the License.  You may obtain a copy of the License at
8
//
9
//   http://www.apache.org/licenses/LICENSE-2.0
10
//
11
// Unless required by applicable law or agreed to in writing,
12
// software distributed under the License is distributed on an
13
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14
// KIND, either express or implied.  See the License for the
15
// specific language governing permissions and limitations
16
// under the License.
17
18
#pragma once
19
20
#include <gen_cpp/olap_file.pb.h>
21
22
#include <atomic>
23
#include <map>
24
#include <memory>
25
#include <mutex>
26
#include <optional>
27
#include <string>
28
#include <vector>
29
30
#include "common/config.h"
31
#include "common/multi_version.h"
32
#include "common/status.h"
33
#include "information_schema/schema_routine_load_job_scanner.h"
34
#include "io/cache/fs_file_cache_storage.h"
35
#include "load/memtable/memtable_memory_limiter.h"
36
#include "runtime/cluster_info.h"
37
#include "runtime/frontend_info.h" // TODO(zhiqiang): find a way to remove this include header
38
#include "storage/index/inverted/inverted_index_writer.h"
39
#include "storage/options.h"
40
#include "storage/tablet/tablet_fwd.h"
41
#include "util/threadpool.h"
42
43
namespace orc {
44
class MemoryPool;
45
}
46
namespace arrow {
47
class MemoryPool;
48
}
49
50
namespace doris {
51
class VDataStreamMgr;
52
class SpillFileManager;
53
class DeltaWriterV2Pool;
54
class DictionaryFactory;
55
class TaskScheduler;
56
struct RuntimeFilterTimerQueue;
57
class WorkloadGroupMgr;
58
struct WriteCooldownMetaExecutors;
59
class TokenBucketRateLimiterHolder;
60
using S3RateLimiterHolder = TokenBucketRateLimiterHolder;
61
class MSRpcRateLimitServices;
62
namespace io {
63
class FileCacheFactory;
64
class HdfsMgr;
65
class PackedFileManager;
66
} // namespace io
67
68
namespace segment_v2 {
69
class InvertedIndexSearcherCache;
70
class InvertedIndexQueryCache;
71
class ConditionCache;
72
class AnnIndexResultCache;
73
class TmpFileDirs;
74
class EncodingInfoResolver;
75
76
namespace inverted_index {
77
class AnalysisFactoryMgr;
78
}
79
80
} // namespace segment_v2
81
82
namespace kerberos {
83
class KerberosTicketMgr;
84
}
85
86
class QueryCache;
87
class WorkloadSchedPolicyMgr;
88
class BfdParser;
89
class BrokerMgr;
90
template <class T>
91
class BrpcClientCache;
92
class ExternalScanContextMgr;
93
class FragmentMgr;
94
class ResultCache;
95
class LoadPathMgr;
96
class NewLoadStreamMgr;
97
class MemTrackerLimiter;
98
class MemTracker;
99
struct TrackerLimiterGroup;
100
class BaseStorageEngine;
101
class ResultBufferMgr;
102
class ResultQueueMgr;
103
class RuntimeQueryStatisticsMgr;
104
class LoadChannelMgr;
105
class LoadStreamMgr;
106
class LoadStreamMapPool;
107
class StreamLoadExecutor;
108
class RoutineLoadTaskExecutor;
109
class StreamLoadRecorderManager;
110
class SmallFileMgr;
111
class BackendServiceClient;
112
class TPaloBrokerServiceClient;
113
class PBackendService_Stub;
114
class PFunctionService_Stub;
115
template <class T>
116
class ClientCache;
117
class HeartbeatFlags;
118
class FrontendServiceClient;
119
class FileMetaCache;
120
class GroupCommitMgr;
121
class CdcClientMgr;
122
class TabletSchemaCache;
123
class TabletColumnObjectPool;
124
class UserFunctionCache;
125
class StoragePageCache;
126
class AnnIndexIVFListCache;
127
class SegmentLoader;
128
class LookupConnectionCache;
129
class RowCache;
130
class DummyLRUCache;
131
class CacheManager;
132
class IdManager;
133
class ProcessProfile;
134
class HeapProfiler;
135
class WalManager;
136
class DNSCache;
137
class IndexPolicyMgr;
138
struct SyncRowsetStats;
139
class DeleteBitmapAggCache;
140
141
// set to true when BE is shutting down
142
inline bool k_doris_exit = false;
143
// set to true after BE start ready
144
inline bool k_is_server_ready = false;
145
146
// Execution environment for queries/plan fragments.
147
// Contains all required global structures, and handles to
148
// singleton services. Clients must call StartServices exactly
149
// once to properly initialise service state.
150
class ExecEnv {
151
public:
152
    // Empty destructor because the compiler-generated one requires full
153
    // declarations for classes in scoped_ptrs.
154
    ~ExecEnv();
155
156
3.29k
    BaseStorageEngine& storage_engine() { return *_storage_engine; }
157
158
    // Initial exec environment. must call this to init all
159
    [[nodiscard]] static Status init(ExecEnv* env, const std::vector<StorePath>& store_paths,
160
                                     const std::vector<StorePath>& spill_store_paths,
161
                                     const std::set<std::string>& broken_paths);
162
163
    // Stop all threads and delete resources.
164
    void destroy();
165
166
    /// Returns the first created exec env instance. In a normal doris, this is
167
    /// the only instance. In test setups with multiple ExecEnv's per process,
168
    /// we return the most recently created instance.
169
901k
    static ExecEnv* GetInstance() {
170
901k
        static ExecEnv s_exec_env;
171
901k
        return &s_exec_env;
172
901k
    }
173
174
    // Requires ExenEnv ready
175
    /*
176
     * Parameters:
177
     * - tablet_id: the id of tablet to get
178
     * - sync_stats: the stats of sync rowset
179
     * - force_use_only_cached: whether only use cached data
180
     * - cache_on_miss: whether cache the tablet meta when missing in cache
181
     */
182
    static Result<BaseTabletSPtr> get_tablet(int64_t tablet_id,
183
                                             SyncRowsetStats* sync_stats = nullptr,
184
                                             bool force_use_only_cached = false,
185
                                             bool cache_on_miss = true);
186
187
    static Status get_tablet_meta(int64_t tablet_id, TabletMetaSharedPtr* tablet_meta,
188
                                  bool force_use_only_cached = false);
189
190
482k
    static bool ready() { return _s_ready.load(std::memory_order_acquire); }
191
74
    static bool tracking_memory() { return _s_tracking_memory.load(std::memory_order_acquire); }
192
1.29k
    static bool get_is_upgrading() { return _s_upgrading.load(std::memory_order_acquire); }
193
0
    static void set_is_upgrading() { _s_upgrading = true; }
194
    const std::string& token() const;
195
0
    ExternalScanContextMgr* external_scan_context_mgr() { return _external_scan_context_mgr; }
196
8
    VDataStreamMgr* vstream_mgr() { return _vstream_mgr; }
197
0
    ResultBufferMgr* result_mgr() { return _result_mgr; }
198
1
    ResultQueueMgr* result_queue_mgr() { return _result_queue_mgr; }
199
0
    ClientCache<BackendServiceClient>* client_cache() { return _backend_client_cache; }
200
0
    ClientCache<FrontendServiceClient>* frontend_client_cache() { return _frontend_client_cache; }
201
0
    ClientCache<TPaloBrokerServiceClient>* broker_client_cache() { return _broker_client_cache; }
202
203
2.26k
    WorkloadGroupMgr* workload_group_mgr() { return _workload_group_manager; }
204
0
    WorkloadSchedPolicyMgr* workload_sched_policy_mgr() { return _workload_sched_mgr; }
205
0
    RuntimeQueryStatisticsMgr* runtime_query_statistics_mgr() {
206
0
        return _runtime_query_statistics_mgr;
207
0
    }
208
209
    // using template to simplify client cache management
210
    template <typename T>
211
    inline ClientCache<T>* get_client_cache() {
212
        return nullptr;
213
    }
214
215
    // Save all MemTrackerLimiters in use.
216
    // Each group corresponds to several MemTrackerLimiters and has a lock.
217
    // Multiple groups are used to reduce the impact of locks.
218
    std::vector<TrackerLimiterGroup> mem_tracker_limiter_pool;
219
    void init_mem_tracker();
220
341k
    std::shared_ptr<MemTrackerLimiter> orphan_mem_tracker() { return _orphan_mem_tracker; }
221
0
    std::shared_ptr<MemTrackerLimiter> brpc_iobuf_block_memory_tracker() {
222
0
        return _brpc_iobuf_block_memory_tracker;
223
0
    }
224
0
    std::shared_ptr<MemTrackerLimiter> segcompaction_mem_tracker() {
225
0
        return _segcompaction_mem_tracker;
226
0
    }
227
8
    std::shared_ptr<MemTrackerLimiter> stream_load_pipe_tracker() {
228
8
        return _stream_load_pipe_tracker;
229
8
    }
230
0
    std::shared_ptr<MemTrackerLimiter> tablets_no_cache_mem_tracker() {
231
0
        return _tablets_no_cache_mem_tracker;
232
0
    }
233
0
    std::shared_ptr<MemTrackerLimiter> rowsets_no_cache_mem_tracker() {
234
0
        return _rowsets_no_cache_mem_tracker;
235
0
    }
236
0
    std::shared_ptr<MemTrackerLimiter> segments_no_cache_mem_tracker() {
237
0
        return _segments_no_cache_mem_tracker;
238
0
    }
239
2.09k
    std::shared_ptr<MemTrackerLimiter> point_query_executor_mem_tracker() {
240
2.09k
        return _point_query_executor_mem_tracker;
241
2.09k
    }
242
12
    std::shared_ptr<MemTrackerLimiter> query_cache_mem_tracker() {
243
12
        return _query_cache_mem_tracker;
244
12
    }
245
20.9k
    std::shared_ptr<MemTrackerLimiter> block_compression_mem_tracker() {
246
20.9k
        return _block_compression_mem_tracker;
247
20.9k
    }
248
0
    std::shared_ptr<MemTrackerLimiter> rowid_storage_reader_tracker() {
249
0
        return _rowid_storage_reader_tracker;
250
0
    }
251
1.86k
    std::shared_ptr<MemTrackerLimiter> subcolumns_tree_tracker() {
252
1.86k
        return _subcolumns_tree_tracker;
253
1.86k
    }
254
2.77k
    std::shared_ptr<MemTrackerLimiter> s3_file_buffer_tracker() { return _s3_file_buffer_tracker; }
255
208
    std::shared_ptr<MemTrackerLimiter> parquet_meta_tracker() { return _parquet_meta_tracker; }
256
257
0
    ThreadPool* send_batch_thread_pool() { return _send_batch_thread_pool.get(); }
258
4
    ThreadPool* buffered_reader_prefetch_thread_pool() {
259
4
        return _buffered_reader_prefetch_thread_pool.get();
260
4
    }
261
0
    ThreadPool* send_table_stats_thread_pool() { return _send_table_stats_thread_pool.get(); }
262
2.38k
    ThreadPool* s3_file_upload_thread_pool() { return _s3_file_upload_thread_pool.get(); }
263
0
    ThreadPool* lazy_release_obj_pool() { return _lazy_release_obj_pool.get(); }
264
    ThreadPool* non_block_close_thread_pool();
265
0
    ThreadPool* s3_file_system_thread_pool() { return _s3_file_system_thread_pool.get(); }
266
0
    ThreadPool* udf_close_workers_pool() { return _udf_close_workers_thread_pool.get(); }
267
0
    ThreadPool* segment_prefetch_thread_pool() { return _segment_prefetch_thread_pool.get(); }
268
0
    ThreadPool* peer_race_s3_thread_pool() { return _peer_race_s3_thread_pool.get(); }
269
270
    void init_file_cache_factory(std::vector<doris::CachePath>& cache_paths);
271
4.17k
    io::FileCacheFactory* file_cache_factory() { return _file_cache_factory; }
272
0
    UserFunctionCache* user_function_cache() { return _user_function_cache; }
273
122k
    FragmentMgr* fragment_mgr() { return _fragment_mgr; }
274
0
    ResultCache* result_cache() { return _result_cache; }
275
8
    ClusterInfo* cluster_info() { return _cluster_info; }
276
1
    LoadPathMgr* load_path_mgr() { return _load_path_mgr; }
277
1
    BfdParser* bfd_parser() const { return _bfd_parser; }
278
0
    BrokerMgr* broker_mgr() const { return _broker_mgr; }
279
0
    BrpcClientCache<PBackendService_Stub>* brpc_internal_client_cache() const {
280
0
        return _internal_client_cache;
281
0
    }
282
0
    BrpcClientCache<PBackendService_Stub>* brpc_streaming_client_cache() const {
283
0
        return _streaming_client_cache;
284
0
    }
285
0
    BrpcClientCache<PFunctionService_Stub>* brpc_function_client_cache() const {
286
0
        return _function_client_cache;
287
0
    }
288
0
    LoadChannelMgr* load_channel_mgr() { return _load_channel_mgr; }
289
0
    LoadStreamMgr* load_stream_mgr() { return _load_stream_mgr.get(); }
290
3
    NewLoadStreamMgr* new_load_stream_mgr() { return _new_load_stream_mgr.get(); }
291
0
    SmallFileMgr* small_file_mgr() { return _small_file_mgr; }
292
293
1.94k
    SpillFileManager* spill_file_mgr() { return _spill_file_mgr; }
294
295
0
    GroupCommitMgr* group_commit_mgr() { return _group_commit_mgr; }
296
0
    CdcClientMgr* cdc_client_mgr() { return _cdc_client_mgr; }
297
298
3
    const std::vector<StorePath>& store_paths() const { return _store_paths; }
299
300
1
    StreamLoadExecutor* stream_load_executor() { return _stream_load_executor.get(); }
301
0
    RoutineLoadTaskExecutor* routine_load_task_executor() { return _routine_load_task_executor; }
302
0
    HeartbeatFlags* heartbeat_flags() { return _heartbeat_flags; }
303
14
    FileMetaCache* file_meta_cache() { return _file_meta_cache; }
304
36
    MemTableMemoryLimiter* memtable_memory_limiter() { return _memtable_memory_limiter.get(); }
305
44
    WalManager* wal_mgr() { return _wal_manager.get(); }
306
8
    DNSCache* dns_cache() { return _dns_cache; }
307
5
    WriteCooldownMetaExecutors* write_cooldown_meta_executors() {
308
5
        return _write_cooldown_meta_executors.get();
309
5
    }
310
311
0
    kerberos::KerberosTicketMgr* kerberos_ticket_mgr() { return _kerberos_ticket_mgr; }
312
0
    io::HdfsMgr* hdfs_mgr() { return _hdfs_mgr; }
313
0
    io::PackedFileManager* packed_file_manager() { return _packed_file_manager; }
314
3
    IndexPolicyMgr* index_policy_mgr() { return _index_policy_mgr; }
315
0
    S3RateLimiterHolder* warmup_download_rate_limiter() { return _warmup_download_rate_limiter; }
316
0
    MSRpcRateLimitServices* ms_rpc_rate_limit_services() {
317
0
        return _ms_rpc_rate_limit_services.get();
318
0
    }
319
320
#ifdef BE_TEST
321
252
    void set_tmp_file_dir(std::unique_ptr<segment_v2::TmpFileDirs> tmp_file_dirs) {
322
252
        this->_tmp_file_dirs = std::move(tmp_file_dirs);
323
252
    }
324
4
    void set_ready() { this->_s_ready = true; }
325
0
    void set_not_ready() { this->_s_ready = false; }
326
68
    void set_memtable_memory_limiter(MemTableMemoryLimiter* limiter) {
327
68
        _memtable_memory_limiter.reset(limiter);
328
68
    }
329
57
    void set_cluster_info(ClusterInfo* cluster_info) { this->_cluster_info = cluster_info; }
330
    void set_new_load_stream_mgr(std::unique_ptr<NewLoadStreamMgr>&& new_load_stream_mgr);
331
    void clear_new_load_stream_mgr();
332
    void set_stream_load_executor(std::unique_ptr<StreamLoadExecutor>&& stream_load_executor);
333
    void clear_stream_load_executor();
334
335
    void set_storage_engine(std::unique_ptr<BaseStorageEngine>&& engine);
336
    void set_inverted_index_searcher_cache(
337
            segment_v2::InvertedIndexSearcherCache* inverted_index_searcher_cache);
338
    void set_inverted_index_query_cache(
339
0
            segment_v2::InvertedIndexQueryCache* inverted_index_query_cache) {
340
0
        _inverted_index_query_cache = inverted_index_query_cache;
341
0
    }
342
1
    void set_process_profile(ProcessProfile* pp) { this->_process_profile = pp; }
343
1
    void set_delete_bitmap_agg_cache(DeleteBitmapAggCache* c) { _delete_bitmap_agg_cache = c; }
344
1
    void set_storage_page_cache(StoragePageCache* c) { this->_storage_page_cache = c; }
345
0
    void set_ann_index_ivf_list_cache(AnnIndexIVFListCache* c) {
346
0
        this->_ann_index_ivf_list_cache = c;
347
0
    }
348
1
    void set_segment_loader(SegmentLoader* sl) { this->_segment_loader = sl; }
349
0
    void set_routine_load_task_executor(RoutineLoadTaskExecutor* r) {
350
0
        this->_routine_load_task_executor = r;
351
0
    }
352
    void set_wal_mgr(std::unique_ptr<WalManager>&& wm);
353
    void clear_wal_mgr();
354
355
    void set_write_cooldown_meta_executors();
356
1
    static void set_tracking_memory(bool tracking_memory) {
357
1
        _s_tracking_memory.store(tracking_memory, std::memory_order_release);
358
1
    }
359
1
    void set_orc_memory_pool(orc::MemoryPool* pool) { _orc_memory_pool = pool; }
360
2
    void set_non_block_close_thread_pool(std::unique_ptr<ThreadPool>&& pool) {
361
2
        _non_block_close_thread_pool = std::move(pool);
362
2
    }
363
2
    void set_s3_file_upload_thread_pool(std::unique_ptr<ThreadPool>&& pool) {
364
2
        _s3_file_upload_thread_pool = std::move(pool);
365
2
    }
366
0
    void set_file_cache_factory(io::FileCacheFactory* factory) { _file_cache_factory = factory; }
367
2
    void set_file_cache_open_fd_cache(std::unique_ptr<io::FDCache>&& fd_cache) {
368
2
        _file_cache_open_fd_cache = std::move(fd_cache);
369
2
    }
370
#endif
371
    // WARN: The following setter methods are intended for use in test code and
372
    // offline tools (like meta_tool) ONLY. They should NOT be called in the
373
    // production environment to avoid thread safety issues and undefined behaviors.
374
1
    void set_cache_manager(CacheManager* cm) { this->_cache_manager = cm; }
375
1
    void set_tablet_schema_cache(TabletSchemaCache* c) { this->_tablet_schema_cache = c; }
376
1
    void set_tablet_column_object_pool(TabletColumnObjectPool* c) {
377
1
        this->_tablet_column_object_pool = c;
378
1
    }
379
380
0
    LoadStreamMapPool* load_stream_map_pool() { return _load_stream_map_pool.get(); }
381
382
0
    DeltaWriterV2Pool* delta_writer_v2_pool() { return _delta_writer_v2_pool.get(); }
383
384
    void wait_for_all_tasks_done();
385
386
    void update_frontends(const std::vector<TFrontendInfo>& new_infos);
387
    std::vector<TFrontendInfo> get_frontends();
388
    std::map<TNetworkAddress, FrontendInfo> get_running_frontends();
389
390
14.2k
    TabletSchemaCache* get_tablet_schema_cache() { return _tablet_schema_cache; }
391
1.09k
    TabletColumnObjectPool* get_tablet_column_object_pool() { return _tablet_column_object_pool; }
392
27.2k
    StoragePageCache* get_storage_page_cache() { return _storage_page_cache; }
393
0
    AnnIndexIVFListCache* get_ann_index_ivf_list_cache() { return _ann_index_ivf_list_cache; }
394
24.0k
    SegmentLoader* segment_loader() { return _segment_loader; }
395
0
    LookupConnectionCache* get_lookup_connection_cache() { return _lookup_connection_cache; }
396
0
    RowCache* get_row_cache() { return _row_cache; }
397
2.60k
    CacheManager* get_cache_manager() { return _cache_manager; }
398
0
    IdManager* get_id_manager() { return _id_manager; }
399
4
    ProcessProfile* get_process_profile() { return _process_profile; }
400
0
    HeapProfiler* get_heap_profiler() { return _heap_profiler; }
401
397
    segment_v2::InvertedIndexSearcherCache* get_inverted_index_searcher_cache() {
402
397
        return _inverted_index_searcher_cache;
403
397
    }
404
360
    segment_v2::InvertedIndexQueryCache* get_inverted_index_query_cache() {
405
360
        return _inverted_index_query_cache;
406
360
    }
407
0
    segment_v2::ConditionCache* get_condition_cache() { return _condition_cache; }
408
0
    segment_v2::EncodingInfoResolver* get_encoding_info_resolver() {
409
0
        return _encoding_info_resolver;
410
0
    }
411
3
    QueryCache* get_query_cache() { return _query_cache; }
412
413
2
    RuntimeFilterTimerQueue* runtime_filter_timer_queue() { return _runtime_filter_timer_queue; }
414
415
2
    DictionaryFactory* dict_factory() { return _dict_factory; }
416
417
606
    segment_v2::TmpFileDirs* get_tmp_file_dirs() { return _tmp_file_dirs.get(); }
418
18.4k
    io::FDCache* file_cache_open_fd_cache() const { return _file_cache_open_fd_cache.get(); }
419
420
21
    segment_v2::AnnIndexResultCache* ann_index_result_cache() { return _ann_index_result_cache; }
421
422
55
    orc::MemoryPool* orc_memory_pool() { return _orc_memory_pool; }
423
0
    arrow::MemoryPool* arrow_memory_pool() { return _arrow_memory_pool; }
424
425
    bool check_auth_token(const std::string& auth_token);
426
10
    void set_stream_mgr(VDataStreamMgr* vstream_mgr) { _vstream_mgr = vstream_mgr; }
427
    void clear_stream_mgr();
428
429
190
    DeleteBitmapAggCache* delete_bitmap_agg_cache() { return _delete_bitmap_agg_cache; }
430
    Status init_mem_env();
431
432
private:
433
    ExecEnv();
434
435
    [[nodiscard]] Status _init(const std::vector<StorePath>& store_paths,
436
                               const std::vector<StorePath>& spill_store_paths,
437
                               const std::set<std::string>& broken_paths);
438
    void _destroy();
439
440
    Status _check_deploy_mode();
441
442
    Status _create_internal_workload_group();
443
    void _init_runtime_filter_timer_queue();
444
445
    inline static std::atomic_bool _s_ready {false};
446
    inline static std::atomic_bool _s_tracking_memory {false};
447
    std::vector<StorePath> _store_paths;
448
    std::vector<StorePath> _spill_store_paths;
449
    inline static std::atomic_bool _s_upgrading {false};
450
451
    io::FileCacheFactory* _file_cache_factory = nullptr;
452
    UserFunctionCache* _user_function_cache = nullptr;
453
    // Leave protected so that subclasses can override
454
    ExternalScanContextMgr* _external_scan_context_mgr = nullptr;
455
    VDataStreamMgr* _vstream_mgr = nullptr;
456
    ResultBufferMgr* _result_mgr = nullptr;
457
    ResultQueueMgr* _result_queue_mgr = nullptr;
458
    ClientCache<BackendServiceClient>* _backend_client_cache = nullptr;
459
    ClientCache<FrontendServiceClient>* _frontend_client_cache = nullptr;
460
    ClientCache<TPaloBrokerServiceClient>* _broker_client_cache = nullptr;
461
462
    // The default tracker consumed by mem hook. If the thread does not attach other trackers,
463
    // by default all consumption will be passed to the process tracker through the orphan tracker.
464
    // In real time, `consumption of all limiter trackers` + `orphan tracker consumption` = `process tracker consumption`.
465
    // Ideally, all threads are expected to attach to the specified tracker, so that "all memory has its own ownership",
466
    // and the consumption of the orphan mem tracker is close to 0, but greater than 0.
467
    std::shared_ptr<MemTrackerLimiter> _orphan_mem_tracker;
468
    std::shared_ptr<MemTrackerLimiter> _brpc_iobuf_block_memory_tracker;
469
    // Count the memory consumption of segment compaction tasks.
470
    std::shared_ptr<MemTrackerLimiter> _segcompaction_mem_tracker;
471
    std::shared_ptr<MemTrackerLimiter> _stream_load_pipe_tracker;
472
473
    std::shared_ptr<MemTrackerLimiter> _tablets_no_cache_mem_tracker;
474
    std::shared_ptr<MemTrackerLimiter> _rowsets_no_cache_mem_tracker;
475
    std::shared_ptr<MemTrackerLimiter> _segments_no_cache_mem_tracker;
476
477
    // Tracking memory may be shared between multiple queries.
478
    std::shared_ptr<MemTrackerLimiter> _point_query_executor_mem_tracker;
479
    std::shared_ptr<MemTrackerLimiter> _block_compression_mem_tracker;
480
    std::shared_ptr<MemTrackerLimiter> _query_cache_mem_tracker;
481
482
    // TODO, looking forward to more accurate tracking.
483
    std::shared_ptr<MemTrackerLimiter> _rowid_storage_reader_tracker;
484
    std::shared_ptr<MemTrackerLimiter> _subcolumns_tree_tracker;
485
    std::shared_ptr<MemTrackerLimiter> _s3_file_buffer_tracker;
486
487
    // Tracking memory consumption of parquet meta
488
    std::shared_ptr<MemTrackerLimiter> _parquet_meta_tracker;
489
490
    std::unique_ptr<ThreadPool> _send_batch_thread_pool;
491
    // Threadpool used to prefetch remote file for buffered reader
492
    std::unique_ptr<ThreadPool> _buffered_reader_prefetch_thread_pool;
493
    // Threadpool used to send TableStats to FE
494
    std::unique_ptr<ThreadPool> _send_table_stats_thread_pool;
495
    // Threadpool used to upload local file to s3
496
    std::unique_ptr<ThreadPool> _s3_file_upload_thread_pool;
497
    // Pool used by join node to build hash table
498
    // Pool to use a new thread to release object
499
    std::unique_ptr<ThreadPool> _lazy_release_obj_pool;
500
    std::unique_ptr<ThreadPool> _non_block_close_thread_pool;
501
    std::unique_ptr<ThreadPool> _s3_file_system_thread_pool;
502
    // for java-udf to close
503
    std::unique_ptr<ThreadPool> _udf_close_workers_thread_pool;
504
    // Threadpool used to prefetch segment file cache blocks
505
    std::unique_ptr<ThreadPool> _segment_prefetch_thread_pool;
506
    std::unique_ptr<ThreadPool> _peer_race_s3_thread_pool;
507
508
    FragmentMgr* _fragment_mgr = nullptr;
509
    WorkloadGroupMgr* _workload_group_manager = nullptr;
510
511
    ResultCache* _result_cache = nullptr;
512
    ClusterInfo* _cluster_info = nullptr;
513
    LoadPathMgr* _load_path_mgr = nullptr;
514
515
    BfdParser* _bfd_parser = nullptr;
516
    BrokerMgr* _broker_mgr = nullptr;
517
    LoadChannelMgr* _load_channel_mgr = nullptr;
518
    std::unique_ptr<LoadStreamMgr> _load_stream_mgr;
519
    std::unique_ptr<NewLoadStreamMgr> _new_load_stream_mgr;
520
    BrpcClientCache<PBackendService_Stub>* _internal_client_cache = nullptr;
521
    BrpcClientCache<PBackendService_Stub>* _streaming_client_cache = nullptr;
522
    BrpcClientCache<PFunctionService_Stub>* _function_client_cache = nullptr;
523
524
    std::unique_ptr<StreamLoadExecutor> _stream_load_executor;
525
    RoutineLoadTaskExecutor* _routine_load_task_executor = nullptr;
526
    StreamLoadRecorderManager* _stream_load_recorder_manager = nullptr;
527
    SmallFileMgr* _small_file_mgr = nullptr;
528
    HeartbeatFlags* _heartbeat_flags = nullptr;
529
530
    // To save meta info of external file, such as parquet footer.
531
    FileMetaCache* _file_meta_cache = nullptr;
532
    std::unique_ptr<MemTableMemoryLimiter> _memtable_memory_limiter;
533
    std::unique_ptr<LoadStreamMapPool> _load_stream_map_pool;
534
    std::unique_ptr<DeltaWriterV2Pool> _delta_writer_v2_pool;
535
    std::unique_ptr<WalManager> _wal_manager;
536
    DNSCache* _dns_cache = nullptr;
537
    std::unique_ptr<WriteCooldownMetaExecutors> _write_cooldown_meta_executors;
538
539
    std::mutex _frontends_lock;
540
    // ip:brpc_port -> frontend_indo
541
    std::map<TNetworkAddress, FrontendInfo> _frontends;
542
    GroupCommitMgr* _group_commit_mgr = nullptr;
543
    CdcClientMgr* _cdc_client_mgr = nullptr;
544
545
    // Maybe we should use unique_ptr, but it need complete type, which means we need
546
    // to include many headers, and for some cpp file that do not need class like TabletSchemaCache,
547
    // these redundancy header could introduce potential bug, at least, more header means slow compile.
548
    // So we choose to use raw pointer, please remember to delete these pointer in deconstructor.
549
    TabletSchemaCache* _tablet_schema_cache = nullptr;
550
    TabletColumnObjectPool* _tablet_column_object_pool = nullptr;
551
    std::unique_ptr<BaseStorageEngine> _storage_engine;
552
    StoragePageCache* _storage_page_cache = nullptr;
553
    AnnIndexIVFListCache* _ann_index_ivf_list_cache = nullptr;
554
    SegmentLoader* _segment_loader = nullptr;
555
    LookupConnectionCache* _lookup_connection_cache = nullptr;
556
    RowCache* _row_cache = nullptr;
557
    CacheManager* _cache_manager = nullptr;
558
    IdManager* _id_manager = nullptr;
559
    ProcessProfile* _process_profile = nullptr;
560
    HeapProfiler* _heap_profiler = nullptr;
561
    segment_v2::InvertedIndexSearcherCache* _inverted_index_searcher_cache = nullptr;
562
    segment_v2::InvertedIndexQueryCache* _inverted_index_query_cache = nullptr;
563
    segment_v2::ConditionCache* _condition_cache = nullptr;
564
    segment_v2::EncodingInfoResolver* _encoding_info_resolver = nullptr;
565
    QueryCache* _query_cache = nullptr;
566
    std::unique_ptr<io::FDCache> _file_cache_open_fd_cache;
567
    DeleteBitmapAggCache* _delete_bitmap_agg_cache {nullptr};
568
    segment_v2::AnnIndexResultCache* _ann_index_result_cache = nullptr;
569
570
    RuntimeFilterTimerQueue* _runtime_filter_timer_queue = nullptr;
571
    DictionaryFactory* _dict_factory = nullptr;
572
573
    WorkloadSchedPolicyMgr* _workload_sched_mgr = nullptr;
574
    IndexPolicyMgr* _index_policy_mgr = nullptr;
575
576
    RuntimeQueryStatisticsMgr* _runtime_query_statistics_mgr = nullptr;
577
578
    std::unique_ptr<segment_v2::TmpFileDirs> _tmp_file_dirs;
579
580
    SpillFileManager* _spill_file_mgr = nullptr;
581
582
    orc::MemoryPool* _orc_memory_pool = nullptr;
583
    arrow::MemoryPool* _arrow_memory_pool = nullptr;
584
585
    kerberos::KerberosTicketMgr* _kerberos_ticket_mgr = nullptr;
586
    io::HdfsMgr* _hdfs_mgr = nullptr;
587
    io::PackedFileManager* _packed_file_manager = nullptr;
588
    S3RateLimiterHolder* _warmup_download_rate_limiter = nullptr;
589
    std::unique_ptr<MSRpcRateLimitServices> _ms_rpc_rate_limit_services;
590
};
591
592
template <>
593
0
inline ClientCache<BackendServiceClient>* ExecEnv::get_client_cache<BackendServiceClient>() {
594
0
    return _backend_client_cache;
595
0
}
596
template <>
597
0
inline ClientCache<FrontendServiceClient>* ExecEnv::get_client_cache<FrontendServiceClient>() {
598
0
    return _frontend_client_cache;
599
0
}
600
template <>
601
inline ClientCache<TPaloBrokerServiceClient>*
602
0
ExecEnv::get_client_cache<TPaloBrokerServiceClient>() {
603
0
    return _broker_client_cache;
604
0
}
605
606
0
inline segment_v2::InvertedIndexQueryCache* GetInvertedIndexQueryCache() {
607
0
    return ExecEnv::GetInstance()->get_inverted_index_query_cache();
608
0
}
609
610
} // namespace doris