tas_spin_lock.cpp

#include "pls/internal/helpers/profiler.h"
#include "pls/internal/base/tas_spin_lock.h"

namespace pls {
namespace internal {
namespace base {

void tas_spin_lock::lock() {
  PROFILE_LOCK("Acquire Lock")
  int tries = 0;
  while (flag_.test_and_set(std::memory_order_acquire)) {
    tries++;
    if (tries % yield_at_tries_ == 0) {
      this_thread::yield();
    }
  }
}

bool tas_spin_lock::try_lock(unsigned int num_tries) {
  PROFILE_LOCK("Try Acquire Lock")
  while (flag_.test_and_set(std::memory_order_acquire)) {
    num_tries--;
    if (num_tries <= 0) {
      return false;
    }
  }
  return true;
}

void tas_spin_lock::unlock() {
  flag_.clear(std::memory_order_release);
}

}
}
}