diff --git a/reverse_engineering/helpers/adaptJsonSchema.js b/reverse_engineering/helpers/adaptJsonSchema.js index e6316d4..1d78cd9 100644 --- a/reverse_engineering/helpers/adaptJsonSchema.js +++ b/reverse_engineering/helpers/adaptJsonSchema.js @@ -20,6 +20,10 @@ const adaptType = field => { return adaptMultiple(field); } + if (type === 'string') { + return handleString(field); + } + if (type === 'number') { return handleNumber(field); } @@ -253,4 +257,16 @@ const convertToValidAvroName = name => { return name.replace(/[^A-Za-z0-9_]/g, '_'); }; +const handleString = field => { + if (Array.isArray(field.enum) && field.enum.length > 0) { + return { + ...field, + type: 'enum', + symbols: field.enum, + }; + } + + return field; +}; + module.exports = { adaptJsonSchema, adaptJsonSchemaName };