Coverage Report

Created: 2026-07-24 09:37

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/runtime/exec_env_init.cpp
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
// IWYU pragma: no_include <bthread/errno.h>
19
#include <gen_cpp/HeartbeatService_types.h>
20
#include <gen_cpp/Metrics_types.h>
21
#include <simdjson.h>
22
#include <sys/resource.h>
23
24
#include <cerrno> // IWYU pragma: keep
25
#include <cstdint>
26
#include <cstdlib>
27
#include <cstring>
28
#include <limits>
29
#include <memory>
30
#include <ostream>
31
#include <string>
32
#include <vector>
33
34
#include "cloud/cloud_cluster_info.h"
35
#include "cloud/cloud_meta_mgr.h"
36
#include "cloud/cloud_ms_rpc_rate_limit_services.h"
37
#include "cloud/cloud_ms_rpc_rate_limiters.h"
38
#include "cloud/cloud_storage_engine.h"
39
#include "cloud/cloud_stream_load_executor.h"
40
#include "cloud/cloud_tablet_hotspot.h"
41
#include "cloud/cloud_warm_up_manager.h"
42
#include "cloud/config.h"
43
#include "common/cast_set.h"
44
#include "common/config.h"
45
#include "common/kerberos/kerberos_ticket_mgr.h"
46
#include "common/logging.h"
47
#include "common/metrics/doris_metrics.h"
48
#include "common/multi_version.h"
49
#include "common/status.h"
50
#include "cpp/token_bucket_rate_limiter.h"
51
#include "exec/exchange/vdata_stream_mgr.h"
52
#include "exec/pipeline/task_queue.h"
53
#include "exec/pipeline/task_scheduler.h"
54
#include "exec/scan/scanner_scheduler.h"
55
#include "exec/sink/delta_writer_v2_pool.h"
56
#include "exec/sink/load_stream_map_pool.h"
57
#include "exec/spill/spill_file_manager.h"
58
#include "exprs/function/dictionary_factory.h"
59
#include "format/orc/orc_memory_pool.h"
60
#include "format/parquet/arrow_memory_pool.h"
61
#include "io/cache/block_file_cache_downloader.h"
62
#include "io/cache/block_file_cache_factory.h"
63
#include "io/cache/fs_file_cache_storage.h"
64
#include "io/fs/file_meta_cache.h"
65
#include "io/fs/local_file_reader.h"
66
#include "load/channel/load_channel_mgr.h"
67
#include "load/channel/load_stream_mgr.h"
68
#include "load/group_commit/group_commit_mgr.h"
69
#include "load/group_commit/wal/wal_manager.h"
70
#include "load/load_path_mgr.h"
71
#include "load/memtable/memtable_memory_limiter.h"
72
#include "load/routine_load/routine_load_task_executor.h"
73
#include "load/stream_load/new_load_stream_mgr.h"
74
#include "load/stream_load/stream_load_executor.h"
75
#include "load/stream_load/stream_load_recorder_manager.h"
76
#include "runtime/broker_mgr.h"
77
#include "runtime/cache/result_cache.h"
78
#include "runtime/cdc_client_mgr.h"
79
#include "runtime/exec_env.h"
80
#include "runtime/external_scan_context_mgr.h"
81
#include "runtime/fragment_mgr.h"
82
#include "runtime/heartbeat_flags.h"
83
#include "runtime/index_policy/index_policy_mgr.h"
84
#include "runtime/memory/cache_manager.h"
85
#include "runtime/memory/heap_profiler.h"
86
#include "runtime/memory/mem_tracker.h"
87
#include "runtime/memory/mem_tracker_limiter.h"
88
#include "runtime/memory/thread_mem_tracker_mgr.h"
89
#include "runtime/process_profile.h"
90
#include "runtime/query_cache/query_cache.h"
91
#include "runtime/result_buffer_mgr.h"
92
#include "runtime/result_queue_mgr.h"
93
#include "runtime/runtime_query_statistics_mgr.h"
94
#include "runtime/small_file_mgr.h"
95
#include "runtime/thread_context.h"
96
#include "runtime/user_function_cache.h"
97
#include "runtime/workload_group/workload_group_manager.h"
98
#include "runtime/workload_management/workload_sched_policy_mgr.h"
99
#include "service/backend_options.h"
100
#include "service/backend_service.h"
101
#include "service/point_query_executor.h"
102
#include "storage/cache/ann_index_ivf_list_cache.h"
103
#include "storage/cache/page_cache.h"
104
#include "storage/id_manager.h"
105
#include "storage/index/ann/ann_index_result_cache/ann_index_result_cache.h"
106
#include "storage/index/inverted/inverted_index_cache.h"
107
#include "storage/olap_define.h"
108
#include "storage/options.h"
109
#include "storage/segment/condition_cache.h"
110
#include "storage/segment/encoding_info.h"
111
#include "storage/segment/segment_loader.h"
112
#include "storage/storage_engine.h"
113
#include "storage/storage_policy.h"
114
#include "storage/tablet/tablet_column_object_pool.h"
115
#include "storage/tablet/tablet_meta.h"
116
#include "storage/tablet/tablet_schema_cache.h"
117
#include "udf/python/python_server.h"
118
#include "util/bfd_parser.h"
119
#include "util/bit_util.h"
120
#include "util/brpc_client_cache.h"
121
#include "util/client_cache.h"
122
#include "util/cpu_info.h"
123
#include "util/disk_info.h"
124
#include "util/dns_cache.h"
125
#include "util/mem_info.h"
126
#include "util/parse_util.h"
127
#include "util/pretty_printer.h"
128
#include "util/threadpool.h"
129
#include "util/thrift_rpc_helper.h"
130
#include "util/timezone_utils.h"
131
132
// clang-format off
133
// this must after util/brpc_client_cache.h
134
// /doris/thirdparty/installed/include/brpc/errno.pb.h:69:3: error: expected identifier
135
//  EINTERNAL = 2001,
136
//   ^
137
//  /doris/thirdparty/installed/include/hadoop_hdfs/hdfs.h:61:19: note: expanded from macro 'EINTERNAL'
138
//  #define EINTERNAL 255
139
#include "io/fs/hdfs/hdfs_mgr.h"
140
#include "io/fs/packed_file_manager.h"
141
// clang-format on
142
143
namespace doris {
144
145
class PBackendService_Stub;
146
class PFunctionService_Stub;
147
148
// Warmup download rate limiter metrics
149
bvar::LatencyRecorder warmup_download_rate_limit_latency("warmup_download_rate_limit_latency");
150
bvar::Adder<int64_t> warmup_download_rate_limit_ns("warmup_download_rate_limit_ns");
151
bvar::Adder<int64_t> warmup_download_rate_limit_exceed_req_num(
152
        "warmup_download_rate_limit_exceed_req_num");
153
154
7
static void init_doris_metrics(const std::vector<StorePath>& store_paths) {
155
7
    bool init_system_metrics = config::enable_system_metrics;
156
7
    std::set<std::string> disk_devices;
157
7
    std::vector<std::string> network_interfaces;
158
7
    std::vector<std::string> paths;
159
11
    for (const auto& store_path : store_paths) {
160
11
        paths.emplace_back(store_path.path);
161
11
    }
162
7
    if (init_system_metrics) {
163
3
        auto st = DiskInfo::get_disk_devices(paths, &disk_devices);
164
3
        if (!st.ok()) {
165
0
            LOG(WARNING) << "get disk devices failed, status=" << st;
166
0
            return;
167
0
        }
168
3
        st = get_inet_interfaces(&network_interfaces, BackendOptions::is_bind_ipv6());
169
3
        if (!st.ok()) {
170
0
            LOG(WARNING) << "get inet interfaces failed, status=" << st;
171
0
            return;
172
0
        }
173
3
    }
174
7
    DorisMetrics::instance()->initialize(init_system_metrics, disk_devices, network_interfaces);
175
7
}
176
177
// Used to calculate the num of min thread and max thread based on the passed config
178
14
static std::pair<size_t, size_t> get_num_threads(size_t min_num, size_t max_num) {
179
14
    auto num_cores = doris::CpuInfo::num_cores();
180
14
    min_num = (min_num == 0) ? num_cores : min_num;
181
14
    max_num = (max_num == 0) ? num_cores : max_num;
182
14
    auto factor = max_num / min_num;
183
14
    min_num = std::min(num_cores * factor, min_num);
184
14
    max_num = std::min(min_num * factor, max_num);
185
14
    return {min_num, max_num};
186
14
}
187
188
20.9k
ThreadPool* ExecEnv::non_block_close_thread_pool() {
189
20.9k
    return _non_block_close_thread_pool.get();
190
20.9k
}
191
192
14
ExecEnv::ExecEnv() = default;
193
194
10
ExecEnv::~ExecEnv() {
195
10
    destroy();
196
10
}
197
198
Status ExecEnv::init(ExecEnv* env, const std::vector<StorePath>& store_paths,
199
                     const std::vector<StorePath>& spill_store_paths,
200
7
                     const std::set<std::string>& broken_paths) {
201
7
    return env->_init(store_paths, spill_store_paths, broken_paths);
202
7
}
203
204
// pick simdjson implementation based on CPU capabilities
205
7
inline void init_simdjson_parser() {
206
    // haswell: AVX2 (2013 Intel Haswell or later, all AMD Zen processors)
207
7
    const auto* haswell_implementation = simdjson::get_available_implementations()["haswell"];
208
7
    if (!haswell_implementation || !haswell_implementation->supported_by_runtime_system()) {
209
        // pick available implementation
210
0
        for (const auto* implementation : simdjson::get_available_implementations()) {
211
0
            if (implementation->supported_by_runtime_system()) {
212
0
                LOG(INFO) << "Using SimdJSON implementation : " << implementation->name() << ": "
213
0
                          << implementation->description();
214
0
                simdjson::get_active_implementation() = implementation;
215
0
                return;
216
0
            }
217
0
        }
218
0
        LOG(WARNING) << "No available SimdJSON implementation found.";
219
7
    } else {
220
7
        LOG(INFO) << "Using SimdJSON Haswell implementation";
221
7
    }
222
7
}
223
224
Status ExecEnv::_init(const std::vector<StorePath>& store_paths,
225
                      const std::vector<StorePath>& spill_store_paths,
226
7
                      const std::set<std::string>& broken_paths) {
227
    //Only init once before be destroyed
228
7
    if (ready()) {
229
0
        return Status::OK();
230
0
    }
231
7
    std::unordered_map<std::string, std::unique_ptr<SpillDataDir>> spill_store_map;
232
11
    for (const auto& spill_path : spill_store_paths) {
233
11
        spill_store_map.emplace(spill_path.path, std::make_unique<SpillDataDir>(
234
11
                                                         spill_path.path, spill_path.capacity_bytes,
235
11
                                                         spill_path.storage_medium));
236
11
    }
237
7
    init_doris_metrics(store_paths);
238
7
    _store_paths = store_paths;
239
7
    _tmp_file_dirs = std::make_unique<segment_v2::TmpFileDirs>(_store_paths);
240
7
    RETURN_IF_ERROR(_tmp_file_dirs->init());
241
7
    _user_function_cache = new UserFunctionCache();
242
7
    static_cast<void>(_user_function_cache->init(doris::config::user_function_dir));
243
7
    _external_scan_context_mgr = new ExternalScanContextMgr(this);
244
7
    set_stream_mgr(new doris::VDataStreamMgr());
245
7
    _result_mgr = new ResultBufferMgr();
246
7
    _result_queue_mgr = new ResultQueueMgr();
247
7
    _backend_client_cache = new BackendServiceClientCache(config::max_client_cache_size_per_host);
248
7
    _frontend_client_cache = new FrontendServiceClientCache(config::max_client_cache_size_per_host);
249
7
    _broker_client_cache = new BrokerServiceClientCache(config::max_client_cache_size_per_host);
250
251
7
    TimezoneUtils::load_timezones_to_cache();
252
253
7
    static_cast<void>(ThreadPoolBuilder("SendBatchThreadPool")
254
7
                              .set_min_threads(config::send_batch_thread_pool_thread_num)
255
7
                              .set_max_threads(config::send_batch_thread_pool_thread_num)
256
7
                              .set_max_queue_size(config::send_batch_thread_pool_queue_size)
257
7
                              .build(&_send_batch_thread_pool));
258
259
7
    static_cast<void>(ThreadPoolBuilder("UDFCloseWorkers")
260
7
                              .set_min_threads(4)
261
7
                              .set_max_threads(std::min(32, CpuInfo::num_cores()))
262
7
                              .build(&_udf_close_workers_thread_pool));
263
264
7
    auto [buffered_reader_min_threads, buffered_reader_max_threads] =
265
7
            get_num_threads(config::num_buffered_reader_prefetch_thread_pool_min_thread,
266
7
                            config::num_buffered_reader_prefetch_thread_pool_max_thread);
267
7
    static_cast<void>(ThreadPoolBuilder("BufferedReaderPrefetchThreadPool")
268
7
                              .set_min_threads(cast_set<int>(buffered_reader_min_threads))
269
7
                              .set_max_threads(cast_set<int>(buffered_reader_max_threads))
270
7
                              .build(&_buffered_reader_prefetch_thread_pool));
271
272
7
    static_cast<void>(ThreadPoolBuilder("SegmentPrefetchThreadPool")
273
7
                              .set_min_threads(cast_set<int>(
274
7
                                      config::segment_prefetch_thread_pool_thread_num_min))
275
7
                              .set_max_threads(cast_set<int>(
276
7
                                      config::segment_prefetch_thread_pool_thread_num_max))
277
7
                              .build(&_segment_prefetch_thread_pool));
278
279
7
    static_cast<void>(ThreadPoolBuilder("SendTableStatsThreadPool")
280
7
                              .set_min_threads(8)
281
7
                              .set_max_threads(32)
282
7
                              .build(&_send_table_stats_thread_pool));
283
284
7
    auto [s3_file_upload_min_threads, s3_file_upload_max_threads] =
285
7
            get_num_threads(config::num_s3_file_upload_thread_pool_min_thread,
286
7
                            config::num_s3_file_upload_thread_pool_max_thread);
287
7
    static_cast<void>(ThreadPoolBuilder("S3FileUploadThreadPool")
288
7
                              .set_min_threads(cast_set<int>(s3_file_upload_min_threads))
289
7
                              .set_max_threads(cast_set<int>(s3_file_upload_max_threads))
290
7
                              .build(&_s3_file_upload_thread_pool));
291
292
    // min num equal to fragment pool's min num
293
    // max num is useless because it will start as many as requested in the past
294
    // queue size is useless because the max thread num is very large
295
7
    static_cast<void>(ThreadPoolBuilder("LazyReleaseMemoryThreadPool")
296
7
                              .set_min_threads(1)
297
7
                              .set_max_threads(1)
298
7
                              .set_max_queue_size(1000000)
299
7
                              .build(&_lazy_release_obj_pool));
300
7
    static_cast<void>(ThreadPoolBuilder("NonBlockCloseThreadPool")
301
7
                              .set_min_threads(cast_set<int>(config::min_nonblock_close_thread_num))
302
7
                              .set_max_threads(cast_set<int>(config::max_nonblock_close_thread_num))
303
7
                              .build(&_non_block_close_thread_pool));
304
7
    static_cast<void>(ThreadPoolBuilder("S3FileSystemThreadPool")
305
7
                              .set_min_threads(config::min_s3_file_system_thread_num)
306
7
                              .set_max_threads(config::max_s3_file_system_thread_num)
307
7
                              .build(&_s3_file_system_thread_pool));
308
7
    static_cast<void>(ThreadPoolBuilder("PeerRaceS3ThreadPool")
309
7
                              .set_min_threads(config::min_peer_race_s3_thread_num)
310
7
                              .set_max_threads(config::max_peer_race_s3_thread_num)
311
7
                              .build(&_peer_race_s3_thread_pool));
312
7
    RETURN_IF_ERROR(init_mem_env());
313
314
    // NOTE: runtime query statistics mgr could be visited by query and daemon thread
315
    // so it should be created before all query begin and deleted after all query and daemon thread stoppped
316
7
    _runtime_query_statistics_mgr = new RuntimeQueryStatisticsMgr();
317
7
    CgroupCpuCtl::init_doris_cgroup_path();
318
7
    _file_cache_open_fd_cache = std::make_unique<io::FDCache>();
319
7
    _file_cache_factory = new io::FileCacheFactory();
320
7
    std::vector<doris::CachePath> cache_paths;
321
7
    init_file_cache_factory(cache_paths);
322
7
    doris::io::BeConfDataDirReader::init_be_conf_data_dir(store_paths, spill_store_paths,
323
7
                                                          cache_paths);
324
7
    _init_runtime_filter_timer_queue();
325
326
7
    _workload_group_manager = new WorkloadGroupMgr();
327
328
7
    _fragment_mgr = new FragmentMgr(this);
329
7
    _result_cache = new ResultCache(config::query_cache_max_size_mb,
330
7
                                    config::query_cache_elasticity_size_mb);
331
7
    if (config::is_cloud_mode()) {
332
1
        _cluster_info = new CloudClusterInfo();
333
6
    } else {
334
6
        _cluster_info = new ClusterInfo();
335
6
    }
336
337
7
    _load_path_mgr = new LoadPathMgr(this);
338
7
    _bfd_parser = BfdParser::create();
339
7
    _broker_mgr = new BrokerMgr(this);
340
7
    _load_channel_mgr = new LoadChannelMgr();
341
7
    auto num_flush_threads = std::min(
342
7
            _store_paths.size() * config::flush_thread_num_per_store,
343
7
            static_cast<size_t>(CpuInfo::num_cores()) * config::max_flush_thread_num_per_cpu);
344
7
    _load_stream_mgr = std::make_unique<LoadStreamMgr>(num_flush_threads);
345
7
    _new_load_stream_mgr = NewLoadStreamMgr::create_unique();
346
7
    _internal_client_cache = new BrpcClientCache<PBackendService_Stub>();
347
7
    _streaming_client_cache =
348
7
            new BrpcClientCache<PBackendService_Stub>("baidu_std", "single", "streaming");
349
7
    _function_client_cache =
350
7
            new BrpcClientCache<PFunctionService_Stub>(config::function_service_protocol);
351
7
    if (config::is_cloud_mode()) {
352
1
        _stream_load_executor = CloudStreamLoadExecutor::create_unique(this);
353
6
    } else {
354
6
        _stream_load_executor = StreamLoadExecutor::create_unique(this);
355
6
    }
356
7
    _routine_load_task_executor = new RoutineLoadTaskExecutor(this);
357
7
    RETURN_IF_ERROR(_routine_load_task_executor->init(MemInfo::mem_limit()));
358
7
    _small_file_mgr = new SmallFileMgr(this, config::small_file_dir);
359
7
    _group_commit_mgr = new GroupCommitMgr(this);
360
7
    _cdc_client_mgr = new CdcClientMgr();
361
7
    _memtable_memory_limiter = std::make_unique<MemTableMemoryLimiter>();
362
7
    _load_stream_map_pool = std::make_unique<LoadStreamMapPool>();
363
7
    _delta_writer_v2_pool = std::make_unique<DeltaWriterV2Pool>();
364
7
    _wal_manager = WalManager::create_unique(this, config::group_commit_wal_path);
365
7
    _dns_cache = new DNSCache();
366
7
    _write_cooldown_meta_executors = std::make_unique<WriteCooldownMetaExecutors>();
367
368
7
    _spill_file_mgr = new SpillFileManager(std::move(spill_store_map));
369
7
    _kerberos_ticket_mgr = new kerberos::KerberosTicketMgr(config::kerberos_ccache_path);
370
7
    _hdfs_mgr = new io::HdfsMgr();
371
7
    _backend_client_cache->init_metrics("backend");
372
7
    _frontend_client_cache->init_metrics("frontend");
373
7
    _broker_client_cache->init_metrics("broker");
374
7
    static_cast<void>(_result_mgr->init());
375
7
    Status status = _load_path_mgr->init();
376
7
    if (!status.ok()) {
377
0
        LOG(ERROR) << "Load path mgr init failed. " << status;
378
0
        return status;
379
0
    }
380
7
    _broker_mgr->init();
381
7
    static_cast<void>(_small_file_mgr->init());
382
7
    if (!status.ok()) {
383
0
        LOG(ERROR) << "Scanner scheduler init failed. " << status;
384
0
        return status;
385
0
    }
386
387
7
    RETURN_IF_ERROR(_memtable_memory_limiter->init(MemInfo::mem_limit()));
388
7
    RETURN_IF_ERROR(_load_channel_mgr->init(MemInfo::mem_limit()));
389
7
    RETURN_IF_ERROR(_wal_manager->init());
390
7
    _heartbeat_flags = new HeartbeatFlags();
391
392
7
    _tablet_schema_cache =
393
7
            TabletSchemaCache::create_global_schema_cache(config::tablet_schema_cache_capacity);
394
395
7
    _tablet_column_object_pool = TabletColumnObjectPool::create_global_column_cache(
396
7
            config::tablet_schema_cache_capacity);
397
398
    // Storage engine
399
7
    doris::EngineOptions options;
400
7
    options.store_paths = store_paths;
401
7
    options.broken_paths = broken_paths;
402
7
    options.backend_uid = doris::UniqueId::gen_uid();
403
    // Check if the startup mode has been modified
404
7
    RETURN_IF_ERROR(_check_deploy_mode());
405
7
    if (config::is_cloud_mode()) {
406
1
        std::cout << "start BE in cloud mode, cloud_unique_id: " << config::cloud_unique_id
407
1
                  << ", meta_service_endpoint: " << config::meta_service_endpoint << std::endl;
408
1
        _storage_engine = std::make_unique<CloudStorageEngine>(options);
409
6
    } else {
410
6
        std::cout << "start BE in local mode" << std::endl;
411
6
        _storage_engine = std::make_unique<StorageEngine>(options);
412
6
    }
413
7
    auto st = _storage_engine->open();
414
7
    if (!st.ok()) {
415
0
        LOG(ERROR) << "Fail to open StorageEngine, res=" << st;
416
0
        return st;
417
0
    }
418
7
    _storage_engine->set_heartbeat_flags(this->heartbeat_flags());
419
7
    if (st = _storage_engine->start_bg_threads(nullptr); !st.ok()) {
420
0
        LOG(ERROR) << "Failed to starge bg threads of storage engine, res=" << st;
421
0
        return st;
422
0
    }
423
424
    // should start after storage_engine->open()
425
7
    _stream_load_recorder_manager = new StreamLoadRecorderManager();
426
7
    _stream_load_recorder_manager->start();
427
428
    // create internal workload group should be after storage_engin->open()
429
7
    RETURN_IF_ERROR(_create_internal_workload_group());
430
7
    _workload_sched_mgr = new WorkloadSchedPolicyMgr();
431
7
    _workload_sched_mgr->start(this);
432
433
    // Initialize packed file manager
434
7
    _packed_file_manager = io::PackedFileManager::instance();
435
7
    if (config::is_cloud_mode()) {
436
1
        RETURN_IF_ERROR(_packed_file_manager->init());
437
1
        _packed_file_manager->start_background_manager();
438
439
        // Start cluster info background worker for compaction read-write separation
440
1
        static_cast<CloudClusterInfo*>(_cluster_info)->start_bg_worker();
441
442
        // Initialize MS RPC rate limiters and table-level backpressure handling.
443
1
        _ms_rpc_rate_limit_services = std::make_unique<MSRpcRateLimitServices>();
444
1
        static_cast<CloudStorageEngine*>(_storage_engine.get())
445
1
                ->meta_mgr()
446
1
                .set_host_level_ms_rpc_rate_limiters(
447
1
                        _ms_rpc_rate_limit_services->host_level_ms_rpc_rate_limiters());
448
1
        static_cast<CloudStorageEngine*>(_storage_engine.get())
449
1
                ->meta_mgr()
450
1
                .set_ms_backpressure_handler(
451
1
                        _ms_rpc_rate_limit_services->ms_backpressure_handler());
452
1
    }
453
454
7
    _index_policy_mgr = new IndexPolicyMgr();
455
456
    // Initialize warmup download rate limiter for cloud mode
457
    // Always create the rate limiter in cloud mode to support dynamic rate limit changes
458
7
    if (config::is_cloud_mode()) {
459
1
        int64_t rate_limit = config::file_cache_warmup_download_rate_limit_bytes_per_second;
460
        // When rate_limit <= 0, pass 0 to disable rate limiting
461
1
        int64_t rate = rate_limit > 0 ? rate_limit : 0;
462
        // max_burst is the same as rate (1 second burst)
463
        // limit is 0 which means no total limit
464
        // When rate is 0, S3RateLimiter will not throttle (no rate limiting)
465
1
        _warmup_download_rate_limiter = new S3RateLimiterHolder(rate, rate, 0, [&](int64_t ns) {
466
0
            if (ns > 0) {
467
0
                warmup_download_rate_limit_latency << ns / 1000;
468
0
                warmup_download_rate_limit_ns << ns;
469
0
                warmup_download_rate_limit_exceed_req_num << 1;
470
0
            }
471
0
        });
472
1
    }
473
474
7
    RETURN_IF_ERROR(_spill_file_mgr->init());
475
7
    RETURN_IF_ERROR(_runtime_query_statistics_mgr->start_report_thread());
476
7
    _dict_factory = new doris::DictionaryFactory();
477
7
    _s_ready = true;
478
479
7
    init_simdjson_parser();
480
481
    // Make aws-sdk-cpp InitAPI and ShutdownAPI called in the same thread
482
7
    S3ClientFactory::instance();
483
7
    return Status::OK();
484
7
}
485
486
// when user not sepcify a workload group in FE, then query could
487
// use dummy workload group.
488
7
Status ExecEnv::_create_internal_workload_group() {
489
7
    LOG(INFO) << "begin create internal workload group.";
490
491
7
    RETURN_IF_ERROR(_workload_group_manager->create_internal_wg());
492
7
    return Status::OK();
493
7
}
494
495
9
void ExecEnv::_init_runtime_filter_timer_queue() {
496
9
    _runtime_filter_timer_queue = new doris::RuntimeFilterTimerQueue();
497
9
    _runtime_filter_timer_queue->run();
498
9
}
499
500
7
void ExecEnv::init_file_cache_factory(std::vector<doris::CachePath>& cache_paths) {
501
    // Load file cache before starting up daemon threads to make sure StorageEngine is read.
502
7
    if (!config::enable_file_cache) {
503
4
        if (config::is_cloud_mode()) {
504
0
            LOG(FATAL) << "Cloud mode requires to enable file cache, plz set "
505
0
                          "config::enable_file_cache "
506
0
                          "= true";
507
0
            exit(-1);
508
0
        }
509
4
        return;
510
4
    }
511
3
    if (config::file_cache_each_block_size > config::s3_write_buffer_size ||
512
3
        config::s3_write_buffer_size % config::file_cache_each_block_size != 0) {
513
0
        LOG_FATAL(
514
0
                "The config file_cache_each_block_size {} must less than or equal to config "
515
0
                "s3_write_buffer_size {} and config::s3_write_buffer_size % "
516
0
                "config::file_cache_each_block_size must be zero",
517
0
                config::file_cache_each_block_size, config::s3_write_buffer_size);
518
0
        exit(-1);
519
0
    }
520
3
    Status rest = doris::parse_conf_cache_paths(doris::config::file_cache_path, cache_paths);
521
3
    if (!rest) {
522
0
        throw Exception(
523
0
                Status::FatalError("parse config file cache path failed, path={}, reason={}",
524
0
                                   doris::config::file_cache_path, rest.msg()));
525
0
    }
526
527
3
    auto cache_status = doris::io::FileCacheFactory::instance()->create_file_caches(
528
3
            cache_paths, [](const std::string&, const Status& status) {
529
0
                if (!doris::config::ignore_broken_disk) {
530
0
                    return false;
531
0
                }
532
0
                LOG(WARNING) << "failed to init file cache, err: " << status;
533
0
                return true;
534
0
            });
535
3
    if (!cache_status.ok()) {
536
0
        throw Exception(Status::FatalError("failed to init file cache, err: {}", cache_status));
537
0
    }
538
3
}
539
540
7
Status ExecEnv::init_mem_env() {
541
7
    bool is_percent = false;
542
7
    std::stringstream ss;
543
    // 1. init mem tracker
544
7
    _process_profile = ProcessProfile::create_global_instance();
545
7
    _heap_profiler = HeapProfiler::create_global_instance();
546
7
    init_mem_tracker();
547
7
    thread_context()->thread_mem_tracker_mgr->init();
548
549
7
    if (!BitUtil::IsPowerOf2(config::min_buffer_size)) {
550
0
        ss << "Config min_buffer_size must be a power-of-two: " << config::min_buffer_size;
551
0
        return Status::InternalError(ss.str());
552
0
    }
553
554
7
    _id_manager = new IdManager();
555
7
    _cache_manager = CacheManager::create_global_instance();
556
557
7
    int64_t storage_cache_limit =
558
7
            ParseUtil::parse_mem_spec(config::storage_page_cache_limit, MemInfo::mem_limit(),
559
7
                                      MemInfo::physical_mem(), &is_percent);
560
7
    while (!is_percent && storage_cache_limit > MemInfo::mem_limit() / 2) {
561
0
        storage_cache_limit = storage_cache_limit / 2;
562
0
    }
563
7
    int32_t index_percentage = config::index_page_cache_percentage;
564
7
    int32_t num_shards = config::storage_page_cache_shard_size;
565
7
    if ((num_shards & (num_shards - 1)) != 0) {
566
0
        int old_num_shards = num_shards;
567
0
        num_shards = cast_set<int>(BitUtil::RoundUpToPowerOfTwo(num_shards));
568
0
        LOG(WARNING) << "num_shards should be power of two, but got " << old_num_shards
569
0
                     << ". Rounded up to " << num_shards
570
0
                     << ". Please modify the 'storage_page_cache_shard_size' parameter in your "
571
0
                        "conf file to be a power of two for better performance.";
572
0
    }
573
7
    if (storage_cache_limit < num_shards * 2) {
574
0
        LOG(WARNING) << "storage_cache_limit(" << storage_cache_limit << ") less than num_shards("
575
0
                     << num_shards
576
0
                     << ") * 2, cache capacity will be 0, continuing to use "
577
0
                        "cache will only have negative effects, will be disabled.";
578
0
    }
579
7
    int64_t pk_storage_page_cache_limit =
580
7
            ParseUtil::parse_mem_spec(config::pk_storage_page_cache_limit, MemInfo::mem_limit(),
581
7
                                      MemInfo::physical_mem(), &is_percent);
582
7
    while (!is_percent && pk_storage_page_cache_limit > MemInfo::mem_limit() / 2) {
583
0
        pk_storage_page_cache_limit = storage_cache_limit / 2;
584
0
    }
585
7
    _storage_page_cache = StoragePageCache::create_global_cache(
586
7
            storage_cache_limit, index_percentage, pk_storage_page_cache_limit, num_shards);
587
7
    LOG(INFO) << "Storage page cache memory limit: "
588
7
              << PrettyPrinter::print(storage_cache_limit, TUnit::BYTES)
589
7
              << ", origin config value: " << config::storage_page_cache_limit;
590
591
    // Init ANN index IVF list cache (dedicated cache for IVF on disk)
592
7
    {
593
7
        int64_t ann_cache_limit = ParseUtil::parse_mem_spec(config::ann_index_ivf_list_cache_limit,
594
7
                                                            MemInfo::mem_limit(),
595
7
                                                            MemInfo::physical_mem(), &is_percent);
596
7
        while (!is_percent && ann_cache_limit > MemInfo::mem_limit() / 2) {
597
0
            ann_cache_limit = ann_cache_limit / 2;
598
0
        }
599
7
        _ann_index_ivf_list_cache = AnnIndexIVFListCache::create_global_cache(ann_cache_limit);
600
7
        LOG(INFO) << "ANN index IVF list cache memory limit: "
601
7
                  << PrettyPrinter::print(ann_cache_limit, TUnit::BYTES)
602
7
                  << ", origin config value: " << config::ann_index_ivf_list_cache_limit;
603
7
    }
604
605
    // Init row cache
606
7
    int64_t row_cache_mem_limit =
607
7
            ParseUtil::parse_mem_spec(config::row_cache_mem_limit, MemInfo::mem_limit(),
608
7
                                      MemInfo::physical_mem(), &is_percent);
609
7
    while (!is_percent && row_cache_mem_limit > MemInfo::mem_limit() / 2) {
610
        // Reason same as buffer_pool_limit
611
0
        row_cache_mem_limit = row_cache_mem_limit / 2;
612
0
    }
613
7
    _row_cache = RowCache::create_global_cache(row_cache_mem_limit);
614
7
    LOG(INFO) << "Row cache memory limit: "
615
7
              << PrettyPrinter::print(row_cache_mem_limit, TUnit::BYTES)
616
7
              << ", origin config value: " << config::row_cache_mem_limit;
617
618
7
    uint64_t fd_number = config::min_file_descriptor_number;
619
7
    struct rlimit l;
620
7
    int ret = getrlimit(RLIMIT_NOFILE, &l);
621
7
    if (ret != 0) {
622
0
        LOG(WARNING) << "call getrlimit() failed. errno=" << strerror(errno)
623
0
                     << ", use default configuration instead.";
624
7
    } else {
625
7
        fd_number = static_cast<uint64_t>(l.rlim_cur);
626
7
    }
627
#ifdef __APPLE__
628
    // On macOS, rlim_cur can be RLIM_INFINITY (INT64_MAX), which causes
629
    // fd_number / 100 * percentage to overflow and crash cast_set<uint32_t>.
630
    // Linux kernels cap this via fs.nr_open (default 1M), so only macOS needs this.
631
    {
632
        constexpr uint64_t max_fd = UINT32_MAX >> 2;
633
        if (fd_number > max_fd) {
634
            fd_number = max_fd;
635
        }
636
    }
637
#endif
638
639
7
    int64_t segment_cache_capacity = 0;
640
7
    if (config::is_cloud_mode()) {
641
        // when in cloud mode, segment cache hold no system FD
642
        // thus the FD num limit makes no sense
643
        // cloud mode use FDCache to control FD
644
1
        segment_cache_capacity = UINT32_MAX;
645
6
    } else {
646
        // SegmentLoader caches segments in rowset granularity. So the size of
647
        // opened files will greater than segment_cache_capacity.
648
6
        segment_cache_capacity = config::segment_cache_capacity;
649
6
        int64_t segment_cache_fd_limit = fd_number / 100 * config::segment_cache_fd_percentage;
650
6
        if (segment_cache_capacity < 0 || segment_cache_capacity > segment_cache_fd_limit) {
651
6
            segment_cache_capacity = segment_cache_fd_limit;
652
6
        }
653
6
    }
654
655
7
    int64_t segment_cache_mem_limit =
656
7
            MemInfo::mem_limit() / 100 * config::segment_cache_memory_percentage;
657
658
7
    _segment_loader = new SegmentLoader(segment_cache_mem_limit, segment_cache_capacity);
659
7
    LOG(INFO) << "segment_cache_capacity <= fd_number * 1 / 5, fd_number: " << fd_number
660
7
              << " segment_cache_capacity: " << segment_cache_capacity
661
7
              << " min_segment_cache_mem_limit " << segment_cache_mem_limit;
662
663
7
    size_t block_file_cache_fd_cache_size =
664
7
            std::min((uint64_t)config::file_cache_max_file_reader_cache_size, fd_number / 3);
665
7
    LOG(INFO) << "max file reader cache size is: " << block_file_cache_fd_cache_size
666
7
              << ", resource hard limit is: " << fd_number
667
7
              << ", config file_cache_max_file_reader_cache_size is: "
668
7
              << config::file_cache_max_file_reader_cache_size;
669
7
    config::file_cache_max_file_reader_cache_size = block_file_cache_fd_cache_size;
670
671
7
    _file_meta_cache = new FileMetaCache(config::max_external_file_meta_cache_num);
672
673
7
    _lookup_connection_cache =
674
7
            LookupConnectionCache::create_global_instance(config::lookup_connection_cache_capacity);
675
676
    // use memory limit
677
7
    int64_t inverted_index_cache_limit =
678
7
            ParseUtil::parse_mem_spec(config::inverted_index_searcher_cache_limit,
679
7
                                      MemInfo::mem_limit(), MemInfo::physical_mem(), &is_percent);
680
7
    while (!is_percent && inverted_index_cache_limit > MemInfo::mem_limit() / 2) {
681
        // Reason same as buffer_pool_limit
682
0
        inverted_index_cache_limit = inverted_index_cache_limit / 2;
683
0
    }
684
7
    _inverted_index_searcher_cache =
685
7
            InvertedIndexSearcherCache::create_global_instance(inverted_index_cache_limit, 256);
686
7
    LOG(INFO) << "Inverted index searcher cache memory limit: "
687
7
              << PrettyPrinter::print(inverted_index_cache_limit, TUnit::BYTES)
688
7
              << ", origin config value: " << config::inverted_index_searcher_cache_limit;
689
690
    // use memory limit
691
7
    int64_t inverted_index_query_cache_limit =
692
7
            ParseUtil::parse_mem_spec(config::inverted_index_query_cache_limit,
693
7
                                      MemInfo::mem_limit(), MemInfo::physical_mem(), &is_percent);
694
7
    while (!is_percent && inverted_index_query_cache_limit > MemInfo::mem_limit() / 2) {
695
        // Reason same as buffer_pool_limit
696
0
        inverted_index_query_cache_limit = inverted_index_query_cache_limit / 2;
697
0
    }
698
7
    _inverted_index_query_cache = InvertedIndexQueryCache::create_global_cache(
699
7
            inverted_index_query_cache_limit, config::inverted_index_query_cache_shards);
700
7
    LOG(INFO) << "Inverted index query match cache memory limit: "
701
7
              << PrettyPrinter::print(inverted_index_query_cache_limit, TUnit::BYTES)
702
7
              << ", origin config value: " << config::inverted_index_query_cache_limit;
703
704
    // ANN index topn result cache
705
7
    int64_t ann_topn_cache_limit =
706
7
            ParseUtil::parse_mem_spec(config::ann_index_result_cache_limit, MemInfo::mem_limit(),
707
7
                                      MemInfo::physical_mem(), &is_percent);
708
7
    _ann_index_result_cache =
709
7
            segment_v2::AnnIndexResultCache::create_global_cache(ann_topn_cache_limit);
710
7
    LOG(INFO) << "ANN index topn result cache memory limit: "
711
7
              << PrettyPrinter::print(ann_topn_cache_limit, TUnit::BYTES)
712
7
              << ", origin config value: " << config::ann_index_result_cache_limit;
713
714
    // use memory limit
715
7
    int64_t condition_cache_limit = config::condition_cache_limit * 1024L * 1024L;
716
7
    _condition_cache = ConditionCache::create_global_cache(condition_cache_limit);
717
7
    LOG(INFO) << "Condition cache memory limit: "
718
7
              << PrettyPrinter::print(condition_cache_limit, TUnit::BYTES)
719
7
              << ", origin config value: " << config::condition_cache_limit;
720
721
    // Initialize encoding info resolver
722
7
    _encoding_info_resolver = new segment_v2::EncodingInfoResolver();
723
724
    // init orc memory pool
725
7
    _orc_memory_pool = new doris::ORCMemoryPool();
726
7
    _arrow_memory_pool = new doris::ArrowMemoryPool();
727
728
7
    _query_cache = QueryCache::create_global_cache(config::query_cache_size * 1024L * 1024L);
729
7
    LOG(INFO) << "query cache memory limit: " << config::query_cache_size << "MB";
730
731
    // The default delete bitmap cache is set to 100MB,
732
    // which can be insufficient and cause performance issues when the amount of user data is large.
733
    // To mitigate the problem of an inadequate cache,
734
    // we will take the larger of 0.5% of the total memory and 100MB as the delete bitmap cache size.
735
7
    int64_t delete_bitmap_agg_cache_cache_limit =
736
7
            ParseUtil::parse_mem_spec(config::delete_bitmap_dynamic_agg_cache_limit,
737
7
                                      MemInfo::mem_limit(), MemInfo::physical_mem(), &is_percent);
738
7
    _delete_bitmap_agg_cache = DeleteBitmapAggCache::create_instance(std::max(
739
7
            delete_bitmap_agg_cache_cache_limit, config::delete_bitmap_agg_cache_capacity));
740
741
7
    return Status::OK();
742
7
}
743
744
8
void ExecEnv::init_mem_tracker() {
745
8
    mem_tracker_limiter_pool.resize(MEM_TRACKER_GROUP_NUM,
746
8
                                    TrackerLimiterGroup()); // before all mem tracker init.
747
8
    _s_tracking_memory = true;
748
8
    _orphan_mem_tracker =
749
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "Orphan");
750
8
    _brpc_iobuf_block_memory_tracker =
751
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "IOBufBlockMemory");
752
8
    _segcompaction_mem_tracker =
