Add support for using datetime type in parameterised statements.

tediousjs · May 7, 2012 · 9ede878 · 9ede878
1 parent 9a8b6d4
commit 9ede878
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/src/data-type.coffee b/src/data-type.coffee
@@ -102,6 +102,30 @@ TYPE =
   0x3D:
     type: 'DATETIME'
     name: 'DateTime'
+    declaration: () ->
+      'datetime'
+    writeParameterData: (buffer, parameter) ->
+      # ParamMetaData (TYPE_INFO)
+      buffer.writeUInt8(typeByName.DateTimeN.id)
+      buffer.writeUInt8(8)
+
+      # ParamLenData
+      if parameter.value?
+        days = (parameter.value.getTime() - EPOCH_DATE.getTime()) / (1000 * 60 * 60 * 24)
+        days = Math.floor(days)
+
+        seconds = parameter.value.getHours() * 60 * 60
+        seconds += parameter.value.getMinutes() * 60
+        seconds += parameter.value.getSeconds()
+        milliseconds = (seconds * 1000) + parameter.value.getMilliseconds()
+        threeHundredthsOfSecond = milliseconds / (3 + (1 / 3))
+        threeHundredthsOfSecond = Math.floor(threeHundredthsOfSecond)
+
+        buffer.writeUInt8(8)
+        buffer.writeUInt32LE(days)
+        buffer.writeUInt32LE(threeHundredthsOfSecond)
+      else
+        buffer.writeUInt8(0)
   0x3E:
     type: 'FLT8'
     name: 'Float'

diff --git a/test/integration/parameterised-statements-test.coffee b/test/integration/parameterised-statements-test.coffee
@@ -69,6 +69,12 @@ exports.smallDateTime = (test) ->
 exports.smallDateTimeNull = (test) ->
   execSql(test, TYPES.SmallDateTime, null)
 
+exports.dateTime = (test) ->
+  execSql(test, TYPES.DateTime, new Date('December 4, 2011 10:04:23'))
+
+exports.dateTimeNull = (test) ->
+  execSql(test, TYPES.DateTime, null)
+
 exports.outputBitTrue = (test) ->
   execSqlOutput(test, TYPES.Bit, true)
 
@@ -114,6 +120,12 @@ exports.outputSmallDateTime = (test) ->
 exports.outputSmallDateTimeNull = (test) ->
   execSqlOutput(test, TYPES.SmallDateTime, null)
 
+exports.outputDateTime = (test) ->
+  execSqlOutput(test, TYPES.DateTime, new Date('December 4, 2011 10:04:23'))
+
+exports.outputDateTimeNull = (test) ->
+  execSqlOutput(test, TYPES.DateTime, null)
+
 exports.multipleParameters = (test) ->
   test.expect(6)