CallbackHelper.java revision 674060f01e9090cd21b3c5656cc3204912ad17a6 
1/*
2 * Copyright 2004 The Apache Software Foundation
3 *
4 *  Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 *  Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16package org.mockito.cglib.proxy;
17
18import java.lang.reflect.Method;
19import java.util.*;
20
21import org.mockito.cglib.core.ReflectUtils;
22
23/**
24 * @version $Id: CallbackHelper.java,v 1.2 2004/06/24 21:15:20 herbyderby Exp $
25 */
26abstract public class CallbackHelper
27implements CallbackFilter
28{
29    private Map methodMap = new HashMap();
30    private List callbacks = new ArrayList();
31
32    public CallbackHelper(Class superclass, Class[] interfaces)
33    {
34        List methods = new ArrayList();
35        Enhancer.getMethods(superclass, interfaces, methods);
36        Map indexes = new HashMap();
37        for (int i = 0, size = methods.size(); i < size; i++) {
38            Method method = (Method)methods.get(i);
39            Object callback = getCallback(method);
40            if (callback == null)
41                throw new IllegalStateException("getCallback cannot return null");
42            boolean isCallback = callback instanceof Callback;
43            if (!(isCallback || (callback instanceof Class)))
44                throw new IllegalStateException("getCallback must return a Callback or a Class");
45            if (i > 0 && ((callbacks.get(i - 1) instanceof Callback) ^ isCallback))
46                throw new IllegalStateException("getCallback must return a Callback or a Class consistently for every Method");
47            Integer index = (Integer)indexes.get(callback);
48            if (index == null) {
49                index = new Integer(callbacks.size());
50                indexes.put(callback, index);
51            }
52            methodMap.put(method, index);
53            callbacks.add(callback);
54        }
55    }
56
57    abstract protected Object getCallback(Method method);
58
59    public Callback[] getCallbacks()
60    {
61        if (callbacks.size() == 0)
62            return new Callback[0];
63        if (callbacks.get(0) instanceof Callback) {
64            return (Callback[])callbacks.toArray(new Callback[callbacks.size()]);
65        } else {
66            throw new IllegalStateException("getCallback returned classes, not callbacks; call getCallbackTypes instead");
67        }
68    }
69
70    public Class[] getCallbackTypes()
71    {
72        if (callbacks.size() == 0)
73            return new Class[0];
74        if (callbacks.get(0) instanceof Callback) {
75            return ReflectUtils.getClasses(getCallbacks());
76        } else {
77            return (Class[])callbacks.toArray(new Class[callbacks.size()]);
78        }
79    }
80
81    public int accept(Method method, List<Method> allMethods)
82    {
83        return ((Integer)methodMap.get(method)).intValue();
84    }
85
86    public int hashCode()
87    {
88        return methodMap.hashCode();
89    }
90
91    public boolean equals(Object o)
92    {
93        if (o == null)
94            return false;
95        if (!(o instanceof CallbackHelper))
96            return false;
97        return methodMap.equals(((CallbackHelper)o).methodMap);
98    }
99}
100