七redis的set数据类型常见命令、内部编码、场景

浅浅的花香味﹌ 2023-05-22 11:28 48阅读 0赞

七redis的set数据类型常见命令、内部编码、场景

  1. 保存多个不重复的字符串的集合。

常见命令

sadd

  • 解释

    往集合中添加元素，如果值已经存在，则忽略。如果键不存在，则创建键后往键里添加元素。

  • 用法 sadd key value

  • 示例

    127.0.0.1:6379> sadd key1 1
    (integer) 1
    127.0.0.1:6379> sadd key2
    (error) ERR wrong number of arguments for ‘sadd’ command

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “sadd”,saddCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1}
    }

1.查询键是否存在
2.如果键不存在则创建数据类型为集合的键（这里完成后键的没有元素）
3.加入到dict字典中
4.键已经存在，如果不是集合类型则返回错误
5.元素加入集合中，返回1表示添加成功，0表示添加失败或元素已存在

  1. /** ** t_set.c **/
  2. void saddCommand(redisClient *c) { 
  3.     robj *set;
  4.     //1.查询键是否存在
  5.     set = lookupKeyWrite(c->db,c->argv[1]);
  6.     c->argv[2] = tryObjectEncoding(c->argv[2]);
  7.     if (set == NULL) { 
  8.     //2.如果键不存在则创建数据类型为集合的键（这里完成后键的没有元素）
  9.         set = setTypeCreate(c->argv[2]);
  10.     //3.加入到dict字典中
  11.         dbAdd(c->db,c->argv[1],set);
  12.     } else { 
  13.     //4.键已经存在，如果不是集合类型则返回错误
  14.         if (set->type != REDIS_SET) { 
  15.             addReply(c,shared.wrongtypeerr);
  16.             return;
  17.         }
  18.     }
  19.     //5. 元素加入集合中，返回1表示添加成功，0表示添加失败或元素已存在
  20.     if (setTypeAdd(set,c->argv[2])) { 
  21.         touchWatchedKey(c->db,c->argv[1]);
  22.         server.dirty++;
  23.         addReply(c,shared.cone);
  24.     } else { 
  25.         addReply(c,shared.czero);
  26.     }
  27. }

scard

  • 解释

    返回集合中元素的个数，如果键不存在返回0。

  • 用法 scard key

  • 示例

    127.0.0.1:6379> sadd key1 1
    (integer) 1
    127.0.01:6379> sadd key1 2
    (integer) 1
    127.0.0.1:6379> scard key1
    (integer) 2
    127.0.0.1:6379> scard key2
    (integer) 0

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “scard”,scardCommand,2,0,NULL,1,1,1}
    }

1.检查键是否存在以及是否数据类型是set集合，否则返回0
2.返回集合中的元素个数
2.1 计算内部编码是hashtable的键的元素个数
2.2 计算内部编码是intset的键的元素个数
2.3 未知的集合内部编码，返回错误信息

  1. /**
  2. ** t_set.c
  3. **/
  5. void scardCommand(redisClient *c) {
  6.     robj *o;
  7.     //1.检查键是否存在以及是否数据类型是set集合，否则返回0
  8.     if ((o = lookupKeyReadOrReply(c,c->argv[1],shared.czero)) == NULL ||
  9.         checkType(c,o,REDIS_SET)) return;
  10.     //2. 返回集合中的元素个数
  11.     addReplyLongLong(c,setTypeSize(o));
  12. }
  14. /** 
  15. ** t_set.c
  16. ** 返回集合中的元素个数
  17. **/
  18. unsigned long setTypeSize(robj *subject) {
  19.     //1.计算内部编码是hashtable的键的元素个数
  20.     if (subject->encoding == REDIS_ENCODING_HT) {
  21.         return dictSize((dict*)subject->ptr);
  22.     } else if (subject->encoding == REDIS_ENCODING_INTSET) {
  23.     //2.计算内部编码是intset的键的元素个数
  24.         return intsetLen((intset*)subject->ptr);
  25.     } else {
  26.     //3. 未知的集合内部编码，返回错误信息
  27.         redisPanic("Unknown set encoding");
  28.     }
  29. }

