Adding domain entitiesv0.1.0

1 files changed, 90 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuicommon/db/MysqlQueryBuilder.scala b/src/main/scala/xyz/driver/pdsuicommon/db/MysqlQueryBuilder.scala
new file mode 100644
index 0000000..672a34e
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/db/MysqlQueryBuilder.scala
@@ -0,0 +1,90 @@
+package xyz.driver.pdsuicommon.db
+
+import java.sql.ResultSet
+
+import io.getquill.{MySQLDialect, MysqlEscape}
+
+import scala.collection.breakOut
+import scala.concurrent.{ExecutionContext, Future}
+
+object MysqlQueryBuilder {
+  import xyz.driver.pdsuicommon.db.QueryBuilder._
+
+  def apply[T](tableName: String,
+               lastUpdateFieldName: Option[String],
+               nullableFields: Set[String],
+               links: Set[TableLink],
+               runner: Runner[T],
+               countRunner: CountRunner)
+              (implicit ec: ExecutionContext): MysqlQueryBuilder[T] = {
+    val parameters = MysqlQueryBuilderParameters(
+      tableData = TableData(tableName, lastUpdateFieldName, nullableFields),
+      links = links.map(x => x.foreignTableName -> x)(breakOut)
+    )
+    new MysqlQueryBuilder[T](parameters)(runner, countRunner, ec)
+  }
+
+  def apply[T](tableName: String,
+               lastUpdateFieldName: Option[String],
+               nullableFields: Set[String],
+               links: Set[TableLink],
+               extractor: (ResultSet) => T)
+              (implicit sqlContext: SqlContext): MysqlQueryBuilder[T] = {
+
+    val runner = (parameters: QueryBuilderParameters) => {
+      Future {
+        val (sql, binder) = parameters.toSql(namingStrategy = MysqlEscape)
+        sqlContext.executeQuery[T](sql, binder, { resultSet =>
+          extractor(resultSet)
+        })
+      }(sqlContext.executionContext)
+    }
+
+    val countRunner = (parameters: QueryBuilderParameters) => {
+      Future {
+        val (sql, binder) = parameters.toSql(countQuery = true, namingStrategy = MysqlEscape)
+        sqlContext.executeQuery[CountResult](sql, binder, { resultSet =>
+          val count = resultSet.getInt(1)
+          val lastUpdate = if (parameters.tableData.lastUpdateFieldName.isDefined) {
+            Option(sqlContext.localDateTimeDecoder.decoder(2, resultSet))
+          } else None
+
+          (count, lastUpdate)
+        }).head
+      }(sqlContext.executionContext)
+    }
+
+    apply[T](
+      tableName = tableName,
+      lastUpdateFieldName = lastUpdateFieldName,
+      nullableFields = nullableFields,
+      links = links,
+      runner = runner,
+      countRunner = countRunner
+    )(sqlContext.executionContext)
+  }
+}
+
+class MysqlQueryBuilder[T](parameters: MysqlQueryBuilderParameters)
+                          (implicit runner: QueryBuilder.Runner[T],
+                           countRunner: QueryBuilder.CountRunner,
+                           ec: ExecutionContext)
+  extends QueryBuilder[T, MySQLDialect, MysqlEscape](parameters) {
+
+  def withFilter(newFilter: SearchFilterExpr): QueryBuilder[T, MySQLDialect, MysqlEscape] = {
+    new MysqlQueryBuilder[T](parameters.copy(filter = newFilter))
+  }
+
+  def withSorting(newSorting: Sorting): QueryBuilder[T, MySQLDialect, MysqlEscape] = {
+    new MysqlQueryBuilder[T](parameters.copy(sorting = newSorting))
+  }
+
+  def withPagination(newPagination: Pagination): QueryBuilder[T, MySQLDialect, MysqlEscape] = {
+    new MysqlQueryBuilder[T](parameters.copy(pagination = Some(newPagination)))
+  }
+
+  def resetPagination: QueryBuilder[T, MySQLDialect, MysqlEscape] = {
+    new MysqlQueryBuilder[T](parameters.copy(pagination = None))
+  }
+
+}