chore: fix comments

Signed-off-by: Roman Gershman <[email protected]>
dragonflydb · Mar 8, 2024 · 0200562 · 0200562
1 parent fae92be
commit 0200562
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 17 deletions.
diff --git a/src/server/common.h b/src/server/common.h
@@ -24,6 +24,7 @@ enum class ListDir : uint8_t { LEFT, RIGHT };
 
 // Dependent on ExpirePeriod representation of the value.
 constexpr int64_t kMaxExpireDeadlineSec = (1u << 28) - 1;  // 8.5 years
+constexpr int64_t kMaxExpireDeadlineMs = kMaxExpireDeadlineSec * 1000;
 
 using DbIndex = uint16_t;
 using ShardId = uint16_t;

diff --git a/src/server/generic_family.cc b/src/server/generic_family.cc
@@ -178,7 +178,7 @@ class RestoreArgs {
 [[nodiscard]] bool RestoreArgs::UpdateExpiration(int64_t now_msec) {
   if (HasExpiration()) {
     int64_t ttl = abs_time_ ? expiration_ - now_msec : expiration_;
-    if (ttl > kMaxExpireDeadlineSec * 1000)
+    if (ttl > kMaxExpireDeadlineMs)
       return false;
 
     expiration_ = ttl < 0 ? -1 : ttl + now_msec;
@@ -856,8 +856,8 @@ void GenericFamily::Pexpire(CmdArgList args, ConnectionContext* cntx) {
   int_arg = std::max<int64_t>(int_arg, -1);
 
   // to be more compatible with redis, we silently cap the expire time to kMaxExpireDeadlineSec
-  if (int_arg > kMaxExpireDeadlineSec * 1000) {
-    int_arg = kMaxExpireDeadlineSec * 1000;
+  if (int_arg > kMaxExpireDeadlineMs) {
+    int_arg = kMaxExpireDeadlineMs;
   }
 
   auto expire_options = ParseExpireOptionsOrReply(args.subspan(2), cntx);

diff --git a/src/server/rdb_load.cc b/src/server/rdb_load.cc
@@ -203,7 +203,7 @@ int ziplistPairsConvertAndValidateIntegrity(const uint8_t* zl, size_t size, unsi
   return ret;
 }
 
-static string ModuleTypeName(uint64_t module_id) {
+string ModuleTypeName(uint64_t module_id) {
   static const char ModuleNameSet[] =
       "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
       "abcdefghijklmnopqrstuvwxyz"

diff --git a/src/server/string_family.cc b/src/server/string_family.cc
@@ -739,16 +739,16 @@ void StringFamily::Set(CmdArgList args, ConnectionContext* cntx) {
           return builder->SendStored();
         }
       }
-      if (!is_ms && int_arg >= kMaxExpireDeadlineSec) {
-        return builder->SendError(InvalidExpireTime("set"));
-      }
-
-      if (!is_ms) {
+      if (is_ms) {
+        if (int_arg > kMaxExpireDeadlineMs) {
+          int_arg = kMaxExpireDeadlineMs;
+        }
+      } else {
+        if (int_arg > kMaxExpireDeadlineSec) {
+          int_arg = kMaxExpireDeadlineSec;
+        }
         int_arg *= 1000;
       }
-      if (int_arg >= kMaxExpireDeadlineSec * 1000) {
-        return builder->SendError(InvalidExpireTime("set"));
-      }
       sparams.expire_after_ms = int_arg;
     } else if (cur_arg == "NX" && !(sparams.flags & SetCmd::SET_IF_EXISTS)) {
       sparams.flags |= SetCmd::SET_IF_NOTEXIST;
@@ -1147,22 +1147,29 @@ void StringFamily::SetExGeneric(bool seconds, CmdArgList args, ConnectionContext
   string_view key = ArgS(args, 0);
   string_view ex = ArgS(args, 1);
   string_view value = ArgS(args, 2);
-  int32_t unit_vals;
+  int64_t unit_vals;
 
   if (!absl::SimpleAtoi(ex, &unit_vals)) {
     return cntx->SendError(kInvalidIntErr);
   }
 
-  if (unit_vals < 1 || unit_vals >= kMaxExpireDeadlineSec) {
+  if (unit_vals < 1) {
     return cntx->SendError(InvalidExpireTime(cntx->cid->name()));
   }
 
   SetCmd::SetParams sparams;
   sparams.flags |= SetCmd::SET_EXPIRE_AFTER_MS;
-  if (seconds)
+  if (seconds) {
+    if (unit_vals > kMaxExpireDeadlineSec) {
+      unit_vals = kMaxExpireDeadlineSec;
+    }
     sparams.expire_after_ms = uint64_t(unit_vals) * 1000;
-  else
+  } else {
+    if (unit_vals > kMaxExpireDeadlineMs) {
+      unit_vals = kMaxExpireDeadlineMs;
+    }
     sparams.expire_after_ms = unit_vals;
+  }
 
   auto cb = [&](Transaction* t, EngineShard* shard) {
     SetCmd sg(t->GetOpArgs(shard), true);

diff --git a/src/server/string_family_test.cc b/src/server/string_family_test.cc
@@ -451,7 +451,8 @@ TEST_F(StringFamilyTest, SetEx) {
   ASSERT_THAT(Run({"ttl", "key"}), IntArg(10));
   ASSERT_THAT(Run({"setex", "key", "0", "val"}), ErrArg("invalid expire time"));
   ASSERT_EQ(Run({"setex", "key", StrCat(5 * 365 * 24 * 3600), "val"}), "OK");
-  ASSERT_THAT(Run({"setex", "key", StrCat(1 << 30), "val"}), ErrArg("invalid expire time"));
+  ASSERT_THAT(Run({"setex", "key", StrCat(1 << 30), "val"}), "OK");
+  ASSERT_THAT(Run({"ttl", "key"}), IntArg(kMaxExpireDeadlineSec));
 }
 
 TEST_F(StringFamilyTest, Range) {