切换到ktor

Author: loheagn

.gitignore

@@ -36,8 +36,10 @@ out/
 ### VS Code ###
 .vscode/
 
-src/main/resources/application.yaml
+src/main/resources/application.conf
 
 logs
 
 config.yaml
+
+.DS_Store

build.gradle.kts

@@ -1,78 +1,56 @@
-import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar
-import org.gradle.api.tasks.testing.logging.TestLogEvent.*
-import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
+val ktor_version: String by project
+val kotlin_version: String by project
+val logback_version: String by project
 
 plugins {
-    kotlin("jvm") version "1.5.10"
     application
-    id("com.github.johnrengelman.shadow") version "7.0.0"
-    kotlin("plugin.serialization") version "1.5.10"
+    kotlin("jvm") version "1.5.31"
+    kotlin("plugin.serialization") version "1.5.31"
 }
 
-group = "cn.edu.buaa.scs"
-version = "1.0.0-SNAPSHOT"
+group = "scs.buaa.edu.cn"
+version = "0.0.1"
+application {
+    mainClass.set("io.ktor.server.netty.EngineMain")
+}
 
 repositories {
     mavenCentral()
 }
 
-val vertxVersion = "4.2.1"
-val junitJupiterVersion = "5.7.0"
+dependencies {
+    implementation("io.ktor:ktor-server-core:$ktor_version")
+    implementation("io.ktor:ktor-server-netty:$ktor_version")
 
-val mainVerticleName = "cn.edu.buaa.scs.cloudapi.MainVerticle"
-val launcherClassName = "cn.edu.buaa.scs.cloudapi.LauncherKt"
+    // coroutine
+    implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.5.2")
+    implementation("org.jetbrains.kotlinx:kotlinx-coroutines-reactive:1.5.2")
 
-val watchForChange = "src/**/*"
-val doOnChange = "${projectDir}/gradlew classes"
+    // kotlin-log
+    implementation("io.github.microutils:kotlin-logging:2.0.11")
+    implementation("ch.qos.logback:logback-classic:$logback_version")
 
-application {
-    mainClass.set(launcherClassName)
-}
+    // auth
+    implementation("io.ktor:ktor-auth:$ktor_version")
 
-dependencies {
-    implementation(platform("io.vertx:vertx-stack-depchain:$vertxVersion"))
-    implementation("io.vertx:vertx-web-client")
-    implementation("io.vertx:vertx-web")
-    implementation("io.vertx:vertx-lang-kotlin")
-    implementation("io.vertx:vertx-redis-client")
-    implementation(kotlin("stdlib-jdk8"))
+    // serialization
+    implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:1.3.0")
+    implementation("io.ktor:ktor-jackson:$ktor_version")
 
-    // kotlin coroutines
-    val coroutinesVersion = "1.5.10"
-    implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.5.2")
-    // yaml
-    implementation("com.charleskorn.kaml:kaml:0.36.0")
+    // Redis
+    implementation("io.lettuce:lettuce-core:6.1.5.RELEASE")
 
     // database
+    val ktorm_version = "3.4.1"
     implementation("com.zaxxer:HikariCP:5.0.0")
-    implementation("org.ktorm:ktorm-core:3.4.1")
+    implementation("org.ktorm:ktorm-core:$ktorm_version")
+    implementation("org.ktorm:ktorm-jackson:$ktorm_version")
+    implementation("org.ktorm:ktorm-support-mysql:$ktorm_version")
+    implementation("org.ktorm:ktorm-support-mysql:3.4.1")
     implementation("mysql:mysql-connector-java:8.0.25")
 
-    // log
-    implementation("io.github.microutils:kotlin-logging:1.6.22")
-    implementation("org.slf4j:slf4j-api:1.7.25")
-    implementation("ch.qos.logback:logback-core:1.2.6")
-    implementation("ch.qos.logback:logback-classic:1.2.6")
 
     // test
-    testImplementation("io.vertx:vertx-junit5")
-    testImplementation("org.junit.jupiter:junit-jupiter:$junitJupiterVersion")
-}
-
-val compileKotlin: KotlinCompile by tasks
-compileKotlin.kotlinOptions.jvmTarget = "11"
-
-tasks.withType<ShadowJar> {
-    archiveClassifier.set("fat")
-    manifest {
-        attributes(mapOf("Main-Verticle" to mainVerticleName))
-    }
-    mergeServiceFiles()
-}
-
-tasks.withType<Test> {
-    useJUnitPlatform()
-    testLogging {
-        events = setOf(PASSED, SKIPPED, FAILED)
-    }
-}
+    testImplementation("io.ktor:ktor-server-tests:$ktor_version")
+    testImplementation("org.jetbrains.kotlin:kotlin-test-junit:$kotlin_version")
+}

