OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "base/threading/thread_id_name_manager.h" |
| 6 |
| 7 #include <stdlib.h> |
| 8 #include <string.h> |
| 9 |
| 10 #include "base/logging.h" |
| 11 #include "base/memory/singleton.h" |
| 12 #include "base/string_util.h" |
| 13 |
| 14 namespace base { |
| 15 namespace { |
| 16 |
| 17 static const char kDefaultName[] = ""; |
| 18 static std::string* g_default_name; |
| 19 |
| 20 typedef std::map<PlatformThreadId, std::string*>::iterator |
| 21 ThreadIdToInternedNameIterator; |
| 22 typedef std::map<std::string, std::string*>::iterator |
| 23 NameToInternedNameIterator; |
| 24 } |
| 25 |
| 26 ThreadIdNameManager::ThreadIdNameManager() { |
| 27 g_default_name = new std::string(kDefaultName); |
| 28 |
| 29 AutoLock locked(lock_); |
| 30 name_to_interned_name_[kDefaultName] = g_default_name; |
| 31 } |
| 32 |
| 33 ThreadIdNameManager::~ThreadIdNameManager() { |
| 34 } |
| 35 |
| 36 ThreadIdNameManager* ThreadIdNameManager::GetInstance() { |
| 37 return Singleton<ThreadIdNameManager, |
| 38 LeakySingletonTraits<ThreadIdNameManager> >::get(); |
| 39 } |
| 40 |
| 41 const char* ThreadIdNameManager::GetDefaultInternedString() { |
| 42 return g_default_name->c_str(); |
| 43 } |
| 44 |
| 45 void ThreadIdNameManager::SetName(PlatformThreadId id, const char* name) { |
| 46 std::string str_name(name); |
| 47 |
| 48 AutoLock locked(lock_); |
| 49 NameToInternedNameIterator iter = name_to_interned_name_.find(str_name); |
| 50 std::string* leaked_str = NULL; |
| 51 if (iter != name_to_interned_name_.end()) { |
| 52 leaked_str = iter->second; |
| 53 } else { |
| 54 leaked_str = new std::string(str_name); |
| 55 name_to_interned_name_[str_name] = leaked_str; |
| 56 } |
| 57 thread_id_to_interned_name_[id] = leaked_str; |
| 58 } |
| 59 |
| 60 const char* ThreadIdNameManager::GetName(PlatformThreadId id) { |
| 61 AutoLock locked(lock_); |
| 62 ThreadIdToInternedNameIterator iter = thread_id_to_interned_name_.find(id); |
| 63 // A platform thread may not have a name set, so return blank. |
| 64 if (iter == thread_id_to_interned_name_.end()) |
| 65 return name_to_interned_name_[kDefaultName]->c_str(); |
| 66 return iter->second->c_str(); |
| 67 } |
| 68 |
| 69 void ThreadIdNameManager::RemoveName(PlatformThreadId id) { |
| 70 if (id == kInvalidThreadId) |
| 71 return; |
| 72 |
| 73 AutoLock locked(lock_); |
| 74 ThreadIdToInternedNameIterator iter = thread_id_to_interned_name_.find(id); |
| 75 DCHECK((iter != thread_id_to_interned_name_.end())); |
| 76 thread_id_to_interned_name_.erase(iter); |
| 77 } |
| 78 |
| 79 } // namespace base |
OLD | NEW |