自行强化HashMap,使得线程安全

Java 数据结构

字数统计: 373阅读时长: 1 min

 2023/10/13 

在Java中,不使用ConcurrentHashMap的前提下,通过自行加强HashMap,使得HashMap线程安全,一般思路都是加锁,可以是普通的lock锁,也可以是其他锁,具体操作如下:

使用HashMap和显示的同步机制，如ReentrantLock来保证多线程安全访问。下面是一个简单的示例：

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.locks.ReentrantLock;

public class CustomThreadSafeHashMap<K, V> {
    private final Map<K, V> map = new HashMap<>();
    private final ReentrantLock lock = new ReentrantLock();

    public void put(K key, V value) {
        lock.lock(); // 获取锁
        try {
            map.put(key, value);
        } finally {
            lock.unlock(); // 释放锁
        }
    }

    public V get(K key) {
        lock.lock(); // 获取锁
        try {
            return map.get(key);
        } finally {
            lock.unlock(); // 释放锁
        }
    }

    public boolean containsKey(K key) {
        lock.lock(); // 获取锁
        try {
            return map.containsKey(key);
        } finally {
            lock.unlock(); // 释放锁
        }
    }

    public void remove(K key) {
        lock.lock(); // 获取锁
        try {
            map.remove(key);
        } finally {
            lock.unlock(); // 释放锁
        }
    }

    // 其他操作方法可以类似实现

    public int size() {
        lock.lock(); // 获取锁
        try {
            return map.size();
        } finally {
            lock.unlock(); // 释放锁
        }
    }
}

上述代码创建了一个自定义的线程安全HashMap类 CustomThreadSafeHashMap，其中使用ReentrantLock来保护对底层HashMap的访问。每个公共方法在执行前都会获取锁，并在执行结束后释放锁，以确保多线程安全。

请注意,生产开发请使用ConcurrentHashMap通常更容易和更安全，因为它内置了复杂的同步逻辑。

原文作者：WinddSnow

原文链接：https://winddsnow.github.io/2023/10/13/security-HashMap/

发表日期：October 13th 2023, 4:40:34 am

更新日期：January 18th 2024, 9:27:25 am

Previous Post

OpenFeign帮我们做了什么

CATALOG