sdiff

  • 解释

    返回第一个键其他键的差集

  • 用法

    sdiff key1 key2 [key3…]
    需要注意: 算法复杂度是O(N),如果集合中的元素过多，会导致卡住redis。

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> sadd key2 c
    (integer) 1
    127.0.0.1:6379> sadd key3 a
    (integer) 1
    127.0.0.1:6379> sadd key3 c
    (integer) 1
    127.0.0.1:6379> sadd key3 e
    (integer) 1
    127.0.0.1:6379> sdiff key1 key2 key3
    1) “b”
    2) “d”

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “sdiff”,sdiffCommand,-2,REDIS_CMD_DENYOOM,NULL,1,-1,1}
    }

    / t_set.c
    **/

    define REDIS_OP_UNION 0

    define REDIS_OP_DIFF 1

    define REDIS_OP_INTER 2

    void sdiffCommand(redisClient *c) {

    1. sunionDiffGenericCommand(c,c->argv+1,c->argc-1,NULL,REDIS_OP_DIFF);

    }

  1. void sunionDiffGenericCommand(redisClient *c, robj **setkeys, int setnum, robj *dstkey, int op) {
  2.     robj **sets = zmalloc(sizeof(robj*)*setnum);
  3.     setTypeIterator *si;
  4.     robj *ele, *dstset = NULL;
  5.     int j, cardinality = 0;
  7.     for (j = 0; j < setnum; j++) {
  8.         robj *setobj = dstkey ?
  9.             lookupKeyWrite(c->db,setkeys[j]) :
  10.             lookupKeyRead(c->db,setkeys[j]);
  11.         if (!setobj) {
  12.             sets[j] = NULL;
  13.             continue;
  14.         }
  15.         if (checkType(c,setobj,REDIS_SET)) {
  16.             zfree(sets);
  17.             return;
  18.         }
  19.         sets[j] = setobj;
  20.     }
  22.     /* We need a temp set object to store our union. If the dstkey
  23.      * is not NULL (that is, we are inside an SUNIONSTORE operation) then
  24.      * this set object will be the resulting object to set into the target key*/
  25.     dstset = createIntsetObject();
  27.     /* Iterate all the elements of all the sets, add every element a single
  28.      * time to the result set */
  29.     for (j = 0; j < setnum; j++) {
  30.         if (op == REDIS_OP_DIFF && j == 0 && !sets[j]) break; /* result set is empty */
  31.         if (!sets[j]) continue; /* non existing keys are like empty sets */
  33.         si = setTypeInitIterator(sets[j]);
  34.         while((ele = setTypeNextObject(si)) != NULL) {
  35.             if (op == REDIS_OP_UNION || j == 0) {
  36.                 if (setTypeAdd(dstset,ele)) {
  37.                     cardinality++;
  38.                 }
  39.             } else if (op == REDIS_OP_DIFF) {
  40.                 if (setTypeRemove(dstset,ele)) {
  41.                     cardinality--;
  42.                 }
  43.             }
  44.             decrRefCount(ele);
  45.         }
  46.         setTypeReleaseIterator(si);
  48.         /* Exit when result set is empty. */
  49.         if (op == REDIS_OP_DIFF && cardinality == 0) break;
  50.     }
  52.     /* Output the content of the resulting set, if not in STORE mode */
  53.     if (!dstkey) {
  54.         addReplyMultiBulkLen(c,cardinality);
  55.         si = setTypeInitIterator(dstset);
  56.         while((ele = setTypeNextObject(si)) != NULL) {
  57.             addReplyBulk(c,ele);
  58.             decrRefCount(ele);
  59.         }
  60.         setTypeReleaseIterator(si);
  61.         decrRefCount(dstset);
  62.     } else {
  63.         /* If we have a target key where to store the resulting set
  64.          * create this key with the result set inside */
  65.         dbDelete(c->db,dstkey);
  66.         if (setTypeSize(dstset) > 0) {
  67.             dbAdd(c->db,dstkey,dstset);
  68.             addReplyLongLong(c,setTypeSize(dstset));
  69.         } else {
  70.             decrRefCount(dstset);
  71.             addReply(c,shared.czero);
  72.         }
  73.         touchWatchedKey(c->db,dstkey);
  74.         server.dirty++;
  75.     }
  76.     zfree(sets);
  77. }

