-
Notifications
You must be signed in to change notification settings - Fork 242
/
HibernateOrmLoadingContext.java
111 lines (95 loc) · 4.55 KB
/
HibernateOrmLoadingContext.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
/*
* Hibernate Search, full-text search for your domain model
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.search.mapper.orm.search.loading.context.impl;
import java.lang.invoke.MethodHandles;
import java.util.Set;
import org.hibernate.engine.spi.SessionImplementor;
import org.hibernate.search.engine.backend.common.spi.DocumentReferenceConverter;
import org.hibernate.search.engine.search.loading.context.spi.LoadingContext;
import org.hibernate.search.engine.search.loading.context.spi.LoadingContextBuilder;
import org.hibernate.search.engine.search.loading.spi.DefaultProjectionHitMapper;
import org.hibernate.search.engine.search.loading.spi.EntityLoader;
import org.hibernate.search.engine.search.loading.spi.ProjectionHitMapper;
import org.hibernate.search.mapper.orm.logging.impl.Log;
import org.hibernate.search.mapper.orm.search.loading.EntityLoadingCacheLookupStrategy;
import org.hibernate.search.mapper.orm.scope.impl.HibernateOrmScopeIndexedTypeContext;
import org.hibernate.search.mapper.orm.search.loading.dsl.SearchLoadingOptionsStep;
import org.hibernate.search.mapper.orm.search.loading.impl.EntityLoaderBuilder;
import org.hibernate.search.mapper.orm.search.loading.impl.HibernateOrmLoadingMappingContext;
import org.hibernate.search.mapper.orm.search.loading.impl.HibernateOrmLoadingSessionContext;
import org.hibernate.search.mapper.orm.search.loading.impl.MutableEntityLoadingOptions;
import org.hibernate.search.mapper.orm.common.EntityReference;
import org.hibernate.search.util.common.logging.impl.LoggerFactory;
public final class HibernateOrmLoadingContext<E> implements LoadingContext<EntityReference, E> {
private static final Log log = LoggerFactory.make( Log.class, MethodHandles.lookup() );
private final SessionImplementor sessionImplementor;
private final DocumentReferenceConverter<EntityReference> referenceHitMapper;
private final EntityLoader<EntityReference, ? extends E> entityLoader;
private final MutableEntityLoadingOptions loadingOptions;
private HibernateOrmLoadingContext(SessionImplementor sessionImplementor,
DocumentReferenceConverter<EntityReference> referenceHitMapper,
EntityLoader<EntityReference, ? extends E> entityLoader,
MutableEntityLoadingOptions loadingOptions) {
this.sessionImplementor = sessionImplementor;
this.referenceHitMapper = referenceHitMapper;
this.entityLoader = entityLoader;
this.loadingOptions = loadingOptions;
}
@Override
public ProjectionHitMapper<EntityReference, E> getProjectionHitMapper() {
try {
sessionImplementor.checkOpen();
}
catch (IllegalStateException e) {
throw log.hibernateSessionIsClosed( e );
}
return new DefaultProjectionHitMapper<>( referenceHitMapper, entityLoader );
}
public SessionImplementor getSessionImplementor() {
return sessionImplementor;
}
public MutableEntityLoadingOptions getLoadingOptions() {
return loadingOptions;
}
public static final class Builder<E>
implements LoadingContextBuilder<EntityReference, E, SearchLoadingOptionsStep>, SearchLoadingOptionsStep {
private final HibernateOrmLoadingSessionContext sessionContext;
private final EntityLoaderBuilder<E> entityLoaderBuilder;
private final MutableEntityLoadingOptions loadingOptions;
public Builder(HibernateOrmLoadingMappingContext mappingContext,
HibernateOrmLoadingSessionContext sessionContext,
Set<HibernateOrmScopeIndexedTypeContext<? extends E>> indexedTypeContexts) {
this.sessionContext = sessionContext;
this.entityLoaderBuilder = new EntityLoaderBuilder<>( mappingContext, sessionContext, indexedTypeContexts );
this.loadingOptions = new MutableEntityLoadingOptions( mappingContext );
}
@Override
public SearchLoadingOptionsStep toAPI() {
return this;
}
@Override
public SearchLoadingOptionsStep fetchSize(int fetchSize) {
loadingOptions.setFetchSize( fetchSize );
return this;
}
@Override
public SearchLoadingOptionsStep cacheLookupStrategy(EntityLoadingCacheLookupStrategy strategy) {
entityLoaderBuilder.cacheLookupStrategy( strategy );
return this;
}
@Override
public LoadingContext<EntityReference, E> build() {
DocumentReferenceConverter<EntityReference> referenceHitMapper = sessionContext.getReferenceHitMapper();
EntityLoader<EntityReference, ? extends E> entityLoader = entityLoaderBuilder.build( loadingOptions );
return new HibernateOrmLoadingContext<>(
sessionContext.getSession(),
referenceHitMapper, entityLoader,
loadingOptions
);
}
}
}