com.vmlens.trace.agent.bootstrap.callback.field.Class2Id Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of trace-agent-bootstrap Show documentation

There is a newer version: 1.1.5

Show newest version

package com.vmlens.trace.agent.bootstrap.callback.field;

import com.vmlens.trace.agent.bootstrap.callback.CallbackState;
import com.vmlens.trace.agent.bootstrap.callback.CallbackStatePerThread;
import com.vmlens.trace.agent.bootstrap.event.AgentLogEvent;
import com.vmlens.trace.agent.bootstrap.event.ClassNameEvent;

import gnu.trove.map.hash.TObjectIntHashMap;

public class Class2Id {

	private int maxId = 1;
	TObjectIntHashMap class2Id = new TObjectIntHashMap();
	private Object LOCK = new Object();
	
	
	public int getId(Class cl)
	{
		synchronized(LOCK)
		{
			int current = class2Id.get(cl);
			
			if( current != 0 )
			{
				return current;
			}
			
			
			current = maxId;
			
			CallbackStatePerThread callbackStatePerThread = CallbackState.callbackStatePerThread.get();
			
			callbackStatePerThread.stackTraceBasedDoNotTrace++;
			
			callbackStatePerThread.queueCollection.putDirect( new ClassNameEvent(cl.getName() ,current )  );
			
			callbackStatePerThread.stackTraceBasedDoNotTrace--;
			
			
			
			maxId++;
			
			class2Id.put(cl, current);
			
			return current;
			
			
		}
		
		
	}
	
	public int getArrayClassId(Class cl)
	{
		synchronized(LOCK)
		{
			int current = class2Id.get(cl);
			
			if( current != 0 )
			{
				return current;
			}
			
			
			current = maxId;
			
			CallbackStatePerThread callbackStatePerThread = CallbackState.callbackStatePerThread.get();
			
			callbackStatePerThread.stackTraceBasedDoNotTrace++;
			
			String name = createName(cl);
			
			
			callbackStatePerThread.queueCollection.putDirect( new ClassNameEvent(name ,current )  );
			
			callbackStatePerThread.stackTraceBasedDoNotTrace--;
			
			
			
			maxId++;
			
			class2Id.put(cl, current);
			
			return current;
			
			
		}
	}
	
	
	
	
	public static String createName(Class cl)
	{
		return createNamePart( cl.getComponentType() , "[]" );
	}

	
	private static String createNamePart(Class cl, String prefix)
	{
		if( cl.isArray() )
		{
			return createNamePart( cl.getComponentType() , prefix + "[]" );
		}
		
		return cl.getName() + prefix;
	}
	
	
	
	
	
}