Make ModuleCache::Get to return instantiated ModuleSP instance so already created in-memory instance can be returned instead of creating a new one.

http://reviews.llvm.org/D8270

llvm-svn: 232075
This commit is contained in:
Oleksiy Vyalov 2015-03-12 18:18:03 +00:00
parent 285dd51b7b
commit eda270ee99
4 changed files with 47 additions and 49 deletions

View File

@ -1134,8 +1134,8 @@ class ModuleCache;
const FileSpec& dst_file_spec);
bool
GetFileFromLocalCache (const ModuleSpec& module_spec,
FileSpec &cached_file_spec);
GetModuleFromLocalCache (const ModuleSpec& module_spec,
lldb::ModuleSP &module_sp);
FileSpec GetModuleCacheRoot ();

View File

@ -1756,22 +1756,12 @@ bool
Platform::GetCachedSharedModule (const ModuleSpec &module_spec,
lldb::ModuleSP &module_sp)
{
FileSpec cached_file_spec;
if (m_module_cache && GetFileFromLocalCache (module_spec, cached_file_spec))
{
auto cached_module_spec (module_spec);
cached_module_spec.GetFileSpec () = cached_file_spec;
cached_module_spec.GetPlatformFileSpec () = module_spec.GetFileSpec ();
module_sp.reset (new Module (cached_module_spec));
return true;
}
return false;
return (m_module_cache && GetModuleFromLocalCache (module_spec, module_sp));
}
bool
Platform::GetFileFromLocalCache (const ModuleSpec& module_spec,
FileSpec &cached_file_spec)
Platform::GetModuleFromLocalCache (const ModuleSpec& module_spec,
lldb::ModuleSP &module_sp)
{
Log *log = GetLogIfAnyCategoriesSet (LIBLLDB_LOG_PLATFORM);
@ -1783,9 +1773,8 @@ Platform::GetFileFromLocalCache (const ModuleSpec& module_spec,
// Check local cache for a module.
auto error = m_module_cache->Get (GetModuleCacheRoot (),
GetHostname (),
resolved_module_spec.GetUUID (),
resolved_module_spec.GetFileSpec (),
cached_file_spec);
resolved_module_spec,
module_sp);
if (error.Success ())
return true;
@ -1825,8 +1814,7 @@ Platform::GetFileFromLocalCache (const ModuleSpec& module_spec,
// Put downloaded file into local module cache.
error = m_module_cache->Put (GetModuleCacheRoot (),
GetHostname (),
resolved_module_spec.GetUUID (),
resolved_module_spec.GetFileSpec (),
resolved_module_spec,
tmp_download_file_spec);
if (error.Fail ())
{
@ -1839,9 +1827,8 @@ Platform::GetFileFromLocalCache (const ModuleSpec& module_spec,
error = m_module_cache->Get (GetModuleCacheRoot (),
GetHostname (),
resolved_module_spec.GetUUID (),
resolved_module_spec.GetFileSpec (),
cached_file_spec);
resolved_module_spec,
module_sp);
return error.Success ();
}

View File

@ -10,6 +10,7 @@
#include "ModuleCache.h"
#include "lldb/Core/Module.h"
#include "lldb/Core/ModuleSpec.h"
#include "lldb/Host/FileSystem.h"
#include "llvm/Support/FileSystem.h"
@ -52,16 +53,15 @@ MakeDirectory (const FileSpec &dir_path)
Error
ModuleCache::Put (const FileSpec &root_dir_spec,
const char *hostname,
const UUID &uuid,
const FileSpec &platform_module_spec,
const ModuleSpec &module_spec,
const FileSpec &tmp_file)
{
const auto module_spec_dir = GetModuleDirectory (root_dir_spec, uuid);
const auto module_spec_dir = GetModuleDirectory (root_dir_spec, module_spec.GetUUID ());
auto error = MakeDirectory (module_spec_dir);
if (error.Fail ())
return error;
const auto module_file_path = JoinPath (module_spec_dir, platform_module_spec.GetFilename ().AsCString ());
const auto module_file_path = JoinPath (module_spec_dir, module_spec.GetFileSpec ().GetFilename ().AsCString ());
const auto tmp_file_path = tmp_file.GetPath ();
const auto err_code = llvm::sys::fs::copy_file (tmp_file_path.c_str (), module_file_path.GetPath ().c_str ());
@ -74,36 +74,45 @@ ModuleCache::Put (const FileSpec &root_dir_spec,
}
// Create sysroot link to a module.
const auto sysroot_module_path_spec = GetHostSysRootModulePath (root_dir_spec, hostname, platform_module_spec);
const auto sysroot_module_path_spec = GetHostSysRootModulePath (root_dir_spec, hostname, module_spec.GetFileSpec ());
return CreateHostSysRootModuleSymLink (sysroot_module_path_spec, module_file_path);
}
Error
ModuleCache::Get (const FileSpec &root_dir_spec,
const char *hostname,
const UUID &uuid,
const FileSpec &platform_module_spec,
FileSpec &cached_module_spec)
const ModuleSpec &module_spec,
ModuleSP &cached_module_sp)
{
cached_module_spec.Clear ();
const auto module_spec_dir = GetModuleDirectory (root_dir_spec, uuid);
const auto module_file_path = JoinPath (module_spec_dir, platform_module_spec.GetFilename ().AsCString ());
Error error;
if (!module_file_path.Exists ())
const auto find_it = m_loaded_modules.find (module_spec.GetUUID ().GetAsString());
if (find_it != m_loaded_modules.end ())
{
error.SetErrorStringWithFormat ("module %s not found", module_file_path.GetPath ().c_str ());
return error;
cached_module_sp = (*find_it).second.lock ();
if (cached_module_sp)
return Error ();
m_loaded_modules.erase (find_it);
}
cached_module_spec = module_file_path;
const auto module_spec_dir = GetModuleDirectory (root_dir_spec, module_spec.GetUUID ());
const auto module_file_path = JoinPath (module_spec_dir, module_spec.GetFileSpec ().GetFilename ().AsCString ());
if (!module_file_path.Exists ())
return Error ("module %s not found", module_file_path.GetPath ().c_str ());
// We may have already cached module but downloaded from an another host - in this case let's create a symlink to it.
const auto sysroot_module_path_spec = GetHostSysRootModulePath (root_dir_spec, hostname, platform_module_spec);
const auto sysroot_module_path_spec = GetHostSysRootModulePath (root_dir_spec, hostname, module_spec.GetFileSpec ());
if (!sysroot_module_path_spec.Exists ())
CreateHostSysRootModuleSymLink (sysroot_module_path_spec, cached_module_spec);
CreateHostSysRootModuleSymLink (sysroot_module_path_spec, module_spec.GetFileSpec ());
return error;
auto cached_module_spec (module_spec);
cached_module_spec.GetUUID ().Clear (); // Clear UUID since it may contain md5 content hash instead of real UUID.
cached_module_spec.GetFileSpec () = module_file_path;
cached_module_spec.GetPlatformFileSpec () = module_spec.GetFileSpec ();
cached_module_sp.reset (new Module (cached_module_spec));
m_loaded_modules.insert (std::make_pair (module_spec.GetUUID ().GetAsString (), cached_module_sp));
return Error ();
}
FileSpec

View File

@ -17,9 +17,11 @@
#include "lldb/Host/FileSpec.h"
#include <string>
#include <unordered_map>
namespace lldb_private {
class Module;
class UUID;
//----------------------------------------------------------------------
@ -45,16 +47,14 @@ public:
Error
Put (const FileSpec &root_dir_spec,
const char *hostname,
const UUID &uuid,
const FileSpec &platform_module_spec,
const ModuleSpec &module_spec,
const FileSpec &tmp_file);
Error
Get (const FileSpec &root_dir_spec,
const char *hostname,
const UUID &uuid,
const FileSpec &platform_module_spec,
FileSpec &cached_module_spec);
const ModuleSpec &module_spec,
lldb::ModuleSP &cached_module_sp);
private:
static FileSpec
@ -65,6 +65,8 @@ private:
static Error
CreateHostSysRootModuleSymLink (const FileSpec &sysroot_module_path_spec, const FileSpec &module_file_path);
std::unordered_map<std::string, lldb::ModuleWP> m_loaded_modules;
};
} // namespace lldb_private