Mercurial > hg > truffle
view graal/Runtime/src/com/oracle/graal/runtime/logging/CountingProxy.java @ 2502:3e3e8bd12730
Copyright rebranding. Update checkstyle file.
author | Thomas Wuerthinger <thomas.wuerthinger@oracle.com> |
---|---|
date | Tue, 26 Apr 2011 21:48:09 +0200 |
parents | 6594e6d8bfe7 |
children | 6bc5ec0255df |
line wrap: on
line source
/* * Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package com.oracle.graal.runtime.logging; import java.lang.reflect.*; import java.util.*; import com.oracle.graal.runtime.server.*; /** * A java.lang.reflect proxy that hierarchically logs all method invocations along with their parameters and return * values. * * @author Lukas Stadler */ public class CountingProxy<T> implements InvocationHandler { public static final boolean ENABLED = Boolean.valueOf(System.getProperty("c1x.countcalls")); private T delegate; private Map<Method, Long> calls = new HashMap<Method, Long>(); public CountingProxy(T delegate) { assert ENABLED; System.out.println("Counting proxy for " + delegate.getClass().getSimpleName() + " created"); this.delegate = delegate; proxies.add(this); } @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { int argCount = args == null ? 0 : args.length; if (method.getParameterTypes().length != argCount) { throw new RuntimeException("wrong parameter count"); } final Object result; long count = calls.containsKey(method) ? calls.get(method) : 0; calls.put(method, count + 1); try { if (args == null) { result = method.invoke(delegate); } else { result = method.invoke(delegate, args); } } catch (InvocationTargetException e) { throw e.getCause(); } return result; } public static <T> T getProxy(Class<T> interf, T delegate) { Class<?>[] interfaces = ReplacingStreams.getAllInterfaces(delegate.getClass()); Object obj = Proxy.newProxyInstance(interf.getClassLoader(), interfaces, new CountingProxy<T>(delegate)); return interf.cast(obj); } private static ArrayList<CountingProxy> proxies = new ArrayList<CountingProxy>(); static { if (ENABLED) { Runtime.getRuntime().addShutdownHook(new Thread() { @Override public void run() { for (CountingProxy proxy : proxies) { proxy.print(); } } }); } } protected void print() { long sum = 0; for (Map.Entry<Method, Long> entry : calls.entrySet()) { Method method = entry.getKey(); long count = entry.getValue(); sum += count; System.out.println(delegate.getClass().getSimpleName() + "." + method.getName() + ": " + count); } System.out.println(delegate.getClass().getSimpleName() + " calls: " + sum); } }