org.apache.calcite.prepare.SqlIdentifierCapturer Class Reference

Collaboration diagram for org.apache.calcite.prepare.SqlIdentifierCapturer:

Public Member Functions
void	scan (SqlNode root)
String	toString ()

Static Public Member Functions
static void	main (String[] args) throws Exception

Public Attributes
Set< ImmutableList< String > >	selects = new HashSet<>()
Set< ImmutableList< String > >	inserts = new HashSet<>()
Set< ImmutableList< String > >	updates = new HashSet<>()
Set< ImmutableList< String > >	deletes = new HashSet<>()

Package Functions
	[instance initializer]
Set< Method >	getRelevantGetters (Object obj)

Static Package Attributes
static final Logger	HEAVYDBLOGGER

Private Attributes
IdentityHashMap< SqlNode, SqlNode >	visitedNodes = new IdentityHashMap<>()
Stack< Set< ImmutableList < String > > >	currentList = new Stack<>()
final Set< ImmutableList < String > >	ignore = new HashSet<>()

Static Private Attributes
static final Map< Class <?>, Set< Method > >	GETTERS_CACHE

Detailed Description

scans from a root SqlNode and retrieves all SqlIdentifiers used in a query.

Definition at line 54 of file SqlIdentifierCapturer.java.

Member Function Documentation

org.apache.calcite.prepare.SqlIdentifierCapturer.[instance initializer] ( )

inlinepackage

Set<Method> org.apache.calcite.prepare.SqlIdentifierCapturer.getRelevantGetters ( Object  obj )

inlinepackage

Definition at line 205 of file SqlIdentifierCapturer.java.

References setup.root.

Referenced by org.apache.calcite.prepare.SqlIdentifierCapturer.scan().

                                             {
    Class<?> root = obj.getClass();

    Set<Method> methods = GETTERS_CACHE.get(root);
    if (null != methods) {
      return methods;
    } else {
      methods = new HashSet<>();
    }

    while (root != null) {
      if (root == SqlNode.class) break;

      for (Method m : root.getDeclaredMethods()) {
        if (m.getParameterTypes().length > 0) continue;

        if (!Modifier.isPublic(m.getModifiers())) continue;

        Class<?> returnType = m.getReturnType();
        if (!SqlNode.class.isAssignableFrom(returnType) && SqlNode[].class != returnType
                && !Collection.class.isAssignableFrom(returnType)) {
          continue;
        }

        methods.add(m);
      }

      root = root.getSuperclass();
    }

    GETTERS_CACHE.put(obj.getClass(), methods);

    return methods;
  }

Here is the caller graph for this function:

static void org.apache.calcite.prepare.SqlIdentifierCapturer.main ( String[]  args ) throws Exception

inlinestatic

Definition at line 251 of file SqlIdentifierCapturer.java.

References run_benchmark_import.parser.

                                                          {
    String sql = "UPDATE sales set f=(SELECT max(r.num) from report as r)";
    sql = "INSER INTO sales (a, b, c ) VALUES(10, (SELECT max(foo) from bob), 0)";
    sql = "SELECT * FROM sales a left outer join (select (select max(id) from rupert) from report2) r on a.id=(select max(r.di) from test)";

    SqlParser parser = SqlParser.create(sql);

    SqlIdentifierCapturer capturer = new SqlIdentifierCapturer();
    capturer.scan(parser.parseQuery());

    System.out.println(capturer.selects);
    System.out.println(capturer.inserts);
    System.out.println(capturer.updates);
    System.out.println(capturer.deletes);
    System.out.println(capturer.ignore);
  }

void org.apache.calcite.prepare.SqlIdentifierCapturer.scan ( SqlNode  root )

inline

Definition at line 75 of file SqlIdentifierCapturer.java.

