// 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 <cstdint>

#include <string>

#include <utility>

#include "runtime/exec_env.h"

#include "runtime/memory/lru_cache_policy.h"

#include "storage/tablet/tablet_fwd.h"

namespace doris {

class OlapTableSchemaParam;

struct OlapTableIndexSchema;

class TabletSchemaCache : public LRUCachePolicy {

public:

    using LRUCachePolicy::insert;

    TabletSchemaCache(size_t capacity)

            : LRUCachePolicy(CachePolicy::CacheType::TABLET_SCHEMA_CACHE, capacity,

                             LRUCacheType::NUMBER, config::tablet_schema_cache_recycle_interval,

                             /*num_shards*/ 32,

                             /*element_count_capacity*/ 0, /*enable_prune*/ true,

                             /*is_lru_k*/ false) {}

    static TabletSchemaCache* create_global_schema_cache(size_t capacity) {

        auto* res = new TabletSchemaCache(capacity);

        return res;

    }

    static TabletSchemaCache* instance() {

        return ExecEnv::GetInstance()->get_tablet_schema_cache();

    }

    std::pair<Cache::Handle*, TabletSchemaSPtr> insert(const std::string& key);

    std::pair<Cache::Handle*, TabletSchemaSPtr> insert(const std::string& key,

                                                       TabletSchemaSPtr tablet_schema);

    std::pair<Cache::Handle*, TabletSchemaSPtr> lookup_schema(const std::string& key);

    static std::string build_load_schema_cache_key(int64_t index_id,

                                                   const OlapTableSchemaParam* table_schema_param,

                                                   const TabletSchema& ori_tablet_schema,

                                                   const OlapTableIndexSchema* index_schema);

    void release(Cache::Handle*);

private:

    class CacheValue : public LRUCacheValueBase {

    public:

        ~CacheValue() override;

        TabletSchemaSPtr tablet_schema;

    };

};

} // namespace doris