sdiffstore

  • 解释

    和sdiff命令的效果一样，不同的是，sdiffstore会存储差集的结果到指定的键，
    如果键已经存在，则覆盖键之前的值。

  • 用法 sdiffstore key key1 key2 [key3…]

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> sadd key2 c
    (integer) 1
    127.0.0.1:6379> sadd key3 a
    (integer) 1
    127.0.0.1:6379> sadd key3 c
    (integer) 1
    127.0.0.1:6379> sadd key3 e
    (integer) 1
    127.0.0.1:6379> sdiff key4 key1 key2 key3
    (integer) 2

    127.0.0.1:6379> smembers key4
    1) “d”
    2) “b”

  • 源码

参见sdiff源码解析

sinter

  • 解释

    返回集合之间的交集

  • 用法

    sinter key1 key2
    需要注意: 算法复杂度是O(N*M),如果集合中的元素过多，会导致卡住redis。

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> sadd key2 c
    (integer) 1
    127.0.0.1:6379> sadd key3 a
    (integer) 1
    127.0.0.1:6379> sadd key3 c
    (integer) 1
    127.0.0.1:6379> sadd key3 e
    (integer) 1
    127.0.0.1:6379> sinter key1 key2 key3
    1) “c”

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “sinter”,sinterCommand,-2,REDIS_CMD_DENYOOM,NULL,1,-1,1}
    }

    / t_set.c
    **/

    void sinterCommand(redisClient *c) {

    1. sinterGenericCommand(c,c->argv+1,c->argc-1,NULL);

    }

sinterstore

  • 解释

    和sinter命令的下效果一样，不同的是，sinterstore会存储交集的结果到指定的键，
    如果键已经存在，则覆盖键之前的值。

  • 用法 sinterstore key key1 key2 [key3…]

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> sadd key2 c
    (integer) 1
    127.0.0.1:6379> sadd key3 a
    (integer) 1
    127.0.0.1:6379> sadd key3 c
    (integer) 1
    127.0.0.1:6379> sadd key3 e
    (integer) 1
    127.0.0.1:6379> sinterstore key4 key1 key2 key3
    (integer) 1
    127.0.0.1:6379> smembers key4
    1) “c”

sismember

  • 解释

    判断元素是否存在于集合中，如果返回1表示存在，0表示不存在,
    如果键不存在则直接返回0

  • 用法 sismember key member

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> sismember key1 c
    (integer) 1
    127.0.0.1:6379> sismember key1 f
    (integer) 0

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “sismember”,sismemberCommand,3,0,NULL,1,1,1}
    }

