| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 |
| * <p> |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * <p> |
| * 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.apache.atlas.web.rest; |
| |
| import org.apache.atlas.AtlasErrorCode; |
| import org.apache.atlas.exception.AtlasBaseException; |
| import org.apache.atlas.discovery.AtlasDiscoveryService; |
| import org.apache.atlas.model.discovery.AtlasSearchResult; |
| import org.apache.atlas.model.discovery.SearchParameters; |
| import org.apache.atlas.utils.AtlasPerfTracer; |
| import org.apache.atlas.web.util.Servlets; |
| import org.apache.commons.collections.CollectionUtils; |
| import org.springframework.stereotype.Service; |
| import org.apache.commons.lang3.StringUtils; |
| import org.slf4j.Logger; |
| |
| import javax.inject.Inject; |
| import javax.inject.Singleton; |
| import javax.ws.rs.Consumes; |
| import javax.ws.rs.GET; |
| import javax.ws.rs.POST; |
| import javax.ws.rs.Path; |
| import javax.ws.rs.Produces; |
| import javax.ws.rs.QueryParam; |
| |
| /** |
| * REST interface for data discovery using dsl or full text search |
| */ |
| @Path("v2/search") |
| @Singleton |
| @Service |
| public class DiscoveryREST { |
| private static final Logger PERF_LOG = AtlasPerfTracer.getPerfLogger("rest.DiscoveryREST"); |
| |
| private final AtlasDiscoveryService atlasDiscoveryService; |
| |
| @Inject |
| public DiscoveryREST(AtlasDiscoveryService discoveryService) { |
| this.atlasDiscoveryService = discoveryService; |
| } |
| |
| /** |
| * Retrieve data for the specified DSL |
| * @param query DSL query |
| * @param typeName limit the result to only entities of specified type or its sub-types |
| * @param classification limit the result to only entities tagged with the given classification or or its sub-types |
| * @param limit limit the result set to only include the specified number of entries |
| * @param offset start offset of the result set (useful for pagination) |
| * @return Search results |
| * @throws AtlasBaseException |
| * @HTTP 200 On successful DSL execution with some results, might return an empty list if execution succeeded |
| * without any results |
| * @HTTP 400 Invalid DSL or query parameters |
| */ |
| @GET |
| @Path("/dsl") |
| @Consumes(Servlets.JSON_MEDIA_TYPE) |
| @Produces(Servlets.JSON_MEDIA_TYPE) |
| public AtlasSearchResult searchUsingDSL(@QueryParam("query") String query, |
| @QueryParam("typeName") String typeName, |
| @QueryParam("classification") String classification, |
| @QueryParam("limit") int limit, |
| @QueryParam("offset") int offset) throws AtlasBaseException { |
| AtlasPerfTracer perf = null; |
| |
| try { |
| if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) { |
| perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "DiscoveryREST.searchUsingDSL(" + query + "," + typeName |
| + "," + classification + "," + limit + "," + offset + ")"); |
| } |
| |
| String queryStr = query == null ? "" : query; |
| |
| if (StringUtils.isNoneEmpty(typeName)) { |
| queryStr = escapeTypeName(typeName) + " " + queryStr; |
| } |
| |
| if (StringUtils.isNoneEmpty(classification)) { |
| // isa works with a type name only - like hive_column isa PII; it doesn't work with more complex query |
| if (StringUtils.isEmpty(query)) { |
| queryStr += (" isa " + classification); |
| } |
| } |
| |
| return atlasDiscoveryService.searchUsingDslQuery(queryStr, limit, offset); |
| } finally { |
| AtlasPerfTracer.log(perf); |
| } |
| } |
| |
| /** |
| * Retrieve data for the specified fulltext query |
| * @param query Fulltext query |
| * @param limit limit the result set to only include the specified number of entries |
| * @param offset start offset of the result set (useful for pagination) |
| * @return Search results |
| * @throws AtlasBaseException |
| * @HTTP 200 On successful FullText lookup with some results, might return an empty list if execution succeeded |
| * without any results |
| * @HTTP 400 Invalid fulltext or query parameters |
| */ |
| @GET |
| @Path("/fulltext") |
| @Consumes(Servlets.JSON_MEDIA_TYPE) |
| @Produces(Servlets.JSON_MEDIA_TYPE) |
| public AtlasSearchResult searchUsingFullText(@QueryParam("query") String query, |
| @QueryParam("excludeDeletedEntities") boolean excludeDeletedEntities, |
| @QueryParam("limit") int limit, |
| @QueryParam("offset") int offset) throws AtlasBaseException { |
| AtlasPerfTracer perf = null; |
| |
| try { |
| if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) { |
| perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "DiscoveryREST.searchUsingFullText(" + query + "," + |
| limit + "," + offset + ")"); |
| } |
| |
| return atlasDiscoveryService.searchUsingFullTextQuery(query, excludeDeletedEntities, limit, offset); |
| } finally { |
| AtlasPerfTracer.log(perf); |
| } |
| } |
| |
| /** |
| * Retrieve data for the specified fulltext query |
| * @param query Fulltext query |
| * @param typeName limit the result to only entities of specified type or its sub-types |
| * @param classification limit the result to only entities tagged with the given classification or or its sub-types |
| * @param limit limit the result set to only include the specified number of entries |
| * @param offset start offset of the result set (useful for pagination) |
| * @return Search results |
| * @throws AtlasBaseException |
| * @HTTP 200 On successful FullText lookup with some results, might return an empty list if execution succeeded |
| * without any results |
| * @HTTP 400 Invalid fulltext or query parameters |
| */ |
| @GET |
| @Path("/basic") |
| @Consumes(Servlets.JSON_MEDIA_TYPE) |
| @Produces(Servlets.JSON_MEDIA_TYPE) |
| public AtlasSearchResult searchUsingBasic(@QueryParam("query") String query, |
| @QueryParam("typeName") String typeName, |
| @QueryParam("classification") String classification, |
| @QueryParam("excludeDeletedEntities") boolean excludeDeletedEntities, |
| @QueryParam("limit") int limit, |
| @QueryParam("offset") int offset) throws AtlasBaseException { |
| AtlasPerfTracer perf = null; |
| |
| try { |
| if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) { |
| perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "DiscoveryREST.searchUsingBasic(" + query + "," + |
| typeName + "," + classification + "," + limit + "," + offset + ")"); |
| } |
| |
| return atlasDiscoveryService.searchUsingBasicQuery(query, typeName, classification, null, null, |
| excludeDeletedEntities, limit, offset); |
| } finally { |
| AtlasPerfTracer.log(perf); |
| } |
| } |
| |
| /** |
| * Retrieve data for the specified attribute search query |
| * @param attrName Attribute name |
| * @param attrValuePrefix Attibute value to search on |
| * @param typeName limit the result to only entities of specified type or its sub-types |
| * @param limit limit the result set to only include the specified number of entries |
| * @param offset start offset of the result set (useful for pagination) |
| * @return Search results |
| * @throws AtlasBaseException |
| * @HTTP 200 On successful FullText lookup with some results, might return an empty list if execution succeeded |
| * without any results |
| * @HTTP 400 Invalid wildcard or query parameters |
| */ |
| @GET |
| @Path("/attribute") |
| @Consumes(Servlets.JSON_MEDIA_TYPE) |
| @Produces(Servlets.JSON_MEDIA_TYPE) |
| public AtlasSearchResult searchUsingAttribute(@QueryParam("attrName") String attrName, |
| @QueryParam("attrValuePrefix") String attrValuePrefix, |
| @QueryParam("typeName") String typeName, |
| @QueryParam("limit") int limit, |
| @QueryParam("offset") int offset) throws AtlasBaseException { |
| AtlasPerfTracer perf = null; |
| |
| try { |
| if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) { |
| perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "DiscoveryREST.searchUsingAttribute(" + attrName + "," + |
| attrValuePrefix + "," + typeName + "," + limit + "," + offset + ")"); |
| } |
| |
| if (StringUtils.isEmpty(attrName) && StringUtils.isEmpty(attrValuePrefix)) { |
| throw new AtlasBaseException(AtlasErrorCode.INVALID_PARAMETERS, |
| String.format("attrName : {0}, attrValue: {1} for attribute search.", attrName, attrValuePrefix)); |
| } |
| |
| return atlasDiscoveryService.searchUsingBasicQuery(null, typeName, null, attrName, attrValuePrefix, true, limit, offset); |
| |
| } finally { |
| AtlasPerfTracer.log(perf); |
| } |
| } |
| |
| /** |
| * Attribute based search for entities satisfying the search parameters |
| * @param parameters Search parameters |
| * @return Atlas search result |
| * @throws AtlasBaseException |
| * |
| * @HTTP 200 On successful search |
| * @HTTP 400 Tag/Entity doesn't exist or Tag/entity filter is present without tag/type name |
| */ |
| @Path("basic") |
| @POST |
| @Consumes(Servlets.JSON_MEDIA_TYPE) |
| @Produces(Servlets.JSON_MEDIA_TYPE) |
| public AtlasSearchResult searchWithParameters(SearchParameters parameters) throws AtlasBaseException { |
| AtlasPerfTracer perf = null; |
| |
| try { |
| if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) { |
| perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "DiscoveryREST.searchWithParameters("+ parameters + ")"); |
| } |
| |
| if (parameters.getLimit() < 0 || parameters.getOffset() < 0) { |
| throw new AtlasBaseException(AtlasErrorCode.BAD_REQUEST, "Limit/offset should be non-negative"); |
| } |
| |
| if (StringUtils.isEmpty(parameters.getTypeName()) && !isEmpty(parameters.getEntityFilters())) { |
| throw new AtlasBaseException(AtlasErrorCode.BAD_REQUEST, "EntityFilters specified without Type name"); |
| } |
| |
| if (StringUtils.isEmpty(parameters.getClassification()) && !isEmpty(parameters.getTagFilters())) { |
| throw new AtlasBaseException(AtlasErrorCode.BAD_REQUEST, "TagFilters specified without tag name"); |
| } |
| |
| return atlasDiscoveryService.searchUsingBasicQuery(parameters); |
| } finally { |
| AtlasPerfTracer.log(perf); |
| } |
| } |
| |
| private boolean isEmpty(SearchParameters.FilterCriteria filterCriteria) { |
| return filterCriteria == null || |
| (StringUtils.isEmpty(filterCriteria.getAttributeName()) && CollectionUtils.isEmpty(filterCriteria.getCriterion())); |
| } |
| |
| private String escapeTypeName(String typeName) { |
| String ret; |
| |
| if (StringUtils.startsWith(typeName, "`") && StringUtils.endsWith(typeName, "`")) { |
| ret = typeName; |
| } else { |
| ret = String.format("`%s`", typeName); |
| } |
| |
| return ret; |
| } |
| } |