补全：连接池

👁️ 4 人浏览 💬 0 人评论 ❤️ 添加收藏

本节的东西：

Gate(n)   最多 n 个同时在里面：Semaphore(n)；顺便用一把小锁维护 inside / peak（峰值）
run_gate(n, workers, hold)   workers 条线程各进一次 Gate 待 hold 秒，交回峰值

补全 Pool：n 个「连接」（就是编号 0..n-1）；take() 用 BoundedSemaphore 等到有空位再从空闲列表拿一个，give(c) 放回去再 release。四条线程各 take-sleep-give 一次，交回 (最多同时借出几个 <= n, 最后空闲个数)。

提交你的答案

请登录后提交答案。

去登录

← 补全：用信号量当信号太粗是什么 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

import threading
import time

class Counter:
    def __init__(self):
        self.n = 0

def inc(self):
        tmp = self.n
        time.sleep(0.001)
        self.n = tmp + 1

class SafeCounter(Counter):
    def __init__(self):
        super().__init__()
        self.lock = threading.Lock()

def inc(self):
        with self.lock:
            tmp = self.n
            time.sleep(0.001)
            self.n = tmp + 1

def hammer(counter, workers=2, times=30):
    def job():
        for _ in range(times):
            counter.inc()
    ts = [threading.Thread(target=job) for _ in range(workers)]
    for t in ts:
        t.start()
    for t in ts:
        t.join()
    return counter.n

def probe(lock):
    """这把锁现在能不能立刻拿到？能就拿了再放回去，交回 True；拿不到交回 False。"""
    if lock.acquire(blocking=False):
        lock.release()
        return True
    return False

class Guard:
    """手写的 with 替身：进就 acquire，出就 release——不管是正常出还是异常出。"""
    def __init__(self, lock):
        self.lock = lock

def __enter__(self):
        self.lock.acquire()
        return self

def __exit__(self, exc_type, exc, tb):
        self.lock.release()
        return False

class Box:
    """有界队列：满了 put 等、空了 get 等；等的时候把锁交出去（wait_for 会放锁、醒来再拿回）。"""
    def __init__(self, cap):
        self.cap = cap
        self.items = []
        self.cond = threading.Condition()

def put(self, x):
        with self.cond:
            self.cond.wait_for(lambda: len(self.items) < self.cap)
            self.items.append(x)
            self.cond.notify_all()

def get(self):
        with self.cond:
            self.cond.wait_for(lambda: len(self.items) > 0)
            x = self.items.pop(0)
            self.cond.notify_all()
            return x

def run_pc(cap, n):
    """一个生产者放 0..n-1，一个消费者取 n 个；交回消费者拿到的顺序。"""
    box = Box(cap)
    got = []

def producer():
        for i in range(n):
            box.put(i)

def consumer():
        for _ in range(n):
            got.append(box.get())
    p = threading.Thread(target=producer, daemon=True)
    c = threading.Thread(target=consumer, daemon=True)
    p.start()
    c.start()
    p.join(timeout=2)      # 写坏了的 get 会让生产者永远等——别把判题机拖死
    c.join(timeout=2)
    return got

class Gate:
    """最多 n 个同时在里面；顺便记下峰值（峰值计数器自己用一把锁保护）。"""
    def __init__(self, n):
        self.sem = threading.Semaphore(n)
        self.lock = threading.Lock()
        self.inside = 0
        self.peak = 0

def __enter__(self):
        self.sem.acquire()
        with self.lock:
            self.inside += 1
            self.peak = max(self.peak, self.inside)

def __exit__(self, *a):
        with self.lock:
            self.inside -= 1
        self.sem.release()
        return False

def run_gate(n, workers, hold=0.03):
    gate = Gate(n)

def job():
        with gate:
            time.sleep(hold)
    ts = [threading.Thread(target=job) for _ in range(workers)]
    for t in ts:
        t.start()
    for t in ts:
        t.join()
    return gate.peak

class Pool:
    def __init__(self, n):
        self.free = list(range(n))
        self.sem = threading.BoundedSemaphore(n)
        self.lock = threading.Lock()
        self.out = 0
        self.peak = 0

def take(self):
        # TODO: sem.acquire()；with lock: 从 free 弹一个、out+1、记 peak；交回连接
        return None

def give(self, c):
        # TODO: with lock: 放回 free、out-1；sem.release()
        pass

p = Pool(2)

def use():
    c = p.take()
    time.sleep(0.02)
    p.give(c)

ts = [threading.Thread(target=use) for _ in range(4)]
for t in ts:
    t.start()
for t in ts:
    t.join()
print(str(1 <= p.peak <= 2) + "/" + str(len(p.free)))

本次输入：

输出：

👩‍🏫

💬 题目评论

补全：连接池

更多题目

全部评论