1.检查键是否存在以及是否数据类型是set集合，否则返回0
2.在键所有数据中查找这个元素是否存在，存在返回1，不存在返回0
1.在内部编码为hashtable中查找
2.在内部编码为intset中查找
3.未知的内部编码

  1. /** ** t_set.c **/
  2. void sismemberCommand(redisClient *c) { 
  3.     robj *set;
  4.     //1.检查键是否存在以及是否数据类型是set集合，否则返回0
  5.     if ((set = lookupKeyReadOrReply(c,c->argv[1],shared.czero)) == NULL ||
  6.         checkType(c,set,REDIS_SET)) return;
  7.     //2.在键所有数据中查找这个元素是否存在，存在返回1，不存在返回0
  8.     c->argv[2] = tryObjectEncoding(c->argv[2]);
  9.     if (setTypeIsMember(set,c->argv[2]))
  10.         addReply(c,shared.cone);
  11.     else
  12.         addReply(c,shared.czero);
  13. }
  16. int setTypeIsMember(robj *subject, robj *value) { 
  17.     long long llval;
  18.     if (subject->encoding == REDIS_ENCODING_HT) { 
  19.         //1. 在内部编码为hashtable中查找
  20.         return dictFind((dict*)subject->ptr,value) != NULL;
  21.     } else if (subject->encoding == REDIS_ENCODING_INTSET) { 
  22.         if (isObjectRepresentableAsLongLong(value,&llval) == REDIS_OK) { 
  23.         //2. 在内部编码为intset中查找
  24.             return intsetFind((intset*)subject->ptr,llval);
  25.         }
  26.     } else { 
  27.         //3.未知的内部编码
  28.         redisPanic("Unknown set encoding");
  29.     }
  30.     return 0;
  31. }

