set correct avro schema for clickhouse date

PeerDB-io · Feb 12, 2024 · c888481 · c888481
1 parent e7dbab2
commit c888481
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/flow/model/qvalue/avro_converter.go b/flow/model/qvalue/avro_converter.go
@@ -85,6 +85,13 @@ func GetAvroSchemaFromQValueKind(kind QValueKind, targetDWH QDWHType, precision
 			if kind == QValueKindTime {
 				return "string", nil
 			}
+			if kind == QValueKindDate {
+				return AvroSchemaField{
+					Name:        "date",
+					Type:        "int",
+					LogicalType: "date",
+				}, nil
+			}
 			return "long", nil
 		}
 		return "string", nil
@@ -267,7 +274,7 @@ func (c *QValueAvroConverter) ToAvroValue() (interface{}, error) {
 			}
 		}
 
-		if c.Nullable && c.TargetDWH == QDWHTypeBigQuery {
+		if c.Nullable {
 			return goavro.Union("int.date", t), nil
 		}
 		return t, nil