[CBRD-25708] Enable partition pruning when the partition key is a function expression and the WHERE clause uses the original column

src/query/partition.c

@@ -137,6 +137,8 @@ static MATCH_STATUS partition_match_key_range (PRUNING_CONTEXT * pinfo, const KE
 					       PRUNING_BITSET * pruned);
 static bool partition_do_regu_variables_match (PRUNING_CONTEXT * pinfo, const REGU_VARIABLE * left,
 					       const REGU_VARIABLE * right);
+static bool partition_do_regu_variables_contain (PRUNING_CONTEXT * pinfo, const REGU_VARIABLE * left,
+						 const REGU_VARIABLE * right);
 static MATCH_STATUS partition_prune (PRUNING_CONTEXT * pinfo, const REGU_VARIABLE * arg, const PRUNING_OP op,
 				     PRUNING_BITSET * pruned);
 static MATCH_STATUS partition_prune_db_val (PRUNING_CONTEXT * pinfo, const DB_VALUE * val, const PRUNING_OP op,
@@ -167,6 +169,11 @@ static int partition_attrinfo_get_key (THREAD_ENTRY * thread_p, PRUNING_CONTEXT
 /* misc pruning functions */
 static bool partition_decrement_value (DB_VALUE * val);
 
+static void partition_cache_dbvalp (REGU_VARIABLE * var, DB_VALUE * val);
+static bool partition_supports_pruning_op_for_function (const PRUNING_OP op, const REGU_VARIABLE * part_expr);
+static MATCH_STATUS partition_prune_arith (PRUNING_CONTEXT * pinfo, const REGU_VARIABLE * left,
+					   const REGU_VARIABLE * right, REGU_VARIABLE * part_expr, const PRUNING_OP op,
+					   PRUNING_BITSET * pruned);
 
 /* PRUNING_BITSET manipulation functions */
 
@@ -1668,6 +1675,14 @@ partition_is_reguvar_const (const REGU_VARIABLE * regu_var)
 	/* either all arguments are constants of this is an expression with no arguments */
 	return true;
       }
+    case TYPE_ATTR_ID:
+      {
+	if (regu_var->value.attr_descr.cache_dbvalp == NULL)
+	  {
+	    return false;
+	  }
+	return true;
+      }
     default:
       return false;
     }
