Java tutorial
/* * * * This file is part of the Hesperides distribution. * * (https://github.com/voyages-sncf-technologies/hesperides) * * Copyright (c) 2016 VSCT. * * * * Hesperides is free software: you can redistribute it and/or modify * * it under the terms of the GNU General Public License as * * published by the Free Software Foundation, version 3. * * * * Hesperides is distributed in the hope that it will be useful, but * * WITHOUT ANY WARRANTY; without even the implied warranty of * * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * * General Public License for more details. * * * * You should have received a copy of the GNU General Public License * * along with this program. If not, see <http://www.gnu.org/licenses/>. * * */ package com.vsct.dt.hesperides.indexation.search; import com.fasterxml.jackson.databind.JavaType; import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.databind.ObjectReader; import com.github.mustachejava.DefaultMustacheFactory; import com.github.mustachejava.Mustache; import com.github.mustachejava.MustacheFactory; import com.vsct.dt.hesperides.indexation.ElasticSearchClient; import com.vsct.dt.hesperides.indexation.model.ElasticSearchResponse; import com.vsct.dt.hesperides.util.TemplateContentGenerator; import io.dropwizard.jackson.Jackson; import java.util.Set; import java.util.stream.Collectors; /** * Created by william_montaz on 09/10/2014. */ public class ApplicationSearch { public static final int SEARCH_SIZE = 50; /** * Mustache factory. */ private final MustacheFactory mustacheFactory = new DefaultMustacheFactory(); /** * The Mustache Template for elastic search requests. */ private final Mustache mustacheSearchByNameLike = mustacheFactory .compile("search.application.name.like.mustache"); private final Mustache mustacheSearchAllPlatform = mustacheFactory .compile("search.platform.name.like.mustache"); private final Mustache mustacheSearchAllPlatformUsingModules = mustacheFactory .compile("search.platform.using.modules.mustache"); /** * The elasticSearch Client Helper. */ private final ElasticSearchClient elasticSearchClient; /** * A jackson reader for elastic search responses. */ private final ObjectReader elasticSearchVsctApplicationReader; private final ObjectReader elasticSearchVsctPlatformReader; private final ObjectReader elasticSearchVsctPlatformApplicationReader; /** * Main constructor. * * @param elasticSearchClient */ public ApplicationSearch(final ElasticSearchClient elasticSearchClient) { ObjectMapper objectMapper = Jackson.newObjectMapper(); JavaType searchType = objectMapper.getTypeFactory().constructParametricType(ElasticSearchResponse.class, ApplicationSearchResponse.class); this.elasticSearchVsctApplicationReader = objectMapper.reader(searchType); JavaType searchTypePlatform = objectMapper.getTypeFactory() .constructParametricType(ElasticSearchResponse.class, PlatformSearchResponse.class); this.elasticSearchVsctPlatformReader = objectMapper.reader(searchTypePlatform); JavaType searchTypePlatformApplication = objectMapper.getTypeFactory() .constructParametricType(ElasticSearchResponse.class, PlatformApplicationSearchResponse.class); this.elasticSearchVsctPlatformApplicationReader = objectMapper.reader(searchTypePlatformApplication); this.elasticSearchClient = elasticSearchClient; } /** * Find an application with a name or just a part of a name. * * @param name * @return a set of applications matching request */ public Set<ApplicationSearchResponse> getApplicationsLike(final String name) { String url = String.format("/platforms/_search?size=%1$s", SEARCH_SIZE); String body = TemplateContentGenerator.from(mustacheSearchByNameLike) .put("applicationName", name.toLowerCase()).generate(); ElasticSearchResponse<ApplicationSearchResponse> esResponse = elasticSearchClient .withResponseReader(elasticSearchVsctApplicationReader).post(url, body); return esResponse.streamOfData().collect(Collectors.toSet()); } /** * Find all platforms for a given application name. * * @return a set of plaforms matching request */ public Set<PlatformSearchResponse> getAllPlatforms(final String name, final String platformName) { String url = String.format("/platforms/_search?size=%1$s", SEARCH_SIZE); String body = TemplateContentGenerator.from(mustacheSearchAllPlatform) .put("applicationName", name.toLowerCase()).put("platformName", platformName).generate(); ElasticSearchResponse<PlatformSearchResponse> esResponse = elasticSearchClient .withResponseReader(elasticSearchVsctPlatformReader).post(url, body); return esResponse.streamOfData().collect(Collectors.toSet()); } /** * Find all platforms using by a module. * * @return a set of plaforms matching request */ public Set<PlatformApplicationSearchResponse> getAllPlatformsUsingModules(final String moduleName, final String moduleVersion, final String isWorkingCopy) { String url = String.format("/platforms/_search?size=%1$s", SEARCH_SIZE); boolean boolIsWorkingCopy = true; if (isWorkingCopy.contains("release")) { boolIsWorkingCopy = false; } String body = TemplateContentGenerator.from(mustacheSearchAllPlatformUsingModules) .put("moduleName", moduleName).put("moduleVersion", moduleVersion) .put("isWorkingCopy", boolIsWorkingCopy).generate(); ElasticSearchResponse<PlatformApplicationSearchResponse> esResponse = elasticSearchClient .withResponseReader(elasticSearchVsctPlatformApplicationReader).post(url, body); return esResponse.streamOfData().collect(Collectors.toSet()); } }