/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* JBoss, Home of Professional Open Source
* Copyright 2011 Red Hat Inc. and/or its affiliates and other contributors
* as indicated by the @authors tag. All rights reserved.
* See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU Lesser General Public License, v. 2.1.
* This program is distributed in the hope that it will be useful, but WITHOUT A
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
* PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
* You should have received a copy of the GNU Lesser General Public License,
* v.2.1 along with this distribution; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
package org.hibernate.search.query.hibernate.impl;
import java.util.ArrayList;
import java.util.List;
import org.hibernate.search.engine.spi.SearchFactoryImplementor;
import org.hibernate.search.util.logging.impl.Log;
import org.hibernate.Criteria;
import org.hibernate.ObjectNotFoundException;
import org.hibernate.Session;
import org.hibernate.search.query.engine.spi.EntityInfo;
import org.hibernate.search.query.engine.spi.TimeoutManager;
import org.hibernate.search.util.logging.impl.LoggerFactory;
/**
* Check if the entity is available in the second level cache and load it if there
* before falling back to the delegate method.
*
* @author Emmanuel Bernard <emmanuel@hibernate.org>
*/
public class SecondLevelCacheObjectsInitializer implements ObjectsInitializer {
private static final Log log = LoggerFactory.make();
private final ObjectsInitializer delegate;
public SecondLevelCacheObjectsInitializer(ObjectsInitializer delegate) {
this.delegate = delegate;
}
public void initializeObjects(EntityInfo[] entityInfos,
Criteria criteria, Class<?> entityType,
SearchFactoryImplementor searchFactoryImplementor,
TimeoutManager timeoutManager,
Session session) {
boolean traceEnabled = log.isTraceEnabled();
//Do not call isTimeOut here as the caller might be the last biggie on the list.
final int maxResults = entityInfos.length;
if ( maxResults == 0 ) {
if ( traceEnabled ) log.tracef( "No object to initialize", maxResults );
return;
}
//check the second-level cache
List<EntityInfo> remainingEntityInfos = new ArrayList<EntityInfo>( entityInfos.length );
for ( EntityInfo entityInfo : entityInfos ) {
if ( ObjectLoaderHelper.areDocIdAndEntityIdIdentical( entityInfo, session ) ) {
final boolean isIn2LCache = session.getSessionFactory().getCache().containsEntity( entityInfo.getClazz(), entityInfo.getId() );
if ( isIn2LCache ) {
try {
//load the object from the second level cache
session.get( entityInfo.getClazz(), entityInfo.getId() );
}
catch (ObjectNotFoundException onfe) {
// Unlikely but needed: an index might be out of sync, and the cache might be as well
remainingEntityInfos.add( entityInfo );
}
}
else {
remainingEntityInfos.add( entityInfo );
}
}
else {
//if document id != entity id we can't use 2LC
remainingEntityInfos.add( entityInfo );
}
}
//update entityInfos to only contains the remaining ones
final int remainingSize = remainingEntityInfos.size();
if ( traceEnabled ) log.tracef( "Initialized %d objects out of %d in the second level cache", maxResults - remainingSize, maxResults );
if ( remainingSize > 0 ) {
delegate.initializeObjects(
remainingEntityInfos.toArray( new EntityInfo[remainingSize] ),
criteria,
entityType,
searchFactoryImplementor,
timeoutManager,
session
);
}
}
}