feat: execute sql with database selection

Author: QuadStingray

src/main/scala/dev/mongocamp/driver/mongodb/sql/MongoSqlQueryHolder.scala

@@ -4,24 +4,25 @@ import dev.mongocamp.driver.mongodb._
 import dev.mongocamp.driver.mongodb.database.DatabaseProvider
 import SQLCommandType.SQLCommandType
 import com.mongodb.client.model.DropIndexOptions
-import net.sf.jsqlparser.statement.{ShowStatement, Statement, UnsupportedStatement}
-import net.sf.jsqlparser.expression.operators.conditional.{AndExpression, OrExpression}
+import dev.mongocamp.driver.mongodb.database.DatabaseProvider.CollectionSeparator
+import net.sf.jsqlparser.statement.{ ShowStatement, Statement, UnsupportedStatement }
+import net.sf.jsqlparser.expression.operators.conditional.{ AndExpression, OrExpression }
 import net.sf.jsqlparser.expression.operators.relational._
-import net.sf.jsqlparser.expression.{Expression, Parenthesis}
-import net.sf.jsqlparser.parser.{CCJSqlParser, StreamProvider}
-import net.sf.jsqlparser.schema.{Column, Table}
+import net.sf.jsqlparser.expression.{ Expression, Parenthesis }
+import net.sf.jsqlparser.parser.{ CCJSqlParser, StreamProvider }
+import net.sf.jsqlparser.schema.{ Column, Table }
 import net.sf.jsqlparser.statement.create.index.CreateIndex
 import net.sf.jsqlparser.statement.delete.Delete
 import net.sf.jsqlparser.statement.drop.Drop
 import net.sf.jsqlparser.statement.insert.Insert
-import net.sf.jsqlparser.statement.select.{AllColumns, FromItem, PlainSelect, Select, SelectItem}
+import net.sf.jsqlparser.statement.select.{ AllColumns, FromItem, PlainSelect, Select, SelectItem }
 import net.sf.jsqlparser.statement.show.ShowTablesStatement
 import net.sf.jsqlparser.statement.truncate.Truncate
 import net.sf.jsqlparser.statement.update.Update
 import org.bson.conversions.Bson
 import org.mongodb.scala.model.IndexOptions
 import org.mongodb.scala.model.Sorts.ascending
-import org.mongodb.scala.{Document, Observable}
+import org.mongodb.scala.{ Document, Observable }
 
 import java.util.concurrent.TimeUnit
 import scala.collection.mutable
@@ -90,11 +91,12 @@ class MongoSqlQueryHolder {
     }
     else if (classOf[UnsupportedStatement].isAssignableFrom(statement.getClass)) {
       val unsupportedStatement = statement.asInstanceOf[UnsupportedStatement]
-      val isShowDatabases = unsupportedStatement.toString.toLowerCase.contains("show databases")
-      val isShowSchemas = unsupportedStatement.toString.toLowerCase.contains("show schemas")
+      val isShowDatabases      = unsupportedStatement.toString.toLowerCase.contains("show databases")
+      val isShowSchemas        = unsupportedStatement.toString.toLowerCase.contains("show schemas")
       if (isShowDatabases | isShowSchemas) {
         sqlCommandType = SQLCommandType.ShowDatabases
-      } else {
+      }
+      else {
         throw new IllegalArgumentException("not supported sql command type")
       }
     }
@@ -104,7 +106,9 @@ class MongoSqlQueryHolder {
     ""
   }
 
-  def getCollection: String = sqlTable.getFullyQualifiedName
+  def getCollection: String = {
+    sqlTable.getFullyQualifiedName.replace(".", CollectionSeparator).replace("'", "").replace("`", "")
+  }
 
   def run(provider: DatabaseProvider, allowDiskUsage: Boolean = true): Observable[Document] = {
     sqlCommandType match {

src/test/scala/dev/mongocamp/driver/mongodb/sql/SelectSqlSpec.scala

@@ -15,6 +15,14 @@ class SelectSqlSpec extends PersonSpecification {
       selectResponse.head.getString("guid") mustEqual "a17be99a-8913-4bb6-8f14-16d4fa1b3559"
     }
 
+    "simple sql with schema" in {
+      val queryConverter = MongoSqlQueryHolder("select * from `mongocamp-unit-test`.`friend`")
+      val selectResponse = queryConverter.run(TestDatabase.provider).resultList()
+      selectResponse.size mustEqual 1327
+      selectResponse.head.getString("name") mustEqual "Castaneda Mccullough"
+      selectResponse.head.getLong("id") mustEqual 33
+    }
+
     "sql with in query" in {
       val queryConverter = MongoSqlQueryHolder("select id, guid, name, age, balance from people where age in (30, 18, 25, 22) order by id asc")
       val selectResponse = queryConverter.run(TestDatabase.provider).resultList()