From d1628cb76ac4f907d8e57346809e3ac0b8e12de5 Mon Sep 17 00:00:00 2001 From: alex Date: Fri, 10 Jul 2026 20:01:03 +0200 Subject: base integraion of command-registry system --- shards.go | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) (limited to 'shards.go') diff --git a/shards.go b/shards.go index ef9f9ef..591e121 100644 --- a/shards.go +++ b/shards.go @@ -92,4 +92,41 @@ func ( db *RedisDB) ExecuteReadAll(fn func([]*Shard) interface{})interface{}{ }() return fn(db.shards) +} + +func ( db *RedisDB) Lock(keys []string){ + if len(keys) == 1{ + shard := db.getShard(keys[0]) + shard.mu.Lock() + }else{ + shardMap := make(map[int]*Shard) + for _, k := range keys { + shard := db.getShard(k) + shardMap[shard.id] = shard + } + var sortedIDs []int + for id := range shardMap { + sortedIDs = append(sortedIDs, id) + } + sort.Ints(sortedIDs) + for _, id := range sortedIDs{ + shardMap[id].mu.Lock() + } + } +} +func ( db *RedisDB) Unlock(keys []string){ + shardMap := make(map[int]*Shard) + for _, k := range keys { + shard := db.getShard(k) + shardMap[shard.id] = shard + } + var sortedIDs []int + for id := range shardMap { + sortedIDs = append(sortedIDs, id) + } + sort.Ints(sortedIDs) + + for i := len(sortedIDs)-1; i >= 0; i --{ + shardMap[sortedIDs[i]].mu.Unlock() + } } \ No newline at end of file -- cgit v1.2.3