Support date/time/datetime/timestamp (#152)

* Support date/time/datetime

* timestamp

examples/v2/date_test.csv

@@ -0,0 +1,2 @@
+1,2020-01-01,18:28:23.284,2020-01-01T18:28:23.284,2020-01-01T18:28:23.284
+2,2020-01-02,18:38:23.284,2020-01-11T19:28:23.284,2020-01-11T19:28:23.284

examples/v2/example.yaml

@@ -25,6 +25,7 @@ clientSettings:
       CREATE TAG course_no_props();
       CREATE TAG building_no_props();
       CREATE EDGE follow_no_props();
+      CREATE TAG date_test(c1 date, c2 time, c3 datetime, c4 timestamp)
     afterPeriod: 8s
   preStop:
     commands: |
@@ -392,3 +393,33 @@ files:
           index: 1
         srcVID:
           index: 0
+
+  - path: ./date_test.csv
+    failDataPath: ./err/date_test.csv
+    batchSize: 2
+    inOrder: true
+    type: csv
+    csv:
+      withHeader: false
+      withLabel: false
+      delimiter: ","
+    schema:
+      type: vertex
+      vertex:
+        vid:
+          index: 0
+        tags:
+          - name: date_test
+            props:
+              - name: c1
+                type: date
+                index: 1
+              - name: c2
+                type: time
+                index: 2
+              - name: c3
+                type: datetime
+                index: 3
+              - name: c4
+                type: timestamp
+                index: 4

pkg/base/tools.go

@@ -29,12 +29,11 @@ func FileExists(filename string) bool {
 
 func IsValidType(t string) bool {
 	switch strings.ToLower(t) {
-	case "string", "int", "float", "double", "bool", "timestamp":
-		break
+	case "string", "int", "float", "double", "bool", "date", "time", "datetime", "timestamp":
+		return true
 	default:
 		return false
 	}
-	return true
 }
 
 func HasHttpPrefix(path string) bool {

pkg/config/config.go

@@ -723,6 +723,11 @@ func (p *Prop) IsStringType() bool {
 	return strings.ToLower(*p.Type) == "string"
 }
 
+func (p *Prop) IsDateOrTimeType() bool {
+	t := strings.ToLower(*p.Type)
+	return t == "date" || t == "time" || t == "datetime" || t == "timestamp"
+}
+
 func (p *Prop) FormatValue(record base.Record) (string, error) {
 	if p.Index != nil && *p.Index >= len(record) {
 		return "", fmt.Errorf("Prop index %d out range %d of record(%v)", *p.Index, len(record), record)
@@ -731,6 +736,10 @@ func (p *Prop) FormatValue(record base.Record) (string, error) {
 	if p.IsStringType() {
 		return fmt.Sprintf("%q", r), nil
 	}
+	if p.IsDateOrTimeType() {
+		return fmt.Sprintf("%s(%q)", strings.ToLower(*p.Type), r), nil
+	}
+
 	return r, nil
 }