References org.apache.calcite.prepare.SqlIdentifierCapturer.deletes, org.apache.calcite.prepare.SqlIdentifierCapturer.getRelevantGetters(), org.apache.calcite.prepare.SqlIdentifierCapturer.ignore, org.apache.calcite.prepare.SqlIdentifierCapturer.inserts, setup.root, org.apache.calcite.prepare.SqlIdentifierCapturer.selects, org.apache.calcite.prepare.SqlIdentifierCapturer.updates, and org.apache.calcite.prepare.SqlIdentifierCapturer.visitedNodes.

                                 {
    if (null == root) {
      return;
    }

    if (root instanceof SqlLiteral || root instanceof SqlDataTypeSpec) {
      return;
    }

    if (null != visitedNodes.put(root, root)) {
      return;
    }

    if (root instanceof SqlNodeList) {
      SqlNodeList snl = (SqlNodeList) root;
      for (SqlNode node : snl) {
        scan(node);
      }
      return;
    }

    if (root instanceof SqlIdentifier) {
      // we need all the hierachy now to deal with multischema
      currentList.peek().add(((SqlIdentifier) root).names.reverse());
      return;
    }

    if (root instanceof SqlBasicCall) {
      SqlBasicCall call = (SqlBasicCall) root;
      if (call.getOperator().getKind() == SqlKind.ARGUMENT_ASSIGNMENT) {
        // We have a => named parameter operator
        // We need to ignore it as otherwise we will pick up literal args
        // as tables, EXCEPT if it points to a CURSOR operator, as there
        // will be at least one table inside of the CURSOR.
        if (call.operandCount() == 0) {
          return;
        }
        if (call.getOperands()[0].getKind() == SqlKind.CURSOR) {
          SqlBasicCall cursor_call = (SqlBasicCall) call.getOperands()[0];
          if (cursor_call.operandCount() == 0) {
            return;
          }
          scan(cursor_call.getOperands()[0]);
          return;
        } else {
          return;
        }
      } else if (call.getOperator().getKind() == SqlKind.AS) {
        // only really interested in the first operand
        scan(call.getOperands()[0]);
        return;
      }
    }

    if (root instanceof SqlOrderBy) {
      scan(((SqlOrderBy) root).fetch);
      scan(((SqlOrderBy) root).offset);
      scan(((SqlOrderBy) root).query);
      return;
    }

    boolean needsPop = false;
    if (root instanceof SqlSelect) {
      currentList.push(selects);
      scan(((SqlSelect) root).getFrom());
      currentList.pop();
      currentList.push(ignore);
      needsPop = true;
    } else if (root instanceof SqlInsert) {
      currentList.push(inserts);
      scan(((SqlInsert) root).getTargetTable());
      currentList.pop();
      currentList.push(ignore);
      needsPop = true;
    } else if (root instanceof SqlUpdate) {
      currentList.push(updates);
      scan(((SqlUpdate) root).getTargetTable());
      currentList.pop();
      currentList.push(ignore);
      needsPop = true;
    } else if (root instanceof SqlDelete) {
      currentList.push(deletes);
      scan(((SqlDelete) root).getTargetTable());
      currentList.pop();
      currentList.push(ignore);
      needsPop = true;
    } else if (root instanceof SqlJoin) {
      currentList.push(ignore);
      scan(((SqlJoin) root).getCondition());
      currentList.pop();
    }

    Set<Method> methods = getRelevantGetters(root);
    for (Method m : methods) {
      Object value = null;
      try {
        value = m.invoke(root);
      } catch (Exception e) {
      }

      if (value instanceof SqlNode[]) {
        SqlNode[] nodes = (SqlNode[]) value;
        for (SqlNode node : nodes) {
          scan(node);
        }
      } else if (value instanceof SqlNode) {
        scan((SqlNode) value);
      } else if (value instanceof Collection) {
        for (Object vobj : ((Collection<?>) value)) {
          if (vobj instanceof SqlNode) {
            scan((SqlNode) vobj);
          }
        }
      }
    }

    if (root instanceof SqlWith) {
      SqlWith with = (SqlWith) root;

      for (SqlNode node : with.withList) {
        SqlWithItem item = (SqlWithItem) node;
        selects.remove(((SqlIdentifier) item.name).names.reverse());
      }
    }

    if (needsPop) {
      currentList.pop();
    }
  }

Here is the call graph for this function:

String org.apache.calcite.prepare.SqlIdentifierCapturer.toString ( )

inline

Definition at line 240 of file SqlIdentifierCapturer.java.

