org.kie.kieora.backend.lucene.fields.SimpleFieldFactory.java Source code

Introduction

Here is the source code for org.kie.kieora.backend.lucene.fields.SimpleFieldFactory.java

Source

/*
 * Copyright 2012 JBoss 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 org.kie.kieora.backend.lucene.fields;

import java.util.Collection;
import java.util.Date;

import org.apache.lucene.document.DoubleField;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.FloatField;
import org.apache.lucene.document.IntField;
import org.apache.lucene.document.LongField;
import org.apache.lucene.document.StoredField;
import org.apache.lucene.document.StringField;
import org.apache.lucene.document.TextField;
import org.apache.lucene.index.IndexableField;
import org.kie.commons.java.nio.base.version.VersionHistory;
import org.kie.commons.java.nio.file.attribute.FileTime;
import org.kie.kieora.backend.lucene.FieldFactory;
import org.kie.kieora.model.KProperty;

import static org.kie.kieora.backend.lucene.util.Serializer.*;

public class SimpleFieldFactory implements FieldFactory {

    @Override
    public IndexableField[] build(final KProperty<?> property) {

        if (Enum.class.isAssignableFrom(property.getValue().getClass())) {
            if (property.isSearchable()) {
                return new IndexableField[] { new TextField(property.getName(),
                        property.getValue().toString().toLowerCase(), Field.Store.YES) };
            }
            return new IndexableField[] { new StringField(property.getName(),
                    property.getValue().toString().toLowerCase(), Field.Store.YES) };
        }

        if (property.getValue().getClass() == String.class) {
            if (property.isSearchable()) {
                return new IndexableField[] {
                        new TextField(property.getName(), property.getValue().toString(), Field.Store.YES) };
            }
            return new IndexableField[] {
                    new StringField(property.getName(), property.getValue().toString(), Field.Store.YES) };
        }

        if (property.getValue().getClass() == Boolean.class) {
            if (property.isSearchable()) {
                return new IndexableField[] { new TextField(property.getName(),
                        ((Boolean) property.getValue()) ? "0" : "1", Field.Store.YES) };
            }
            return new IndexableField[] { new StringField(property.getName(),
                    ((Boolean) property.getValue()) ? "0" : "1", Field.Store.YES) };
        }

        if (property.getValue().getClass() == Integer.class) {
            if (property.isSearchable()) {
                return new IndexableField[] {
                        new IntField(property.getName(), (Integer) property.getValue(), Field.Store.YES) };
            }
            return new IndexableField[] { new StoredField(property.getName(), (Integer) property.getValue()) };
        }

        if (property.getValue().getClass() == Long.class) {
            if (property.isSearchable()) {
                return new IndexableField[] {
                        new LongField(property.getName(), (Long) property.getValue(), Field.Store.YES) };
            }
            return new IndexableField[] { new StoredField(property.getName(), (Long) property.getValue()) };
        }

        if (property.getValue().getClass() == Double.class) {
            if (property.isSearchable()) {
                return new IndexableField[] {
                        new DoubleField(property.getName(), (Double) property.getValue(), Field.Store.YES) };
            }
            return new IndexableField[] { new StoredField(property.getName(), (Double) property.getValue()) };
        }

        if (property.getValue().getClass() == Float.class) {
            if (property.isSearchable()) {
                return new IndexableField[] {
                        new FloatField(property.getName(), (Float) property.getValue(), Field.Store.YES) };
            }
            return new IndexableField[] { new StoredField(property.getName(), (Float) property.getValue()) };
        }

        if (FileTime.class.isAssignableFrom(property.getValue().getClass())) {
            if (property.isSearchable()) {
                return new IndexableField[] { new LongField(property.getName(),
                        ((FileTime) property.getValue()).toMillis(), Field.Store.YES) };
            }
            return new IndexableField[] {
                    new StoredField(property.getName(), ((FileTime) property.getValue()).toMillis()) };
        }

        if (Date.class.isAssignableFrom(property.getValue().getClass())) {
            if (property.isSearchable()) {
                return new IndexableField[] { new LongField(property.getName(),
                        ((Date) property.getValue()).getTime(), Field.Store.YES) };
            }
            return new IndexableField[] {
                    new StoredField(property.getName(), ((Date) property.getValue()).getTime()) };
        }

        if (VersionHistory.class.isAssignableFrom(property.getValue().getClass())) {
            final VersionHistory versionHistory = (VersionHistory) property.getValue();
            return build(versionHistory);
        }

        if (Collection.class.isAssignableFrom(property.getValue().getClass())) {
            final StringBuilder sb = new StringBuilder();
            for (final java.lang.Object ovalue : (Collection) property.getValue()) {
                sb.append(ovalue).append(' ');
            }

            if (property.isSearchable()) {
                return new IndexableField[] { new TextField(property.getName(), sb.toString(), Field.Store.YES) };
            }
            return new IndexableField[] { new StringField(property.getName(), sb.toString(), Field.Store.YES) };
        }

        try {
            return new IndexableField[] { new StoredField(property.getName(), toByteArray(property.getValue())) };
        } catch (final Exception ex) {
            return new IndexableField[] { new StoredField(property.getName(), property.getValue().toString()) };
        }
    }

    private IndexableField[] build(final VersionHistory versionHistory) {
        if (versionHistory.records().size() == 0) {
            return new IndexableField[] {};
        }

        final int lastIndex = versionHistory.records().size() - 1;
        final IndexableField checkinComment = build(new KProperty<String>() {
            @Override
            public String getName() {
                return "checkinComment";
            }

            @Override
            public String getValue() {
                return versionHistory.records().get(lastIndex).comment();
            }

            @Override
            public boolean isSearchable() {
                return true;
            }
        })[0];

        final IndexableField lastModifiedBy = build(new KProperty<String>() {
            @Override
            public String getName() {
                return "lastModifiedBy";
            }

            @Override
            public String getValue() {
                return versionHistory.records().get(lastIndex).author();
            }

            @Override
            public boolean isSearchable() {
                return true;
            }
        })[0];

        final IndexableField createdBy = build(new KProperty<String>() {
            @Override
            public String getName() {
                return "createdBy";
            }

            @Override
            public String getValue() {
                return versionHistory.records().get(0).author();
            }

            @Override
            public boolean isSearchable() {
                return true;
            }
        })[0];

        final IndexableField createdDate = build(new KProperty<Date>() {
            @Override
            public String getName() {
                return "createdDate";
            }

            @Override
            public Date getValue() {
                return versionHistory.records().get(0).date();
            }

            @Override
            public boolean isSearchable() {
                return true;
            }
        })[0];

        final IndexableField lastModifiedDate = build(new KProperty<Date>() {
            @Override
            public String getName() {
                return "lastModifiedDate";
            }

            @Override
            public Date getValue() {
                return versionHistory.records().get(lastIndex).date();
            }

            @Override
            public boolean isSearchable() {
                return true;
            }
        })[0];

        return new IndexableField[] { checkinComment, lastModifiedBy, createdBy, createdDate, lastModifiedDate };

    }

}