au.org.consumerdatastandards.support.util.CustomAttributesUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of api-model Show documentation

This artefact describes the Consumer Data Standards using Java classes and annotations in a way suitable for automatic generation of Open API Specification, documentation, Server Stub, Client Libraries and Reference Test.

There is a newer version: 1.1.1

Show newest version

package au.org.consumerdatastandards.support.util;

import au.org.consumerdatastandards.support.data.CDSDataType;
import au.org.consumerdatastandards.support.data.CustomAttribute;
import au.org.consumerdatastandards.support.data.CustomAttributes;
import au.org.consumerdatastandards.support.model.ModelBase;

import java.lang.reflect.AnnotatedElement;
import java.util.*;

public class CustomAttributesUtil {

    public static void addCustomAttributes(AnnotatedElement annotatedElement, ModelBase model) {

        CustomAttribute customAttribute = annotatedElement.getAnnotation(CustomAttribute.class);
        if (customAttribute != null) {
            model.add(customAttribute);
        }
        CustomAttributes customAttributes = annotatedElement.getAnnotation(CustomAttributes.class);
        if (customAttributes != null) {
            model.addAll(customAttributes.value());
        }
    }

    public static Map getGroupedAttributes(AnnotatedElement annotatedElement) {

        Set attributes = new TreeSet<>(Comparator.comparing(attribute -> (attribute.name() + attribute.value())));
        CustomAttribute customAttribute = annotatedElement.getAnnotation(CustomAttribute.class);
        if (customAttribute != null) {
            attributes.add(customAttribute);
        }
        CustomAttributes customAttributes = annotatedElement.getAnnotation(CustomAttributes.class);
        if (customAttributes != null) {
            Collections.addAll(attributes, customAttributes.value());
        }
        
        Map groupedAttributes = getGroupedAttributes(attributes);
        
        /**
         * Now inject cds-type
         */
        
        CDSDataType cdsDataType = annotatedElement.getAnnotation(CDSDataType.class);
        if(cdsDataType != null) {
            groupedAttributes.put("x-cds-type", cdsDataType.value().getName());
        }
        
        return groupedAttributes;
        
    }

    @SuppressWarnings("unchecked")
    public static Map getGroupedAttributes(Set attributes) {
        Map map = new LinkedHashMap<>();
        for (CustomAttribute attribute : attributes) {
            if (attribute.multiple()) {
                Set values = new TreeSet<>();
                if (map.get(attribute.name()) != null) {
                    values = (Set)map.get(attribute.name());
                }
                values.add(attribute.value());
                map.put(attribute.name(), values);
            } else {
                map.put(attribute.name(), attribute.value());
            }
        }
        return map;
    }

}