关于redis:Redis-的-set-nx-底层怎么实现的

2次阅读

共计 2722 个字符，预计需要花费 7 分钟才能阅读完成。

这个问题其实自身很 ” 简略 ”，那么只有两种可能一种你看过 Redis 的源码，一种是没看过进行猜想。不过首先咱们能够说一些前奏：

Redis 一开始是有 setnx 这个命令的，起初废除了，而将 nx 作为 set 的一个参数项，同时也就反对指定过期工夫
这个命令的性能就是 set 一个 kv，如果 k 存在则失败，如果 k 不存在就胜利 set 这个 kv
咱们经常会用 Redis 的 set nx 来实现分布式锁，所以预计提问者想确保你理解原理，从而应用分布式锁的时候更加安心，或者想通过这个问题来引出分布式锁的问题

咱们首先能够大胆猜想一下实现形式。能够间接先简化一下问题，实质就是给你一个 map，而后实现一个 setnx 办法，当 k 存在则间接失败。最要害的问题就是解决并发问题。

那解决并发的问题，能想到的就是要么锁，要么 cas，要么间接队列卡死，对吧。

而后，联合所晓得的，redis 自身执行命令就是单线程，不须要锁，没有并发，那么间接查一把，而后解决就完事了。如果不看源码应该没有别的坑了吧。

话不多说，间接上源码：

/* Forward declaration */
static int getExpireMillisecondsOrReply(client *c, robj *expire, int flags, int unit, long long *milliseconds);

void setGenericCommand(client *c, int flags, robj *key, robj *val, robj *expire, int unit, robj *ok_reply, robj *abort_reply) {
    long long milliseconds = 0; /* initialized to avoid any harmness warning */
    int found = 0;
    int setkey_flags = 0;

    if (expire && getExpireMillisecondsOrReply(c, expire, flags, unit, &milliseconds) != C_OK) {return;}

    if (flags & OBJ_SET_GET) {if (getGenericCommand(c) == C_ERR) return;
    }

    found = (lookupKeyWrite(c->db,key) != NULL);

    if ((flags & OBJ_SET_NX && found) ||
        (flags & OBJ_SET_XX && !found))
    {if (!(flags & OBJ_SET_GET)) {addReply(c, abort_reply ? abort_reply : shared.null[c->resp]);
        }
        return;
    }

    /* When expire is not NULL, we avoid deleting the TTL so it can be updated later instead of being deleted and then created again. */
    setkey_flags |= ((flags & OBJ_KEEPTTL) || expire) ? SETKEY_KEEPTTL : 0;
    setkey_flags |= found ? SETKEY_ALREADY_EXIST : SETKEY_DOESNT_EXIST;

    setKey(c,c->db,key,val,setkey_flags);
    server.dirty++;
    notifyKeyspaceEvent(NOTIFY_STRING,"set",key,c->db->id);

    if (expire) {setExpire(c,c->db,key,milliseconds);
        /* Propagate as SET Key Value PXAT millisecond-timestamp if there is
         * EX/PX/EXAT flag. */
        if (!(flags & OBJ_PXAT)) {robj *milliseconds_obj = createStringObjectFromLongLong(milliseconds);
            rewriteClientCommandVector(c, 5, shared.set, key, val, shared.pxat, milliseconds_obj);
            decrRefCount(milliseconds_obj);
        }
        notifyKeyspaceEvent(NOTIFY_GENERIC,"expire",key,c->db->id);
    }

    if (!(flags & OBJ_SET_GET)) {addReply(c, ok_reply ? ok_reply : shared.ok);
    }

    /* Propagate without the GET argument (Isn't needed if we had expire since in that case we completely re-written the command argv) */
    if ((flags & OBJ_SET_GET) && !expire) {
        int argc = 0;
        int j;
        robj **argv = zmalloc((c->argc-1)*sizeof(robj*));
        for (j=0; j < c->argc; j++) {char *a = c->argv[j]->ptr;
            /* Skip GET which may be repeated multiple times. */
            if (j >= 3 &&
                (a[0] == 'g' || a[0] == 'G') &&
                (a[1] == 'e' || a[1] == 'E') &&
                (a[2] == 't' || a[2] == 'T') && a[3] == '\0')
                continue;
            argv[argc++] = c->argv[j];
            incrRefCount(c->argv[j]);
        }
        replaceClientCommandVector(c, argc, argv);
    }
}

看到代码间接吓坏了吧，不敢看了对吧？间接来看答案了对吧？其实这代码挺好懂的，没啥坑点与想的差不多。抓住 OBJ_SET_NX 示意用户输出的命令带有 NX 示意，其余一律不论。

lookupKeyWrite 间接利用这个办法查一把，如果 found(找到了) 并且是 OBJ_SET_NX 也就是 NX，那么间接返回了。这里能够多说一句，留神到这个办法名称带有了 Write 了吧，所以这个办法是专门用来写 key 的时候用的，是保障并发下可用的。
如果没找到就直接插入呗，还能有啥，和一般的 kv 插入也没啥区别了。（前面也没有 OBJ_SET_NX 标识了不是吗？）

所以其实这个问题真的没有那么简单，所以提问者还有一个考察点就是你对于 Redis 单线程执行命令是否有强烈的信赖或者是相熟细节。如果你信，那么其实猜也能齐全猜出来的。

正文完