diff --git a/springboot/src/main/java/com/example/controller/WebController.java b/springboot/src/main/java/com/example/controller/WebController.java index 04a087da..86cd0428 100644 --- a/springboot/src/main/java/com/example/controller/WebController.java +++ b/springboot/src/main/java/com/example/controller/WebController.java @@ -1,8 +1,11 @@ package com.example.controller; +import cn.hutool.core.util.ObjectUtil; import com.example.common.Result; -import com.example.entity.Admin; +import com.example.common.RoleEnum; +import com.example.entity.Account; import com.example.service.AdminService; +import com.example.service.StudentService; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; @@ -16,6 +19,8 @@ public class WebController { @Resource private AdminService adminService; + @Resource + private StudentService studentService; /** * 默认请求接口 @@ -29,9 +34,31 @@ public class WebController { * 登录接口 */ @PostMapping("/login") - public Result login(@RequestBody Admin admin) { - Admin dbAdmin = adminService.login(admin); - return Result.success(dbAdmin); + public Result login(@RequestBody Account account) { + Account dbAccount; + if (RoleEnum.ADMIN.name().equals(account.getRole())) { // 管理员登录 + dbAccount = adminService.login(account); + } else if (RoleEnum.STUDENT.name().equals(account.getRole())) { // 学生登录 + dbAccount = studentService.login(account); + } else { + return Result.error("角色错误"); + } + return Result.success(dbAccount); + } + + + + + /** + * 注册接口 + */ + @PostMapping("/register") + public Result register(@RequestBody Account account) { + if (ObjectUtil.isEmpty(account.getUsername()) || ObjectUtil.isEmpty(account.getPassword())) { + return Result.error("账号或密码必填"); + } + studentService.register(account); + return Result.success(); } }