This repository has been archived by the owner on Aug 11, 2020. It is now read-only.
/
ConstructionContext.java
97 lines (81 loc) · 2.88 KB
/
ConstructionContext.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
/*
* Copyright (C) 2006 Google Inc.
*
* 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 com.google.inject.internal;
import com.google.inject.internal.InjectorImpl.InjectorOptions;
import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.List;
/**
* Context of a dependency construction. Used to manage circular references.
*
* @author crazybob@google.com (Bob Lee)
*/
final class ConstructionContext<T> {
T currentReference;
boolean constructing;
List<DelegatingInvocationHandler<T>> invocationHandlers;
public T getCurrentReference() {
return currentReference;
}
public void removeCurrentReference() {
this.currentReference = null;
}
public void setCurrentReference(T currentReference) {
this.currentReference = currentReference;
}
public boolean isConstructing() {
return constructing;
}
public void startConstruction() {
this.constructing = true;
}
public void finishConstruction() {
this.constructing = false;
invocationHandlers = null;
}
public Object createProxy(InjectorOptions injectorOptions, Class<?> expectedType)
throws InternalProvisionException {
if (injectorOptions.disableCircularProxies) {
throw InternalProvisionException.circularDependenciesDisabled(expectedType);
}
if (!expectedType.isInterface()) {
throw InternalProvisionException.cannotProxyClass(expectedType);
}
if (invocationHandlers == null) {
invocationHandlers = new ArrayList<>();
}
DelegatingInvocationHandler<T> invocationHandler = new DelegatingInvocationHandler<>();
invocationHandlers.add(invocationHandler);
// TODO: if I create a proxy which implements all the interfaces of
// the implementation type, I'll be able to get away with one proxy
// instance (as opposed to one per caller).
ClassLoader classLoader = BytecodeGen.getClassLoader(expectedType);
return expectedType.cast(
Proxy.newProxyInstance(
classLoader,
new Class[] {expectedType, CircularDependencyProxy.class},
invocationHandler));
}
public void setProxyDelegates(T delegate) {
if (invocationHandlers != null) {
for (DelegatingInvocationHandler<T> handler : invocationHandlers) {
handler.setDelegate(delegate);
}
// initialization of each handler can happen no more than once
invocationHandlers = null;
}
}
}