org.jpmml.model.visitors.AttributeInterner Maven / Gradle / Ivy

Go to download

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

JPMML class model

There is a newer version: 1.6.6

/*
 * Copyright (c) 2022 Villu Ruusmann
 */
package org.jpmml.model.visitors;

import java.lang.reflect.Field;
import java.util.List;

import jakarta.xml.bind.annotation.XmlAttribute;
import jakarta.xml.bind.annotation.XmlList;
import org.dmg.pmml.PMMLObject;
import org.dmg.pmml.VisitorAction;
import org.jpmml.model.ReflectionUtil;

abstract
public class AttributeInterner extends Interner {

	public AttributeInterner(Class type){
		super(type);
	}

	@Override
	public VisitorAction visit(PMMLObject object){
		List fields = ReflectionUtil.getFields(object.getClass());

		for(Field field : fields){
			XmlAttribute attribute = field.getAnnotation(XmlAttribute.class);
			XmlList list = field.getAnnotation(XmlList.class);

			if((attribute == null) && (list == null)){
				continue;
			}

			apply(field, object);
		}

		return super.visit(object);
	}
}