content: /**
   * 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.util;
  
  public class AtlasGremlin2QueryProvider extends AtlasGremlinQueryProvider {
      @Override
      public String getQuery(final AtlasGremlinQuery gremlinQuery) {
          switch (gremlinQuery) {
              case ENTITY_ACTIVE_METRIC:
                  return "g.V().has('__typeName', T.in, [%s]).has('__state', 'ACTIVE').groupCount{it.getProperty('__typeName')}.cap.toList()";
              case ENTITY_DELETED_METRIC:
                  return "g.V().has('__typeName', T.in, [%s]).has('__state', 'DELETED').groupCount{it.getProperty('__typeName')}.cap.toList()";
  
              case EXPORT_BY_GUID_FULL:
                  return "g.V('__guid', startGuid).bothE().bothV().has('__guid').transform{[__guid:it.__guid,isProcess:(it.__superTypeNames != null) ? it.__superTypeNames.contains('Process') : false ]}.dedup().toList()";
              case EXPORT_BY_GUID_CONNECTED_IN_EDGE:
                  return "g.V('__guid', startGuid).inE().outV().has('__guid').transform{[__guid:it.__guid,isProcess:(it.__superTypeNames != null) ? it.__superTypeNames.contains('Process') : false ]}.dedup().toList()";
              case EXPORT_BY_GUID_CONNECTED_OUT_EDGE:
                  return "g.V('__guid', startGuid).outE().inV().has('__guid').transform{[__guid:it.__guid,isProcess:(it.__superTypeNames != null) ? it.__superTypeNames.contains('Process') : false ]}.dedup().toList()";
              case EXPORT_TYPE_ALL_FOR_TYPE:
                  return "g.V().has('__typeName',T.in,typeName).has('__guid').__guid.toList()";
              case EXPORT_TYPE_STARTS_WITH:
                  return "g.V().has('__typeName',typeName).filter({it.getProperty(attrName).startsWith(attrValue)}).has('__guid').__guid.toList()";
              case EXPORT_TYPE_ENDS_WITH:
                  return "g.V().has('__typeName',typeName).filter({it.getProperty(attrName).endsWith(attrValue)}).has('__guid').__guid.toList()";
              case EXPORT_TYPE_CONTAINS:
                  return "g.V().has('__typeName',typeName).filter({it.getProperty(attrName).contains(attrValue)}).has('__guid').__guid.toList()";
              case EXPORT_TYPE_MATCHES:
                  return "g.V().has('__typeName',typeName).filter({it.getProperty(attrName).matches(attrValue)}).has('__guid').__guid.toList()";
              case EXPORT_TYPE_DEFAULT:
                  return "g.V().has('__typeName',typeName).has(attrName, attrValue).has('__guid').__guid.toList()";
              case FULL_LINEAGE_DATASET:
                  return "g.V('__guid', '%s').as('src').in('%s').out('%s')." +
                          "loop('src', {((it.path.contains(it.object)) ? false : true)}, " +
                          "{((it.object.'__superTypeNames') ? " +
                          "(it.object.'__superTypeNames'.contains('DataSet')) : false)})." +
                          "path().toList()";
              case PARTIAL_LINEAGE_DATASET:
                  return "g.V('__guid', '%s').as('src').in('%s').out('%s')." +
                          "loop('src', {it.loops <= %s}, {((it.object.'__superTypeNames') ? " +
                          "(it.object.'__superTypeNames'.contains('DataSet')) : false)})." +
                          "path().toList()";
  
              case BASIC_SEARCH_TYPE_FILTER:
                  return ".has('__typeName', within(typeNames))";
              case BASIC_SEARCH_CLASSIFICATION_FILTER:
                  return ".or(has('__traitNames', within(traitNames)), has('__propagatedTraitNames', within(traitNames)))";
              case BASIC_SEARCH_STATE_FILTER:
                  return ".has('__state', state)";
              case TO_RANGE_LIST:
                  return " [startIdx..<endIdx].toList()";
              case GUID_PREFIX_FILTER:
                  return ".filter{it.'__guid'.matches(guid)}";
              case COMPARE_LT:
                  return ".has('%s', lt(%s))";
              case COMPARE_LTE:
                  return ".has('%s', lte(%s))";
              case COMPARE_GT:
                  return ".has('%s', gt(%s))";
              case COMPARE_GTE:
                  return ".has('%s', gte(%s))";
              case COMPARE_EQ:
                  return ".has('%s', eq(%s))";
              case COMPARE_NEQ:
                  return ".has('%s', neq(%s))";
              case COMPARE_MATCHES:
                  return ".filter({it.getProperty('%s').matches(%s)})";
              case COMPARE_STARTS_WITH:
                  return ".filter({it.getProperty('%s').startsWith(%s)})";
              case COMPARE_ENDS_WITH:
                  return ".filter({it.getProperty('%s').endsWith(%s)})";
              case COMPARE_CONTAINS:
                  return ".filter({it.getProperty('%s').contains(%s)})";
              case COMPARE_IS_NULL:
                  return ".hasNot('%s')";
              case COMPARE_NOT_NULL:
                  return ".has('%s')";
              case RELATIONSHIP_SEARCH:
                  return "g.V('__guid', guid).both(relation).has('__state', within(states))";
              case RELATIONSHIP_SEARCH_ASCENDING_SORT:
                  return ".order{it.a.getProperty(sortAttributeName) <=> it.b.getProperty(sortAttributeName)}";
              case RELATIONSHIP_SEARCH_DESCENDING_SORT:
                  return ".order{it.b.getProperty(sortAttributeName) <=> it.a.getProperty(sortAttributeName)}";
              case GREMLIN_SEARCH_RETURNS_VERTEX_ID:
                  return "g.V.range(0,0).collect()";
              case GREMLIN_SEARCH_RETURNS_EDGE_ID:
                  return "g.E.range(0,0).collect()";
          }
          // Should never reach this point
          return null;
      }
  }
  