blob: 791ef0ff3e70db8f11c9d4a78e529d8ccd5c79c7 (
plain)
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
|
/**
* Copyright (C) 2008 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.common.collect.Maps;
import com.google.common.collect.Sets;
import com.google.inject.Key;
import com.google.inject.internal.util.SourceProvider;
import java.util.Map;
import java.util.Set;
/**
* Minimal set that doesn't hold strong references to the contained keys.
*
* @author jessewilson@google.com (Jesse Wilson)
*/
final class WeakKeySet {
/**
* We store strings rather than keys so we don't hold strong references.
*
* <p>One potential problem with this approach is that parent and child injectors cannot define
* keys whose class names are equal but class loaders are different. This shouldn't be an issue
* in practice.
*/
private Map<String, Set<Object>> backingSet;
public void add(Key<?> key, Object source) {
if (backingSet == null) {
backingSet = Maps.newHashMap();
}
// if it's an instanceof Class, it was a JIT binding, which we don't
// want to retain.
if (source instanceof Class || source == SourceProvider.UNKNOWN_SOURCE) {
source = null;
}
String k = key.toString();
Set<Object> sources = backingSet.get(k);
if (sources == null) {
sources = Sets.newLinkedHashSet();
backingSet.put(k, sources);
}
sources.add(Errors.convert(source));
}
public boolean contains(Key<?> key) {
// avoid calling key.toString() if the backing set is empty. toString is expensive in aggregate,
// and most WeakKeySets are empty in practice (because they're used by top-level injectors)
return backingSet != null && backingSet.containsKey(key.toString());
}
public Set<Object> getSources(Key<?> key) {
return backingSet.get(key.toString());
}
}
|
