#include <iostream>
#include <atomic>
#include <thread>
#include <vector>
#include <unistd.h>
using namespace std;

class SpinLock {
	atomic_flag f_;
public:
	SpinLock():f_(ATOMIC_FLAG_INIT) {}
	virtual ~SpinLock() {}
	void lock() { while(f_.test_and_set(memory_order_acquire)); }
	void unlock() { f_.clear(memory_order_release); }
};

class SpinLockWithYield {
	atomic_flag f_;
public:
	SpinLockWithYield():f_(ATOMIC_FLAG_INIT) {}
	virtual ~SpinLockWithYield() {}
	void lock() {
		unsigned short spinCount = 0;
		while(f_.test_and_set(memory_order_acquire)) {
			if ( (spinCount++) == 0 ) {
#ifdef _WIN32
				SwitchToThread();
#else
				sched_yield();
#endif
			}
		}
	}
	void unlock() { f_.clear(memory_order_release); }
};

SpinLock lck;

thread thd([&]{
	cout << "thread lock..." << endl;
	lck.lock();
	sleep(3);
	cout << "thread unlock..." << endl;
	lck.unlock();
});
thd.detach();

this_thread::sleep_for(chrono::milliseconds(50));
cout << "main thread lock... wait." << endl;
lck.lock();
cout << "SpinLock done." << endl;