753
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::COMPACTION, "SegCompaction");
754
8
    _tablets_no_cache_mem_tracker = MemTrackerLimiter::create_shared(
755
8
            MemTrackerLimiter::Type::METADATA, "Tablets(not in TabletSchemaCache)");
756
8
    _segments_no_cache_mem_tracker = MemTrackerLimiter::create_shared(
757
8
            MemTrackerLimiter::Type::METADATA, "Segments(not in SegmentCache)");
758
8
    _rowsets_no_cache_mem_tracker =
759
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::METADATA, "Rowsets");
760
8
    _point_query_executor_mem_tracker =
761
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "PointQueryExecutor");
762
8
    _query_cache_mem_tracker =
763
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::CACHE, "QueryCache");
764
8
    _block_compression_mem_tracker =
765
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "BlockCompression");
766
8
    _rowid_storage_reader_tracker =
767
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "RowIdStorageReader");
768
8
    _subcolumns_tree_tracker =
769
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "SubcolumnsTree");
770
8
    _s3_file_buffer_tracker =
771
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::GLOBAL, "S3FileBuffer");
772
8
    _stream_load_pipe_tracker =
773
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::LOAD, "StreamLoadPipe");
774
8
    _parquet_meta_tracker =
775
8
            MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::METADATA, "ParquetMeta");