@@ -1910,7 +1925,6 @@ partition_match_pred_expr (PRUNING_CONTEXT * pinfo, const PRED_EXPR * pr, PRUNIN
 	    /* see if part_expr matches right or left */
 	    REGU_VARIABLE *left, *right;
 	    PRUNING_OP op;
-
 	    left = pr->pe.m_eval_term.et.et_comp.lhs;
 	    right = pr->pe.m_eval_term.et.et_comp.rhs;
 	    op = partition_rel_op_to_pruning_op (pr->pe.m_eval_term.et.et_comp.rel_op);
@@ -1924,6 +1938,17 @@ partition_match_pred_expr (PRUNING_CONTEXT * pinfo, const PRED_EXPR * pr, PRUNIN
 	      {
 		status = partition_prune (pinfo, left, op, pruned);
 	      }
+	    else if (partition_supports_pruning_op_for_function (op, part_expr))
+	      {
+		if (partition_do_regu_variables_contain (pinfo, left, part_expr))
+		  {
+		    status = partition_prune_arith (pinfo, left, right, part_expr, op, pruned);
+		  }
+		else if (partition_do_regu_variables_contain (pinfo, right, part_expr))
+		  {
+		    status = partition_prune_arith (pinfo, right, left, part_expr, op, pruned);
+		  }
+	      }
 	    break;
 	  }
 
@@ -1951,6 +1976,13 @@ partition_match_pred_expr (PRUNING_CONTEXT * pinfo, const PRED_EXPR * pr, PRUNIN
 	      {
 		status = partition_prune (pinfo, list, op, pruned);
 	      }
+	    else if (partition_supports_pruning_op_for_function (op, part_expr))
+	      {
+		if (partition_do_regu_variables_contain (pinfo, regu, part_expr))
+		  {
+		    status = partition_prune_arith (pinfo, regu, list, part_expr, op, pruned);
+		  }
+	      }
 	  }
 	  break;
 
@@ -1971,6 +2003,105 @@ partition_match_pred_expr (PRUNING_CONTEXT * pinfo, const PRED_EXPR * pr, PRUNIN
   return status;
 }
 
+static MATCH_STATUS
+partition_prune_arith (PRUNING_CONTEXT * pinfo, const REGU_VARIABLE * left, const REGU_VARIABLE * right,
+		       REGU_VARIABLE * part_expr, const PRUNING_OP op, PRUNING_BITSET * pruned)
+{
+  MATCH_STATUS status = MATCH_NOT_FOUND;
+  DB_VALUE val, casted_val;
+  TP_DOMAIN_STATUS dom_status;
+  bool is_value;
+
+  if (partition_get_value_from_regu_var (pinfo, right, &val, &is_value) == NO_ERROR)
+    {
+      if (TP_DOMAIN_TYPE (left->domain) != DB_VALUE_TYPE (&val))
+	{
+	  dom_status = tp_value_cast (&val, &casted_val, left->domain, false);
+
+	  if (dom_status != DOMAIN_COMPATIBLE)
+	    {
+	      (void) tp_domain_status_er_set (dom_status, ARG_FILE_LINE, &val, left->domain);
+
+	      pinfo->error_code = ER_FAILED;
+	      pr_clear_value (&val);
+	      pr_clear_value (&casted_val);
+	      return MATCH_NOT_FOUND;
+	    }
+
+	  partition_cache_dbvalp (part_expr, &casted_val);
+	}
+      else
+	{
+	  partition_cache_dbvalp (part_expr, &val);
+	}
+
+      status = partition_prune (pinfo, part_expr, op, pruned);
+      partition_cache_dbvalp (part_expr, NULL);
+    }
+  pr_clear_value (&val);
+  pr_clear_value (&casted_val);
+
+  return status;
+}
+
+static bool
+partition_supports_pruning_op_for_function (const PRUNING_OP op, const REGU_VARIABLE * part_expr)
+{
+  OPERATOR_TYPE opcode;
+
+  if (part_expr->type != TYPE_INARITH)
+    {
+      return false;
+    }
+
+  opcode = part_expr->value.arithptr->opcode;
+
+  switch (op)
+    {
+    case PO_LT:
+    case PO_LE:
+    case PO_GT:
+    case PO_GE:
+      switch (opcode)
+	{
+	case T_YEAR:
+	case T_TODAYS:
+	case T_UNIX_TIMESTAMP:
+	  return true;
+	default:
+	  return false;
+	}
+    case PO_EQ:
+    case PO_NE:
+    case PO_IN:
+    case PO_NOT_IN:
+    case PO_IS_NULL:
+      return true;
+    default:
+      return false;
+    }
+
+}
+
+static bool
+partition_do_regu_variables_contain (PRUNING_CONTEXT * pinfo, const REGU_VARIABLE * left, const REGU_VARIABLE * right)
+{
+  if (left == NULL || right == NULL)
+    {
+      return left == right;
+    }
+
+  if (left->type == TYPE_ATTR_ID && right->type == TYPE_INARITH)
+    {
+      if (left->value.attr_descr.id == pinfo->attr_id)
+	{
+	  return true;
+	}
+    }
+
+  return false;
+}
+
 /*
  * partition_match_key_range () - perform pruning using a key_range
  * return : pruned list
@@ -2575,6 +2706,36 @@ partition_set_cache_info_for_expr (REGU_VARIABLE * var, ATTR_ID attr_id, HEAP_CA
     }
 }
 
+static void
+partition_cache_dbvalp (REGU_VARIABLE * var, DB_VALUE * val)
+{
+  assert (var != NULL);
+  switch (var->type)
+    {
+    case TYPE_ATTR_ID:
+      var->value.attr_descr.cache_dbvalp = val;
+      break;
+
+    case TYPE_INARITH:
+      if (var->value.arithptr->leftptr != NULL)
+	{
+	  partition_cache_dbvalp (var->value.arithptr->leftptr, val);
+	}
+
+      if (var->value.arithptr->rightptr != NULL)
+	{
+	  partition_cache_dbvalp (var->value.arithptr->rightptr, val);
+	}
+
+      if (var->value.arithptr->thirdptr != NULL)
+	{
+	  partition_cache_dbvalp (var->value.arithptr->thirdptr, val);
+	}
+    default:
+      break;
+    }
+}
+
 /*
  * partition_get_attribute_id () - get the id of the attribute of the
  *				   partition expression