feat: Add setObjectEx, setObjectEXAsync methods

Author: codeba

redis-keeper-support/src/main/java/org/codeba/redis/keeper/support/DefaultRedissonTemplate.java

@@ -2047,12 +2047,24 @@ public void setObject(String key, Object value) {
         kString.setObject(key, value);
     }
 
+    @Override
+    public void setObjectEx(String key, Object value, Duration duration) {
+        log("setObjectEx", key, value);
+        kString.setObjectEx(key, value, duration);
+    }
+
     @Override
     public CompletableFuture<Void> setObjectAsync(String key, Object value) {
         log("setObjectAsync", key, value);
         return kString.setObjectAsync(key, value);
     }
 
+    @Override
+    public CompletableFuture<Void> setObjectEXAsync(String key, Object value, Duration duration) {
+        log("setObjectAsync", key, value);
+        return kString.setObjectEXAsync(key, value, duration);
+    }
+
     @Override
     public Map<String, Object> mGet(String... keys) {
         log("mGet", keys);

redis-keeper-support/src/main/java/org/codeba/redis/keeper/support/KRedissonBatch.java

@@ -36,7 +36,7 @@
 /**
  * The type K redisson batch.
  */
-class KRedissonBatch implements KBatch {
+public class KRedissonBatch implements KBatch {
     /**
      * The R batch.
      */

redis-keeper-support/src/main/java/org/codeba/redis/keeper/support/KRedissonString.java

@@ -137,6 +137,23 @@ public void setObject(String key, Object value) {
         }
     }
 
+    @Override
+    public void setObjectEx(String key, Object value, Duration duration) {
+        if (value instanceof String) {
+            setEX(key, value.toString(), duration);
+        } else if (value instanceof Integer || value instanceof Long) {
+            final RAtomicLong rAtomicLong = getRAtomicLong(key);
+            rAtomicLong.set(Long.parseLong(value.toString()));
+            rAtomicLong.expire(duration.toMillis(), TimeUnit.MILLISECONDS);
+        } else if (value instanceof Float || value instanceof Double) {
+            final RAtomicDouble rAtomicDouble = getRAtomicDouble(key);
+            rAtomicDouble.set(Double.parseDouble(value.toString()));
+            rAtomicDouble.expire(duration.toMillis(), TimeUnit.MILLISECONDS);
+        } else {
+            getRBucket(key).set(value, duration.toMillis(), TimeUnit.MILLISECONDS);
+        }
+    }
+
     @Override
     public CompletableFuture<Map<String, Object>> mGetAsync(String... keys) {
         return getRBuckets(getCodec()).getAsync(keys).toCompletableFuture();

redis-keeper-support/src/main/java/org/codeba/redis/keeper/support/KRedissonStringAsync.java

@@ -137,6 +137,25 @@ public CompletableFuture<Void> setObjectAsync(String key, Object value) {
         }
     }
 
+    @Override
+    public CompletableFuture<Void> setObjectEXAsync(String key, Object value, Duration duration) {
+        if (value instanceof String) {
+            return setEXAsync(key, value.toString(), duration);
+        } else if (value instanceof Integer || value instanceof Long) {
+            final RAtomicLongAsync rAtomicLong = getRAtomicLong(key);
+            return rAtomicLong.setAsync(Long.parseLong(value.toString()))
+                    .thenRunAsync(() -> rAtomicLong.expireAsync(duration.toMillis(), TimeUnit.MILLISECONDS))
+                    .toCompletableFuture();
+        } else if (value instanceof Float || value instanceof Double) {
+            final RAtomicDoubleAsync rAtomicDouble = getRAtomicDouble(key);
+            return rAtomicDouble.setAsync(Double.parseDouble(value.toString()))
+                    .thenRunAsync(() -> rAtomicDouble.expireAsync(duration.toMillis(), TimeUnit.MILLISECONDS))
+                    .toCompletableFuture();
+        } else {
+            return getRBucket(key).setAsync(value, duration.toMillis(), TimeUnit.MILLISECONDS).toCompletableFuture();
+        }
+    }
+
     @Override
     public CompletableFuture<Void> setAsync(String key, String value) {
         return getRBucket(key, getCodec()).setAsync(value).toCompletableFuture();