Java 基于CAS手动实现简单连接池

最新推荐文章于 2024-09-08 21:42:31 发布

.inevitable

最新推荐文章于 2024-09-08 21:42:31 发布

阅读量57

点赞数

分类专栏： 2023年1月文章标签： java 数据库服务器

本文链接：https://blog.csdn.net/qq_40670452/article/details/128609461

版权

2023年1月专栏收录该内容

5 篇文章 0 订阅

订阅专栏

该代码示例展示了一个简单的线程池实现，使用了AtomicIntegerArray来管理连接状态，当获取或释放连接时，会进行状态的原子性更新。线程池未处理动态扩容和收缩，没有超时等待机制，且在等待连接时可能会发生不可恢复的中断异常。

摘要由CSDN通过智能技术生成

并未考虑动态扩容与收缩，可用性检测，超时等待

public class Test {

    @SneakyThrows
    public static void main(String[] args) {
        // new Thread(() -> {}, "").start();
        // log.debug("");
        Pool pool = new Pool(2);
        for (int i = 0; i < 5; i++) {
            new Thread(() -> {
                Connection connection = pool.getConnection();
                try {
                    Thread.sleep(new Random().nextInt(2));
                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
                pool.free(connection);
            }, String.valueOf(i)).start();

        }
    }
}

@Slf4j(topic = "c.Pool")
class Pool {
    private static int poolSize;
    private static Connection[] connections;
    private static AtomicIntegerArray states;  // 0未连接 1 已连接

    public Pool(int poolSize) {
        this.poolSize = poolSize;
        connections = new Connection[poolSize];
        for (int i = 0; i < poolSize; i++) {
            connections[i] = new MockConnection("Connection -> " + i);  //模拟已经连接
        }
        states = new AtomicIntegerArray(poolSize);
    }

    public Connection getConnection() {
        while (true) {
            for (int i = 0; i < poolSize; i ++) {
                if (states.get(i) == 0) {
                    if (states.compareAndSet(i, 0, 1)) {
                        log.debug("getConnection -> {}", i);
                        return connections[i];
                    }
                }
            }
            synchronized (this) {
                try {
                    log.debug("wait");
                    this.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    public void free(Connection connection) {
        for (int i = 0; i < poolSize; i ++) {
            if (connection == connections[i]) {
                states.set(i, 0);
                log.debug("free -> {}", i);
                synchronized (this) {
                    this.notifyAll();
                    log.debug("notifyAll {}", i);
                }
                break;
            }
        }
    }
}

class MockConnection implements Connection{

    private String name;

    public MockConnection (String name) {
        this.name = name;
    }
	//...implements methods
}