forked from micronaut-projects/micronaut-core
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BeanProviderDefinition.java
129 lines (112 loc) · 4.3 KB
/
BeanProviderDefinition.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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
* Copyright 2017-2021 original 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
*
* https://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 io.micronaut.inject.provider;
import io.micronaut.context.*;
import io.micronaut.context.exceptions.NoSuchBeanException;
import io.micronaut.core.annotation.Internal;
import io.micronaut.core.annotation.NonNull;
import io.micronaut.core.annotation.Nullable;
import io.micronaut.core.type.Argument;
import io.micronaut.inject.BeanDefinition;
import io.micronaut.inject.qualifiers.AnyQualifier;
import io.micronaut.inject.qualifiers.Qualifiers;
import java.util.Iterator;
import java.util.Optional;
import java.util.stream.Stream;
/**
* Implementation for {@link BeanProvider} bean lookups.
*
* @author graemerocher
* @since 3.0.0
*/
@Internal
public final class BeanProviderDefinition extends AbstractProviderDefinition<BeanProvider<Object>> {
@Override
public boolean isEnabled(BeanContext context, BeanResolutionContext resolutionContext) {
return true;
}
@SuppressWarnings({"unchecked", "rawtypes"})
@Override
public Class<BeanProvider<Object>> getBeanType() {
return (Class) BeanProvider.class;
}
@Override
public boolean isPresent() {
return true;
}
@Override
protected BeanProvider<Object> buildProvider(
@NonNull BeanResolutionContext resolutionContext,
@NonNull BeanContext context,
@NonNull Argument<Object> argument,
@Nullable Qualifier<Object> qualifier,
boolean singleton) {
return new BeanProvider<Object>() {
private final Qualifier<Object> finalQualifier =
qualifier instanceof AnyQualifier ? null : qualifier;
private Qualifier<Object> qualify(Qualifier<Object> qualifier) {
if (finalQualifier == null) {
return qualifier;
} else if (qualifier == null) {
return finalQualifier;
}
//noinspection unchecked
return Qualifiers.byQualifiers(finalQualifier, qualifier);
}
@Override
public Object get() {
return ((DefaultBeanContext) context).getBean(resolutionContext, argument, finalQualifier);
}
@Override
public Optional<Object> find(Qualifier<Object> qualifier) {
return ((DefaultBeanContext) context).findBean(resolutionContext, argument, qualify(qualifier));
}
@Override
public BeanDefinition<Object> getDefinition() {
return context.getBeanDefinition(argument, finalQualifier);
}
@Override
public Object get(Qualifier<Object> qualifier) {
return ((DefaultBeanContext) context).getBean(resolutionContext, argument, qualify(qualifier));
}
@Override
public boolean isUnique() {
try {
return context.getBeanDefinitions(argument, finalQualifier).size() == 1;
} catch (NoSuchBeanException e) {
return false;
}
}
@Override
public boolean isPresent() {
return context.containsBean(argument, finalQualifier);
}
@NonNull
@Override
public Iterator<Object> iterator() {
return ((DefaultBeanContext) context).getBeansOfType(resolutionContext, argument, finalQualifier).iterator();
}
@Override
public Stream<Object> stream() {
return ((DefaultBeanContext) context).streamOfType(resolutionContext, argument, finalQualifier);
}
};
}
@Override
protected boolean isAllowEmptyProviders(BeanContext context) {
return true;
}
}