class UserRepartitionNode extends RepartitionNode {

  public Expression resolveSelect(final int idx, final Expression expression) {
    return getPartitionBy().equals(expression)
        ? new UnqualifiedColumnReferenceExp(Iterables.getOnlyElement(getSchema().key()).name())
        : expression;
  }

  public  UserRepartitionNode(
      final PlanNodeId id,
      final PlanNode source,
      final LogicalSchema schema,
      final Expression originalPartitionBy,
      final Expression partitionBy
  );

}