@Override public Part copy() { FunctionCall nfc = (FunctionCall) getParent().copy(null); ColumnInstance ci = (ColumnInstance) nfc.getParametersEdge().get(0); ConstantExpression litex = (ConstantExpression) nfc.getParametersEdge().get(1); return new EqualityPart(nfc, ci, litex, complete); }
private boolean applies(SchemaContext sc, DMLStatement stmt) throws PEException { if (stmt instanceof UpdateStatement) { UpdateStatement us = (UpdateStatement) stmt; for (Iterator<ExpressionNode> iter = us.getUpdateExpressionsEdge().iterator(); iter.hasNext(); ) { FunctionCall fc = (FunctionCall) iter.next(); ColumnInstance lhs = (ColumnInstance) fc.getParametersEdge().get(0); if (lhs.getPEColumn().isPartOfContainerDistributionVector()) throw new SchemaException( new ErrorInfo( AvailableErrors.INVALID_CONTAINER_DISCRIMINANT_COLUMN_UPDATE, lhs.getPEColumn().getName().getUnquotedName().get(), lhs.getPEColumn().getTable().getName().getUnquotedName().get())); } return false; } else if (stmt instanceof DeleteStatement) { DeleteStatement ds = (DeleteStatement) stmt; PETable pet = ds.getTargetDeleteEdge().get().getAbstractTable().asTable(); if (pet.isContainerBaseTable(sc)) { List<Part> parts = DiscriminantCollector.getDiscriminants(sc, ds.getWhereClause()); if (parts == null || parts.isEmpty()) throw new SchemaException( new ErrorInfo( AvailableErrors.INVALID_CONTAINER_DELETE, pet.getName().getUnquotedName().get())); else { List<SchemaCacheKey<PEContainerTenant>> matchingTenants = convert(sc, parts); stmt.getBlock().store(ContainerBaseTableRewriteTransformFactory.class, matchingTenants); return true; } } return false; } return false; }
private List<SchemaCacheKey<PEContainerTenant>> getPertinentTenants( PlannerContext pc, DMLStatement stmt) throws PEException { SchemaContext sc = pc.getContext(); if (stmt instanceof UpdateStatement) { UpdateStatement us = (UpdateStatement) stmt; for (Iterator<ExpressionNode> iter = us.getUpdateExpressionsEdge().iterator(); iter.hasNext(); ) { FunctionCall fc = (FunctionCall) iter.next(); ColumnInstance lhs = (ColumnInstance) fc.getParametersEdge().get(0); if (lhs.getPEColumn().isPartOfContainerDistributionVector()) throw new SchemaException( Pass.PLANNER, "Invalid update: discriminant column " + lhs.getPEColumn().getName().getSQL() + " of container base table " + lhs.getPEColumn().getTable().getName().getSQL() + " cannot be updated"); } } else if (stmt instanceof DeleteStatement) { DeleteStatement ds = (DeleteStatement) stmt; PETable pet = ds.getTargetDeleteEdge().get().getAbstractTable().asTable(); if (pet.isContainerBaseTable(sc)) { List<Part> parts = DiscriminantCollector.getDiscriminants(sc, ds.getWhereClause()); if (parts == null || parts.isEmpty()) throw new SchemaException( Pass.PLANNER, "Invalid delete on container base table " + pet.getName().getSQL() + ". Not restricted by discriminant columns"); else { List<SchemaCacheKey<PEContainerTenant>> matchingTenants = convert(sc, parts); return matchingTenants; } } } return null; }