1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
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 {
final case class DbCredentials(user: String,
password: String,
host: String,
port: Int,
dbName: String,
dbCreateFlag: Boolean,
dbContext: String,
connectionParams: String,
url: String)
final case class Settings(credentials: DbCredentials, connection: Config, 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])
}
}
|
