add logical types

mattiabasone · Aug 12, 2024 · de3f4dc · de3f4dc
1 parent 724c2f9
commit de3f4dc
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/src/Objects/Schema/Generation/TypeMapper.php b/src/Objects/Schema/Generation/TypeMapper.php
@@ -31,6 +31,13 @@ public function __construct(SchemaGenerator $generator)
             TypeName::MAP => $this->simpleType(Schema::map()),
             TypeName::ENUM => $this->simpleType(Schema::enum()),
             TypeName::FIXED => $this->simpleType(Schema::fixed()),
+            TypeName::UUID => $this->simpleType(Schema::uuid()),
+            TypeName::TIME_MICROS => $this->simpleType(Schema::timeMicros()),
+            TypeName::TIME_MILLIS => $this->simpleType(Schema::timeMillis()),
+            TypeName::TIMESTAMP_MICROS => $this->simpleType(Schema::timestampMicros()),
+            TypeName::TIMESTAMP_MILLIS => $this->simpleType(Schema::timestampMillis()),
+            TypeName::LOCAL_TIMESTAMP_MICROS => $this->simpleType(Schema::localTimestampMicros()),
+            TypeName::LOCAL_TIMESTAMP_MILLIS => $this->simpleType(Schema::localTimestampMillis()),
         ];
     }
 

diff --git a/test/Objects/Schema/Generation/Fixture/SimpleRecord.php b/test/Objects/Schema/Generation/Fixture/SimpleRecord.php
@@ -27,4 +27,16 @@ class SimpleRecord
     #[AvroType("int")]
     #[AvroDefault(42)]
     private $intType;
+
+    /**
+     * @SerDe\AvroType("uuid")
+     */
+    #[AvroType("uuid")]
+    private $uuidType;
+
+    /**
+     * @SerDe\AvroType("timestamp-millis")
+     */
+    #[AvroType("timestamp-millis")]
+    private $timestampMillisType;
 }
diff --git a/test/Objects/Schema/Generation/SchemaGeneratorTest.php b/test/Objects/Schema/Generation/SchemaGeneratorTest.php
@@ -157,6 +157,14 @@ public function it_should_generate_records_containing_records()
                         'intType',
                         Schema::int(),
                         FieldOption::default(42)
+                    )
+                    ->field(
+                        'uuidType',
+                        Schema::uuid()
+                    )
+                    ->field(
+                        'timestampMillisType',
+                        Schema::timestampMillis()
                     ),
             )
             ->field(
@@ -186,6 +194,13 @@ public function test_it_should_generate_records_containing_records_using_attribu
                         'intType',
                         Schema::int(),
                         FieldOption::default(42)
+                    )->field(
+                        'uuidType',
+                        Schema::uuid()
+                    )
+                    ->field(
+                        'timestampMillisType',
+                        Schema::timestampMillis()
                     ),
             )
             ->field(