mirror of
https://github.com/MaskRay/ccls.git
synced 2024-11-22 23:55:08 +00:00
83 lines
2.3 KiB
C++
83 lines
2.3 KiB
C++
#include "file_consumer.h"
|
|
|
|
#include "indexer.h"
|
|
#include "platform.h"
|
|
#include "utils.h"
|
|
|
|
namespace {
|
|
|
|
std::string FileName(CXFile file) {
|
|
CXString cx_name = clang_getFileName(file);
|
|
std::string name = clang::ToString(cx_name);
|
|
return NormalizePath(name);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
bool operator==(const CXFileUniqueID& a, const CXFileUniqueID& b) {
|
|
return a.data[0] == b.data[0] && a.data[1] == b.data[1] && a.data[2] == b.data[2];
|
|
}
|
|
|
|
FileConsumer::FileConsumer(SharedState* shared_state) : shared_(shared_state) {}
|
|
|
|
IndexedFile* FileConsumer::TryConsumeFile(CXFile file, bool* is_first_ownership) {
|
|
assert(is_first_ownership);
|
|
|
|
CXFileUniqueID file_id;
|
|
if (clang_getFileUniqueID(file, &file_id) != 0) {
|
|
std::cerr << "Could not get unique file id for " << FileName(file) << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
// Try to find cached local result.
|
|
auto it = local_.find(file_id);
|
|
if (it != local_.end()) {
|
|
*is_first_ownership = false;
|
|
return it->second.get();
|
|
}
|
|
|
|
std::string file_name = FileName(file);
|
|
|
|
// No result in local; we need to query global.
|
|
bool did_insert = false;
|
|
{
|
|
std::lock_guard<std::mutex> lock(shared_->mutex);
|
|
did_insert = shared_->files.insert(file_name).second;
|
|
}
|
|
*is_first_ownership = did_insert;
|
|
local_[file_id] = did_insert ? MakeUnique<IndexedFile>(file_name) : nullptr;
|
|
return local_[file_id].get();
|
|
}
|
|
|
|
IndexedFile* FileConsumer::ForceLocal(CXFile file) {
|
|
// Try to fetch the file using the normal system, which will insert the file
|
|
// usage into global storage.
|
|
{
|
|
bool is_first;
|
|
IndexedFile* cache = TryConsumeFile(file, &is_first);
|
|
if (cache)
|
|
return cache;
|
|
}
|
|
|
|
// It's already been taken before, just create a local copy.
|
|
CXFileUniqueID file_id;
|
|
if (clang_getFileUniqueID(file, &file_id) != 0) {
|
|
std::cerr << "Could not get unique file id for " << FileName(file) << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
auto it = local_.find(file_id);
|
|
if (it == local_.end() || !it->second)
|
|
local_[file_id] = MakeUnique<IndexedFile>(FileName(file));
|
|
assert(local_.find(file_id) != local_.end());
|
|
return local_[file_id].get();
|
|
}
|
|
|
|
std::vector<std::unique_ptr<IndexedFile>> FileConsumer::TakeLocalState() {
|
|
std::vector<std::unique_ptr<IndexedFile>> result;
|
|
for (auto& entry : local_) {
|
|
if (entry.second)
|
|
result.push_back(std::move(entry.second));
|
|
}
|
|
return result;
|
|
} |