776
8
}
777
778
7
Status ExecEnv::_check_deploy_mode() {
779
11
    for (auto _path : _store_paths) {
780
11
        auto deploy_mode_path = fmt::format("{}/{}", _path.path, DEPLOY_MODE_PREFIX);
781
11
        std::string expected_mode = doris::config::is_cloud_mode() ? "cloud" : "local";
782
11
        bool exists = false;
783
11
        RETURN_IF_ERROR(io::global_local_filesystem()->exists(deploy_mode_path, &exists));
784
11
        if (exists) {
785
            // check if is ok
786
8
            io::FileReaderSPtr reader;
787
8
            RETURN_IF_ERROR(io::global_local_filesystem()->open_file(deploy_mode_path, &reader));
788
8
            size_t fsize = reader->size();
789
8
            if (fsize > 0) {
790
8
                std::string actual_mode;
791
8
                actual_mode.resize(fsize, '\0');
792
8
                size_t bytes_read = 0;
793
8
                RETURN_IF_ERROR(reader->read_at(0, {actual_mode.data(), fsize}, &bytes_read));
794
8
                DCHECK_EQ(fsize, bytes_read);
795
8
                if (expected_mode != actual_mode) {
796
0
                    return Status::InternalError(
797
0
                            "You can't switch deploy mode from {} to {}, "
798
0
                            "maybe you need to check be.conf\n",
799
0
                            actual_mode.c_str(), expected_mode.c_str());
800
0
                }
801
8
                LOG(INFO) << "The current deployment mode is " << expected_mode << ".";
802
8
            }
803
8
        } else {
804
3
            io::FileWriterPtr file_writer;
805
3
            RETURN_IF_ERROR(
806
3
                    io::global_local_filesystem()->create_file(deploy_mode_path, &file_writer));
807
3
            RETURN_IF_ERROR(file_writer->append(expected_mode));
808
3
            RETURN_IF_ERROR(file_writer->close());
809
3
            LOG(INFO) << "The file deploy_mode doesn't exist, create it.";
810
3
            auto cluster_id_path = fmt::format("{}/{}", _path.path, CLUSTER_ID_PREFIX);
811
3
            RETURN_IF_ERROR(io::global_local_filesystem()->exists(cluster_id_path, &exists));
812
3
            if (exists) {
813
0
                LOG(WARNING) << "This may be an upgrade from old version,"
814
0
                             << "or the deploy_mode file has been manually deleted";
815
0
            }
816
3
        }
817
11
    }
818
7
    return Status::OK();
819
7
}
820
821
#ifdef BE_TEST
822
void ExecEnv::set_new_load_stream_mgr(std::unique_ptr<NewLoadStreamMgr>&& new_load_stream_mgr) {
823
    this->_new_load_stream_mgr = std::move(new_load_stream_mgr);
824
}
825
826
void ExecEnv::clear_new_load_stream_mgr() {
827
    this->_new_load_stream_mgr.reset();
828
}
829
830
void ExecEnv::set_stream_load_executor(std::unique_ptr<StreamLoadExecutor>&& stream_load_executor) {
831
    this->_stream_load_executor = std::move(stream_load_executor);
832
}
833
834
void ExecEnv::clear_stream_load_executor() {
835
    this->_stream_load_executor.reset();
836
}
837
838
void ExecEnv::set_wal_mgr(std::unique_ptr<WalManager>&& wm) {
839
    this->_wal_manager = std::move(wm);
840
}
841
void ExecEnv::clear_wal_mgr() {
842
    this->_wal_manager.reset();
843
}
844
#endif
845
// TODO(zhiqiang): Need refactor all thread pool. Each thread pool must have a Stop method.
846
// We need to stop all threads before releasing resource.
847
18
void ExecEnv::destroy() {
848
    //Only destroy once after init
849
18
    if (!ready()) {
850
11
        return;
851
11
    }
852
    // Memory barrier to prevent other threads from accessing destructed resources
853
7
    _s_ready = false;
854
855
7
    SAFE_STOP(_wal_manager);
856
7
    _wal_manager.reset();
857
7
    SAFE_STOP(_load_channel_mgr);
858
7
    SAFE_STOP(_broker_mgr);
859
7
    SAFE_STOP(_load_path_mgr);
860
7
    SAFE_STOP(_result_mgr);
861
7
    SAFE_STOP(_group_commit_mgr);
862
    // _routine_load_task_executor should be stopped before _new_load_stream_mgr.
863
7
    SAFE_STOP(_routine_load_task_executor);
864
7
    SAFE_STOP(_stream_load_recorder_manager);
865
    // stop workload scheduler
866
7
    SAFE_STOP(_workload_sched_mgr);
867
    // Stop workload group execution threads before FragmentMgr. Running pipeline tasks can still
868
    // report status through FragmentMgr's async thread pool.
869
7
    SAFE_STOP(_workload_group_manager);
870
871
7
    SAFE_STOP(_external_scan_context_mgr);
872
7
    SAFE_STOP(_fragment_mgr);
873
7
    SAFE_STOP(_runtime_filter_timer_queue);
874
    // NewLoadStreamMgr should be destoried before storage_engine & after fragment_mgr stopped.
875
7
    _load_stream_mgr.reset();
876
7
    _new_load_stream_mgr.reset();
877
7
    _stream_load_executor.reset();
878
7
    _memtable_memory_limiter.reset();
879
7
    _delta_writer_v2_pool.reset();
880
7
    _load_stream_map_pool.reset();
881
    // Workload group schedulers own the query pipeline, scan, and memtable flush queues.
882
    // Release them after fragment/load resources have stopped submitting cleanup work.
883
7
    if (_workload_group_manager) {
884
3
        _workload_group_manager->destroy_schedulers();
885
3
    }
886
7
    SAFE_STOP(_write_cooldown_meta_executors);
887
888
    // _id_manager must be destoried before tablet schema cache
889
7
    SAFE_DELETE(_id_manager);
890
891
    // Stop cluster info background worker before storage engine is destroyed
892
7
    if (config::is_cloud_mode() && _cluster_info) {
893
0
        static_cast<CloudClusterInfo*>(_cluster_info)->stop_bg_worker();
894
0
    }
895
896
    // StorageEngine must be destoried before _cache_manager destory.
897
7
    SAFE_STOP(_storage_engine);
898
7
    _storage_engine.reset();
899
900
7
    SAFE_STOP(_spill_file_mgr);
901
7
    if (_runtime_query_statistics_mgr) {
902
3
        _runtime_query_statistics_mgr->stop_report_thread();
903
3
    }
904
7
    SAFE_SHUTDOWN(_buffered_reader_prefetch_thread_pool);
905
7
    SAFE_SHUTDOWN(_segment_prefetch_thread_pool);
906
7
    SAFE_SHUTDOWN(_s3_file_upload_thread_pool);
907
7
    SAFE_SHUTDOWN(_lazy_release_obj_pool);
908
7
    SAFE_SHUTDOWN(_non_block_close_thread_pool);
909
7
    SAFE_SHUTDOWN(_s3_file_system_thread_pool);
910
7
    SAFE_SHUTDOWN(_peer_race_s3_thread_pool);
911
7
    SAFE_SHUTDOWN(_send_batch_thread_pool);
912
7
    SAFE_SHUTDOWN(_udf_close_workers_thread_pool);
913
7
    SAFE_SHUTDOWN(_send_table_stats_thread_pool);
914
915
7
    SAFE_DELETE(_load_channel_mgr);
916
917
7
    SAFE_DELETE(_inverted_index_query_cache);
918
7
    SAFE_DELETE(_inverted_index_searcher_cache);
919
7
    SAFE_DELETE(_condition_cache);
920
7
    SAFE_DELETE(_encoding_info_resolver);
921
7
    SAFE_DELETE(_lookup_connection_cache);
922
7
    SAFE_DELETE(_segment_loader);
923
7
    SAFE_DELETE(_row_cache);
924
7
    SAFE_DELETE(_query_cache);
925
7
    SAFE_DELETE(_delete_bitmap_agg_cache);
926
927
    // Free resource after threads are stopped.
928
    // Some threads are still running, like threads created by _new_load_stream_mgr ...
929
7
    SAFE_DELETE(_tablet_schema_cache);
930
7
    SAFE_DELETE(_tablet_column_object_pool);
931
932
    // _storage_page_cache must be destoried before _cache_manager
933
7
    SAFE_DELETE(_ann_index_ivf_list_cache);
934
7
    SAFE_DELETE(_storage_page_cache);
935
936
7
    SAFE_DELETE(_small_file_mgr);
937
7
    SAFE_DELETE(_broker_mgr);
938
7
    SAFE_DELETE(_load_path_mgr);
939
7
    SAFE_DELETE(_result_mgr);
940
7
    SAFE_DELETE(_file_meta_cache);
941
7
    SAFE_DELETE(_group_commit_mgr);
942
7
    SAFE_DELETE(_cdc_client_mgr);
943
7
    SAFE_DELETE(_routine_load_task_executor);
944
7
    SAFE_DELETE(_stream_load_recorder_manager);
945
    // _stream_load_executor
946
7
    SAFE_DELETE(_function_client_cache);
947
7
    SAFE_DELETE(_streaming_client_cache);
948
7
    SAFE_DELETE(_internal_client_cache);
949
950
7
    SAFE_DELETE(_bfd_parser);
951
7
    SAFE_DELETE(_result_cache);
952
7
    SAFE_DELETE(_vstream_mgr);
953
    // When _vstream_mgr is deconstructed, it will try call query context's dctor and will
954
    // access spill stream mgr, so spill stream mgr should be deconstructed after data stream manager
955
7
    SAFE_DELETE(_spill_file_mgr);
956
7
    SAFE_DELETE(_fragment_mgr);
957
7
    SAFE_DELETE(_workload_sched_mgr);
958
7
    SAFE_DELETE(_workload_group_manager);
959
7
    SAFE_DELETE(_file_cache_factory);
960
7
    SAFE_DELETE(_runtime_filter_timer_queue);
961
7
    SAFE_DELETE(_dict_factory);
962
    // TODO(zhiqiang): Maybe we should call shutdown before release thread pool?
963
7
    _lazy_release_obj_pool.reset(nullptr);
964
7
    _non_block_close_thread_pool.reset(nullptr);
965
7
    _s3_file_system_thread_pool.reset(nullptr);
966
7
    _peer_race_s3_thread_pool.reset(nullptr);
967
7
    _send_table_stats_thread_pool.reset(nullptr);
968
7
    _buffered_reader_prefetch_thread_pool.reset(nullptr);
969
7
    _segment_prefetch_thread_pool.reset(nullptr);
970
7
    _s3_file_upload_thread_pool.reset(nullptr);
971
7
    _send_batch_thread_pool.reset(nullptr);
972
7
    _udf_close_workers_thread_pool.reset(nullptr);
973
7
    _write_cooldown_meta_executors.reset(nullptr);
974
975
7
    SAFE_DELETE(_broker_client_cache);
976
7
    SAFE_DELETE(_frontend_client_cache);
977
7
    SAFE_DELETE(_backend_client_cache);
978
7
    SAFE_DELETE(_result_queue_mgr);
979
980
7
    SAFE_DELETE(_external_scan_context_mgr);
981
7
    SAFE_DELETE(_user_function_cache);
982
983
    // cache_manager must be destoried after all cache.
984
    // https://github.com/apache/doris/issues/24082#issuecomment-1712544039
985
7
    SAFE_DELETE(_cache_manager);
986
7
    _file_cache_open_fd_cache.reset(nullptr);
987
988
    // _heartbeat_flags must be destoried after staroge engine
989
7
    SAFE_DELETE(_heartbeat_flags);
990
991
    // Master Info is a thrift object, it could be the last one to deconstruct.
992
    // Master info should be deconstruct later than fragment manager, because fragment will
993
    // access cluster_info.backend_id to access some info. If there is a running query and master
994
    // info is deconstructed then BE process will core at coordinator back method in fragment mgr.
995
7
    SAFE_DELETE(_cluster_info);
996
997
    // NOTE: runtime query statistics mgr could be visited by query and daemon thread
998
    // so it should be created before all query begin and deleted after all query and daemon thread stoppped
999
7
    SAFE_DELETE(_runtime_query_statistics_mgr);
1000
1001
7
    SAFE_DELETE(_arrow_memory_pool);
1002
1003
7
    SAFE_DELETE(_orc_memory_pool);
1004
1005
    // dns cache is a global instance and need to be released at last
1006
7
    SAFE_DELETE(_dns_cache);
1007
7
    SAFE_DELETE(_kerberos_ticket_mgr);
1008
7
    SAFE_DELETE(_hdfs_mgr);
1009
    // PackedFileManager is a singleton, just stop its background thread
1010
7
    if (_packed_file_manager) {
1011
3
        _packed_file_manager->stop_background_manager();
1012
3
        _packed_file_manager = nullptr;
1013
3
    }
1014
1015
7
    SAFE_DELETE(_process_profile);
1016
7
    SAFE_DELETE(_heap_profiler);
1017
1018
7
    SAFE_DELETE(_index_policy_mgr);
1019
7
    SAFE_DELETE(_warmup_download_rate_limiter);
1020
1021
7
    _s_tracking_memory = false;
1022
1023
7
    clear_storage_resource();
1024
7
    PythonServerManager::instance().shutdown();
1025
7
    LOG(INFO) << "Doris exec envorinment is destoried.";
1026
7
}
1027
1028
} // namespace doris
1029
1030
namespace doris::config {
1031
namespace {
1032
1033
0
void refresh_ms_rpc_rate_limiters() {
1034
0
    auto* services = ExecEnv::GetInstance()->ms_rpc_rate_limit_services();
1035
0
    if (services != nullptr) {
1036
0
        services->reset_host_level_rate_limiters();
1037
0
    }
1038
0
}
1039
1040
0
void refresh_ms_backpressure_throttle_params() {
1041
0
    auto* services = ExecEnv::GetInstance()->ms_rpc_rate_limit_services();
1042
0
    if (services != nullptr) {
1043
0
        services->update_backpressure_throttle_params(ms_backpressure_upgrade_top_k,
1044
0
                                                      ms_backpressure_throttle_ratio,
1045
0
                                                      ms_rpc_table_qps_limit_floor);
1046
0
    }
1047
0
}
1048
1049
0
void refresh_ms_backpressure_coordinator_params() {
1050
0
    auto* services = ExecEnv::GetInstance()->ms_rpc_rate_limit_services();
1051
0
    if (services != nullptr) {
1052
0
        services->update_backpressure_coordinator_params(ms_backpressure_upgrade_interval_ms,
1053
0
                                                         ms_backpressure_downgrade_interval_ms);
1054
0
    }
1055
0
}
1056
1057
} // namespace
1058
1059
// Callback to update warmup download rate limiter when config changes is registered
1060
DEFINE_ON_UPDATE(file_cache_warmup_download_rate_limit_bytes_per_second,
1061
                 [](int64_t old_val, int64_t new_val) {
1062
                     auto* rate_limiter = ExecEnv::GetInstance()->warmup_download_rate_limiter();
1063
                     if (rate_limiter != nullptr && new_val != old_val) {
1064
                         // Reset rate limiter with new rate limit value
1065
                         // When new_val <= 0, pass 0 to disable rate limiting
1066
                         int64_t rate = new_val > 0 ? new_val : 0;
1067
                         rate_limiter->reset(rate, rate, 0);
1068
                         if (rate > 0) {
1069
                             LOG(INFO) << "Warmup download rate limiter updated from " << old_val
1070
                                       << " to " << new_val << " bytes/s";
1071
                         } else {
1072
                             LOG(INFO) << "Warmup download rate limiter disabled";
1073
                         }
1074
                     }
1075
                 });
1076
1077
DEFINE_ON_UPDATE(ms_rpc_qps_default, [](int32_t old_val, int32_t new_val) {
1078
    if (old_val != new_val) {
1079
        refresh_ms_rpc_rate_limiters();
1080
    }
1081
});
1082
1083
#define DEFINE_MS_RPC_QPS_ON_UPDATE(enum_name, config_suffix, display_name)             \
1084
    DEFINE_ON_UPDATE(ms_rpc_qps_##config_suffix, [](int32_t old_val, int32_t new_val) { \
1085
        if (old_val != new_val) {                                                       \
1086
            refresh_ms_rpc_rate_limiters();                                             \
1087
        }                                                                               \
1088
    });
