content: /*
   * Copyright 1999-2017 Alibaba Group Holding Ltd.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package com.alibaba.druid.sql.ast.statement;
  
  import com.alibaba.druid.DbType;
  import com.alibaba.druid.sql.ast.SQLCommentHint;
  import com.alibaba.druid.sql.ast.SQLExpr;
  import com.alibaba.druid.sql.ast.SQLName;
  import com.alibaba.druid.sql.ast.SQLStatement;
  import com.alibaba.druid.sql.ast.SQLStatementImpl;
  import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
  import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
  import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
  import com.alibaba.druid.sql.visitor.SQLASTVisitor;
  
  import java.util.ArrayList;
  import java.util.List;
  
  public class SQLSetStatement extends SQLStatementImpl {
      private Option option;
  
      private List<SQLAssignItem> items = new ArrayList<SQLAssignItem>();
  
      private List<SQLCommentHint> hints;
  
      private boolean useSet;
  
      SQLStatement maridbSetForStatement;
  
      public SQLSetStatement() {
      }
  
      public SQLSetStatement(DbType dbType) {
          super(dbType);
      }
  
      public SQLSetStatement(SQLExpr target, SQLExpr value) {
          this(target, value, null);
      }
  
      public SQLSetStatement(SQLExpr target, SQLExpr value, DbType dbType) {
          super(dbType);
          SQLAssignItem item = new SQLAssignItem(target, value);
          item.setParent(this);
          this.items.add(item);
      }
  
      public static SQLSetStatement plus(SQLName target) {
          SQLExpr value = new SQLBinaryOpExpr(target.clone(), SQLBinaryOperator.Add, new SQLIntegerExpr(1));
          return new SQLSetStatement(target, value);
      }
  
      public List<SQLAssignItem> getItems() {
          return items;
      }
  
      public void setItems(List<SQLAssignItem> items) {
          this.items = items;
      }
  
      public List<SQLCommentHint> getHints() {
          return hints;
      }
  
      public void setHints(List<SQLCommentHint> hints) {
          this.hints = hints;
      }
  
      public Option getOption() {
          return option;
      }
  
      public void setOption(Option option) {
          this.option = option;
      }
  
      public boolean isUseSet() {
          return useSet;
      }
  
      public void setUseSet(boolean useSet) {
          this.useSet = useSet;
      }
  
      public void set(SQLExpr target, SQLExpr value) {
          SQLAssignItem assignItem = new SQLAssignItem(target, value);
          assignItem.setParent(this);
          this.items.add(assignItem);
      }
  
      public SQLStatement getMaridbSetForStatement() {
          return maridbSetForStatement;
      }
  
      public void setMaridbSetForStatement(SQLStatement maridbSetForStatement) {
          this.maridbSetForStatement = maridbSetForStatement;
      }
  
      @Override
      protected void accept0(SQLASTVisitor visitor) {
          if (visitor.visit(this)) {
              acceptChild(visitor, this.items);
              acceptChild(visitor, this.hints);
              if (maridbSetForStatement != null) {
                  maridbSetForStatement.accept(visitor);
              }
          }
          visitor.endVisit(this);
      }
  
      public void output(StringBuilder buf) {
          buf.append("SET ");
  
          for (int i = 0; i < items.size(); ++i) {
              if (i != 0) {
                  buf.append(", ");
              }
  
              SQLAssignItem item = items.get(i);
              item.output(buf);
          }
      }
  
      public SQLSetStatement clone() {
          SQLSetStatement x = new SQLSetStatement();
          for (SQLAssignItem item : items) {
              SQLAssignItem item2 = item.clone();
              item2.setParent(x);
              x.items.add(item2);
          }
          if (hints != null) {
              for (SQLCommentHint hint : hints) {
                  SQLCommentHint h2 = hint.clone();
                  h2.setParent(x);
                  x.hints.add(h2);
              }
          }
          return x;
      }
  
      public List getChildren() {
          return this.items;
      }
  
      public static enum Option {
          IDENTITY_INSERT,
          PASSWORD, // mysql
          GLOBAL,
          SESSION,
          LOCAL,
          PROJECT
      }
  }
  