io.curly.bloodhound.service.ElasticQueryService.java Source code

Introduction

Here is the source code for io.curly.bloodhound.service.ElasticQueryService.java

Source

/*
 *        Copyright 2015 the original author or authors.
 *
 *    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 io.curly.bloodhound.service;

import com.netflix.hystrix.contrib.javanica.annotation.HystrixCommand;
import com.netflix.hystrix.contrib.javanica.command.ObservableResult;
import io.curly.bloodhound.model.Artifact;
import io.curly.bloodhound.model.ArtifactRepository;
import io.curly.bloodhound.query.search.ElasticQueryBuilder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import rx.Observable;

import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;

/**
 * @author Joo Pedro Evangelista
 */
@Slf4j
@Service
public class ElasticQueryService {

    private final ArtifactRepository repository;
    private final ElasticQueryBuilder elasticQueryBuilder;

    @Autowired
    public ElasticQueryService(ArtifactRepository repository, ElasticQueryBuilder elasticQueryBuilder) {
        this.repository = repository;
        this.elasticQueryBuilder = elasticQueryBuilder;
    }

    @HystrixCommand(fallbackMethod = "noResults")
    public Observable<List<Artifact>> getResults(String query) {
        log.debug("Querying data for {}", query);
        return new ObservableResult<List<Artifact>>() {
            @Override
            public List<Artifact> invoke() {
                return StreamSupport
                        .stream(repository.search(elasticQueryBuilder.build(query)).spliterator(), false)
                        .collect(Collectors.toList());
            }
        };
    }

    List<Artifact> noResults(String query) {
        log.warn("getResults failed!! the requested query was {}", query);
        return Collections.emptyList();
    }
}