-
Notifications
You must be signed in to change notification settings - Fork 1k
/
BaseApplicationEngine.kt
136 lines (118 loc) · 4.63 KB
/
BaseApplicationEngine.kt
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
/*
* Copyright 2014-2019 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
*/
package io.ktor.server.engine
import io.ktor.http.*
import io.ktor.http.content.*
import io.ktor.server.application.*
import io.ktor.server.engine.internal.*
import io.ktor.server.http.content.*
import io.ktor.server.plugins.*
import io.ktor.server.response.*
import io.ktor.server.routing.*
import io.ktor.util.*
import io.ktor.util.pipeline.*
import io.ktor.utils.io.concurrent.*
import kotlinx.coroutines.*
import kotlin.native.concurrent.*
/**
* Base class for implementing [ApplicationEngine]
*
* It creates default engine pipeline, provides [application] property and installs default transformations
* on respond and receive
*
* @param environment instance of [ApplicationEngineEnvironment] for this engine
* @param pipeline pipeline to use with this engine
*/
public abstract class BaseApplicationEngine(
public final override val environment: ApplicationEngineEnvironment,
public val pipeline: EnginePipeline = defaultEnginePipeline(environment)
) : ApplicationEngine {
/**
* Configuration for the [BaseApplicationEngine]
*/
public open class Configuration : ApplicationEngine.Configuration()
protected val resolvedConnectors: CompletableDeferred<List<EngineConnectorConfig>> = CompletableDeferred()
init {
val environment = environment
val info = StartupInfo()
val pipeline = pipeline
BaseApplicationResponse.setupSendPipeline(pipeline.sendPipeline)
environment.monitor.subscribe(ApplicationStarting) {
if (!info.isFirstLoading) {
info.initializedStartAt = currentTimeMillisBridge()
}
it.receivePipeline.merge(pipeline.receivePipeline)
it.sendPipeline.merge(pipeline.sendPipeline)
it.receivePipeline.installDefaultTransformations()
it.sendPipeline.installDefaultTransformations()
it.installDefaultInterceptors()
it.installDefaultTransformationChecker()
}
environment.monitor.subscribe(ApplicationStarted) {
val finishedAt = currentTimeMillisBridge()
val elapsedTimeInSeconds = (finishedAt - info.initializedStartAt) / 1_000.0
if (info.isFirstLoading) {
environment.log.info("Application started in $elapsedTimeInSeconds seconds.")
info.isFirstLoading = false
} else {
environment.log.info("Application auto-reloaded in $elapsedTimeInSeconds seconds.")
}
}
val connectors = resolvedConnectors
val log = environment.log
CoroutineScope(environment.application.coroutineContext).launch {
connectors.await().forEach {
val host = escapeHostname(it.host)
log.info(
"Responding at ${it.type.name.lowercase()}://$host:${it.port}"
)
}
}
}
override suspend fun resolvedConnectors(): List<EngineConnectorConfig> {
return resolvedConnectors.await()
}
}
private suspend fun PipelineContext<Unit, ApplicationCall>.verifyHostHeader() {
val hostHeaders = call.request.headers.getAll(HttpHeaders.Host) ?: return
if (hostHeaders.size > 1) {
call.respond(HttpStatusCode.BadRequest)
finish()
}
}
private class StartupInfo {
var isFirstLoading = true
var initializedStartAt = currentTimeMillisBridge()
}
@OptIn(InternalAPI::class)
private fun Application.installDefaultInterceptors() {
intercept(ApplicationCallPipeline.Fallback) {
if (call.isHandled) return@intercept
val status = call.response.status()
?: call.attributes.getOrNull(RoutingFailureStatusCode)
?: HttpStatusCode.NotFound
call.respond(status)
}
intercept(ApplicationCallPipeline.Call) {
verifyHostHeader()
}
}
@OptIn(InternalAPI::class)
private fun Application.installDefaultTransformationChecker() {
// Respond with "415 Unsupported Media Type" if content cannot be transformed on receive
intercept(ApplicationCallPipeline.Plugins) {
try {
proceed()
} catch (e: CannotTransformContentToTypeException) {
call.respond(HttpStatusCode.UnsupportedMediaType)
}
}
val checkBodyPhase = PipelinePhase("BodyTransformationCheckPostRender")
sendPipeline.insertPhaseAfter(ApplicationSendPipeline.Render, checkBodyPhase)
sendPipeline.intercept(checkBodyPhase) { subject ->
if (subject !is OutgoingContent) {
proceedWith(HttpStatusCodeContent(HttpStatusCode.NotAcceptable))
}
}
}