mirror of
https://github.com/MaskRay/ccls.git
synced 2024-11-26 09:31:59 +00:00
96 lines
2.7 KiB
C++
96 lines
2.7 KiB
C++
#include "file_consumer.h"
|
|
|
|
#include "clang_utils.h"
|
|
#include "indexer.h"
|
|
#include "platform.h"
|
|
#include "utils.h"
|
|
|
|
#include <loguru.hpp>
|
|
|
|
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];
|
|
}
|
|
|
|
bool FileConsumerSharedState::Mark(const std::string& file) {
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
return files.insert(file).second;
|
|
}
|
|
|
|
void FileConsumerSharedState::Reset(const std::string& file) {
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
auto it = files.find(file);
|
|
if (it != files.end())
|
|
files.erase(it);
|
|
}
|
|
|
|
FileConsumer::FileConsumer(FileConsumerSharedState* shared_state,
|
|
const std::string& parse_file)
|
|
: shared_(shared_state), parse_file_(parse_file) {}
|
|
|
|
IndexFile* FileConsumer::TryConsumeFile(CXFile file, bool* is_first_ownership) {
|
|
assert(is_first_ownership);
|
|
|
|
CXFileUniqueID file_id;
|
|
if (clang_getFileUniqueID(file, &file_id) != 0) {
|
|
EmitError(file);
|
|
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 = shared_->Mark(file_name);
|
|
*is_first_ownership = did_insert;
|
|
local_[file_id] = did_insert ? MakeUnique<IndexFile>(file_name) : nullptr;
|
|
return local_[file_id].get();
|
|
}
|
|
|
|
IndexFile* 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;
|
|
IndexFile* 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) {
|
|
EmitError(file);
|
|
return nullptr;
|
|
}
|
|
|
|
auto it = local_.find(file_id);
|
|
if (it == local_.end() || !it->second)
|
|
local_[file_id] = MakeUnique<IndexFile>(FileName(file));
|
|
assert(local_.find(file_id) != local_.end());
|
|
return local_[file_id].get();
|
|
}
|
|
|
|
std::vector<std::unique_ptr<IndexFile>> FileConsumer::TakeLocalState() {
|
|
std::vector<std::unique_ptr<IndexFile>> result;
|
|
for (auto& entry : local_) {
|
|
if (entry.second)
|
|
result.push_back(std::move(entry.second));
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void FileConsumer::EmitError(CXFile file) const {
|
|
std::string file_name = ToString(clang_getFileName(file));
|
|
// TODO: Investigate this more, why can we get an empty file name?
|
|
if (!file_name.empty()) {
|
|
LOG_S(ERROR) << "Could not get unique file id for " << file_name
|
|
<< " when parsing " << parse_file_;
|
|
}
|
|
} |