org.hotrod.runtime.livesql.metadata.AllColumns Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hotrod-livesql Show documentation
HotRod is an ORM for Java, Spring and SpringBoot.
There is a newer version: 4.8.1
package org.hotrod.runtime.livesql.metadata;

import java.util.Arrays;
import java.util.List;
import java.util.function.Predicate;
import java.util.stream.Collectors;

import org.hotrod.runtime.livesql.expressions.AliasedExpression;
import org.hotrod.runtime.livesql.expressions.ComparableExpression;
import org.hotrod.runtime.livesql.expressions.ResultSetColumn;
import org.hotrod.runtime.livesql.queries.select.AbstractSelectObject.AliasGenerator;
import org.hotrod.runtime.livesql.queries.select.AbstractSelectObject.TableReferences;
import org.hotrod.runtime.livesql.queries.select.QueryWriter;

public class AllColumns implements ResultSetColumn {

  private List columns;

  public AllColumns(final Column... columns) {
    this.columns = Arrays.asList(columns);
  }

  public ColumnSubset filter(final Predicate predicate) {
    return new ColumnSubset(this.columns.stream().filter(predicate).collect(Collectors.toList()));
  }

  public static interface ColumnRenamer {
    String newName(Column c);
  }

  public ColumnAliased as(final ColumnRenamer aliaser) {
    return new ColumnAliased(this.columns.stream() //
        .map(c -> {
          return new AliasedExpression((ComparableExpression) c, aliaser.newName(c));
        }) //
        .collect(Collectors.toList()));
  }

  @Override
  public void renderTo(QueryWriter w) {
    throw new UnsupportedOperationException();
  }

  public static interface ColumnList extends ResultSetColumn {
    boolean isEmpty();
  }

  public static class ColumnSubset implements ColumnList {

    private List columns;

    protected ColumnSubset(final List columns) {
      this.columns = columns;
    }

    public boolean isEmpty() {
      return this.columns.isEmpty();
    }

    public ColumnAliased as(final ColumnRenamer aliaser) {
      return new ColumnAliased(this.columns.stream() //
          .map(c -> {
            return new AliasedExpression((ComparableExpression) c, aliaser.newName(c));
          }) //
          .collect(Collectors.toList()));
    }

    @Override
    public void renderTo(QueryWriter w) {
      boolean first = true;
      for (int i = 0; i < this.columns.size(); i++) {
        if (first) {
          first = false;
        } else {
          w.write(", ");
        }
        this.columns.get(i).renderTo(w);
      }
    }

    @Override
    public void validateTableReferences(TableReferences tableReferences, AliasGenerator ag) {
      // Nothing to do
    }

  }

  public static class ColumnAliased implements ColumnList {

    private List columns;

    protected ColumnAliased(final List columns) {
      this.columns = columns;
    }

    public boolean isEmpty() {
      return this.columns.isEmpty();
    }

    @Override
    public void renderTo(QueryWriter w) {
      boolean first = true;
      for (int i = 0; i < this.columns.size(); i++) {
        if (first) {
          first = false;
        } else {
          w.write(", ");
        }
        this.columns.get(i).renderTo(w);
      }

    }

    @Override
    public void validateTableReferences(TableReferences tableReferences, AliasGenerator ag) {
      // Nothing to do
    }

  }

  @Override
  public void validateTableReferences(TableReferences tableReferences, AliasGenerator ag) {
    // Nothing to do
  }

}