deploy.sh

@@ -1 +1,4 @@
-kotlin.code.style=official
+ktor_version=1.6.5
+kotlin_version=1.5.31
+logback_version=1.2.3
+kotlin.code.style=official

gradle.properties

@@ -1,5 +1,5 @@
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-7.0-bin.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-7.1-bin.zip
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists

gradle/wrapper/gradle-wrapper.jar

@@ -72,7 +72,7 @@ case "`uname`" in
   Darwin* )
     darwin=true
     ;;
-  MINGW* )
+  MSYS* | MINGW* )
     msys=true
     ;;
   NONSTOP* )

gradle/wrapper/gradle-wrapper.properties

@@ -1 +1 @@
-rootProject.name = "cloudapi"
+rootProject.name = "cloudapi"

gradlew

@@ -0,0 +1,60 @@
+package cn.edu.buaa.scs
+
+import cn.edu.buaa.scs.plugins.*
+import cn.edu.buaa.scs.utils.logger
+import com.zaxxer.hikari.HikariConfig
+import com.zaxxer.hikari.HikariDataSource
+import io.ktor.application.*
+import io.lettuce.core.RedisClient
+import io.lettuce.core.RedisURI
+import org.ktorm.database.Database
+import org.ktorm.logging.Slf4jLoggerAdapter
+
+fun main(args: Array<String>): Unit =
+    io.ktor.server.netty.EngineMain.main(args)
+
+
+@Suppress("unused")
+fun Application.webModule() {
+    configureCORS()
+    configureCallID()
+    configureMonitoring()
+    configureContentNegotiation()
+    configureStatusPage()
+    configureRouting()
+}
+
+fun Application.getConfig(name: String, default: String = ""): String =
+    this.environment.config.propertyOrNull(name)?.getString() ?: default
+
+lateinit var authRedis: RedisClient
+
+@Suppress("unused")
+fun Application.redisModule() {
+    // redis for auth
+    authRedis = RedisClient.create(
+        RedisURI.Builder.redis(getConfig("redis.auth.host", "localhost"))
+            .withPort(getConfig("redis.auth.port", "6379").toInt())
+            .withPassword(getConfig("redis.auth.password", "").toCharArray())
+            .build()
+    )
+    logger("auth-redis")().info { "auth redis connected" }
+}
+
+lateinit var db: Database
+fun Application.dbModule() {
+    val mainDBHost = getConfig("db.main.host", "localhost")
+    val mainDBPort = getConfig("db.main.port", "3306")
+    val mainDBName = getConfig("db.main.name")
+    val mainDBUser = getConfig("db.main.username", "root")
+    val mainDBPassword = getConfig("db.main.password")
+    db = Database.connect(
+        dataSource = HikariDataSource(HikariConfig().apply {
+            jdbcUrl = "jdbc:mysql://$mainDBHost:$mainDBPort/$mainDBName"
+            username = mainDBUser
+            password = mainDBPassword
+        }),
+        logger = Slf4jLoggerAdapter(logger("mainDB")().underlyingLogger)
+    )
+    logger("mainDB")().info { "main database connected" }
+}

settings.gradle.kts

@@ -0,0 +1,10 @@
+package cn.edu.buaa.scs.error
+
+@Suppress("unused")
+class BadRequestException(message: String = "400") : io.ktor.features.BadRequestException(message)
+
+class AuthenticationException(message: String = "401") : Exception(message)
+
+class AuthorizationException(message: String = "403") : Exception(message)
+
+class NotFoundException(message: String = "404") : Exception(message)