Fix 228: Table: DateTime serialization for milliseonds

microsoft-azure-api/src/main/java/com/microsoft/windowsazure/services/blob/implementation/ISO8601DateConverter.java

@@ -17,6 +17,7 @@
 import java.text.DateFormat;
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
+import java.util.Calendar;
 import java.util.Date;
 import java.util.Locale;
 import java.util.TimeZone;
@@ -26,38 +27,54 @@
  */
 public class ISO8601DateConverter {
     // Note: because of the trailing "0000000", this is not quite ISO 8601 compatible
-    private static final String DATETIME_PATTERN = "yyyy-MM-dd'T'HH:mm:ss.SSSSSSS'Z'";
-    private static final String SHORT_DATETIME_PATTERN = "yyyy-MM-dd'T'HH:mm:ss'Z'";
+    private static final String DATETIME_PATTERN = "yyyy-MM-dd'T'HH:mm:ss.SSS'Z'";
+    private static final String DATETIME_PATTERN_NO_S = "yyyy-MM-dd'T'HH:mm'Z'";
+    private static final String DATETIME_PATTERN_NO_MS = "yyyy-MM-dd'T'HH:mm:ss'Z'";
+    private static final String DATETIME_PATTERN_TO_DECIMAL = "yyyy-MM-dd'T'HH:mm:ss.";
 
     public String format(Date date) {
-        return getFormat().format(date);
-    }
-
-    public String shortFormat(Date date) {
-        return getShortFormat().format(date);
+        DateFormat iso8601Format = new SimpleDateFormat(DATETIME_PATTERN, Locale.US);
+        iso8601Format.setTimeZone(TimeZone.getTimeZone("GMT"));
+        return iso8601Format.format(date);
     }
 
     public Date parse(String date) throws ParseException {
         if (date == null)
             return null;
 
-        // Sometimes, the date comes back without the ".SSSSSSS" part (presumably when the decimal value
-        // of the date is "0". Use the short format in that case.
-        if (date.indexOf('.') < 0)
-            return getShortFormat().parse(date);
-        else
-            return getFormat().parse(date);
-    }
+        int length = date.length();
+        if (length == 17) {
+            // [2012-01-04T23:21Z] length = 17
+            return parseDateFromString(date, DATETIME_PATTERN_NO_S);
+        }
+        else if (length == 20) {
+            // [2012-01-04T23:21:59Z] length = 20
+            return parseDateFromString(date, DATETIME_PATTERN_NO_MS);
+        }
+        else if (length >= 22 && length <= 28) {
+            // [2012-01-04T23:21:59.1Z] length = 22
+            // [2012-01-04T23:21:59.1234567Z] length = 28
+            // Need to handle the milliseconds gently.
 
-    private DateFormat getFormat() {
-        DateFormat iso8601Format = new SimpleDateFormat(DATETIME_PATTERN, Locale.US);
-        iso8601Format.setTimeZone(TimeZone.getTimeZone("GMT"));
-        return iso8601Format;
+            Date allExceptMilliseconds = parseDateFromString(date, DATETIME_PATTERN_TO_DECIMAL);
+            long timeWithSecondGranularity = allExceptMilliseconds.getTime();
+            // Decimal point is at 19
+            String secondDecimalString = date.substring(19, date.indexOf('Z'));
+            Float secondDecimal = Float.parseFloat(secondDecimalString);
+            int milliseconds = Math.round(secondDecimal * 1000);
+            long timeInMS = timeWithSecondGranularity + milliseconds;
+            Calendar cal = Calendar.getInstance();
+            cal.setTimeInMillis(timeInMS);
+            return cal.getTime();
+        }
+        else {
+            throw new IllegalArgumentException(String.format("Invalid Date String: %s", date));
+        }
     }
 
-    private DateFormat getShortFormat() {
-        DateFormat iso8601Format = new SimpleDateFormat(SHORT_DATETIME_PATTERN, Locale.US);
+    public static Date parseDateFromString(final String value, final String pattern) throws ParseException {
+        DateFormat iso8601Format = new SimpleDateFormat(pattern, Locale.US);
         iso8601Format.setTimeZone(TimeZone.getTimeZone("GMT"));
-        return iso8601Format;
+        return iso8601Format.parse(value);
     }
 }

microsoft-azure-api/src/main/java/com/microsoft/windowsazure/services/table/implementation/AtomReaderWriter.java

@@ -193,7 +193,7 @@ private InputStream generateEntry(PropertiesWriter propertiesWriter) {
             writer.writeEndElement(); // title
 
             writer.writeStartElement("updated");
-            writer.writeCharacters(iso8601DateConverter.shortFormat(dateFactory.getDate()));
+            writer.writeCharacters(iso8601DateConverter.format(dateFactory.getDate()));
             writer.writeEndElement(); // updated
 
             writer.writeStartElement("author");

microsoft-azure-api/src/main/java/com/microsoft/windowsazure/services/table/implementation/DefaultEdmValueConterter.java

@@ -39,7 +39,7 @@ public String serialize(String edmType, Object value) {
 
         String serializedValue;
         if (value instanceof Date) {
-            serializedValue = iso8601DateConverter.shortFormat((Date) value);
+            serializedValue = iso8601DateConverter.format((Date) value);
         }
         else {
             serializedValue = value.toString();

microsoft-azure-api/src/test/java/com/microsoft/windowsazure/services/blob/implementation/ISO8601DateConverterTests.java

@@ -55,7 +55,7 @@ public void shortFormatRoundTrips() throws Exception {
 
         // Act
         Date result = converter.parse(value);
-        String value2 = converter.shortFormat(result);
+        String value2 = converter.format(result);
 
         // Assert
         assertNotNull(result);