Extend the BsonWriter to write TimeSpan as Ticks (which is its minima…

…l unit).
andoco · May 21, 2010 · 9f226e1 · 9f226e1
1 parent 1f53b0e
commit 9f226e1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/source/MongoDB.Tests/UnitTests/Bson/TestBsonWriter.cs b/source/MongoDB.Tests/UnitTests/Bson/TestBsonWriter.cs
@@ -200,5 +200,15 @@ public void TestWriteBytesAsBinary()
 
             Assert.AreEqual("FwAAAAVieXRlcwAGAAAAAgIAAAAKDAA=",bson);
         }
+
+        [Test]
+        public void TestWriteTimeSpanAsTicks()
+        {
+            var span = TimeSpan.FromSeconds(123456);
+
+            var bson = Serialize(new Document("span", span));
+
+            Assert.AreEqual("EwAAABJzcGFuAACggnEfAQAAAA==",bson);
+        }
     }
 }
diff --git a/source/MongoDB/Bson/BsonWriter.cs b/source/MongoDB/Bson/BsonWriter.cs
@@ -68,7 +68,10 @@ public void WriteValue(BsonType type, Object obj)
                     _writer.Write((int)obj);
                     return;
                 case BsonType.Long:
-                    _writer.Write((long)obj);
+                    if(obj is TimeSpan)
+                        _writer.Write(((TimeSpan)obj).Ticks);
+                    else
+                        _writer.Write((long)obj);
                     return;
                 case BsonType.Date:
                     Write((DateTime)obj);
@@ -579,6 +582,8 @@ protected BsonType TranslateToBsonType(object obj)
                 return BsonType.Oid;
             if(type == typeof(DateTime))
                 return BsonType.Date;
+            if(type == typeof(TimeSpan))
+                return BsonType.Long;
             if(type == typeof(MongoRegex))
                 return BsonType.Regex;
             if(type == typeof(DBRef))