org.kie.kieora.io.IOSearchIndex Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kieora-commons-io Show documentation
Kieora Commons IO
There is a newer version: 6.0.0.CR5
/*
 * Copyright 2013 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.io;

import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.kie.commons.io.IOSearchService;
import org.kie.commons.io.IOService;
import org.kie.commons.java.nio.base.FileSystemId;
import org.kie.commons.java.nio.base.SegmentedPath;
import org.kie.commons.java.nio.file.Path;
import org.kie.kieora.model.KObject;
import org.kie.kieora.search.ClusterSegment;
import org.kie.kieora.search.SearchIndex;

import static org.kie.commons.validation.PortablePreconditions.*;

/**
 *
 */
public class IOSearchIndex implements IOSearchService {

    private final SearchIndex searchIndex;
    private final IOService   ioService;

    public IOSearchIndex( final SearchIndex searchIndex,
                          final IOService ioService ) {
        this.searchIndex = checkNotNull( "searchIndex", searchIndex );
        this.ioService = checkNotNull( "ioService", ioService );
    }

    @Override
    public List searchByAttrs( final Map attrs,
                                     final int pageSize,
                                     final int startIndex,
                                     final Path... roots ) {
        final List kObjects = searchIndex.searchByAttrs( attrs, pageSize, startIndex, buildClusterSegments( roots ) );
        return new ArrayList() {{
            for ( KObject kObject : kObjects ) {
                add( ioService.get( URI.create( kObject.getKey() ) ) );
            }
        }};
    }

    @Override
    public List fullTextSearch( final String term,
                                      final int pageSize,
                                      final int startIndex,
                                      final Path... roots ) {
        final List kObjects = searchIndex.fullTextSearch( term, pageSize, startIndex, buildClusterSegments( roots ) );
        return new ArrayList() {{
            for ( KObject kObject : kObjects ) {
                add( ioService.get( URI.create( kObject.getKey() ) ) );
            }
        }};
    }

    @Override
    public int searchByAttrsHits( final Map attrs,
                                  final Path... roots ) {
        return searchIndex.searchByAttrsHits( attrs, buildClusterSegments( roots ) );
    }

    @Override
    public int fullTextSearchHits( final String term,
                                   final Path... roots ) {
        return searchIndex.fullTextSearchHits( term, buildClusterSegments( roots ) );
    }

    private ClusterSegment[] buildClusterSegments( final Path[] roots ) {
        if ( roots == null || roots.length == 0 ) {
            return new ClusterSegment[ 0 ];
        }
        final ClusterSegment[] clusterSegments = new ClusterSegment[ roots.length ];
        for ( int i = 0; i < roots.length; i++ ) {
            final Path root = roots[ i ];
            final SegmentedPath segmentedPath = (SegmentedPath) root;
            final FileSystemId fsId = (FileSystemId) root.getFileSystem();
            clusterSegments[ i ] = new ClusterSegment() {
                @Override
                public String getClusterId() {
                    return fsId.id();
                }

                @Override
                public String[] segmentIds() {
                    return new String[]{ segmentedPath.getSegmentId() };
                }
            };
        }
        return clusterSegments;
    }
}