Merge pull request ServiceStack#161 from msarchet/New_Set_Options

Initial Commands Added - looking for feedback

Author: mythz

src/ServiceStack.Redis/Commands.cs

@@ -174,5 +174,10 @@ public static class Commands
         //public readonly static byte[] Exists = "EXISTS".ToUtf8Bytes();
         public readonly static byte[] Flush = "FLUSH".ToUtf8Bytes();
         public readonly static byte[] Slowlog = "SLOWLOG".ToUtf8Bytes();
+
+        public readonly static byte[] ExpireInSeconds = "EX".ToUtf8Bytes();
+        public readonly static byte[] ExpireInMilliseconds = "PX".ToUtf8Bytes();
+        public readonly static byte[] SetIfKeyDoesNotExist = "NX".ToUtf8Bytes();
+        public readonly static byte[] SetIfKeyExists = "XX".ToUtf8Bytes();
     }
 }

src/ServiceStack.Redis/RedisClient.cs

@@ -122,6 +122,47 @@ public void SetEntry(string key, string value)
             base.Set(key, bytesValue);
         }
 
+        public void SetEntry(string key, string value, TimeSpan expireIn)
+        {
+            var bytesValue = value != null ? value.ToUtf8Bytes() : null;
+
+            if (expireIn.Milliseconds == 0)
+            {
+                base.Set(key, bytesValue, (int)expireIn.TotalSeconds);
+            }
+            else { 
+                base.Set(key, bytesValue, null, (long)expireIn.TotalMilliseconds);
+            }
+        }
+
+        public void SetEntryIfExists(string key, string value, TimeSpan expireIn)
+        {
+            var bytesValue = value != null ? value.ToUtf8Bytes() : null;
+
+            if (expireIn.Milliseconds == 0)
+            {
+                base.Set(key, bytesValue, (int)expireIn.TotalSeconds, null, true);
+            }
+            else
+            {
+                base.Set(key, bytesValue, null, (long)expireIn.TotalMilliseconds, true);
+            }
+        }
+
+        public void SetEntryIfNotExists(string key, string value, TimeSpan expireIn)
+        {
+            var bytesValue = value != null ? value.ToUtf8Bytes() : null;
+
+            if (expireIn.Milliseconds == 0)
+            {
+                base.Set(key, bytesValue, (int)expireIn.TotalSeconds, null, false);
+            }
+            else
+            {
+                base.Set(key, bytesValue, null, (long)expireIn.TotalMilliseconds, false);
+            }
+        }
+
         public void ChangeDb(long db)
         {
             Db = db;

src/ServiceStack.Redis/RedisNativeClient.cs

@@ -315,7 +315,7 @@ public long StrLen(string key)
 			return SendExpectLong(Commands.StrLen, key.ToUtf8Bytes());
 		}
 
-    	public void Set(string key, byte[] value)
+    	public void Set(string key, byte[] value, int? expirySeconds = null, long? expiryMs = null, bool? exists = null)
         {
             if (key == null)
                 throw new ArgumentNullException("key");
@@ -324,7 +324,28 @@ public void Set(string key, byte[] value)
             if (value.Length > OneGb)
                 throw new ArgumentException("value exceeds 1G", "value");
 
-            SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value);
+            if (!expirySeconds.HasValue && !exists.HasValue)
+            {
+                SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value);
+            }
+            else if (!exists.HasValue && expiryMs.HasValue)
+            {
+                SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value, Commands.ExpireInMilliseconds, expiryMs.Value.ToUtf8Bytes());
+            }
+            else if (!exists.HasValue && expirySeconds.HasValue)
+            {
+                SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value, Commands.ExpireInSeconds, expirySeconds.Value.ToUtf8Bytes());
+            }
+            else if (exists.HasValue && !expirySeconds.HasValue && !expiryMs.HasValue)
+            {
+                SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value, exists.Value ? Commands.SetIfKeyExists : Commands.SetIfKeyDoesNotExist);
+            }
+            else if (exists.HasValue && expirySeconds.HasValue) {
+                SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value, exists.Value ? Commands.SetIfKeyExists : Commands.SetIfKeyDoesNotExist, Commands.ExpireInSeconds, expirySeconds.Value.ToUtf8Bytes());
+            }
+            else if (exists.HasValue && expiryMs.HasValue) {
+                SendExpectSuccess(Commands.Set, key.ToUtf8Bytes(), value, exists.Value ? Commands.SetIfKeyExists : Commands.SetIfKeyDoesNotExist, Commands.ExpireInMilliseconds, expiryMs.Value.ToUtf8Bytes());
+            }
         }
 
         public void SetEx(string key, int expireInSeconds, byte[] value)