首发于Enaium的个人博客
引言
在本文中,我们将使用 Kotlin 编写一个简单的 HTTP 服务器。我们将使用 Java 的 ServerSocket
类来实现这个服务器。我们将创建一个简单的服务器,它将监听端口 8000,并在接收到请求时返回一个简单的响应。
Http 的格式
HTTP 请求和响应都是文本格式的。HTTP 请求由请求行、请求头和请求体组成。HTTP 响应由状态行、响应头和响应体组成。
具体可以到 MDN 查看。
代码实现
首先我们需要创建一个Method
枚举和一个Version
枚举,用于表示请求的方法和版本。
enum class Method {GET,POST,UNKNOWN;companion object {fun parse(method: String): Method =when (method) {"GET" -> GET"POST" -> POSTelse -> UNKNOWN}}
}
enum class Version {HTTP_1_1,UNKNOWN;companion object {fun parse(version: String): Version = when (version) {"HTTP/1.1" -> HTTP_1_1else -> UNKNOWN}}override fun toString(): String {return when (this) {HTTP_1_1 -> "HTTP/1.1"UNKNOWN -> "UNKNOWN"}}
}
然后我们创建一个HttpRequest
类,用于表示 HTTP 请求。
data class HttpRequest(val method: Method,val path: String,val version: Version,val headers: Map<String, String>,val body: String
)
接着我们为HttpRequest
类添加一个静态方法parse
,用于解析 HTTP 请求。
companion object {fun parse(reader: BufferedReader): HttpRequest {}
}
在parse
方法中,我们首先读取请求行,然后读取请求头,最后读取请求体。
首先我们先来定义一些默认值。
var method = Method.UNKNOWN
var path = ""
var version = Version.UNKNOWN
val headers = mutableMapOf<String, String>()
var body = ""
然后我们读取请求的所有数据,不过需要注意这里不不能使用readText
和readLines
,这些方法会导致Socket
被关闭。
val request = StringBuilder()
var readed: String?
while (reader.readLine().also { readed = it } != null && readed!!.isNotEmpty()) {request.append(readed).append("\n")
}
接着我们解析请求的数据。
request.lines().forEachIndexed { index, line ->if (index == 0) {val splitWithSpace = line.split(" ")method = Method.parse(splitWithSpace[0])path = splitWithSpace[1]version = Version.parse(splitWithSpace[2])} else if (line.contains(": ")) {val split = line.split(": ")headers[split[0]] = split[1]} else if (line.isEmpty()) {} else {body = line}
}
首先是第一行,我们使用空格分割,然后解析请求方法、路径和版本。然后是请求头,我们使用冒号空格分割,然后解析请求头。最后是请求体,如果不为空,我们就保存请求体。
最后我们将解析的数据返回。
return HttpRequest(method,path,version,headers,body
)
接着我们创建一个HttpResponse
类,用于表示 HTTP 响应。
data class HttpResponse(val version: Version = Version.HTTP_1_1,val statusCode: Int,val statusText: String,val headers: Map<String, String> = emptyMap(),val body: String = ""
)
然后我们为HttpResponse
类重写toString
方法,用于将响应转换为字符串。
override fun toString(): String {return """$version $statusCode $statusText${headers.map { "${it.key}: ${it.value}" }.joinToString("\n")}$body""".trimIndent()
}
接着我们创建一个Handler
类型,用于处理请求。
typealias Handler = (HttpRequest) -> HttpResponse
然后我们创建一个Route
类,用于表示路由。
data class Route(val method: Method, val path: String, val handler: Handler)
接着创建一个Router
类,用于管理路由。
class Router {private val routes = mutableListOf<Route>()fun get(path: String, handler: Handler) {routes.add(Route(Method.GET, path, handler))}fun handle(socket: ServerSocket) {}
}
在Router
类中,我们定义了一个routes
属性,用于保存所有的路由。然后我们定义了一个get
方法,用于添加一个 GET 请求的路由。最后我们定义了一个handle
方法,用于处理请求。
接着我们需要实现handle
方法。
fun handle(socket: ServerSocket) {while (true) {val client = socket.accept()val reader = client.getInputStream().bufferedReader()val writer = client.getOutputStream().bufferedWriter()val httpRequest = HttpRequest.parse(reader)routes.findLast { it.method == httpRequest.method && it.path == httpRequest.path }?.let {val toString = it.handler.invoke(httpRequest).toString()writer.write(toString)writer.flush()} ?: let {writer.write(HttpResponse(Version.HTTP_1_1,404,"NotFound",headers = mapOf("Content-Type" to "text/html"),body = "<h1>404 Not Found</h1>").toString())writer.flush()}client.close()}
}
在handle
方法中,我们首先创建一个ServerSocket
,然后进入一个无限循环。在循环中,我们首先接受一个客户端连接,然后创建一个BufferedReader
和一个BufferedWriter
,用于读取请求和写入响应。然后我们解析请求,然后查找路由,如果找到了路由,我们就调用路由的处理函数,然后将响应写入到客户端。如果没有找到路由,我们就返回一个 404 响应。最后我们关闭客户端连接。
最后我们创建一个Server
类,用于创建服务器。
class HttpServer(port: Int) {private val serverSocket: ServerSocket = ServerSocket(port)fun start() {}
}
在start
方法中,我们创建一个Router
,然后添加一个路由,最后调用Router
的handle
方法。
val router = Router()
router.get("/") { _ ->HttpResponse(Version.HTTP_1_1, 200, "OK")
}
router.get("/hello") { _ ->HttpResponse(Version.HTTP_1_1, 200, "OK",headers = mapOf("Content-Type" to "text/html"),body = "<h1>Hello World!</h1>")
}
router.handle(serverSocket)
这样我们就完成了一个简单的 HTTP 服务器。
现在来测试一下我们的服务器。
fun main() {HttpServer(8080).start()
}
在终端中运行main
函数,然后在浏览器中打开http://localhost:8080
和http://localhost:8080/hello
,你应该能看到一个简单的页面。
总结
在本文中,我们使用 Kotlin 编写了一个简单的 HTTP 服务器。我们使用 Java 的 ServerSocket
类来实现这个服务器。我们创建了一个简单的服务器,它监听端口 8080,并在接收到请求时返回一个简单的响应。我们还创建了一个简单的路由系统,用于处理不同的请求。
完整源码