ConstraintToJsonConverter.java

package com.sequenceiq.cloudbreak.converter;

import org.springframework.stereotype.Component;

import com.sequenceiq.cloudbreak.api.model.ConstraintJson;
import com.sequenceiq.cloudbreak.domain.Constraint;

@Component
public class ConstraintToJsonConverter extends AbstractConversionServiceAwareConverter<Constraint, ConstraintJson> {

    @Override
    public ConstraintJson convert(Constraint source) {
        ConstraintJson constraintJson = new ConstraintJson();
        if (source.getConstraintTemplate() != null) {
            constraintJson.setConstraintTemplateName(source.getConstraintTemplate().getName());
        }
        if (source.getInstanceGroup() != null) {
            constraintJson.setInstanceGroupName(source.getInstanceGroup().getGroupName());
        }
        constraintJson.setHostCount(source.getHostCount());
        return constraintJson;
    }
}