之前发过一个Java利用Instrumentation获取对象大小的博客,这次修复了一些Bug,应该是非常准确了,除了一种情况,那就是下面代码里面main方法里的情况:
/**
* 因为要用到java.lang.instrument.Instrumentation,
* 本代码需要打成jar包以javaagent运行,manifest.mf文件内容如下
Manifest-Version: 1.0
Premain-Class: com.teasp.mem.SizeOfAgent
Boot-Class-Path:
Can-Redefine-Classes: false
* 运行方式:打包成sizeof.jar后,
* 执行java -javaagent:sizeof.jar com.teasp.mem.SizeOfAgent
*/
public class SizeOfAgent
{
private static Map<Object, Object> visited = new IdentityHashMap<Object, Object>();
private static Instrumentation inst;
/** initializes agent */
public static void premain(String agentArgs, Instrumentation instP)
{
inst = instP;
}
/**
* Returns object size without member sub-objects.
* @param o object to get size of
* @return object size
*/
public static long sizeOf(Object o)
{
if(inst == null)
{
throw new IllegalStateException("Can not access instrumentation environment.\n" +
"Please check if jar file containing SizeOfAgent class is \n" +
"specified in the java's \"-javaagent\" command line argument.");
}
return inst.getObjectSize(o);
}
/**
* Calculates full size of object iterating over
* its hierarchy graph.
* @param obj object to calculate size of
* @return object size
*/
public static long fullSizeOf(Object obj)
{
// Map<Object, Object> visited = new IdentityHashMap<Object, Object>();
Stack<Object> stack = new Stack<Object>();
long result = internalSizeOf(obj, stack/*, visited*/);
while (!stack.isEmpty())
{
result += internalSizeOf(stack.pop(), stack/*, visited*/);
}
// visited.clear();
return result;
}
private static boolean skipObject(Object obj/*, Map<Object, Object> visited*/)
{
// if (obj instanceof String) {
// // skip interned string
// if (obj == ((String) obj).intern()) {
// return true;
// }
// }
return (obj == null) || visited.containsKey(obj);
}
@SuppressWarnings("rawtypes")
private static long internalSizeOf(Object obj, Stack<Object> stack/*, Map<Object, Object> visited*/)
{
if (skipObject(obj/*, visited*/))
{
return 0;
}
visited.put(obj, null);
long result = 0;
// get size of object + primitive variables + member pointers
result += SizeOfAgent.sizeOf(obj);
// process all array elements
Class clazz = obj.getClass();
if (clazz.isArray())
{
if(clazz.getName().length() != 2)
{// skip primitive type array
int length = Array.getLength(obj);
for (int i = 0; i < length; i++)
{
stack.add(Array.get(obj, i));
}
}
return result;
}
// process all fields of the object
while (clazz != null)
{
Field[] fields = clazz.getDeclaredFields();
for (int i = 0; i < fields.length; i++)
{
if (!Modifier.isStatic(fields[i].getModifiers()))
{
if (fields[i].getType().isPrimitive())
{
continue; // skip primitive fields
}
else
{
fields[i].setAccessible(true);
try
{
// objects to be estimated are put to stack
Object objectToAdd = fields[i].get(obj);
if (objectToAdd != null)
{
stack.add(objectToAdd);
}
}
catch (IllegalAccessException ex)
{
assert false;
}
}
}
}
clazz = clazz.getSuperclass();
}
return result;
}
public static void main(String[] args) throws Exception
{
Object a = new Object();
Object b = a;
System.out.println(System.identityHashCode(a) + ":" + System.identityHashCode(b));
String stra = new String("abc");
String strb = new String("abc");
System.out.println("stra : " + fullSizeOf(stra)); //第一次计算时把"abc"这个常量池里面的字符串的字符数组给算进去了
System.out.println("strb : " + fullSizeOf(strb)); //第二次计算时就不会
}
}