protobuf4j.orm.sql.clause.SetClause Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of protobuf4j-sql Show documentation

A Facility Framework to Develop with Google Protobuf

The newest version!

package protobuf4j.orm.sql.clause;

import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import lombok.Data;
import protobuf4j.orm.sql.AbstractSqlObject;
import protobuf4j.orm.sql.IExpression;
import protobuf4j.orm.sql.ISqlValue;
import protobuf4j.orm.sql.expr.Column;
import protobuf4j.orm.sql.expr.Value;

import javax.annotation.Nonnull;
import java.util.List;

/**
 * author: yuanwq
 * date: 2018/7/12
 */
@Data
public class SetClause extends AbstractSqlObject {
  private final List setItems = Lists.newArrayList();

  public boolean isEmpty() {
    return setItems.isEmpty();
  }

  public SetClause clear() {
    setItems.clear();
    return this;
  }

  private SetClause addSetItem(SetItem setItem) {
    this.setItems.add(setItem);
    return this;
  }

  public SetClause setExpression(String column, IExpression expression) {
    return addSetItem(new SetItem(Column.of(column), expression));
  }

  public SetClause setValue(String column, Object value) {
    return setExpression(column, Value.of(value, column));
  }

  public SetClause setColumn(String column, String otherColumn) {
    return setExpression(column, Column.of(otherColumn));
  }

  @Override
  public StringBuilder toSqlTemplate(@Nonnull StringBuilder sb) {
    Preconditions.checkArgument(!setItems.isEmpty(), "nothing to set");
    sb.append("SET ");
    boolean first = true;
    for (SetItem setItem : setItems) {
      if (first) {
        first = false;
      } else {
        sb.append(",");
      }
      setItem.toSqlTemplate(sb);
    }
    return sb;
  }

  @Override
  public StringBuilder toSolidSql(@Nonnull StringBuilder sb) {
    sb.append("SET ");
    boolean first = true;
    for (SetItem setItem : setItems) {
      if (first) {
        first = false;
      } else {
        sb.append(",");
      }
      setItem.toSolidSql(sb);
    }
    return sb;
  }

  @Override
  public List collectSqlValue(@Nonnull List sqlValues) {
    for (SetItem setItem : setItems) {
      setItem.collectSqlValue(sqlValues);
    }
    return sqlValues;
  }
}