// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements.  See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License.  You may obtain a copy of the License at

//

//   http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied.  See the License for the

// specific language governing permissions and limitations

// under the License.

#pragma once

#include <gen_cpp/BackendService.h>

#include <gen_cpp/Status_types.h>

#include <memory>

#include <string>

#include <vector>

#include "agent/agent_server.h"

#include "agent/topic_subscriber.h"

#include "common/status.h"

#include "load/stream_load/stream_load_recorder.h"

namespace doris {

class StorageEngine;

class ExecEnv;

class TAgentResult;

class TAgentTaskRequest;

class TAgentPublishRequest;

class TStreamLoadRecordResult;

class TDiskTrashInfo;

class TCheckStorageFormatResult;

class TRoutineLoadTask;

class TScanBatchResult;

class TScanCloseParams;

class TScanCloseResult;

class TScanNextBatchParams;

class TScanOpenParams;

class TScanOpenResult;

class TSnapshotRequest;

class TStatus;

class TTabletStatResult;

class TUniqueId;

class TIngestBinlogRequest;

class TIngestBinlogResult;

class ThreadPool;

// This class just forward rpc for actual handler

// make this class because we can bind multiple service on single point

class BaseBackendService : public BackendServiceIf {

public:

    BaseBackendService(ExecEnv* exec_env);

    ~BaseBackendService() override;

    // Start runtime workers that the thrift server depends on (agent workers,

    // ingest-binlog thread pool, etc.). Must be called before constructing the

    // thrift server. The name makes the side effects explicit: this is not a

    // lightweight preparation hook.

    virtual Status start_thrift_dependencies() = 0;

    // Agent service

    void submit_tasks(TAgentResult& return_value,

                      const std::vector<TAgentTaskRequest>& tasks) override {

        _agent_server->submit_tasks(return_value, tasks);

    }

    void publish_cluster_state(TAgentResult& result, const TAgentPublishRequest& request) override {

        _agent_server->publish_cluster_state(result, request);

    }

    void publish_topic_info(TPublishTopicResult& result,

                            const TPublishTopicRequest& topic_request) override {

        _agent_server->get_topic_subscriber()->handle_topic_info(topic_request);

    }

    void submit_routine_load_task(TStatus& t_status,

                                  const std::vector<TRoutineLoadTask>& tasks) override;

    // used for external service, open means start the scan procedure

    void open_scanner(TScanOpenResult& result_, const TScanOpenParams& params) override;

    // used for external service, external use getNext to fetch data batch after batch until eos = true

    void get_next(TScanBatchResult& result_, const TScanNextBatchParams& params) override;

    // used for external service, close some context and release resource related with this context

    void close_scanner(TScanCloseResult& result_, const TScanCloseParams& params) override;

    ////////////////////////////////////////////////////////////////////////////

    // begin local backend functions

    ////////////////////////////////////////////////////////////////////////////

    void get_tablet_stat(TTabletStatResult& result) override;

    int64_t get_trash_used_capacity() override;

    void get_stream_load_record(TStreamLoadRecordResult& result,

                                int64_t last_stream_record_time) override;

    void get_disk_trash_used_capacity(std::vector<TDiskTrashInfo>& diskTrashInfos) override;

    void make_snapshot(TAgentResult& return_value,

                       const TSnapshotRequest& snapshot_request) override;

    void release_snapshot(TAgentResult& return_value, const std::string& snapshot_path) override;

    void check_storage_format(TCheckStorageFormatResult& result) override;

    void ingest_binlog(TIngestBinlogResult& result, const TIngestBinlogRequest& request) override;

    void query_ingest_binlog(TQueryIngestBinlogResult& result,

                             const TQueryIngestBinlogRequest& request) override;

    void get_realtime_exec_status(TGetRealtimeExecStatusResponse& response,

                                  const TGetRealtimeExecStatusRequest& request) override;

    void get_dictionary_status(TDictionaryStatusList& result,

                               const std::vector<int64_t>& dictionary_id) override;

    void test_storage_connectivity(TTestStorageConnectivityResponse& response,

                                   const TTestStorageConnectivityRequest& request) override;

    void get_python_envs(std::vector<TPythonEnvInfo>& result) override;

    void get_python_packages(std::vector<TPythonPackageInfo>& result,

                             const std::string& python_version) override;

    ////////////////////////////////////////////////////////////////////////////

    // begin cloud backend functions

    ////////////////////////////////////////////////////////////////////////////

    void warm_up_cache_async(TWarmUpCacheAsyncResponse& response,

                             const TWarmUpCacheAsyncRequest& request) override;

    void check_warm_up_cache_async(TCheckWarmUpCacheAsyncResponse& response,

                                   const TCheckWarmUpCacheAsyncRequest& request) override;

    // If another cluster load, FE need to notify the cluster to sync the load data

    void sync_load_for_tablets(TSyncLoadForTabletsResponse& response,

                               const TSyncLoadForTabletsRequest& request) override;

    void get_top_n_hot_partitions(TGetTopNHotPartitionsResponse& response,

                                  const TGetTopNHotPartitionsRequest& request) override;

    void warm_up_tablets(TWarmUpTabletsResponse& response,

                         const TWarmUpTabletsRequest& request) override;

    void stop_works() { _agent_server->stop_report_workers(); }

protected:

    void get_stream_load_record(TStreamLoadRecordResult& result, int64_t last_stream_record_time,

                                std::shared_ptr<StreamLoadRecorder> stream_load_recorder);

    ExecEnv* _exec_env = nullptr;

    std::unique_ptr<AgentServer> _agent_server;

    std::unique_ptr<ThreadPool> _ingest_binlog_workers;

};

// `StorageEngine` mixin for `BaseBackendService`

class BackendService final : public BaseBackendService {

public:

    BackendService(StorageEngine& engine, ExecEnv* exec_env);

    ~BackendService() override;

    Status start_thrift_dependencies() override;

    void get_tablet_stat(TTabletStatResult& result) override;

    int64_t get_trash_used_capacity() override;

    void get_stream_load_record(TStreamLoadRecordResult& result,

                                int64_t last_stream_record_time) override;

    void get_disk_trash_used_capacity(std::vector<TDiskTrashInfo>& diskTrashInfos) override;

    void make_snapshot(TAgentResult& return_value,

                       const TSnapshotRequest& snapshot_request) override;

    void release_snapshot(TAgentResult& return_value, const std::string& snapshot_path) override;

    void check_storage_format(TCheckStorageFormatResult& result) override;

    void ingest_binlog(TIngestBinlogResult& result, const TIngestBinlogRequest& request) override;

    void query_ingest_binlog(TQueryIngestBinlogResult& result,

                             const TQueryIngestBinlogRequest& request) override;

private:

    StorageEngine& _engine;

};

} // namespace doris