mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-27 02:44:18 +00:00
d616ebea23
Summary: Closes https://github.com/facebook/rocksdb/pull/2226 Differential Revision: D4967547 Pulled By: siying fbshipit-source-id: dd3b58ae1e7a106ab6bb6f37ab5c88575b125ab4
41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
// This source code is also licensed under the GPLv2 license found in the
|
|
// COPYING file in the root directory of this source tree.
|
|
//
|
|
|
|
#include "util/random.h"
|
|
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <thread>
|
|
#include <utility>
|
|
|
|
#include "port/likely.h"
|
|
#include "util/thread_local.h"
|
|
|
|
#ifdef ROCKSDB_SUPPORT_THREAD_LOCAL
|
|
#define STORAGE_DECL static __thread
|
|
#else
|
|
#define STORAGE_DECL static
|
|
#endif
|
|
|
|
namespace rocksdb {
|
|
|
|
Random* Random::GetTLSInstance() {
|
|
STORAGE_DECL Random* tls_instance;
|
|
STORAGE_DECL std::aligned_storage<sizeof(Random)>::type tls_instance_bytes;
|
|
|
|
auto rv = tls_instance;
|
|
if (UNLIKELY(rv == nullptr)) {
|
|
size_t seed = std::hash<std::thread::id>()(std::this_thread::get_id());
|
|
rv = new (&tls_instance_bytes) Random((uint32_t)seed);
|
|
tls_instance = rv;
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
} // namespace rocksdb
|