All Downloads are FREE. Search and download functionalities are using the official Maven repository.

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

Go to download

JPMML class model

There is a newer version: 1.6.6
Show newest version
/*
 * Copyright (c) 2015 Villu Ruusmann
 */
package org.jpmml.model.visitors;

import java.lang.instrument.Instrumentation;
import java.lang.reflect.Field;
import java.util.Collections;
import java.util.IdentityHashMap;
import java.util.List;
import java.util.Set;

import org.dmg.pmml.PMMLObject;
import org.dmg.pmml.VisitorAction;
import org.jpmml.agent.InstrumentationProvider;
import org.jpmml.model.ReflectionUtil;

/**
 * 

 * A Visitor that measures the deep size of a class model object.
 * 

 *
 * 

 * The object size is measured using {@link Instrumentation#getObjectSize(Object)} method.
 * 

 */
public class MemoryMeasurer extends AbstractVisitor implements Resettable {

	private Instrumentation instrumentation = InstrumentationProvider.getInstrumentation();

	private long size = 0L;

	private Set objects = Collections.newSetFromMap(new IdentityHashMap<>());


	/**
	 * @throws IllegalStateException If the JPMML agent is not available.
	 */
	public MemoryMeasurer(){
	}

	@Override
	public void reset(){
		this.size = 0L;

		this.objects.clear();
	}

	@Override
	public VisitorAction visit(PMMLObject object){
		measure(object);

		return super.visit(object);
	}

	public long getSize(){
		return this.size;
	}

	public Set getObjects(){
		return this.objects;
	}

	private void measure(Object object){
		boolean status = this.objects.add(object);

		if(!status){
			return;
		}

		this.size += this.instrumentation.getObjectSize(object);

		Class clazz = object.getClass();

		// Wrapper objects for primitive values do not have Object-type instance fields
		if(ReflectionUtil.isPrimitiveWrapper(clazz)){
			return;
		}

		List fields = ReflectionUtil.getFields(clazz);
		for(Field field : fields){
			Class type = field.getType();

			if(type.isPrimitive()){
				continue;
			}

			Object value = ReflectionUtil.getFieldValue(field, object);
			if(shouldMeasure(value)){
				measure(value);
			}
		}

		if(object instanceof Object[]){
			Object[] values = (Object[])object;

			for(int i = 0; i < values.length; i++){
				Object value = values[i];

				if(shouldMeasure(value)){
					measure(value);
				}
			}
		}
	}

	static
	private boolean shouldMeasure(Object object){

		if(object != null){

			if(object instanceof Enum){
				return false;
			} // End if

			if(object instanceof PMMLObject){
				return false;
			}

			return true;
		}

		return false;
	}
}    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy