/*
* 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.indexes.serialization.impl;
import java.io.Serializable;
import java.util.List;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.FieldType;
import org.apache.lucene.document.FieldType.NumericType;
import org.apache.lucene.index.IndexableField;
import org.hibernate.search.SearchException;
import org.hibernate.search.backend.AddLuceneWork;
import org.hibernate.search.backend.DeleteLuceneWork;
import org.hibernate.search.backend.LuceneWork;
import org.hibernate.search.backend.OptimizeLuceneWork;
import org.hibernate.search.backend.PurgeAllLuceneWork;
import org.hibernate.search.backend.UpdateLuceneWork;
import org.hibernate.search.engine.spi.SearchFactoryImplementor;
import org.hibernate.search.indexes.serialization.spi.Deserializer;
import org.hibernate.search.indexes.serialization.spi.LuceneFieldContext;
import org.hibernate.search.indexes.serialization.spi.LuceneNumericFieldContext;
import org.hibernate.search.indexes.serialization.spi.LuceneWorkSerializer;
import org.hibernate.search.indexes.serialization.spi.SerializationProvider;
import org.hibernate.search.indexes.serialization.spi.Serializer;
import org.hibernate.search.util.logging.impl.Log;
import org.hibernate.search.util.logging.impl.LoggerFactory;
import static org.hibernate.search.indexes.serialization.impl.SerializationHelper.toByteArray;
/**
* Serializes {@code List<LuceneWork>} instances back and forth using a pluggable {@code SerializerProvider}.
*
* This class controls the overall traversal process and delegates true serialization work to the {@code SerializerProvider}.
*
* @author Emmanuel Bernard <emmanuel@hibernate.org>
*/
public class LuceneWorkSerializerImpl implements LuceneWorkSerializer {
private static Log log = LoggerFactory.make();
private final SearchFactoryImplementor searchFactory;
private SerializationProvider provider;
public LuceneWorkSerializerImpl(SerializationProvider provider, SearchFactoryImplementor searchFactory) {
this.provider = provider;
this.searchFactory = searchFactory;
}
/**
* Convert a List of LuceneWork into a byte[]
*/
@Override
public byte[] toSerializedModel(List<LuceneWork> works) {
try {
Serializer serializer = provider.getSerializer();
serializer.luceneWorks( works );
for ( LuceneWork work : works ) {
if ( work instanceof OptimizeLuceneWork ) {
serializer.addOptimizeAll();
}
else if (work instanceof PurgeAllLuceneWork) {
serializer.addPurgeAll( work.getEntityClass().getName() );
}
else if (work instanceof DeleteLuceneWork) {
processId( work, serializer );
serializer.addDelete( work.getEntityClass().getName() );
}
else if (work instanceof AddLuceneWork ) {
buildDocument( work.getDocument(), serializer );
processId( work, serializer );
serializer.addAdd( work.getEntityClass().getName(), work.getFieldToAnalyzerMap() );
}
else if (work instanceof UpdateLuceneWork ) {
buildDocument( work.getDocument(), serializer );
processId( work, serializer );
serializer.addUpdate( work.getEntityClass().getName(), work.getFieldToAnalyzerMap() );
}
}
return serializer.serialize();
}
catch (RuntimeException e) {
if ( e instanceof SearchException ) {
throw e;
}
else {
throw log.unableToSerializeLuceneWorks( e );
}
}
}
private void processId(LuceneWork work, Serializer serializer) {
Serializable id = work.getId();
if ( id instanceof Integer ) {
serializer.addIdAsInteger( (Integer) id );
}
else if ( id instanceof Long ) {
serializer.addIdAsLong( (Long) id );
}
else if ( id instanceof Float ) {
serializer.addIdAsFloat( (Float) id );
}
else if ( id instanceof Double ) {
serializer.addIdAsDouble( (Double) id );
}
else if ( id instanceof String ) {
serializer.addIdAsString( id.toString() );
}
else {
serializer.addIdSerializedInJava( toByteArray( id ) );
}
}
/**
* Convert a byte[] to a List of LuceneWork (assuming the same SerializationProvider is used of course)
*/
@Override
public List<LuceneWork> toLuceneWorks(byte[] data) {
try {
Deserializer deserializer = provider.getDeserializer();
LuceneWorkHydrator hydrator = new LuceneWorkHydrator( searchFactory );
deserializer.deserialize( data, hydrator );
return hydrator.getLuceneWorks();
}
catch (RuntimeException e) {
if ( e instanceof SearchException ) {
throw e;
}
else {
throw log.unableToReadSerializedLuceneWorks( e );
}
}
}
private void buildDocument(Document document, Serializer serializer) {
final List<IndexableField> docFields = document.getFields();
serializer.fields( docFields );
for ( IndexableField fieldable : docFields ) {
final FieldType fieldType = (FieldType) fieldable.fieldType();
final NumericType numericType = fieldType.numericType();
if ( numericType != null ) {
LuceneNumericFieldContext context = new LuceneNumericFieldContext( fieldType, fieldable.name(), fieldable.boost() );
switch ( numericType ) {
case INT:
serializer.addIntNumericField( fieldable.numericValue().intValue(), context );
break;
case LONG:
serializer.addLongNumericField( fieldable.numericValue().longValue(), context );
break;
case FLOAT:
serializer.addFloatNumericField( fieldable.numericValue().floatValue(), context );
break;
case DOUBLE:
serializer.addDoubleNumericField( fieldable.numericValue().doubleValue(), context );
break;
default:
String dataType = numericType.toString();
throw log.unknownNumericFieldType( dataType );
}
}
else if (fieldable instanceof Field) {
Field safeField = (Field) fieldable;
//FIXME it seems like in new Field implementation it's possible to have multiple data types at the same time. Investigate?
//The following sequence of else/ifs would not be appropriate.
if ( safeField.binaryValue() != null ) {
serializer.addFieldWithBinaryData( new LuceneFieldContext( safeField ) );
}
else if ( safeField.stringValue() != null ) {
serializer.addFieldWithStringData( new LuceneFieldContext( safeField ) );
}
else if ( safeField.readerValue() != null && safeField.readerValue() instanceof Serializable ) {
serializer.addFieldWithSerializableReaderData( new LuceneFieldContext( safeField ) );
}
else if ( safeField.readerValue() != null ) {
throw log.conversionFromReaderToStringNotYetImplemented();
}
else if ( safeField.tokenStreamValue() != null ) {
serializer.addFieldWithTokenStreamData( new LuceneFieldContext( safeField ) );
}
else {
throw log.unknownFieldType( safeField.getClass() );
}
}
else {
throw log.cannotSerializeCustomField( fieldable.getClass() );
}
}
serializer.addDocument();
}
@Override
public String describeSerializer() {
return provider.toString();
}
}