commit 4667bce9a98b3f56db909c608c71ec2c852ec79f
Author: pan <1029559041@qq.com>
Date: Thu Apr 29 01:16:02 2021 +0800
init
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..fae6b74
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,7 @@
+/.gradle
+/.idea
+/out
+/build
+*.iml
+*.ipr
+*.iws
diff --git a/build.gradle.kts b/build.gradle.kts
new file mode 100644
index 0000000..1395968
--- /dev/null
+++ b/build.gradle.kts
@@ -0,0 +1,44 @@
+val ktor_version: String by project
+val kotlin_version: String by project
+val logback_version: String by project
+val exposedVersion: String by project
+
+plugins {
+ application
+ kotlin("jvm") version "1.4.32"
+ kotlin("plugin.serialization") version "1.4.32"
+}
+
+group = "com.gyf.csams"
+version = "0.0.1"
+
+application {
+ mainClassName = "io.ktor.server.netty.EngineMain"
+}
+
+repositories {
+ mavenLocal()
+ jcenter()
+ maven { url = uri("https://kotlin.bintray.com/ktor") }
+}
+
+dependencies {
+ implementation("org.jetbrains.kotlin:kotlin-stdlib-jdk8:$kotlin_version")
+ implementation("io.ktor:ktor-server-netty:$ktor_version")
+ implementation("ch.qos.logback:logback-classic:$logback_version")
+ implementation("io.ktor:ktor-server-core:$ktor_version")
+ implementation("io.ktor:ktor-gson:$ktor_version")
+ implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:1.1.0")
+ implementation("org.jetbrains.exposed:exposed-core:$exposedVersion")
+ implementation("org.jetbrains.exposed:exposed-dao:$exposedVersion")
+ implementation("org.jetbrains.exposed:exposed-jdbc:$exposedVersion")
+ implementation("mysql:mysql-connector-java:8.0.19")
+ implementation("com.zaxxer:HikariCP:3.4.2")
+ testImplementation("io.ktor:ktor-server-tests:$ktor_version")
+}
+
+kotlin.sourceSets["main"].kotlin.srcDirs("src")
+kotlin.sourceSets["test"].kotlin.srcDirs("test")
+
+sourceSets["main"].resources.srcDirs("resources")
+sourceSets["test"].resources.srcDirs("testresources")
diff --git a/gradle.properties b/gradle.properties
new file mode 100644
index 0000000..fb2a5e3
--- /dev/null
+++ b/gradle.properties
@@ -0,0 +1,5 @@
+ktor_version=1.5.1
+kotlin.code.style=official
+kotlin_version=1.4.32
+logback_version=1.2.1
+exposedVersion=0.31.1
diff --git a/gradle/wrapper/gradle-wrapper.jar b/gradle/wrapper/gradle-wrapper.jar
new file mode 100644
index 0000000..28861d2
Binary files /dev/null and b/gradle/wrapper/gradle-wrapper.jar differ
diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties
new file mode 100644
index 0000000..33682bb
--- /dev/null
+++ b/gradle/wrapper/gradle-wrapper.properties
@@ -0,0 +1,5 @@
+distributionBase=GRADLE_USER_HOME
+distributionPath=wrapper/dists
+distributionUrl=https\://services.gradle.org/distributions/gradle-6.6.1-all.zip
+zipStoreBase=GRADLE_USER_HOME
+zipStorePath=wrapper/dists
diff --git a/gradlew b/gradlew
new file mode 100644
index 0000000..cccdd3d
--- /dev/null
+++ b/gradlew
@@ -0,0 +1,172 @@
+#!/usr/bin/env sh
+
+##############################################################################
+##
+## Gradle start up script for UN*X
+##
+##############################################################################
+
+# Attempt to set APP_HOME
+# Resolve links: $0 may be a link
+PRG="$0"
+# Need this for relative symlinks.
+while [ -h "$PRG" ] ; do
+ ls=`ls -ld "$PRG"`
+ link=`expr "$ls" : '.*-> \(.*\)$'`
+ if expr "$link" : '/.*' > /dev/null; then
+ PRG="$link"
+ else
+ PRG=`dirname "$PRG"`"/$link"
+ fi
+done
+SAVED="`pwd`"
+cd "`dirname \"$PRG\"`/" >/dev/null
+APP_HOME="`pwd -P`"
+cd "$SAVED" >/dev/null
+
+APP_NAME="Gradle"
+APP_BASE_NAME=`basename "$0"`
+
+# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+DEFAULT_JVM_OPTS=""
+
+# Use the maximum available, or set MAX_FD != -1 to use that value.
+MAX_FD="maximum"
+
+warn () {
+ echo "$*"
+}
+
+die () {
+ echo
+ echo "$*"
+ echo
+ exit 1
+}
+
+# OS specific support (must be 'true' or 'false').
+cygwin=false
+msys=false
+darwin=false
+nonstop=false
+case "`uname`" in
+ CYGWIN* )
+ cygwin=true
+ ;;
+ Darwin* )
+ darwin=true
+ ;;
+ MINGW* )
+ msys=true
+ ;;
+ NONSTOP* )
+ nonstop=true
+ ;;
+esac
+
+CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
+
+# Determine the Java command to use to start the JVM.
+if [ -n "$JAVA_HOME" ] ; then
+ if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
+ # IBM's JDK on AIX uses strange locations for the executables
+ JAVACMD="$JAVA_HOME/jre/sh/java"
+ else
+ JAVACMD="$JAVA_HOME/bin/java"
+ fi
+ if [ ! -x "$JAVACMD" ] ; then
+ die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+ fi
+else
+ JAVACMD="java"
+ which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+fi
+
+# Increase the maximum file descriptors if we can.
+if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
+ MAX_FD_LIMIT=`ulimit -H -n`
+ if [ $? -eq 0 ] ; then
+ if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
+ MAX_FD="$MAX_FD_LIMIT"
+ fi
+ ulimit -n $MAX_FD
+ if [ $? -ne 0 ] ; then
+ warn "Could not set maximum file descriptor limit: $MAX_FD"
+ fi
+ else
+ warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
+ fi
+fi
+
+# For Darwin, add options to specify how the application appears in the dock
+if $darwin; then
+ GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
+fi
+
+# For Cygwin, switch paths to Windows format before running java
+if $cygwin ; then
+ APP_HOME=`cygpath --path --mixed "$APP_HOME"`
+ CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
+ JAVACMD=`cygpath --unix "$JAVACMD"`
+
+ # We build the pattern for arguments to be converted via cygpath
+ ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
+ SEP=""
+ for dir in $ROOTDIRSRAW ; do
+ ROOTDIRS="$ROOTDIRS$SEP$dir"
+ SEP="|"
+ done
+ OURCYGPATTERN="(^($ROOTDIRS))"
+ # Add a user-defined pattern to the cygpath arguments
+ if [ "$GRADLE_CYGPATTERN" != "" ] ; then
+ OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
+ fi
+ # Now convert the arguments - kludge to limit ourselves to /bin/sh
+ i=0
+ for arg in "$@" ; do
+ CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
+ CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option
+
+ if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition
+ eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
+ else
+ eval `echo args$i`="\"$arg\""
+ fi
+ i=$((i+1))
+ done
+ case $i in
+ (0) set -- ;;
+ (1) set -- "$args0" ;;
+ (2) set -- "$args0" "$args1" ;;
+ (3) set -- "$args0" "$args1" "$args2" ;;
+ (4) set -- "$args0" "$args1" "$args2" "$args3" ;;
+ (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
+ (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
+ (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
+ (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
+ (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
+ esac
+fi
+
+# Escape application args
+save () {
+ for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
+ echo " "
+}
+APP_ARGS=$(save "$@")
+
+# Collect all arguments for the java command, following the shell quoting and substitution rules
+eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
+
+# by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong
+if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then
+ cd "$(dirname "$0")"
+fi
+
+exec "$JAVACMD" "$@"
diff --git a/gradlew.bat b/gradlew.bat
new file mode 100644
index 0000000..f955316
--- /dev/null
+++ b/gradlew.bat
@@ -0,0 +1,84 @@
+@if "%DEBUG%" == "" @echo off
+@rem ##########################################################################
+@rem
+@rem Gradle startup script for Windows
+@rem
+@rem ##########################################################################
+
+@rem Set local scope for the variables with windows NT shell
+if "%OS%"=="Windows_NT" setlocal
+
+set DIRNAME=%~dp0
+if "%DIRNAME%" == "" set DIRNAME=.
+set APP_BASE_NAME=%~n0
+set APP_HOME=%DIRNAME%
+
+@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+set DEFAULT_JVM_OPTS=
+
+@rem Find java.exe
+if defined JAVA_HOME goto findJavaFromJavaHome
+
+set JAVA_EXE=java.exe
+%JAVA_EXE% -version >NUL 2>&1
+if "%ERRORLEVEL%" == "0" goto init
+
+echo.
+echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+echo.
+echo Please set the JAVA_HOME variable in your environment to match the
+echo location of your Java installation.
+
+goto fail
+
+:findJavaFromJavaHome
+set JAVA_HOME=%JAVA_HOME:"=%
+set JAVA_EXE=%JAVA_HOME%/bin/java.exe
+
+if exist "%JAVA_EXE%" goto init
+
+echo.
+echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
+echo.
+echo Please set the JAVA_HOME variable in your environment to match the
+echo location of your Java installation.
+
+goto fail
+
+:init
+@rem Get command-line arguments, handling Windows variants
+
+if not "%OS%" == "Windows_NT" goto win9xME_args
+
+:win9xME_args
+@rem Slurp the command line arguments.
+set CMD_LINE_ARGS=
+set _SKIP=2
+
+:win9xME_args_slurp
+if "x%~1" == "x" goto execute
+
+set CMD_LINE_ARGS=%*
+
+:execute
+@rem Setup the command line
+
+set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
+
+@rem Execute Gradle
+"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
+
+:end
+@rem End local scope for the variables with windows NT shell
+if "%ERRORLEVEL%"=="0" goto mainEnd
+
+:fail
+rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
+rem the _cmd.exe /c_ return code!
+if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
+exit /b 1
+
+:mainEnd
+if "%OS%"=="Windows_NT" endlocal
+
+:omega
diff --git a/resources/application.conf b/resources/application.conf
new file mode 100644
index 0000000..2e2b760
--- /dev/null
+++ b/resources/application.conf
@@ -0,0 +1,21 @@
+ktor {
+ deployment {
+ port = 8080
+ port = ${?PORT}
+ mysql = {
+ username = root
+ password = 123456
+ jdbcUrl = "jdbc:mysql://localhost:3306/csams"
+ driverClassName = "com.mysql.cj.jdbc.Driver"
+ }
+
+ watch = [ classes ]
+ }
+ application {
+ modules = [
+ com.gyf.csams.ApplicationKt.module,
+ com.gyf.csams.MySQLKt.MySQL,
+ com.gyf.csams.ApplicationKt.Controller
+ ]
+ }
+}
diff --git a/resources/logback.xml b/resources/logback.xml
new file mode 100644
index 0000000..bdbb64e
--- /dev/null
+++ b/resources/logback.xml
@@ -0,0 +1,12 @@
+
+
+
+ %d{YYYY-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %msg%n
+
+
+
+
+
+
+
+
diff --git a/settings.gradle.kts b/settings.gradle.kts
new file mode 100644
index 0000000..2b5cd2c
--- /dev/null
+++ b/settings.gradle.kts
@@ -0,0 +1 @@
+rootProject.name = "server"
diff --git a/src/AccountController.kt b/src/AccountController.kt
new file mode 100644
index 0000000..90d559a
--- /dev/null
+++ b/src/AccountController.kt
@@ -0,0 +1,46 @@
+package com.gyf.csams
+
+import io.ktor.application.*
+import io.ktor.request.*
+import io.ktor.response.*
+import io.ktor.routing.*
+
+
+fun Application.AccountController() {
+
+ routing {
+ route(path = "$ApiPathPrefix/account"){
+ route(path="/register") {
+ /**
+ * 检测学号是否已注册
+ */
+ get(path = "/checkId") {
+ val studentId = call.request.queryParameters["studentId"]
+ if (studentId?.isNotEmpty() == true) {
+ if (Service.registered(studentId)) {
+ call.respond(ApiResponse(code = 200, message = "学号已注册", body = true))
+ } else {
+ call.respond(ApiResponse(code = 200, message = "学号可注册", body = false))
+ }
+ } else {
+ call.respond(Simple.error("学号检测失败，请联系管理员"))
+ }
+ }
+
+ /**
+ * 注册账号
+ */
+ post {
+ val userVo = call.receive()
+ val userResDto = Service.register(userVo)
+ if (userResDto != null) {
+ call.respond(ApiResponse(code = 200, message = "注册成功", userResDto))
+ } else {
+ call.respond(ApiResponse(code = 400, message = "注册失败", body = null))
+ }
+ }
+ }
+ }
+ }
+}
+
diff --git a/src/Application.kt b/src/Application.kt
new file mode 100644
index 0000000..76ab661
--- /dev/null
+++ b/src/Application.kt
@@ -0,0 +1,34 @@
+package com.gyf.csams
+
+import io.ktor.application.*
+import io.ktor.features.*
+import io.ktor.gson.*
+import io.ktor.http.*
+
+fun main(args: Array): Unit = io.ktor.server.netty.EngineMain.main(args)
+
+@Suppress("unused") // Referenced in application.conf
+@kotlin.jvm.JvmOverloads
+fun Application.module(testing: Boolean = false) {
+ install(CORS) {
+ method(HttpMethod.Options)
+ method(HttpMethod.Put)
+ method(HttpMethod.Delete)
+ method(HttpMethod.Patch)
+ header(HttpHeaders.Authorization)
+ header("MyCustomHeader")
+ allowCredentials = true
+ anyHost() // @TODO: Don't do this in production if possible. Try to limit it.
+ }
+
+ install(ContentNegotiation) {
+ gson {
+ }
+ }
+
+}
+
+
+fun Application.Controller(testing: Boolean = false){
+ this.AccountController()
+}
diff --git a/src/Dao.kt b/src/Dao.kt
new file mode 100644
index 0000000..206b363
--- /dev/null
+++ b/src/Dao.kt
@@ -0,0 +1,12 @@
+package com.gyf.csams
+
+import org.jetbrains.exposed.sql.Column
+import org.jetbrains.exposed.sql.Table
+
+object User: Table(){
+ val studentId:Column = varchar(name="student_id",length = 8)
+ val name:Column = varchar(name="name",length = 10)
+ val password:Column = varchar(name="password",length = 32)
+ override val primaryKey: PrimaryKey
+ get() = PrimaryKey(studentId)
+}
\ No newline at end of file
diff --git a/src/MySQL.kt b/src/MySQL.kt
new file mode 100644
index 0000000..2313a24
--- /dev/null
+++ b/src/MySQL.kt
@@ -0,0 +1,28 @@
+package com.gyf.csams
+
+import com.zaxxer.hikari.HikariConfig
+import com.zaxxer.hikari.HikariDataSource
+import io.ktor.application.*
+import org.jetbrains.exposed.sql.Database
+import org.jetbrains.exposed.sql.SchemaUtils
+import org.jetbrains.exposed.sql.transactions.transaction
+
+fun Application.MySQL(testing: Boolean = false){
+ val config = HikariConfig().apply {
+ jdbcUrl = environment.config.property("ktor.deployment.mysql.jdbcUrl").getString()
+ driverClassName = environment.config.property("ktor.deployment.mysql.driverClassName").getString()
+ username = environment.config.property("ktor.deployment.mysql.username").getString()
+ password = environment.config.property("ktor.deployment.mysql.password").getString()
+ maximumPoolSize = 10
+ }
+ val dataSource = HikariDataSource(config)
+ Database.connect(dataSource)
+
+ initTable()
+}
+
+fun initTable(){
+ transaction {
+ SchemaUtils.createMissingTablesAndColumns(User)
+ }
+}
\ No newline at end of file
diff --git a/src/Service.kt b/src/Service.kt
new file mode 100644
index 0000000..0bb604f
--- /dev/null
+++ b/src/Service.kt
@@ -0,0 +1,46 @@
+package com.gyf.csams
+
+import org.jetbrains.exposed.sql.insert
+import org.jetbrains.exposed.sql.select
+import org.jetbrains.exposed.sql.transactions.transaction
+import org.slf4j.LoggerFactory
+
+
+class Service {
+
+
+ companion object {
+ private val logger = LoggerFactory.getLogger(Service::class.java)
+ /**
+ * 检查学号是否已注册
+ */
+ fun registered(selectId: String): Boolean {
+ return transaction {
+ return@transaction User.select { User.studentId eq selectId }.count().toInt() > 0
+ }
+ }
+
+ /**
+ * 注册
+ */
+ fun register(userVo: UserVo): UserResDto? {
+ var userResDto: UserResDto? = null
+ try {
+ transaction {
+ val _pwd = randomNum(8)
+ val result = User.insert {
+ it[studentId] = userVo.studentId
+ it[name] = userVo.name
+ it[password] = _pwd.md5()
+ }.resultedValues
+ if (result?.isNotEmpty() == true) {
+ userResDto = UserResDto(password = _pwd)
+ }
+ }
+ } catch (e: Exception) {
+ logger.error("注册失败,发生异常：$e")
+ }
+ return userResDto
+ }
+ }
+}
\ No newline at end of file
diff --git a/src/Util.kt b/src/Util.kt
new file mode 100644
index 0000000..e5c1549
--- /dev/null
+++ b/src/Util.kt
@@ -0,0 +1,21 @@
+package com.gyf.csams
+
+import java.math.BigInteger
+import java.security.MessageDigest
+
+const val ApiPathPrefix="/api"
+/**
+ * 从0-9生成随机字符串
+ */
+fun randomNum(length:Int=8):String{
+ return List(length) {('0' .. '9').random()}.joinToString("")
+}
+
+/**
+ * md5加密字符串
+ */
+fun String.md5(): String {
+ val md = MessageDigest.getInstance("MD5")
+ return BigInteger(1, md.digest(toByteArray())).toString(16).padStart(32, '0')
+}
+
diff --git a/src/Vo.kt b/src/Vo.kt
new file mode 100644
index 0000000..e441f1f
--- /dev/null
+++ b/src/Vo.kt
@@ -0,0 +1,24 @@
+package com.gyf.csams
+
+import kotlinx.serialization.Serializable
+
+
+data class ApiResponse(val code:Int,val message:String,val body:T?=null)
+
+class Simple {
+ companion object {
+ fun success(message: String):ApiResponse{
+ return ApiResponse(code = 200, message = message, body = null)
+ }
+
+ fun error(message: String): ApiResponse {
+ return ApiResponse(code = 500, message = message, body = null)
+ }
+ }
+}
+
+
+@Serializable
+data class UserVo(val studentId:String,val name:String)
+
+data class UserResDto(val password:String)
\ No newline at end of file
diff --git a/test/ApplicationTest.kt b/test/ApplicationTest.kt
new file mode 100644
index 0000000..124f25d
--- /dev/null
+++ b/test/ApplicationTest.kt
@@ -0,0 +1,50 @@
+package com.gyf.csams
+
+import io.ktor.config.*
+import io.ktor.http.*
+import io.ktor.server.testing.*
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+class ApplicationTest {
+ @Test
+ fun testRoot() {
+ withTestApplication({ module(testing = true) }) {
+ handleRequest(HttpMethod.Get, "/").apply {
+ assertEquals(HttpStatusCode.OK, response.status())
+ assertEquals("HELLO WORLD!", response.content)
+ }
+ }
+ }
+
+ @Test
+ fun testRandom(){
+ val c= randomNum(8)
+ println(c)
+ val d= randomNum(6)
+ println(d)
+ }
+
+ @Test
+ fun testMd5(){
+ println("admin".md5())
+ }
+
+ @Test
+ fun testInsertUser(){
+ withTestApplication({
+ (environment.config as MapApplicationConfig).apply {
+ // Set here the properties
+ put("ktor.deployment.mysql.jdbcUrl", "jdbc:mysql://localhost:3306/csams")
+ put("ktor.deployment.mysql.driverClassName", "com.mysql.cj.jdbc.Driver")
+ put("ktor.deployment.mysql.username", "root")
+ put("ktor.deployment.mysql.password", "123456")
+ }
+ MySQL(testing = true)
+ }) {
+ val c=Service.registered(selectId = "77889")
+ println(c)
+ }
+ }
+
+}