feat: added avro schema map datatype (#2344)

Author: Exar04

src/helpers/AvroToMetaModel.ts

@@ -1,6 +1,7 @@
 import {
   AnyModel,
   ArrayModel,
+  DictionaryModel,
   AvroSchema,
   BooleanModel,
   EnumModel,
@@ -109,6 +110,14 @@ export function AvroToMetaModel(
   if (unionModel !== undefined) {
     return unionModel;
   }
+  const dictionaryModel = toDictionaryModel(
+    avroSchemaModel,
+    modelName,
+    alreadySeenModels
+  );
+  if (dictionaryModel !== undefined) {
+    return dictionaryModel;
+  }
 
   Logger.warn('Failed to convert to MetaModel, defaulting to AnyModel.');
   return new AnyModel(
@@ -366,3 +375,40 @@ export function toArrayModel(
   }
   return undefined;
 }
+export function toDictionaryModel(
+  avroSchemaModel: AvroSchema,
+  name: string,
+  alreadySeenModels: Map<AvroSchema, MetaModel>
+): DictionaryModel | undefined {
+  if (
+    (typeof avroSchemaModel.type === 'string' ||
+      Array.isArray(avroSchemaModel.type)) &&
+    avroSchemaModel.type?.includes('map')
+  ) {
+    const keyModel = new StringModel(
+      '',
+      undefined,
+      getMetaModelOptions(avroSchemaModel)
+    );
+    let valueModel = new AnyModel(
+      '',
+      undefined,
+      getMetaModelOptions(avroSchemaModel)
+    );
+    if (avroSchemaModel.values !== undefined) {
+      const AvroModel = new AvroSchema();
+      AvroModel.name = `${name}_${avroSchemaModel.values}`;
+      AvroModel.type = avroSchemaModel.values;
+      valueModel = AvroToMetaModel(AvroModel, alreadySeenModels);
+    }
+
+    return new DictionaryModel(
+      name,
+      avroSchemaModel.originalInput,
+      getMetaModelOptions(avroSchemaModel),
+      keyModel,
+      valueModel
+    );
+  }
+  return undefined;
+}

src/models/AvroSchema.ts

@@ -22,4 +22,5 @@ export class AvroSchema {
   exclusiveMinimum?: unknown;
   exclusiveMaximum?: unknown;
   logicalType?: unknown;
+  values?: string;
 }

test/helpers/AvroToMetaModel.spec.ts

@@ -5,6 +5,7 @@ import {
   BooleanModel,
   EnumModel,
   FloatModel,
+  DictionaryModel,
   IntegerModel,
   ObjectModel,
   StringModel
@@ -208,4 +209,14 @@ describe('AvroToMetaModel', () => {
     expect(model).not.toBeUndefined();
     expect(model instanceof IntegerModel).toBeTruthy();
   });
+  test('should convert to dictionary model', () => {
+    const av = new AvroSchema();
+    av.name = 'test';
+    av.type = { type: 'map', values: 'integer' };
+
+    const model = AvroToMetaModel(av);
+
+    expect(model).not.toBeUndefined();
+    expect(model instanceof DictionaryModel).toBeTruthy();
+  });
 });