Commit c220e0e1 authored by groot's avatar groot
Browse files

refine code

parent 5380e0c9

core/src/db/IndexFailedChecker.cpp

+2 −2
Original line number Diff line number Diff line
@@ -38,7 +38,7 @@ IndexFailedChecker::GetFailedIndexFileOfTable(const std::string& table_id, std:: 
    std::lock_guard<std::mutex> lck(mutex_);

    auto iter = index_failed_files_.find(table_id);

    if (iter != index_failed_files_.end()) {

        meta::File2RefCount& failed_map = iter->second;

        File2RefCount& failed_map = iter->second;

        for (auto it_file = failed_map.begin(); it_file != failed_map.end(); ++it_file) {

            failed_files.push_back(it_file->first);

        }
@@ -53,7 +53,7 @@ IndexFailedChecker::MarkFailedIndexFile(const meta::TableFileSchema& file) { 


    auto iter = index_failed_files_.find(file.table_id_);

    if (iter == index_failed_files_.end()) {

        meta::File2RefCount failed_files;

        File2RefCount failed_files;

        failed_files.insert(std::make_pair(file.file_id_, 1));

        index_failed_files_.insert(std::make_pair(file.table_id_, failed_files));

    } else {

core/src/db/IndexFailedChecker.h

+2 −1
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ 


#pragma once



#include "db/Types.h"

#include "meta/Meta.h"

#include "utils/Status.h"
@@ -47,7 +48,7 @@ class IndexFailedChecker { 


 private:

    std::mutex mutex_;

    meta::Table2Files index_failed_files_;  // table id mapping to (file id mapping to failed times)

    Table2Files index_failed_files_;  // table id mapping to (file id mapping to failed times)

};



}  // namespace engine

core/src/db/OngoingFileChecker.cpp

+1 −1
Original line number Diff line number Diff line
@@ -82,7 +82,7 @@ OngoingFileChecker::MarkOngoingFileNoLock(const meta::TableFileSchema& table_fil 


    auto iter = ongoing_files_.find(table_file.table_id_);

    if (iter == ongoing_files_.end()) {

        meta::File2RefCount files_refcount;

        File2RefCount files_refcount;

        files_refcount.insert(std::make_pair(table_file.file_id_, 1));

        ongoing_files_.insert(std::make_pair(table_file.table_id_, files_refcount));

    } else {

core/src/db/OngoingFileChecker.h

+2 −1
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ 


#pragma once



#include "db/Types.h"

#include "meta/Meta.h"

#include "utils/Status.h"
@@ -54,7 +55,7 @@ class OngoingFileChecker : public meta::Meta::CleanUpFilter { 


 private:

    std::mutex mutex_;

    meta::Table2Files ongoing_files_;  // table id mapping to (file id mapping to ongoing ref-count)

    Table2Files ongoing_files_;  // table id mapping to (file id mapping to ongoing ref-count)

};



}  // namespace engine

core/src/db/Types.h

+6 −0
Original line number Diff line number Diff line
@@ -21,6 +21,9 @@ 


#include <faiss/Index.h>

#include <stdint.h>

#include <map>

#include <set>

#include <string>

#include <utility>

#include <vector>
@@ -40,5 +43,8 @@ struct TableIndex { 
    int32_t metric_type_ = (int)MetricType::L2;

};



using File2RefCount = std::map<std::string, int64_t>;

using Table2Files = std::map<std::string, File2RefCount>;



}  // namespace engine

}  // namespace milvus