smembers

  • 解释

    返回集合中的所有元素

  • 用法 smembers key

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> smembers key1
    1) “a”
    2) “b”
    3) “c”
    4) “d”

  • 源码

    / redis.c **/
    // 其中2 表示这个客户端只能输入两个字符串
    // 第一个字符串： smembers 第二个字符串: key
    struct redisCommand readonlyCommandTable[] = {
    { “smembers”,sinterCommand,2,0,NULL,1,1,1}
    }

    / t_set.c */
    void sinterCommand(redisClient     c) { 

    1. sinterGenericCommand(c,c->argv+1,c->argc-1,NULL);

    }

  1. void sinterGenericCommand(redisClient *c, robj **setkeys, unsigned long setnum, robj *dstkey) { 
  2.     // 1.暂存有效的键
  3.     robj **sets = zmalloc(sizeof(robj*)*setnum);
  4.     setTypeIterator *si;
  5.     robj *eleobj, *dstset = NULL;
  6.     int64_t intobj;
  7.     void *replylen = NULL;
  8.     unsigned long j, cardinality = 0;
  9.     int encoding;
  11.     for (j = 0; j < setnum; j++) { 
  12.     //2. 循环输入的参数，查询键是否存在
  13.         robj *setobj = dstkey ?
  14.             lookupKeyWrite(c->db,setkeys[j]) :
  15.             lookupKeyRead(c->db,setkeys[j]);
  16.         if (!setobj) { 
  17.             zfree(sets);
  18.             if (dstkey) { 
  19.                 if (dbDelete(c->db,dstkey)) { 
  20.                     touchWatchedKey(c->db,dstkey);
  21.                     server.dirty++;
  22.                 }
  23.                 addReply(c,shared.czero);
  24.             } else { 
  25.                 addReply(c,shared.emptymultibulk);
  26.             }
  27.             return;
  28.         }
  29.         if (checkType(c,setobj,REDIS_SET)) { 
  30.             zfree(sets);
  31.             return;
  32.         }
  33.         sets[j] = setobj;
  34.     }
  35.     /* Sort sets from the smallest to largest, this will improve our * algorithm's performace */
  36.     qsort(sets,setnum,sizeof(robj*),qsortCompareSetsByCardinality);
  38.     /* The first thing we should output is the total number of elements... * since this is a multi-bulk write, but at this stage we don't know * the intersection set size, so we use a trick, append an empty object * to the output list and save the pointer to later modify it with the * right length */
  39.     if (!dstkey) { 
  40.         replylen = addDeferredMultiBulkLength(c);
  41.     } else { 
  42.         /* If we have a target key where to store the resulting set * create this key with an empty set inside */
  43.         dstset = createIntsetObject();
  44.     }
  46.     /* Iterate all the elements of the first (smallest) set, and test * the element against all the other sets, if at least one set does * not include the element it is discarded */
  47.     si = setTypeInitIterator(sets[0]);
  48.     while((encoding = setTypeNext(si,&eleobj,&intobj)) != -1) { 
  49.         for (j = 1; j < setnum; j++) { 
  50.             if (sets[j] == sets[0]) continue;
  51.             if (encoding == REDIS_ENCODING_INTSET) { 
  52.                 /* intset with intset is simple... and fast */
  53.                 if (sets[j]->encoding == REDIS_ENCODING_INTSET &&
  54.                     !intsetFind((intset*)sets[j]->ptr,intobj))
  55.                 { 
  56.                     break;
  57.                 /* in order to compare an integer with an object we * have to use the generic function, creating an object * for this */
  58.                 } else if (sets[j]->encoding == REDIS_ENCODING_HT) { 
  59.                     eleobj = createStringObjectFromLongLong(intobj);
  60.                     if (!setTypeIsMember(sets[j],eleobj)) { 
  61.                         decrRefCount(eleobj);
  62.                         break;
  63.                     }
  64.                     decrRefCount(eleobj);
  65.                 }
  66.             } else if (encoding == REDIS_ENCODING_HT) { 
  67.                 /* Optimization... if the source object is integer * encoded AND the target set is an intset, we can get * a much faster path. */
  68.                 if (eleobj->encoding == REDIS_ENCODING_INT &&
  69.                     sets[j]->encoding == REDIS_ENCODING_INTSET &&
  70.                     !intsetFind((intset*)sets[j]->ptr,(long)eleobj->ptr))
  71.                 { 
  72.                     break;
  73.                 /* else... object to object check is easy as we use the * type agnostic API here. */
  74.                 } else if (!setTypeIsMember(sets[j],eleobj)) { 
  75.                     break;
  76.                 }
  77.             }
  78.         }
  80.         /* Only take action when all sets contain the member */
  81.         if (j == setnum) { 
  82.             if (!dstkey) { 
  83.                 if (encoding == REDIS_ENCODING_HT)
  84.                     addReplyBulk(c,eleobj);
  85.                 else
  86.                     addReplyBulkLongLong(c,intobj);
  87.                 cardinality++;
  88.             } else { 
  89.                 if (encoding == REDIS_ENCODING_INTSET) { 
  90.                     eleobj = createStringObjectFromLongLong(intobj);
  91.                     setTypeAdd(dstset,eleobj);
  92.                     decrRefCount(eleobj);
  93.                 } else { 
  94.                     setTypeAdd(dstset,eleobj);
  95.                 }
  96.             }
  97.         }
  98.     }
  99.     setTypeReleaseIterator(si);
  101.     if (dstkey) { 
  102.         /* Store the resulting set into the target, if the intersection * is not an empty set. */
  103.         dbDelete(c->db,dstkey);
  104.         if (setTypeSize(dstset) > 0) { 
  105.             dbAdd(c->db,dstkey,dstset);
  106.             addReplyLongLong(c,setTypeSize(dstset));
  107.         } else { 
  108.             decrRefCount(dstset);
  109.             addReply(c,shared.czero);
  110.         }
  111.         touchWatchedKey(c->db,dstkey);
  112.         server.dirty++;
  113.     } else { 
  114.         setDeferredMultiBulkLength(c,replylen,cardinality);
  115.     }
  116.     zfree(sets);
  117. }

smove

  • 解释

    原子的移动集合(source)中的一个元素到另一个集合(destination)中。 如果元素或者source键不存在，命令将不做任何操作直接返回0;
    如果destionation不存在则新建键;
    如果元素在destination中已经存在，则只做source删除这个元素操作;
    如果destination不是集合类型，则报错。

  • 用法 smove source destination member

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> smove key1 key2 a
    (integer) 1
    127.0.0.1:6379> smembers key2
    1) “a”

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “smove”,smoveCommand,4,0,NULL,1,2,1}
    }

