001/* 002 * (C) Copyright 2018 Nuxeo (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * pierre 018 */ 019package org.nuxeo.ecm.core.io.avro; 020 021import java.util.ArrayList; 022import java.util.List; 023 024import org.apache.avro.Schema; 025import org.apache.avro.Schema.Field; 026import org.nuxeo.ecm.core.schema.types.ComplexType; 027import org.nuxeo.runtime.avro.AvroSchemaFactory; 028import org.nuxeo.runtime.avro.AvroSchemaFactoryContext; 029 030/** 031 * @since 10.2 032 */ 033public class ComplexTypeSchemaFactory extends AvroSchemaFactory<ComplexType> { 034 035 public ComplexTypeSchemaFactory(AvroSchemaFactoryContext context) { 036 super(context); 037 } 038 039 @Override 040 public Schema createSchema(ComplexType input) { 041 Schema schema = Schema.createRecord(getName(input), null, input.getNamespace().prefix, false); 042 List<Field> fields = new ArrayList<>(input.getFields().size()); 043 for (org.nuxeo.ecm.core.schema.types.Field f : input.getFields()) { 044 String fieldName = context.getService().encodeName(f.getName().getLocalName()); 045 Schema fieldSchema = context.createSchema(f.getType()); 046 if (f.isNillable()) { 047 fieldSchema = nullable(fieldSchema); 048 } 049 fields.add(new Field(fieldName, fieldSchema, null, (Object) null)); 050 } 051 schema.setFields(fields); 052 return schema; 053 } 054 055 @Override 056 public String getName(ComplexType input) { 057 return context.getService().encodeName(input.getName()); 058 } 059 060 @Override 061 public String getQualifiedName(ComplexType input) { 062 return context.getService().encodeName(input.getNamespace().prefix) + ":" + getName(input); 063 } 064 065}