add code for hset function

Signed-off-by: hwware <[email protected]>

Author: hwware

src/t_hash.c

@@ -1109,13 +1109,24 @@ void hsetnxCommand(client *c) {
 void hsetCommand(client *c) {
     int i, created = 0;
     robj *o;
+    long previous_element_number;
+    long current_element_number;
 
     if ((c->argc % 2) == 1) {
         addReplyErrorArity(c);
         return;
     }
 
-    if ((o = hashTypeLookupWriteOrCreate(c, c->argv[1])) == NULL) return;
+    o = lookupKeyWrite(c->db, c->argv[1]);
+    if (checkType(c, o, OBJ_HASH)) return;
+    if (o == NULL) {
+        o = createHashObject();
+        dbAdd(c->db, c->argv[1], &o);
+        previous_element_number = 0;
+    } else {
+        previous_element_number = hashTypeLength(o);
+    }
+
     hashTypeTryConversion(o, c->argv, 2, c->argc - 1);
     bool has_volatile_fields = hashTypeHasVolatileFields(o);
     for (i = 2; i < c->argc; i += 2) {
@@ -1128,6 +1139,11 @@ void hsetCommand(client *c) {
     notifyKeyspaceEvent(NOTIFY_HASH, "hset", c->argv[1], c->db->id);
     server.dirty += (c->argc - 2) / 2;
 
+    if (created > 0) {
+        current_element_number = previous_element_number + created;
+        updateBigKeyList(c->argv[1], previous_element_number, current_element_number, HASH_TYPE);
+    }
+
     /* HMSET (deprecated) and HSET return value is different. */
     char *cmdname = c->argv[0]->ptr;
     if (cmdname[1] == 's' || cmdname[1] == 'S') {

src/t_list.c

@@ -521,6 +521,8 @@ void linsertCommand(client *c) {
     listTypeIterator *iter;
     listTypeEntry entry;
     int inserted = 0;
+    long previous_element_number;
+    long current_element_number;
 
     if (strcasecmp(c->argv[2]->ptr, "after") == 0) {
         where = LIST_TAIL;
@@ -539,6 +541,7 @@ void linsertCommand(client *c) {
      * the list twice (once to see if the value can be inserted and once
      * to do the actual insert), so we assume this value can be inserted
      * and convert the listpack to a regular list if necessary. */
+    previous_element_number = listTypeLength(subject);
     listTypeTryConversionAppend(subject, c->argv, 4, 4, NULL, NULL);
 
     /* Seek pivot from head to tail */
@@ -556,6 +559,8 @@ void linsertCommand(client *c) {
         signalModifiedKey(c, c->db, c->argv[1]);
         notifyKeyspaceEvent(NOTIFY_LIST, "linsert", c->argv[1], c->db->id);
         server.dirty++;
+        current_element_number = previous_element_number + 1;
+        updateBigKeyList(c->argv[1], previous_element_number, current_element_number, LIST_TYPE);
     } else {
         /* Notify client of a failed insert */
         addReplyLongLong(c, -1);