1.在dict字典中查找source,destination的键，以及编码要转移的元素
2.如果source的键不存在则直接返回0
3.检查source和destination,如果类型不是集合返回错误
4.如果source和destination是同一键则不操作直接返回0
5.从source中移除元素
6.移除后source中的元素为空，则删除键
7.如果destination不存在创建这个键
8.destination插入被source移除的元素

  1. /**
  2. ** t_set.c
  3. **/
  4. void smoveCommand(redisClient *c) {
  5.     //1.在dict字典中查找source,destination的键，以及编码要转移的元素
  6.     robj *srcset, *dstset, *ele;
  7.     srcset = lookupKeyWrite(c->db,c->argv[1]);
  8.     dstset = lookupKeyWrite(c->db,c->argv[2]);
  9.     ele = c->argv[3] = tryObjectEncoding(c->argv[3]);
  11.        //2.如果source的键不存在则直接返回0
  12.     if (srcset == NULL) {
  13.         addReply(c,shared.czero);
  14.         return;
  15.     }
  18.      //3.检查source和destination,如果类型不是集合返回错误，
  19.      //    destination不存在不检查数据类型
  20.     if (checkType(c,srcset,REDIS_SET) ||
  21.         (dstset && checkType(c,dstset,REDIS_SET))) return;
  23.     //4.如果source和destination是同一键则不操作直接返回0
  24.     if (srcset == dstset) {
  25.         addReply(c,shared.cone);
  26.         return;
  27.     }
  29.     //5.从source中移除元素
  30.     if (!setTypeRemove(srcset,ele)) {
  31.         addReply(c,shared.czero);
  32.         return;
  33.     }
  35.     //6.移除后source中的元素为空，则删除键
  36.     if (setTypeSize(srcset) == 0) dbDelete(c->db,c->argv[1]);
  37.     touchWatchedKey(c->db,c->argv[1]);
  38.     touchWatchedKey(c->db,c->argv[2]);
  39.     server.dirty++;
  41.     //7.如果destination不存在创建这个键
  42.     if (!dstset) {
  43.         dstset = setTypeCreate(ele);
  44.         dbAdd(c->db,c->argv[2],dstset);
  45.     }
  47.     //8.destination插入被source移除的元素
  48.     if (setTypeAdd(dstset,ele)) server.dirty++;
  49.     addReply(c,shared.cone);
  50. }

spop

  • 解释

    随机移除集合中指定数量的元素并且被移除的元素作为这个命令的结果返回。
    如果没有指定数量，则移除一个元素。

  • 用法 spop key [count]

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> spop key1
    “d”
    127.0.0.1:6379> smembers key1
    1) “c”
    2) “b”
    3) “a”
    127.0.0.1:6379> spop key1 2
    1) “c”
    2) “a”
    127.0.0.1:6379> smembers key1
    1) “b”

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “spop”,spopCommand,2,0,NULL,1,1,1}
    }

1.查找键是否存在以及是否是集合类型
2.返回内部编码，如果是hashtable/intset返回一个随机元素
2.1 如果内部编码是hashtable，返回一个随机元素
2.2 如果内部编码是intset，返回一个随机元素
2.3 如果内部编码都不是返回错误
3.内部编码是intset移除上面返回的索引元素
4.内部编码是hashtableu移除元素
5.没看明白为什么这里还要再删除一下元素呢(有看懂的朋友可以告知我一下哈
6.如果集合的元素个数为0，则删除键

  1. /** ** t_set.c **/
  2. void spopCommand(redisClient *c) { 
  3.     robj *set, *ele, *aux;
  4.     int64_t llele;
  5.     int encoding;
  6.     //1. 查找键是否存在以及是否是集合类型
  7.     if ((set = lookupKeyWriteOrReply(c,c->argv[1],shared.nullbulk)) == NULL ||
  8.         checkType(c,set,REDIS_SET)) return;
  9.     //2. 返回内部编码，如果是hashtable/intset返回一个随机元素
  10.     encoding = setTypeRandomElement(set,&ele,&llele);
  11.     if (encoding == REDIS_ENCODING_INTSET) { 
  12.     //3.内部编码是intset移除上面返回的元素
  13.         ele = createStringObjectFromLongLong(llele);
  14.         set->ptr = intsetRemove(set->ptr,llele,NULL);
  15.     } else { 
  16.     //4. 内部编码是hashtableu移除元素
  17.         incrRefCount(ele);
  18.         setTypeRemove(set,ele);
  19.     }
  21.     //5.没看明白为什么这里还要再删除一下元素呢(有看懂的朋友可以告知我一下哈)
  22.     aux = createStringObject("SREM",4);
  23.     rewriteClientCommandVector(c,3,aux,c->argv[1],ele);
  24.     decrRefCount(ele);
  25.     decrRefCount(aux);
  26.     //6. 如果集合的元素个数为0，则删除键
  27.     addReplyBulk(c,ele);
  28.     if (setTypeSize(set) == 0) dbDelete(c->db,c->argv[1]);
  29.     touchWatchedKey(c->db,c->argv[1]);
  30.     server.dirty++;
  31. }
  33. int setTypeRandomElement(robj *setobj, robj **objele, int64_t *llele) { 
  34.     if (setobj->encoding == REDIS_ENCODING_HT) { 
  35.         //1.如果内部编码是hashtable，返回一个随机元素
  36.         dictEntry *de = dictGetRandomKey(setobj->ptr);
  37.         *objele = dictGetEntryKey(de);
  38.     } else if (setobj->encoding == REDIS_ENCODING_INTSET) { 
  39.        //2.如果内部编码是intset，返回一个随机元素
  40.         *llele = intsetRandom(setobj->ptr);
  41.     } else { 
  42.         //3.如果内部编码都不是返回错误
  43.         redisPanic("Unknown set encoding");
  44.     }
  45.     return setobj->encoding;
  46. }

srandmember

  • 解释

    随机返回集合中指定数量的元素(数量的绝对值个元素)，
    如果未指定数量，则返回一个元素。

  • 用法 srandmember key [count]

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> srandmember key1
    1) “a”

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “srandmember”,srandmemberCommand,2,0,NULL,1,1,1}
    }

    / t_set.c */
    void srandmemberCommand(redisClient     c) { 

    1. robj *set, *ele;
    2. int64_t llele;
    3. int encoding;
    4. 1. 查找键是否存在以及是否是集合类型
    5. if ((set = lookupKeyReadOrReply(c,c->argv[1],shared.nullbulk)) == NULL ||
    6.     checkType(c,set,REDIS_SET)) return;
    7. 2. 返回内部编码，如果是hashtable返回一个随机元素，如果是intset返回一个随机索引
    8. encoding = setTypeRandomElement(set,&ele,&llele);
    9. if (encoding == REDIS_ENCODING_INTSET) { 
    10.     addReplyBulkLongLong(c,llele);
    11. } else { 
    12.     addReplyBulk(c,ele);
    13. }

    }

