org.apache.calcite.rel.rules.InjectFilterRule Class Reference

Inheritance diagram for org.apache.calcite.rel.rules.InjectFilterRule:

Collaboration diagram for org.apache.calcite.rel.rules.InjectFilterRule:

Classes
interface	Config

Public Member Functions
	InjectFilterRule (Config config, List< Restriction > restrictions)
void	onMatch (RelOptRuleCall call)

Static Public Attributes
static Set< String >	visitedMemo = new HashSet<>()

Package Functions
void	clearMemo ()

Package Attributes
final List< Restriction >	restrictions

Static Package Attributes
static final Logger	HEAVYDBLOGGER = LoggerFactory.getLogger(InjectFilterRule.class)

Detailed Description

Definition at line 38 of file InjectFilterRule.java.

Constructor & Destructor Documentation

org.apache.calcite.rel.rules.InjectFilterRule.InjectFilterRule ( Config  config, List< Restriction >  restrictions  )

inline

Definition at line 45 of file InjectFilterRule.java.

References org.apache.calcite.rel.rules.InjectFilterRule.clearMemo(), and org.apache.calcite.rel.rules.InjectFilterRule.restrictions.

Referenced by org.apache.calcite.rel.rules.InjectFilterRule.Config.toRule().

                                                                         {
    super(config);
    this.restrictions = restrictions;
    clearMemo();
  }

Here is the call graph for this function:

Here is the caller graph for this function:

Member Function Documentation

void org.apache.calcite.rel.rules.InjectFilterRule.clearMemo ( )

inlinepackage

Definition at line 51 of file InjectFilterRule.java.

Referenced by org.apache.calcite.rel.rules.InjectFilterRule.InjectFilterRule().

                   {
    visitedMemo.clear();
  }

Here is the caller graph for this function:

void org.apache.calcite.rel.rules.InjectFilterRule.onMatch ( RelOptRuleCall  call )

inline

Definition at line 56 of file InjectFilterRule.java.

References field(), org.apache.calcite.rel.rules.InjectFilterRule.restrictions, and org.apache.calcite.rel.rules.InjectFilterRule.visitedMemo.

                                           {
    LogicalTableScan childScanNode = call.rel(0);
    String scanNodeString = childScanNode.toString();
    if (visitedMemo.contains(scanNodeString)) {
      return;
    } else {
      visitedMemo.add(scanNodeString);
    }
    RelOptTable table = childScanNode.getTable();
    List<String> qname = table.getQualifiedName();

    String query_database = null;
    String query_table = null;
    if (qname.size() == 2) {
      query_database = qname.get(0);
      query_table = qname.get(1);
    }
    if (query_database == null || query_database.isEmpty() || query_table == null
            || query_table.isEmpty()) {
      throw new RuntimeException(
              "Restrictions: Expected qualified name as [database, table] but got: "
              + qname);
    }

    ArrayList<RexNode> orList = new ArrayList<RexNode>();
    RelBuilder builder = call.builder();
    RexBuilder rBuilder = builder.getRexBuilder();
    builder = builder.push(childScanNode);
    boolean found = false;
    for (Restriction restriction : restrictions) {
      // Match the database name.
      String rest_database = restriction.getRestrictionDatabase();
      if (rest_database != null && !rest_database.isEmpty()
              && !rest_database.equals(query_database)) {
        // TODO(sy): Maybe remove the isEmpty() wildcarding in HEAVY.AI 6.0.
        HEAVYDBLOGGER.debug("RLS row-level security restriction for database "
                + rest_database + " ignored because this query is on database "
                + query_database);
        continue;
      }

      // Match the table name.
      String rest_table = restriction.getRestrictionTable();
      if (rest_table != null && !rest_table.isEmpty()
              && !rest_table.equals(query_table)) {
        // TODO(sy): Maybe remove the isEmpty() wildcarding in HEAVY.AI 6.0.
        HEAVYDBLOGGER.debug("RLS row-level security restriction for table " + rest_table
                + " ignored because this query is on table " + query_table);
        continue;
      }

      // Match the column name.
      RelDataTypeField field = table.getRowType().getField(
              restriction.getRestrictionColumn(), false, false);
      if (field == null) {
        HEAVYDBLOGGER.debug("RLS row-level security restriction for column "
                + restriction.getRestrictionColumn()
                + " ignored because column not present in query table " + query_table);
        continue;
      }

      // Generate the RLS row-level security filter for one Restriction.
      found = true;
      HEAVYDBLOGGER.debug(
              "Scan is " + childScanNode.toString() + " TABLE is " + table.toString());
      HEAVYDBLOGGER.debug("Column " + restriction.getRestrictionColumn()
              + " exists in table " + table.getQualifiedName());

      for (String val : restriction.getRestrictionValues()) {
        HEAVYDBLOGGER.debug("Column is " + restriction.getRestrictionColumn()
                + " literal is '" + val + "'");
        RexNode lit;
        if (SqlTypeName.NUMERIC_TYPES.indexOf(field.getType().getSqlTypeName()) == -1) {
          if (val.length() < 2 || val.charAt(0) != '\''
                  || val.charAt(val.length() - 1) != '\'') {
            throw new RuntimeException(
                    "Restrictions: Expected a CREATE POLICY VALUES string with single quotes.");
          }
          lit = rBuilder.makeLiteral(
                  val.substring(1, val.length() - 1), field.getType(), false);
        } else {
          lit = rBuilder.makeLiteral(Integer.parseInt(val), field.getType(), false);
        }
        RexNode rn = builder.call(SqlStdOperatorTable.EQUALS,
                builder.field(restriction.getRestrictionColumn()),
                lit);
        orList.add(rn);
      }
    }

    if (found) {
      RexNode relOr = builder.call(SqlStdOperatorTable.OR, orList);
      final RelNode newNode = builder.filter(relOr).build();
      call.transformTo(newNode);
    }
  };

Here is the call graph for this function:

Member Data Documentation

final Logger org.apache.calcite.rel.rules.InjectFilterRule.HEAVYDBLOGGER = LoggerFactory.getLogger(InjectFilterRule.class)

staticpackage

Definition at line 42 of file InjectFilterRule.java.

final List<Restriction> org.apache.calcite.rel.rules.InjectFilterRule.restrictions

package

Definition at line 43 of file InjectFilterRule.java.

Referenced by org.apache.calcite.rel.rules.InjectFilterRule.InjectFilterRule(), and org.apache.calcite.rel.rules.InjectFilterRule.onMatch().

Set<String> org.apache.calcite.rel.rules.InjectFilterRule.visitedMemo = new HashSet<>()

static

Definition at line 41 of file InjectFilterRule.java.

Referenced by org.apache.calcite.rel.rules.InjectFilterRule.onMatch().

---

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/rel/rules/InjectFilterRule.java