1089
META_SERVICE_RPC_TYPES(DEFINE_MS_RPC_QPS_ON_UPDATE)
1090
#undef DEFINE_MS_RPC_QPS_ON_UPDATE
1091
1092
DEFINE_ON_UPDATE(ms_backpressure_upgrade_interval_ms, [](int32_t old_val, int32_t new_val) {
1093
    if (old_val != new_val) {
1094
        refresh_ms_backpressure_coordinator_params();
1095
    }
1096
});
1097
DEFINE_ON_UPDATE(ms_backpressure_downgrade_interval_ms, [](int32_t old_val, int32_t new_val) {
1098
    if (old_val != new_val) {
1099
        refresh_ms_backpressure_coordinator_params();
1100
    }
1101
});
1102
DEFINE_ON_UPDATE(ms_backpressure_upgrade_top_k, [](int32_t old_val, int32_t new_val) {
1103
    if (old_val != new_val) {
1104
        refresh_ms_backpressure_throttle_params();
1105
    }
1106
});
1107
DEFINE_ON_UPDATE(ms_backpressure_throttle_ratio, [](double old_val, double new_val) {
1108
    if (old_val != new_val) {
1109
        refresh_ms_backpressure_throttle_params();
1110
    }
1111
});
1112
DEFINE_ON_UPDATE(ms_rpc_table_qps_limit_floor, [](double old_val, double new_val) {
1113
    if (old_val != new_val) {
1114
        refresh_ms_backpressure_throttle_params();
1115
    }
1116
});
1117
} // namespace doris::config