pan
4 years ago
commit
4667bce9a9
@ -0,0 +1,7 @@ |
|||||||
|
/.gradle |
||||||
|
/.idea |
||||||
|
/out |
||||||
|
/build |
||||||
|
*.iml |
||||||
|
*.ipr |
||||||
|
*.iws |
||||||
@ -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") |
||||||
@ -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 |
||||||
Binary file not shown.
@ -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 |
||||||
@ -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" "$@" |
||||||
@ -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 |
||||||
@ -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 |
||||||
|
] |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,12 @@ |
|||||||
|
<configuration> |
||||||
|
<appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender"> |
||||||
|
<encoder> |
||||||
|
<pattern>%d{YYYY-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %msg%n</pattern> |
||||||
|
</encoder> |
||||||
|
</appender> |
||||||
|
<root level="trace"> |
||||||
|
<appender-ref ref="STDOUT"/> |
||||||
|
</root> |
||||||
|
<logger name="org.eclipse.jetty" level="INFO"/> |
||||||
|
<logger name="io.netty" level="INFO"/> |
||||||
|
</configuration> |
||||||
@ -0,0 +1 @@ |
|||||||
|
rootProject.name = "server" |
||||||
@ -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<UserVo>() |
||||||
|
val userResDto = Service.register(userVo) |
||||||
|
if (userResDto != null) { |
||||||
|
call.respond(ApiResponse(code = 200, message = "注册成功", userResDto)) |
||||||
|
} else { |
||||||
|
call.respond(ApiResponse(code = 400, message = "注册失败", body = null)) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
@ -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<String>): 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() |
||||||
|
} |
||||||
@ -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<String> = varchar(name="student_id",length = 8) |
||||||
|
val name:Column<String> = varchar(name="name",length = 10) |
||||||
|
val password:Column<String> = varchar(name="password",length = 32) |
||||||
|
override val primaryKey: PrimaryKey |
||||||
|
get() = PrimaryKey(studentId) |
||||||
|
} |
||||||
@ -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) |
||||||
|
} |
||||||
|
} |
||||||
@ -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 |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -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') |
||||||
|
} |
||||||
|
|
||||||
@ -0,0 +1,24 @@ |
|||||||
|
package com.gyf.csams |
||||||
|
|
||||||
|
import kotlinx.serialization.Serializable |
||||||
|
|
||||||
|
|
||||||
|
data class ApiResponse<T>(val code:Int,val message:String,val body:T?=null) |
||||||
|
|
||||||
|
class Simple { |
||||||
|
companion object { |
||||||
|
fun success(message: String):ApiResponse<Any>{ |
||||||
|
return ApiResponse(code = 200, message = message, body = null) |
||||||
|
} |
||||||
|
|
||||||
|
fun error(message: String): ApiResponse<Any> { |
||||||
|
return ApiResponse(code = 500, message = message, body = null) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
@Serializable |
||||||
|
data class UserVo(val studentId:String,val name:String) |
||||||
|
|
||||||
|
data class UserResDto(val password:String) |
||||||
@ -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) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue