Add column checking

src/main/scala/com/github/takezoe/scala/jdbc/validation/Models.scala

@@ -15,14 +15,17 @@ class TableModel {
 class SelectModel {
   var from = new ListBuffer[TableModel]
   val columns = new ListBuffer[ColumnModel]
+  val orderBy = new ListBuffer[ColumnModel]
+  val where = new ListBuffer[ColumnModel]
 
   override def toString(): String = {
-    s"Select(from: $from, columns: $columns)"
+    s"Select(from: $from, columns: $columns, orderBy; $orderBy, where: $where)"
   }
 
   // TODO
   def validate() = {
-    columns.foreach { column =>
+    (columns ++ where ++ orderBy).foreach { column =>
+      println(column)
       val table = if (column.table != null){
         from.find(x => x.name == column.table || x.alias == column.table)
       } else {

src/main/scala/com/github/takezoe/scala/jdbc/validation/SelectVisitor.scala

@@ -48,6 +48,30 @@ class SelectVisitor extends SelectVisitorAdapter {
 
     }
 
+    Option(plainSelect.getWhere).foreach(_.accept(new ExpressionVisitorAdapter {
+      override def visit(column: Column): Unit = {
+        val c = new ColumnModel()
+        c.name = column.getColumnName
+        c.table = Option(column.getTable).map(_.getName).orNull
+        select.where += c
+      }
+    }))
+    Option(plainSelect.getOrderByElements).foreach(_.asScala.foreach { orderBy =>
+      println("OrderBy: " + orderBy)
+      orderBy.accept(new OrderByVisitor {
+        override def visit(orderBy: OrderByElement): Unit = {
+          orderBy.getExpression.accept(new ExpressionVisitorAdapter(){
+            override def visit(column: Column): Unit = {
+              val c = new ColumnModel()
+              c.name = column.getColumnName
+              c.table = Option(column.getTable).map(_.getName).orNull
+              select.orderBy += c
+            }
+          })
+        }
+      })
+    })
+
     println(select)
     select.validate()
   }