提交 c6d38f0a 编写于 作者: P peng.xu

feat(db): impl build index


Former-commit-id: 3fd41e5fa861957e108a4d077a152dc0c30f2691
上级 710ae19d
...@@ -408,10 +408,10 @@ void DBImpl::BackgroundCompaction(std::set<std::string> table_ids) { ...@@ -408,10 +408,10 @@ void DBImpl::BackgroundCompaction(std::set<std::string> table_ids) {
meta_ptr_->CleanUpFilesWithTTL(ttl); meta_ptr_->CleanUpFilesWithTTL(ttl);
} }
void DBImpl::StartBuildIndexTask() { void DBImpl::StartBuildIndexTask(bool force) {
static uint64_t index_clock_tick = 0; static uint64_t index_clock_tick = 0;
index_clock_tick++; index_clock_tick++;
if(index_clock_tick%INDEX_ACTION_INTERVAL != 0) { if(!force && (index_clock_tick%INDEX_ACTION_INTERVAL != 0)) {
return; return;
} }
...@@ -431,8 +431,20 @@ void DBImpl::StartBuildIndexTask() { ...@@ -431,8 +431,20 @@ void DBImpl::StartBuildIndexTask() {
} }
Status DBImpl::BuildIndex(const std::string& table_id) { Status DBImpl::BuildIndex(const std::string& table_id) {
bool has = false;
meta_ptr_->HasNonIndexFiles(table_id, has);
int times = 1;
while (has) {
ENGINE_LOG_DEBUG << "Non index files detected! Will build index " << times;
meta_ptr_->UpdateTableFilesToIndex(table_id); meta_ptr_->UpdateTableFilesToIndex(table_id);
return BuildIndexByTable(table_id); StartBuildIndexTask(true);
std::this_thread::sleep_for(std::chrono::milliseconds(std::min(10*1000, times*100)));
meta_ptr_->HasNonIndexFiles(table_id, has);
times++;
}
return Status::OK();
/* return BuildIndexByTable(table_id); */
} }
Status DBImpl::BuildIndex(const meta::TableFileSchema& file) { Status DBImpl::BuildIndex(const meta::TableFileSchema& file) {
......
...@@ -109,7 +109,7 @@ class DBImpl : public DB { ...@@ -109,7 +109,7 @@ class DBImpl : public DB {
Status BackgroundMergeFiles(const std::string &table_id); Status BackgroundMergeFiles(const std::string &table_id);
void BackgroundCompaction(std::set<std::string> table_ids); void BackgroundCompaction(std::set<std::string> table_ids);
void StartBuildIndexTask(); void StartBuildIndexTask(bool force=false);
void BackgroundBuildIndex(); void BackgroundBuildIndex();
Status Status
......
...@@ -316,6 +316,28 @@ Status DBMetaImpl::DescribeTable(TableSchema &table_schema) { ...@@ -316,6 +316,28 @@ Status DBMetaImpl::DescribeTable(TableSchema &table_schema) {
return Status::OK(); return Status::OK();
} }
Status DBMetaImpl::HasNonIndexFiles(const std::string& table_id, bool& has) {
has = false;
try {
auto selected = ConnectorPtr->select(columns(&TableFileSchema::id_),
where((c(&TableFileSchema::file_type_) == (int) TableFileSchema::RAW
or
c(&TableFileSchema::file_type_) == (int) TableFileSchema::TO_INDEX)
and c(&TableFileSchema::table_id_) == table_id
));
if (selected.size() >= 1) {
has = true;
} else {
has = false;
}
} catch (std::exception &e) {
return HandleException("Encounter exception when check non index files", e);
}
return Status::OK();
}
Status DBMetaImpl::HasTable(const std::string &table_id, bool &has_or_not) { Status DBMetaImpl::HasTable(const std::string &table_id, bool &has_or_not) {
has_or_not = false; has_or_not = false;
......
...@@ -35,6 +35,8 @@ public: ...@@ -35,6 +35,8 @@ public:
const std::vector<size_t>& ids, const std::vector<size_t>& ids,
TableFilesSchema& table_files) override; TableFilesSchema& table_files) override;
virtual Status HasNonIndexFiles(const std::string& table_id, bool& has) override;
virtual Status UpdateTableFilesToIndex(const std::string& table_id) override; virtual Status UpdateTableFilesToIndex(const std::string& table_id) override;
virtual Status UpdateTableFile(TableFileSchema& file_schema) override; virtual Status UpdateTableFile(TableFileSchema& file_schema) override;
......
...@@ -58,6 +58,8 @@ public: ...@@ -58,6 +58,8 @@ public:
virtual Status FilesToIndex(TableFilesSchema&) = 0; virtual Status FilesToIndex(TableFilesSchema&) = 0;
virtual Status HasNonIndexFiles(const std::string& table_id, bool& has) = 0;
virtual Status CleanUp() = 0; virtual Status CleanUp() = 0;
virtual Status CleanUpFilesWithTTL(uint16_t) = 0; virtual Status CleanUpFilesWithTTL(uint16_t) = 0;
......
...@@ -436,6 +436,11 @@ namespace meta { ...@@ -436,6 +436,11 @@ namespace meta {
return Status::OK(); return Status::OK();
} }
Status MySQLMetaImpl::HasNonIndexFiles(const std::string& table_id, bool& has) {
// TODO
return Status::OK();
}
Status MySQLMetaImpl::DeleteTable(const std::string& table_id) { Status MySQLMetaImpl::DeleteTable(const std::string& table_id) {
// std::lock_guard<std::recursive_mutex> lock(mysql_mutex); // std::lock_guard<std::recursive_mutex> lock(mysql_mutex);
......
...@@ -40,6 +40,8 @@ namespace meta { ...@@ -40,6 +40,8 @@ namespace meta {
const std::vector<size_t>& ids, const std::vector<size_t>& ids,
TableFilesSchema& table_files) override; TableFilesSchema& table_files) override;
virtual Status HasNonIndexFiles(const std::string& table_id, bool& has) override;
virtual Status UpdateTableFile(TableFileSchema& file_schema) override; virtual Status UpdateTableFile(TableFileSchema& file_schema) override;
virtual Status UpdateTableFilesToIndex(const std::string& table_id) override; virtual Status UpdateTableFilesToIndex(const std::string& table_id) override;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册