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);
            }
        }
    }
}