自旋锁测试脚本

以下是一个简单的自旋锁测试脚本，可以用于测试自旋锁的正确性和性能。

import threading
import time

# 自旋锁类
class SpinLock:
    def __init__(self):
        self.lock = threading.Lock()
        self.counter = 0

    def acquire(self):
        while True:
            if self.lock.acquire(False):
                if self.counter == 0:
                    return True
                else:
                    self.lock.release()
            self.counter += 1
            time.sleep(0.001)

    def release(self):
        self.counter -= 1
        self.lock.release()

# 测试函数
def test(lock, num_threads, num_iterations):
    def worker():
        for i in range(num_iterations):
            lock.acquire()
            time.sleep(0.001)
            lock.release()

    threads = []
    for i in range(num_threads):
        threads.append(threading.Thread(target=worker))
    start_time = time.time()
    for thread in threads:
        thread.start()
    for thread in threads:
        thread.join()
    end_time = time.time()
    elapsed_time = end_time - start_time
    print(f"Num threads: {num_threads}, Num iterations: {num_iterations}, Elapsed time: {elapsed_time:.3f} seconds")

# 运行测试
lock = SpinLock()
test(lock, num_threads=1, num_iterations=1000)
test(lock, num_threads=2, num_iterations=1000)
test(lock, num_threads=4, num_iterations=1000)
test(lock, num_threads=8, num_iterations=1000)

这个脚本实现了一个自旋锁类 `SpinLock`，并且定义了一个测试函数 `test`，用于测试不同线程数和迭代次数下的自旋锁性能。在测试函数中，我们创建了多个线程，并且让每个线程重复获取和释放自旋锁一定次数。最后我们记录了运行测试的时间，并且输出测试结果。

你可以运行这个脚本来测试自旋锁的性能和正确性。如果自旋锁实现正确，那么在多个线程同时请求自旋锁时，只有一个线程能够成功获取到自旋锁，其他线程会在自旋中等待。同时，自旋锁的性能应该随着线程数的增加而增加，但是如果线程数过多，自旋锁的性能会有所下降。