redis 锁集合元素不能重复

原创

mob649e816594b7 2024-06-05 05:10:50 ©著作权

文章标签 Redis redis 客户端 文章分类 Redis 数据库

©著作权归作者所有：来自51CTO博客作者mob649e816594b7的原创作品，请联系作者获取转载授权，否则将追究法律责任

Redis 分布式锁及集合元素不重复

在分布式系统中，锁和集合元素不重复是两个常见的需求。Redis 是一个高性能的键值存储数据库，提供了丰富的数据结构和功能，能够很好地满足这两种需求。

Redis 锁

在多个客户端同时对同一个资源进行操作时，为了避免竞争条件（Race Condition）和数据不一致的问题，我们需要使用锁来保护共享资源。Redis 提供了 SETNX 命令，可以实现简单的分布式锁。

具体实现如下：

SETNX lock_key 1

如果返回值为 1，表示获得锁，可以执行操作；如果返回值为 0，则表示锁已被其他客户端持有，此时应该等待或重试。

为了防止死锁，我们还可以给锁设置过期时间，即使客户端异常退出或忘记释放锁，也能保证锁最终会被释放。

Redis 集合

Redis 的集合数据结构 Set 是一个无序、元素不重复的数据集合，支持添加、删除、查找等操作。我们可以利用集合来存储唯一的元素，实现元素不重复的功能。

SADD set_key element1
SADD set_key element2

添加元素到集合中，如果元素已存在，则添加操作会被忽略。这样就可以保证集合中的元素不重复。

示例代码

下面我们通过一个简单的示例来演示 Redis 锁和集合元素不重复的应用：

// 获取 Redis 连接
const redis = require('redis');
const client = redis.createClient();

// 加锁
function acquireLock(lockKey) {
    return new Promise((resolve, reject) => {
        client.SETNX(lockKey, 1, (err, result) => {
            if (err) {
                reject(err);
            } else {
                resolve(result === 1);
            }
        });
    });
}

// 释放锁
function releaseLock(lockKey) {
    client.del(lockKey);
}

// 添加元素到集合
function addElementToSet(setKey, element) {
    client.SADD(setKey, element);
}

// 示例
const lockKey = 'resource_lock';
const setKey = 'unique_elements_set';
const element = 'element1';

acquireLock(lockKey)
    .then((success) => {
        if (success) {
            addElementToSet(setKey, element);
            releaseLock(lockKey);
        } else {
            console.log('Failed to acquire lock');
        }
    });