srem

  • 解释

    移除集合中的指定元素移除成功然后1
    如果元素不存在则返回0；
    如果键不存在则返回0；
    当键不是集合类型则返回错误信息。

  • 用法 srem key member

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key1 c
    (integer) 1
    127.0.0.1:6379> sadd key1 d
    (integer) 1
    127.0.0.1:6379> srem key1 a
    (integer) 1
    127.0.0.1:6379> srem key1 r
    (integer) 1

  • 源码

    / redis.c **/
    struct redisCommand readonlyCommandTable[] = {
    { “srem”,sremCommand,3,0,NULL,1,1,1}
    }

  1. 查找键是否存在以及是否是集合类型
  2. 移除元素
    2.1 内部编码是hashtable,在hashbable中删除元素
    2.2 hashtable如果需要重哈希则进行重哈希
    2.3 内部编码是intset,移除intset中的元素

  3. 如果移除元素后，集合的元素个数为0，则删除键

    / t_set.c */
    void sremCommand(redisClient     c) { 

    1. robj *set;
    2. 1. 查找键是否存在以及是否是集合类型
    3. if ((set = lookupKeyWriteOrReply(c,c->argv[1],shared.czero)) == NULL ||
    4.     checkType(c,set,REDIS_SET)) return;
    5. c->argv[2] = tryObjectEncoding(c->argv[2]);
    6. //2. 移除元素
    7. if (setTypeRemove(set,c->argv[2])) { 
    8. //3. 如果移除元素后，集合的元素个数为0，则删除键
    9.     if (setTypeSize(set) == 0) dbDelete(c->db,c->argv[1]);
    10.     touchWatchedKey(c->db,c->argv[1]);
    11.     server.dirty++;
    12.     addReply(c,shared.cone);
    13. } else { 
    14.     addReply(c,shared.czero);
    15. }

    }

    int setTypeRemove(robj setobj, robj value) { 

    1. long long llval;
    2. if (setobj->encoding == REDIS_ENCODING_HT) { 
    3.     //1.内部编码是hashtable,在hashbable中删除元素
    4.     if (dictDelete(setobj->ptr,value) == DICT_OK) { 
    5.         //2. hashtable如果需要重哈希则进行重哈希
    6.         if (htNeedsResize(setobj->ptr)) dictResize(setobj->ptr);
    7.         return 1;
    8.     }
    9. } else if (setobj->encoding == REDIS_ENCODING_INTSET) { 
    10.     //3.内部编码是intset,移除intset中的元素
    11.     if (isObjectRepresentableAsLongLong(value,&llval) == REDIS_OK) { 
    12.         int success;
    13.         setobj->ptr = intsetRemove(setobj->ptr,llval,&success);
    14.         if (success) return 1;
    15.     }
    16. } else { 
    17.     //3.内部编码不是hashtable或intset则返回错误
    18.     redisPanic("Unknown set encoding");
    19. }
    20. return 0;

    }

sunion

  • 解释

    对多个键进行并集操作并且返回结果

  • 用法 sunion key1 key2 [key3…]

  • 示例

    127.0.0.1:6379> sadd key1 a
    (integer) 1
    127.0.0.1:6379> sadd key1 b
    (integer) 1
    127.0.0.1:6379> sadd key2 c
    (integer) 1
    127.0.0.1:6379> sadd key2 d
    (integer) 1
    127.0.0.1:6379> sadd key3 r
    (integer) 1
    127.0.0.1:6379> sadd key3 a
    (integer) 1
    127.0.0.1:6379> sunion key1 key2 key3
    1) “a”
    2) “b”
    3) “c”
    4) “d”
    5) “r”

  • 源码

sunionstore

  • 解释

    存储sunion的结果到指定键

  • 用法 sunionstore key key1 key2 key3

内部编码

  1. hashtable、intset

使用场景

场景一：黑名单

存储用户id黑名单、ip黑名单、设备黑名单等，通过sismember命令判断是否在set中。

场景二: 访问数据统计

统计每日访问UV，每日访问ip

随机抽奖

随机抽奖，参与用户加入集合：
1.抽中用户不能参与多轮抽奖: 使用spop命令随机移除用户
2.抽中用户能参与多轮抽奖: 使用srandmember命令随机返回用户

https://blog.csdn.net/wizblack/article/details/78796557

https://redis.io/commands/sadd

发表评论

表情：
评论列表 （有 0 条评论，48人围观）

还没有评论，来说两句吧...

相关阅读