mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
97 lines
2.9 KiB
C++
97 lines
2.9 KiB
C++
/*
|
|
This file is part of TON Blockchain Library.
|
|
|
|
TON Blockchain Library is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TON Blockchain Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Copyright 2017-2020 Telegram Systems LLP
|
|
*/
|
|
#pragma once
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/Slice.h"
|
|
|
|
#include <map>
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
|
|
namespace td {
|
|
|
|
// TODO template KeyT
|
|
class Hints {
|
|
using KeyT = int64;
|
|
using RatingT = int64;
|
|
|
|
public:
|
|
void add(KeyT key, Slice name);
|
|
|
|
void remove(KeyT key) {
|
|
add(key, "");
|
|
}
|
|
|
|
void set_rating(KeyT key, RatingT rating);
|
|
|
|
std::pair<size_t, vector<KeyT>> search(
|
|
Slice query, int32 limit,
|
|
bool return_all_for_empty_query = false) const; // TODO sort by name instead of sort by rating
|
|
|
|
bool has_key(KeyT key) const;
|
|
|
|
string key_to_string(KeyT key) const;
|
|
|
|
std::pair<size_t, vector<KeyT>> search_empty(int32 limit) const; // == search("", limit, true)
|
|
|
|
size_t size() const;
|
|
|
|
private:
|
|
std::map<string, vector<KeyT>> word_to_keys_;
|
|
std::map<string, vector<KeyT>> translit_word_to_keys_;
|
|
std::unordered_map<KeyT, string> key_to_name_;
|
|
std::unordered_map<KeyT, RatingT> key_to_rating_;
|
|
|
|
static void add_word(const string &word, KeyT key, std::map<string, vector<KeyT>> &word_to_keys);
|
|
static void delete_word(const string &word, KeyT key, std::map<string, vector<KeyT>> &word_to_keys);
|
|
|
|
static vector<string> fix_words(vector<string> words);
|
|
|
|
static vector<string> get_words(Slice name, bool is_search);
|
|
|
|
static void add_search_results(vector<KeyT> &results, const string &word,
|
|
const std::map<string, vector<KeyT>> &word_to_keys);
|
|
|
|
vector<KeyT> search_word(const string &word) const;
|
|
|
|
class CompareByRating {
|
|
const std::unordered_map<KeyT, RatingT> &key_to_rating_;
|
|
|
|
RatingT get_rating(const KeyT &key) const {
|
|
auto it = key_to_rating_.find(key);
|
|
if (it == key_to_rating_.end()) {
|
|
return RatingT();
|
|
}
|
|
return it->second;
|
|
}
|
|
|
|
public:
|
|
explicit CompareByRating(const std::unordered_map<KeyT, RatingT> &key_to_rating) : key_to_rating_(key_to_rating) {
|
|
}
|
|
|
|
bool operator()(const KeyT &lhs, const KeyT &rhs) const {
|
|
auto lhs_rating = get_rating(lhs);
|
|
auto rhs_rating = get_rating(rhs);
|
|
return lhs_rating < rhs_rating || (lhs_rating == rhs_rating && lhs < rhs);
|
|
}
|
|
};
|
|
};
|
|
|
|
} // namespace td
|