package com.chenyang.druid.sql.ast.statement;

import com.chenyang.druid.sql.ast.SQLExpr;
import com.chenyang.druid.sql.ast.SQLObjectImpl;
import com.chenyang.druid.sql.ast.expr.SQLBinaryOperator;
import com.chenyang.druid.sql.ast.expr.SQLIdentifierExpr;
import com.chenyang.druid.sql.visitor.SQLASTVisitor;
import java.util.ArrayList;
import java.util.List;

public class SQLPartitionRef extends SQLObjectImpl {
   private List<Item> items = new ArrayList();

   protected void accept0(SQLASTVisitor v) {
      if (v.visit(this)) {
         this.acceptChild(v, this.items);
      }

      v.endVisit(this);
   }

   public List<Item> getItems() {
      return this.items;
   }

   public void addItem(Item item) {
      item.setParent(this);
      this.items.add(item);
   }

   public void addItem(SQLIdentifierExpr name, SQLExpr value) {
      Item item = new Item();
      item.setColumnName(name);
      item.setValue(value);
      item.setParent(this);
      this.items.add(item);
   }

   public static class Item extends SQLObjectImpl {
      private SQLIdentifierExpr columnName;
      private SQLExpr value;
      private SQLBinaryOperator operator;

      public Item() {
      }

      public Item(SQLIdentifierExpr columnName) {
         this.setColumnName(columnName);
      }

      protected void accept0(SQLASTVisitor v) {
      }

      public SQLIdentifierExpr getColumnName() {
         return this.columnName;
      }

      public void setColumnName(SQLIdentifierExpr x) {
         if (x != null) {
            x.setParent(this);
         }

         this.columnName = x;
      }

      public SQLExpr getValue() {
         return this.value;
      }

      public void setValue(SQLExpr x) {
         if (x != null) {
            x.setParent(this);
         }

         this.value = x;
      }

      public SQLBinaryOperator getOperator() {
         return this.operator;
      }

      public void setOperator(SQLBinaryOperator operator) {
         this.operator = operator;
      }
   }
}
