rocksdb/util/random.cc

//  Copyright (c) 2013, 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.
//

#include "util/random.h"

#include <pthread.h>
#include <stdint.h>
#include <string.h>

#include "port/likely.h"
#include "util/thread_local.h"

#if 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)) {
    const pthread_t self = pthread_self();
    uint32_t seed = 0;
    memcpy(&seed, &self, sizeof(seed));
    rv = new (&tls_instance_bytes) Random(seed);
    tls_instance = rv;
  }
  return rv;
}

}  // namespace rocksdb