org.codehaus.groovy.runtime.GroovyCategorySupport Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of groovy-all Show documentation
Show all versions of groovy-all Show documentation
Groovy: A powerful, dynamic language for the JVM
/*
* Copyright 2003-2007 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.codehaus.groovy.runtime;
import groovy.lang.Closure;
import java.lang.ref.SoftReference;
import org.codehaus.groovy.reflection.CachedClass;
import org.codehaus.groovy.reflection.CachedMethod;
import org.codehaus.groovy.reflection.ReflectionCache;
import org.codehaus.groovy.runtime.metaclass.NewInstanceMetaMethod;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
/**
* @author sam
* @author Paul King
* @author Alex Tkachman
*/
public class GroovyCategorySupport {
private static AtomicInteger categoriesInUse = new AtomicInteger();
public static class CategoryMethodList extends ArrayList {
public final int level;
final CategoryMethodList previous;
final AtomicInteger usage;
public CategoryMethodList(String name, int level, CategoryMethodList previous) {
this.level = level;
this.previous = previous;
if (previous != null) {
addAll(previous);
usage = previous.usage;
}
else {
usage = getCategoryNameUsage (name);
}
}
public boolean add(CategoryMethod o) {
usage.incrementAndGet();
return super.add(o);
}
}
public static class ThreadCategoryInfo extends HashMap{
int level;
private void newScope () {
categoriesInUse.incrementAndGet();
level++;
}
private void endScope () {
for (Iterator> it = entrySet().iterator(); it.hasNext(); ) {
final Map.Entry e = it.next();
final CategoryMethodList list = e.getValue();
if (list.level == level) {
final CategoryMethodList prev = list.previous;
if (prev == null) {
it.remove();
list.usage.addAndGet(-list.size());
}
else {
e.setValue(prev);
list.usage.addAndGet(prev.size()-list.size());
}
}
}
level--;
categoriesInUse.getAndDecrement();
}
private Object use(Class categoryClass, Closure closure) {
newScope();
try {
use(categoryClass);
return closure.call();
} finally {
endScope();
}
}
public Object use(List categoryClasses, Closure closure) {
newScope();
try {
for (Iterator i = categoryClasses.iterator(); i.hasNext(); ) {
Class clazz = (Class) i.next();
use(clazz);
}
return closure.call();
} finally {
endScope();
}
}
private void applyUse(CachedClass cachedClass) {
CachedMethod[] methods = cachedClass.getMethods();
for (int i = 0; i < methods.length; i++) {
CachedMethod cachedMethod = methods[i];
if (cachedMethod.isStatic() && cachedMethod.isPublic()) {
CachedClass[] paramTypes = cachedMethod.getParameterTypes();
if (paramTypes.length > 0) {
CachedClass metaClass = paramTypes[0];
CategoryMethod mmethod = new CategoryMethod(cachedMethod, metaClass.getTheClass());
final String name = cachedMethod.getName();
CategoryMethodList list = get(name);
if (list == null || list.level != level) {
list = new CategoryMethodList(name,level,list);
put (name, list);
}
list.add(mmethod);
Collections.sort(list);
}
}
}
}
private void use(Class categoryClass) {
CachedClass cachedClass = ReflectionCache.getCachedClass(categoryClass);
LinkedList classStack = new LinkedList();
for (CachedClass superClass = cachedClass; superClass.getTheClass()!=Object.class; superClass = superClass.getCachedSuperClass()) {
classStack.add(superClass);
}
while (!classStack.isEmpty()) {
CachedClass klazz = (CachedClass) classStack.removeLast();
applyUse(klazz);
}
}
public CategoryMethodList getCategoryMethods(String name) {
return level == 0 ? null : get(name);
}
}
private static final MyThreadLocal threadInfo = new MyThreadLocal();
private static class CategoryMethod extends NewInstanceMetaMethod implements Comparable {
private final Class metaClass;
public CategoryMethod(CachedMethod metaMethod, Class metaClass) {
super(metaMethod);
this.metaClass = metaClass;
}
public boolean isCacheable() { return false; }
/**
* Sort by most specific to least specific.
*
* @param o the object to compare against
*/
public int compareTo(Object o) {
CategoryMethod thatMethod = (CategoryMethod) o;
Class thisClass = metaClass;
Class thatClass = thatMethod.metaClass;
if (thisClass == thatClass) return 0;
if (isChildOfParent(thisClass, thatClass)) return -1;
if (isChildOfParent(thatClass, thisClass)) return 1;
return 0;
}
private boolean isChildOfParent(Class candidateChild, Class candidateParent) {
Class loop = candidateChild;
while(loop != null && loop != Object.class) {
loop = loop.getSuperclass();
if (loop == candidateParent) {
return true;
}
}
return false;
}
}
public static AtomicInteger getCategoryNameUsage (String name) {
return threadInfo.getUsage (name);
}
/**
* Create a scope based on given categoryClass and invoke closure within that scope.
*
* @param categoryClass the class containing category methods
* @param closure the closure during which to make the category class methods available
* @return the value returned from the closure
*/
public static Object use(Class categoryClass, Closure closure) {
return threadInfo.getInfo().use(categoryClass, closure);
}
/**
* Create a scope based on given categoryClasses and invoke closure within that scope.
*
* @param categoryClasses the list of classes containing category methods
* @param closure the closure during which to make the category class methods available
* @return the value returned from the closure
*/
public static Object use(List categoryClasses, Closure closure) {
return threadInfo.getInfo().use(categoryClasses, closure);
}
public static boolean hasCategoryInCurrentThread() {
return categoriesInUse.get() != 0 && threadInfo.getInfo().level != 0;
}
public static boolean hasCategoryInAnyThread() {
return categoriesInUse.get() != 0;
}
/**
* This method is used to pull all the new methods out of the local thread context with a particular name.
*
* @param name the method name of interest
* @return the list of methods
*/
public static CategoryMethodList getCategoryMethods(String name) {
return threadInfo.getInfo().getCategoryMethods(name);
}
private static class MyThreadLocal extends ThreadLocal {
ConcurrentHashMap usage = new ConcurrentHashMap ();
protected SoftReference initialValue() {
return new SoftReference(new ThreadCategoryInfo());
}
public ThreadCategoryInfo getInfo() {
ThreadCategoryInfo tcinfo = (ThreadCategoryInfo)get().get();
if( tcinfo == null ) {
tcinfo = new ThreadCategoryInfo();
set(new SoftReference(tcinfo));
}
return tcinfo;
}
public AtomicInteger getUsage (String name) {
AtomicInteger u = usage.get(name);
if (u != null) {
return u;
}
final AtomicInteger ai = new AtomicInteger();
final AtomicInteger prev = usage.putIfAbsent(name, ai);
return prev == null ? ai : prev;
}
}
}