com.sap.csc.poc.ems.persistence.initial.entitlement.EntitlementHeaderAttributeDataInitializer.java Source code

Introduction

Here is the source code for com.sap.csc.poc.ems.persistence.initial.entitlement.EntitlementHeaderAttributeDataInitializer.java

Source

package com.sap.csc.poc.ems.persistence.initial.entitlement;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;

import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;
import javax.persistence.OneToOne;

import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.reflect.FieldUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Component;

import com.sap.csc.poc.ems.model.jpa.entitlement.EntitlementAttribute;
import com.sap.csc.poc.ems.model.jpa.entitlement.EntitlementHeaderAttribute;
import com.sap.csc.poc.ems.model.jpa.entitlement.EntitlementType;
import com.sap.csc.poc.ems.model.jpa.entitlement.softwarelicense.SoftwareLicenseEntitlementHeader;
import com.sap.csc.poc.ems.persistence.initial.JpaDataInitializer;
import com.sap.csc.poc.ems.persistence.repository.entitlement.EntitlementHeaderAttributeRepository;

@Component
public class EntitlementHeaderAttributeDataInitializer extends JpaDataInitializer<EntitlementHeaderAttribute> {

    private static final List<Class<?>> EXCLUSION_PROPERTY_ATTRIBUTES = Arrays
            .asList(new Class<?>[] { ManyToOne.class, OneToMany.class, OneToOne.class, Class.class });

    @Autowired
    private EntitlementHeaderAttributeRepository entitlementHeaderAttributeRepository;

    @Autowired
    private EntitlementTypeDataInitializer entitlementTypeDataInitializer;

    @Override
    public JpaRepository<EntitlementHeaderAttribute, Long> getRepository() {
        return entitlementHeaderAttributeRepository;
    }

    @Override
    public Collection<EntitlementHeaderAttribute> create() {
        ArrayList<EntitlementHeaderAttribute> attributes = new ArrayList<>();
        // Software License
        EntitlementType sl = entitlementTypeDataInitializer.getByName("SL");
        // Software License - Header Attributes
        Field[] headerFields = FieldUtils.getAllFields(SoftwareLicenseEntitlementHeader.class);
        sl.setHeaderAttributes(new ArrayList<>(headerFields.length));
        for (Field headerField : headerFields) {
            if (!Modifier.isStatic(headerField.getModifiers()) &&
            // Exclusion from JPA annotations
                    CollectionUtils.isEmpty(CollectionUtils.intersection(
                            // Annotations
                            Arrays.asList(headerField.getAnnotations()).stream()
                                    .map(annotation -> annotation.annotationType()).collect(Collectors.toList()),
                            // Exclusions
                            EXCLUSION_PROPERTY_ATTRIBUTES)))
                sl.getHeaderAttributes()
                        .add(generateEntitlementAttributes(headerField, new EntitlementHeaderAttribute()));
        }
        for (EntitlementHeaderAttribute headerAttribute : sl.getHeaderAttributes()) {
            headerAttribute.setEntitlementType(sl);
            attributes.add(headerAttribute);
        }

        return attributes;
    }

    public EntitlementHeaderAttribute getByName(String name) {
        return get().stream().filter(type -> StringUtils.equalsIgnoreCase(type.getName(), name)).findFirst().get();
    }

    private <TEntitlementAttribute extends EntitlementAttribute> TEntitlementAttribute generateEntitlementAttributes(
            Field field, TEntitlementAttribute entitlementAttribute) {
        entitlementAttribute.setName(field.getName());
        entitlementAttribute.setType(field.getType().getSimpleName());
        entitlementAttribute.setLength(50);
        entitlementAttribute.setValidation("NOT_NULL");
        return entitlementAttribute;
    }
}