References org.apache.calcite.prepare.SqlIdentifierCapturer.deletes, org.apache.calcite.prepare.SqlIdentifierCapturer.ignore, org.apache.calcite.prepare.SqlIdentifierCapturer.inserts, org.apache.calcite.prepare.SqlIdentifierCapturer.selects, and org.apache.calcite.prepare.SqlIdentifierCapturer.updates.

                           {
    String out = "";
    out += " Selects: " + selects + "\n";
    out += " Inserts: " + inserts + "\n";
    out += " Updates: " + updates + "\n";
    out += " Deletes: " + deletes + "\n";
    out += " Ignore : " + ignore + "\n";

    return out;
  }

Member Data Documentation

Stack<Set<ImmutableList<String> > > org.apache.calcite.prepare.SqlIdentifierCapturer.currentList = new Stack<>()

private

Definition at line 63 of file SqlIdentifierCapturer.java.

Set<ImmutableList<String> > org.apache.calcite.prepare.SqlIdentifierCapturer.deletes = new HashSet<>()

Definition at line 69 of file SqlIdentifierCapturer.java.

Referenced by com.mapd.calcite.parser.HeavyDBParser.captureIdentifiers(), org.apache.calcite.prepare.SqlIdentifierCapturer.scan(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testDelete(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testInsert(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testSelect(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testUpdate(), and org.apache.calcite.prepare.SqlIdentifierCapturer.toString().

final Map<Class<?>, Set<Method> > org.apache.calcite.prepare.SqlIdentifierCapturer.GETTERS_CACHE

staticprivate

Initial value:

=
          new ConcurrentHashMap<>()

Definition at line 58 of file SqlIdentifierCapturer.java.

final Logger org.apache.calcite.prepare.SqlIdentifierCapturer.HEAVYDBLOGGER

staticpackage

Initial value:

=
          LoggerFactory.getLogger(SqlIdentifierCapturer.class)

Definition at line 55 of file SqlIdentifierCapturer.java.

final Set<ImmutableList<String> > org.apache.calcite.prepare.SqlIdentifierCapturer.ignore = new HashSet<>()

private

Definition at line 71 of file SqlIdentifierCapturer.java.

Referenced by org.apache.calcite.prepare.SqlIdentifierCapturer.scan(), and org.apache.calcite.prepare.SqlIdentifierCapturer.toString().

Set<ImmutableList<String> > org.apache.calcite.prepare.SqlIdentifierCapturer.inserts = new HashSet<>()

Definition at line 67 of file SqlIdentifierCapturer.java.

Referenced by com.mapd.calcite.parser.HeavyDBParser.captureIdentifiers(), org.apache.calcite.prepare.SqlIdentifierCapturer.scan(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testDelete(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testInsert(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testSelect(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testUpdate(), and org.apache.calcite.prepare.SqlIdentifierCapturer.toString().

Set<ImmutableList<String> > org.apache.calcite.prepare.SqlIdentifierCapturer.selects = new HashSet<>()

Definition at line 66 of file SqlIdentifierCapturer.java.

Referenced by com.mapd.calcite.parser.HeavyDBParser.captureIdentifiers(), com.mapd.calcite.parser.HeavyDBParser.resolveSelectIdentifiers(), org.apache.calcite.prepare.SqlIdentifierCapturer.scan(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testDelete(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testInsert(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testSelect(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testUpdate(), and org.apache.calcite.prepare.SqlIdentifierCapturer.toString().

Set<ImmutableList<String> > org.apache.calcite.prepare.SqlIdentifierCapturer.updates = new HashSet<>()

Definition at line 68 of file SqlIdentifierCapturer.java.

Referenced by com.mapd.calcite.parser.HeavyDBParser.captureIdentifiers(), org.apache.calcite.prepare.SqlIdentifierCapturer.scan(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testDelete(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testInsert(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testSelect(), com.mapd.parser.server.test.SqlIdentifierCapturerTest.testUpdate(), and org.apache.calcite.prepare.SqlIdentifierCapturer.toString().

IdentityHashMap<SqlNode, SqlNode> org.apache.calcite.prepare.SqlIdentifierCapturer.visitedNodes = new IdentityHashMap<>()

private

Definition at line 61 of file SqlIdentifierCapturer.java.

Referenced by org.apache.calcite.prepare.SqlIdentifierCapturer.scan().

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/java/calcite/src/main/java/org/apache/calcite/prepare/SqlIdentifierCapturer.java