bun.sh/src/thread_safe_hash_map.zig

const std = @import("std");
const sync = @import("sync.zig");
usingnamespace @import("global.zig");
const hash_map = @import("hash_map.zig");

pub fn ThreadSafeStringHashMap(comptime Value: type) type {
    const HashMapType = hash_map.StringHashMap(Value);
    return struct {
        backing: HashMapType,
        lock: sync.RwLock,
        pub const HashMap = @This();

        pub fn init(allocator: *std.mem.Allocator) !*HashMap {
            var self = try allocator.create(HashMap);
            self.* = HashMap{ .backing = HashMapType.init(allocator), .lock = sync.RwLock.init() };

            return self;
        }

        pub fn get(self: *HashMap, key: string) ?Value {
            self.lock.lockShared();
            defer self.lock.unlockShared();
            return self.backing.get(key);
        }

        pub fn getHash(key: string) u64 {
            return HashMapType.getHash(key);
        }

        pub fn contains(self: *HashMap, str: string) bool {
            self.lock.lockShared();
            defer self.lock.unlockShared();
            return self.backing.contains(str);
        }

        pub fn containsHash(self: *HashMap, hash: u64) bool {
            self.lock.lockShared();
            defer self.lock.unlockShared();
            return self.backing.contains(str);
        }

        pub fn deinit(self: *HashMap, allocator: *std.mem.Allocator) void {
            self.backing.deinit();
        }

        pub fn put(self: *HashMap, key: string, value: Value) !void {
            self.lock.lock();
            defer self.lock.unlock();
            try self.backing.put(key, value);
        }

        pub fn putWithHash(self: *HashMap, key: string, hash: u64, value: Value) !void {
            self.lock.lock();
            defer self.lock.unlock();
            try self.backing.put(key, value);
        }
    };
}