io.atlasmap.v2.AtlasSchemaFactoryWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of atlas-model Show documentation
There is a newer version: 2.5.2
package io.atlasmap.v2;

import java.util.HashMap;

import com.fasterxml.jackson.annotation.JsonAnyGetter;
import com.fasterxml.jackson.annotation.JsonAnySetter;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.databind.BeanProperty;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.module.jsonSchema.factories.JsonSchemaFactory;
import com.fasterxml.jackson.module.jsonSchema.factories.SchemaFactoryWrapper;
import com.fasterxml.jackson.module.jsonSchema.factories.VisitorContext;
import com.fasterxml.jackson.module.jsonSchema.factories.WrapperFactory;
import com.fasterxml.jackson.module.jsonSchema.types.SimpleTypeSchema;
import com.fasterxml.jackson.module.jsonSchema.types.StringSchema;

public class AtlasSchemaFactoryWrapper extends SchemaFactoryWrapper {

    public AtlasSchemaFactoryWrapper() {
        super(new AtlasSchemaFactoryWrapper.AtlasWrapperFactory());
        schemaProvider = new JsonSchemaFactory() {

            @JsonProperty
            protected FieldType atlasFieldType;

            @SuppressWarnings("unused")
            public FieldType getAtlasFieldType() {
                return atlasFieldType;
            }

            @SuppressWarnings("unused")
            public void setAtlasFieldType(FieldType atlasFieldType) {
                this.atlasFieldType = atlasFieldType;
            }

            @Override
            public StringSchema stringSchema() {
                return new ExtendedStringSchema();
            }
        };
    }

    private static class AtlasWrapperFactory extends WrapperFactory {
        private AtlasWrapperFactory() {
        }

        public SchemaFactoryWrapper getWrapper(SerializerProvider p) {
            SchemaFactoryWrapper wrapper = new AtlasSchemaFactoryWrapper();
            if (p != null) {
                wrapper.setProvider(p);
            }

            return wrapper;
        }

        public SchemaFactoryWrapper getWrapper(SerializerProvider p, VisitorContext rvc) {
            SchemaFactoryWrapper wrapper = new AtlasSchemaFactoryWrapper();
            if (p != null) {
                wrapper.setProvider(p);
            }
            wrapper.setVisitorContext(rvc);
            return wrapper;
        }
    }

    interface ExtendedJsonSchema {
        @JsonAnyGetter HashMap getMetadata();

        @JsonAnySetter
        default void setMetadata(String name, Object value) {
            getMetadata().put(name, value);
        }
    }

    private static class ExtendedStringSchema extends StringSchema implements ExtendedJsonSchema {
        HashMap metadata = new HashMap<>();
        @JsonAnyGetter
        @Override
        public HashMap getMetadata() {
            return metadata;
        }

        @Override
        public void enrichWithBeanProperty(BeanProperty beanProperty) {
            enrichMetadata(this, beanProperty, metadata);
            super.enrichWithBeanProperty(beanProperty);
        }

    }

    private static void enrichMetadata(SimpleTypeSchema schema, BeanProperty property, HashMap metadata) {
        AtlasActionProperty atlasField = property.getAnnotation(AtlasActionProperty.class);
        if (atlasField != null) {
            schema.setTitle(atlasField.title());
            metadata.put("atlas-field-type", atlasField.type());
            metadata.put("atlas-collection-type", atlasField.collectionType());
        }
    }
}