1 files changed, 94 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/pdsuicommon/db/MySqlContext.scala b/src/main/scala/xyz/driver/pdsuicommon/db/MySqlContext.scala
new file mode 100644
index 0000000..768d1e3
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/db/MySqlContext.scala
@@ -0,0 +1,94 @@
+package xyz.driver.pdsuicommon.db
+
+import java.io.Closeable
+import java.time._
+import java.util.concurrent.Executors
+import javax.sql.DataSource
+
+import com.typesafe.config.Config
+import io.getquill._
+import xyz.driver.pdsuicommon.concurrent.MdcExecutionContext
+import xyz.driver.pdsuicommon.db.MySqlContext.Settings
+import xyz.driver.pdsuicommon.error.IncorrectIdException
+import xyz.driver.pdsuicommon.logging._
+
+import scala.concurrent.ExecutionContext
+import scala.util.control.NonFatal
+import scala.util.{Failure, Success, Try}
+
+object MySqlContext extends PhiLogging {
+
+  case class DbCredentials(user: String,
+                           password: String,
+                           host: String,
+                           port: Int,
+                           dbName: String,
+                           dbCreateFlag: Boolean,
+                           dbContext: String,
+                           connectionParams: String,
+                           url: String)
+
+  case class Settings(credentials: DbCredentials,
+                      connection: Config,
+                      connectionAttemptsOnStartup: Int,
+                      threadPoolSize: Int)
+
+  def apply(settings: Settings): MySqlContext = {
+    // Prevent leaking credentials to a log
+    Try(JdbcContextConfig(settings.connection).dataSource) match {
+      case Success(dataSource) => new MySqlContext(dataSource, settings)
+      case Failure(NonFatal(e)) =>
+        logger.error(phi"Can not load dataSource, error: ${Unsafe(e.getClass.getName)}")
+        throw new IllegalArgumentException("Can not load dataSource from config. Check your database and config")
+    }
+  }
+}
+
+class MySqlContext(dataSource: DataSource with Closeable, settings: Settings)
+  extends MysqlJdbcContext[MysqlEscape](dataSource)
+    with TransactionalContext
+    with EntityExtractorDerivation[Literal] {
+
+  private val tpe = Executors.newFixedThreadPool(settings.threadPoolSize)
+
+  implicit val executionContext: ExecutionContext = {
+    val orig = ExecutionContext.fromExecutor(tpe)
+    MdcExecutionContext.from(orig)
+  }
+
+  override def close(): Unit = {
+    super.close()
+    tpe.shutdownNow()
+  }
+
+  /**
+    * Overrode, because Quill JDBC optionDecoder pass null inside decoders.
+    * If custom decoder don't have special null handler, it will failed.
+    *
+    * @see https://github.com/getquill/quill/issues/535
+    */
+  implicit override def optionDecoder[T](implicit d: Decoder[T]): Decoder[Option[T]] =
+    decoder(
+      sqlType = d.sqlType,
+      row =>
+        index => {
+          try {
+            val res = d(index - 1, row)
+            if (row.wasNull) {
+              None
+            } else {
+              Some(res)
+            }
+          } catch {
+            case _: NullPointerException => None
+            case _: IncorrectIdException => None
+          }
+      }
+    )
+
+  final implicit class LocalDateTimeDbOps(val left: LocalDateTime) {
+
+    // scalastyle:off
+    def <=(right: LocalDateTime): Quoted[Boolean] = quote(infix"$left <= $right".as[Boolean])
+  }
+}