实现登录注册接口
Change-Id: I3d57cca89cac8945d562f6a39127b3454c1cd9ac
diff --git a/src/main/java/edu/bjtu/groupone/backend/BackendApplication.java b/src/main/java/edu/bjtu/groupone/backend/BackendApplication.java
index 32033e0..1cc2086 100644
--- a/src/main/java/edu/bjtu/groupone/backend/BackendApplication.java
+++ b/src/main/java/edu/bjtu/groupone/backend/BackendApplication.java
@@ -1,13 +1,19 @@
package edu.bjtu.groupone.backend;
+import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.web.servlet.ServletComponentScan;
+import org.springframework.scheduling.annotation.EnableScheduling;
+@EnableScheduling
+@ServletComponentScan
@SpringBootApplication
+@MapperScan("edu.bjtu.groupone.backend.mapper")
public class BackendApplication {
- public static void main(String[] args) {
- SpringApplication.run(BackendApplication.class, args);
- }
+ public static void main(String[] args) {
+ SpringApplication.run(BackendApplication.class, args);
+ }
}
diff --git a/src/main/java/edu/bjtu/groupone/backend/config/Config.java b/src/main/java/edu/bjtu/groupone/backend/config/Config.java
new file mode 100644
index 0000000..cfc3745
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/config/Config.java
@@ -0,0 +1,34 @@
+// src/main/java/edu/bjtu/groupone/backend/config/Config.java
+package edu.bjtu.groupone.backend.config;
+
+import edu.bjtu.groupone.backend.interceptor.Interceptor;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.servlet.config.annotation.*;
+
+@Configuration
+public class Config implements WebMvcConfigurer {
+
+ @Autowired
+ private Interceptor authInterceptor;
+
+ /** 全局 CORS 配置 */
+ @Override
+ public void addCorsMappings(CorsRegistry registry) {
+ registry.addMapping("/**")
+ // Vite 默认端口 5173，如果你的 React 在 3000 也一并写上
+ .allowedOrigins("http://localhost:5173", "http://localhost:3000")
+ .allowedMethods("GET", "POST", "PUT", "DELETE", "OPTIONS")
+ .allowedHeaders("*")
+ // 如果需要前端带 Cookie（如 JWT 存在 Cookie）就打开
+ .allowCredentials(true)
+ .maxAge(3600);
+ }
+
+ @Override
+ public void addInterceptors(InterceptorRegistry registry) {
+ registry.addInterceptor(authInterceptor)
+ .addPathPatterns("/**");
+ }
+ // 其它 CORS、静态资源、Swagger 配置保持不动
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/config/SwaggerConfig.java b/src/main/java/edu/bjtu/groupone/backend/config/SwaggerConfig.java
new file mode 100644
index 0000000..042c278
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/config/SwaggerConfig.java
@@ -0,0 +1,39 @@
+package edu.bjtu.groupone.backend.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import springfox.documentation.builders.PathSelectors;
+import springfox.documentation.builders.RequestHandlerSelectors;
+import springfox.documentation.service.ApiInfo;
+import springfox.documentation.spi.DocumentationType;
+import springfox.documentation.spring.web.plugins.Docket;
+
+@Configuration
+//@EnableOpenApi
+public class SwaggerConfig {
+ @Bean
+ public Docket docket(){
+ return new Docket(DocumentationType.OAS_30)
+ .apiInfo(apiInfo())
+ .enable(true)
+ .groupName("CJB")
+ .select()
+ .apis(RequestHandlerSelectors.basePackage("com.example.eliteedu_prism.controller"))
+ .paths(PathSelectors.ant("/controller/**"))
+ .build();
+ }
+
+
+ @SuppressWarnings("all")
+ public ApiInfo apiInfo(){
+ return new ApiInfo(
+ "zrj's api",
+ "redis project",
+ "v1.0",
+ "2261839618@qq.com", //开发者团队的邮箱
+ "ZRJ",
+ "Apache 2.0", //许可证
+ "http://www.apache.org/licenses/LICENSE-2.0" //许可证链接
+ );
+ }
+}
\ No newline at end of file
diff --git a/src/main/java/edu/bjtu/groupone/backend/controller/UserController.java b/src/main/java/edu/bjtu/groupone/backend/controller/UserController.java
new file mode 100644
index 0000000..1388940
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/controller/UserController.java
@@ -0,0 +1,113 @@
+package edu.bjtu.groupone.backend.controller;
+
+import edu.bjtu.groupone.backend.model.InviteCode;
+import edu.bjtu.groupone.backend.model.Result;
+import edu.bjtu.groupone.backend.model.User;
+import edu.bjtu.groupone.backend.service.InviteCodeService;
+import edu.bjtu.groupone.backend.service.UserService;
+import edu.bjtu.groupone.backend.utils.JwtUtils;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.util.HashMap;
+import java.util.Map;
+
+@CrossOrigin
+@Slf4j
+@RestController
+public class UserController {
+
+ @Autowired
+ private UserService userService;
+ @Autowired
+ private InviteCodeService inviteCodeService;
+
+ /** 登录：接收表单字段 identificationNumber、password */
+ @PostMapping("/login")
+ public Result login(
+ @RequestParam Integer identificationNumber,
+ @RequestParam String password
+ ) {
+ User u = new User();
+ u.setIdentificationNumber(identificationNumber);
+ u.setPassword(password);
+ User user = userService.login(u);
+ if (user != null) {
+ Map<String, Object> payload = new HashMap<>();
+ payload.put("id", user.getUserId());
+ payload.put("username", user.getUsername());
+ payload.put("identificationNumber", user.getIdentificationNumber());
+ String token = JwtUtils.generateJwt(payload);
+
+ // 打印到控制台
+ log.info("【登录成功】生成的 JWT token = {}", token);
+ return Result.success(token);
+ }
+ return Result.error("用户名或密码错误");
+ }
+
+ /** 注册：接收表单字段 username、email、password、verificationCode、identificationNumber、inviteCode */
+ @PostMapping("/register")
+ public Result register(
+ @RequestParam String username,
+ @RequestParam String email,
+ @RequestParam String password,
+ @RequestParam String verificationCode,
+ @RequestParam Integer identificationNumber,
+ @RequestParam String inviteCode
+ ) {
+ // 1. 验证邀请码
+ if (inviteCodeService.validateInviteCode(inviteCode) == null) {
+ return Result.error("无效邀请码或已过期");
+ }
+ // 2. 验证邮箱验证码
+ if (!userService.verifyCode(email, verificationCode)) {
+ return Result.error("验证码错误或已过期");
+ }
+ // 3. 检查邮箱唯一
+ if (userService.isEmailExists(email)) {
+ return Result.error("邮箱已被注册");
+ }
+ // 4. 构造用户实体并设置所有非空字段
+ User newUser = new User();
+ newUser.setUsername(username);
+ newUser.setEmail(email);
+ newUser.setPassword(password);
+ newUser.setIdentificationNumber(identificationNumber);
+ // registrationDate 格式要和你的 Mapper SQL 对应，如果是 String 存储：
+ newUser.setRegistrationDate(
+ LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"))
+ );
+ // 5. 插入到数据库
+ userService.register(newUser);
+ // 6. 扣减邀请码次数
+ if (!inviteCodeService.useInviteCode(inviteCode)) {
+ log.error("邀请码扣减失败, code={}", inviteCode);
+ }
+ return Result.success("注册成功");
+ }
+
+ /** 发送邮箱验证码：接收表单字段 email、inviteCode */
+ @PostMapping("/sendVerification")
+ public Result sendVerification(
+ @RequestParam String email,
+ @RequestParam String inviteCode
+ ) {
+ if (inviteCodeService.validateInviteCode(inviteCode) == null) {
+ return Result.error("无效邀请码");
+ }
+ userService.sendVerificationCode(email);
+ return Result.success("验证码已发送");
+ }
+
+ @GetMapping("/api/me")
+ public Result me(@RequestHeader("token") String token) {
+ // 从 token 解析出用户信息，JwtUtils.parseJwt 返回一个 Map
+ var claims = JwtUtils.parseJwt(token);
+ // 你可以根据需要返回完整 User，也可以直接返回用户名
+ return Result.success(claims);
+ }
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/interceptor/Interceptor.java b/src/main/java/edu/bjtu/groupone/backend/interceptor/Interceptor.java
new file mode 100644
index 0000000..f3b9d2d
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/interceptor/Interceptor.java
@@ -0,0 +1,58 @@
+// src/main/java/edu/bjtu/groupone/backend/interceptor/Interceptor.java
+package edu.bjtu.groupone.backend.interceptor;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import edu.bjtu.groupone.backend.model.Result;
+import edu.bjtu.groupone.backend.utils.JwtUtils;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import org.springframework.util.StringUtils;
+import org.springframework.web.servlet.HandlerInterceptor;
+
+@Slf4j
+@Component
+public class Interceptor implements HandlerInterceptor {
+ @Override
+ public boolean preHandle(HttpServletRequest req,
+ HttpServletResponse resp,
+ Object handler) throws Exception {
+ String url = req.getRequestURI();
+ log.info("Request URI: {}", url);
+ // 放行无需鉴权的路径：登录、注册、发码、静态资源、HTML 页面
+ if (url.equals("/") ||
+ url.endsWith(".html") ||
+ url.contains("login") ||
+ url.contains("register") ||
+ url.equals("/api/me") || // 放行用户信息接口
+ url.equals("/error") || // 放行错误页面
+ url.contains("sendVerification") ||
+ url.startsWith("/swagger-ui") ||
+ url.startsWith("/v3/api-docs") ||
+ url.startsWith("/favicon.ico") ||
+ url.startsWith("/static/")) {
+ return true;
+ }
+ // 从请求头读取 token
+ String token = req.getHeader("token");
+ if (!StringUtils.hasLength(token)) {
+ resp.setContentType("application/json;charset=UTF-8");
+ resp.getWriter()
+ .write(new ObjectMapper()
+ .writeValueAsString(Result.error("NOT_LOGIN")));
+ return false;
+ }
+ try {
+ JwtUtils.parseJwt(token);
+ return true;
+ } catch (Exception e) {
+ resp.setContentType("application/json;charset=UTF-8");
+ resp.getWriter()
+ .write(new ObjectMapper()
+ .writeValueAsString(Result.error("NOT_LOGIN")));
+ return false;
+ }
+ }
+
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/mapper/InviteCodeMapper.java b/src/main/java/edu/bjtu/groupone/backend/mapper/InviteCodeMapper.java
new file mode 100644
index 0000000..78bb7e0
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/mapper/InviteCodeMapper.java
@@ -0,0 +1,14 @@
+package edu.bjtu.groupone.backend.mapper;
+
+import edu.bjtu.groupone.backend.model.InviteCode;
+import org.apache.ibatis.annotations.*;
+
+@Mapper
+public interface InviteCodeMapper {
+ @Select("SELECT * FROM invite_code WHERE code = #{code}")
+ InviteCode selectByCode(String code);
+
+ @Update("UPDATE invite_code SET remaining_uses = remaining_uses - 1 " +
+ "WHERE code = #{code} AND remaining_uses > 0")
+ int decrementRemainingUses(String code);
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/mapper/UserMapper.java b/src/main/java/edu/bjtu/groupone/backend/mapper/UserMapper.java
new file mode 100644
index 0000000..3cbf2c1
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/mapper/UserMapper.java
@@ -0,0 +1,22 @@
+package edu.bjtu.groupone.backend.mapper;
+
+
+import edu.bjtu.groupone.backend.model.User;
+import org.apache.ibatis.annotations.*;
+
+@Mapper
+public interface UserMapper {
+ @Select("select * from user where identification_number=#{identificationNumber} and password=#{password}")
+ User login(User user);
+
+ @Select("SELECT * FROM user WHERE username = #{username}")
+ User selectByUsername(String username);
+
+ @Select("SELECT * FROM user WHERE email = #{email}")
+ User selectByEmail(String email);
+
+ @Insert("INSERT INTO user(username, email, password, registration_date, identification_number) " +
+ "VALUES(#{username}, #{email}, #{password}, #{registrationDate}, #{identificationNumber})")
+ @Options(useGeneratedKeys = true, keyProperty = "userId")
+ void insertUser(User user);
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/model/InviteCode.java b/src/main/java/edu/bjtu/groupone/backend/model/InviteCode.java
new file mode 100644
index 0000000..f1019f9
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/model/InviteCode.java
@@ -0,0 +1,18 @@
+package edu.bjtu.groupone.backend.model;
+import com.fasterxml.jackson.annotation.JsonFormat;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import java.time.LocalDateTime;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+public class InviteCode {
+ private String code;
+ private int maxUses;
+ private int remainingUses;
+ @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
+ private LocalDateTime expiryTime; //失效时间
+ private int createdBy; //创建者的id
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/model/Result.java b/src/main/java/edu/bjtu/groupone/backend/model/Result.java
new file mode 100644
index 0000000..f3d44ec
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/model/Result.java
@@ -0,0 +1,27 @@
+package edu.bjtu.groupone.backend.model;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class Result {
+ private Integer code;//0表示成功，1表示失败
+ private String msg;//提示信息
+ private Object data;//要返回的数据
+ public static Result success() {//增删改响应成功
+
+ return new Result(0, "success", null);
+ }
+ public static Result success(Object data) {//增删改响应成功
+
+ return new Result(0, "success", data);
+ }
+ public static Result error(String msg) {//增删改响应成功
+
+ return new Result(1, msg, null);
+ }
+
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/model/User.java b/src/main/java/edu/bjtu/groupone/backend/model/User.java
new file mode 100644
index 0000000..95b2f00
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/model/User.java
@@ -0,0 +1,23 @@
+package edu.bjtu.groupone.backend.model;
+
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+public class User {
+ private int userId;
+ private String username;
+ private String email;
+ private String password;
+ private String address;
+ private String role;
+ private String profilePic;
+ private String registrationDate;
+ private int identificationNumber;
+ private String avatar;// 头像
+ private boolean isfollowed = false;
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/model/UserDTO.java b/src/main/java/edu/bjtu/groupone/backend/model/UserDTO.java
new file mode 100644
index 0000000..279ac26
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/model/UserDTO.java
@@ -0,0 +1,16 @@
+package edu.bjtu.groupone.backend.model;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class UserDTO {
+ private User user; // 原来的 User 对象
+ private boolean isFollowed; // 是否关注的状态
+
+ // 构造函数
+
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/service/InviteCodeService.java b/src/main/java/edu/bjtu/groupone/backend/service/InviteCodeService.java
new file mode 100644
index 0000000..9391e14
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/service/InviteCodeService.java
@@ -0,0 +1,8 @@
+package edu.bjtu.groupone.backend.service;
+
+import edu.bjtu.groupone.backend.model.InviteCode;
+
+public interface InviteCodeService {
+ InviteCode validateInviteCode(String code);
+ boolean useInviteCode(String code);
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/service/UserService.java b/src/main/java/edu/bjtu/groupone/backend/service/UserService.java
new file mode 100644
index 0000000..662dafe
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/service/UserService.java
@@ -0,0 +1,15 @@
+package edu.bjtu.groupone.backend.service;
+
+
+import edu.bjtu.groupone.backend.model.User;
+
+public interface UserService {
+
+ User login(User user);
+
+ boolean isUsernameExists(String username);
+ boolean isEmailExists(String email);
+ void register(User user);
+ void sendVerificationCode(String email);
+ boolean verifyCode(String email, String code);
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/service/impl/InviteCodeServiceImpl.java b/src/main/java/edu/bjtu/groupone/backend/service/impl/InviteCodeServiceImpl.java
new file mode 100644
index 0000000..b08b7a5
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/service/impl/InviteCodeServiceImpl.java
@@ -0,0 +1,45 @@
+package edu.bjtu.groupone.backend.service.impl;
+
+import edu.bjtu.groupone.backend.mapper.InviteCodeMapper;
+import edu.bjtu.groupone.backend.model.InviteCode;
+import edu.bjtu.groupone.backend.service.InviteCodeService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.time.LocalDateTime;
+
+@Service
+public class InviteCodeServiceImpl implements InviteCodeService {
+
+ @Autowired
+ private InviteCodeMapper inviteCodeMapper;
+
+ /**
+ * 验证邀请码是否存在、未过期且剩余次数大于 0
+ */
+ @Override
+ public InviteCode validateInviteCode(String code) {
+ InviteCode inviteCode = inviteCodeMapper.selectByCode(code);
+ if (inviteCode == null
+ || inviteCode.getRemainingUses() <= 0
+ || inviteCode.getExpiryTime().isBefore(LocalDateTime.now())) {
+ return null;
+ }
+ return inviteCode;
+ }
+
+ /**
+ * 使用一次邀请码：先验证再扣减剩余次数
+ */
+ @Override
+ public boolean useInviteCode(String code) {
+ // 先调用 validateInviteCode 保证业务合理性
+ InviteCode valid = validateInviteCode(code);
+ if (valid == null) {
+ return false;
+ }
+ // 再执行扣减
+ int updated = inviteCodeMapper.decrementRemainingUses(code);
+ return updated > 0;
+ }
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/service/impl/UserServImpl.java b/src/main/java/edu/bjtu/groupone/backend/service/impl/UserServImpl.java
new file mode 100644
index 0000000..828b78d
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/service/impl/UserServImpl.java
@@ -0,0 +1,91 @@
+package edu.bjtu.groupone.backend.service.impl;
+
+import edu.bjtu.groupone.backend.mapper.UserMapper;
+import edu.bjtu.groupone.backend.model.User;
+import edu.bjtu.groupone.backend.service.UserService;
+import edu.bjtu.groupone.backend.utils.EmailUtil;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.util.Random;
+import java.util.concurrent.ConcurrentHashMap;
+
+@Service
+public class UserServImpl implements UserService {
+
+ @Autowired
+ private UserMapper userMapper;
+ @Autowired
+ private EmailUtil emailUtil;
+ private final ConcurrentHashMap<String, String> emailCodes = new ConcurrentHashMap<>();
+ private static final long CODE_EXPIRE_MINUTES = 5;
+
+ @Override
+ public User login(User user) {
+ return userMapper.login(user);
+ }
+
+ @Override
+ public boolean isUsernameExists(String username) {
+ return userMapper.selectByUsername(username) != null;
+ }
+
+ @Override
+ public boolean isEmailExists(String email) {
+ return userMapper.selectByEmail(email) != null;
+ }
+
+ @Override
+ public void register(User user) {
+ // 如果前端未提供用户名，可生成默认
+ String username = user.getUsername();
+ if (username == null || username.isBlank()) {
+ do {
+ username = generateDefaultUsername(user.getEmail());
+ } while (isUsernameExists(username));
+ user.setUsername(username);
+ }
+ // 设置注册时间
+ user.setRegistrationDate(
+ LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"))
+ );
+ // 插入用户记录
+ userMapper.insertUser(user);
+ }
+
+ @Override
+ public void sendVerificationCode(String email) {
+ String code = generateRandomCode();
+ emailUtil.sendVerificationEmail(email, code);
+ long expireAt = System.currentTimeMillis() + CODE_EXPIRE_MINUTES * 60 * 1000;
+ emailCodes.put(email, code + "|" + expireAt);
+ }
+
+ @Override
+ public boolean verifyCode(String email, String inputCode) {
+ String stored = emailCodes.get(email);
+ if (stored == null) {
+ return false;
+ }
+ String[] parts = stored.split("\\|");
+ if (parts.length != 2) {
+ return false;
+ }
+ String code = parts[0];
+ long expireTime = Long.parseLong(parts[1]);
+ return System.currentTimeMillis() <= expireTime && code.equals(inputCode);
+ }
+
+ private String generateRandomCode() {
+ return String.format("%06d", new Random().nextInt(1_000000));
+ }
+
+ private String generateDefaultUsername(String email) {
+ String prefix = email != null && email.contains("@")
+ ? email.substring(0, email.indexOf("@"))
+ : "user";
+ return "user_" + prefix + "_" + String.format("%04d", new Random().nextInt(10000));
+ }
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/utils/AiUtils.java b/src/main/java/edu/bjtu/groupone/backend/utils/AiUtils.java
new file mode 100644
index 0000000..8dd8a8b
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/utils/AiUtils.java
@@ -0,0 +1,55 @@
+// ChatCompletionsService.java
+package edu.bjtu.groupone.backend.utils;
+
+import com.volcengine.ark.runtime.model.completion.chat.ChatCompletionRequest;
+import com.volcengine.ark.runtime.model.completion.chat.ChatMessage;
+import com.volcengine.ark.runtime.model.completion.chat.ChatMessageRole;
+import com.volcengine.ark.runtime.service.ArkService;
+import org.springframework.stereotype.Component;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Component
+public class AiUtils {
+
+ private ArkService service;
+
+ public void ChatCompletionsService(String apiKey) {
+ this.service = new ArkService(apiKey);
+ }
+
+ public String getAIResponse(String userInput) {
+ List<ChatMessage> messages = new ArrayList<>();
+ messages.add(ChatMessage.builder()
+ .role(ChatMessageRole.SYSTEM)
+ .content("你是豆包，是由字节跳动开发的 AI 人工智能助手")
+ .build());
+
+ messages.add(ChatMessage.builder()
+ .role(ChatMessageRole.USER)
+ .content(userInput)
+ .build());
+
+ ChatCompletionRequest request = ChatCompletionRequest.builder()
+ .model("ep-20241106142047-5g46t") // 使用指定的模型
+ .messages(messages)
+ .build();
+
+ final StringBuilder aiResponse = new StringBuilder();
+
+ service.streamChatCompletion(request)
+ .doOnError(Throwable::printStackTrace)
+ .blockingForEach(choice -> {
+ if (choice.getChoices().size() > 0) {
+ aiResponse.append(choice.getChoices().get(0).getMessage().getContent());
+ }
+ });
+
+ return aiResponse.toString();
+ }
+
+ public void shutdown() {
+ service.shutdownExecutor();
+ }
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/utils/AliOSSUtils.java b/src/main/java/edu/bjtu/groupone/backend/utils/AliOSSUtils.java
new file mode 100644
index 0000000..bedda4f
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/utils/AliOSSUtils.java
@@ -0,0 +1,45 @@
+package edu.bjtu.groupone.backend.utils;
+
+import com.aliyun.oss.OSS;
+import com.aliyun.oss.OSSClientBuilder;
+import org.springframework.stereotype.Component;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.UUID;
+
+/**
+ * 阿里云 OSS 工具类
+ */
+@Component
+public class AliOSSUtils {
+
+ private String endpoint = "https://oss-cn-beijing.aliyuncs.com";
+ private String accessKeyId = "LTAI5tQjb3fy8hTxxRGoS9b3";
+ private String accessKeySecret = "luBkGCqe3mqfYuiSOawC0KgPCi1yI4";
+ private String bucketName = "avatar-wzh";
+
+ /**
+ * 实现上传图片到OSS
+ */
+ public String upload(MultipartFile file) throws IOException {
+ // 获取上传的文件的输入流
+ InputStream inputStream = file.getInputStream();
+
+ // 避免文件覆盖
+ String originalFilename = file.getOriginalFilename();
+ String fileName = UUID.randomUUID().toString() + originalFilename.substring(originalFilename.lastIndexOf("."));
+
+ //上传文件到 OSS
+ OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
+ ossClient.putObject(bucketName, fileName, inputStream);
+
+ //文件访问路径
+ String url = endpoint.split("//")[0] + "//" + bucketName + "." + endpoint.split("//")[1] + "/" + fileName;
+ // 关闭ossClient
+ ossClient.shutdown();
+ return url;// 把上传到oss的路径返回
+ }
+
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/utils/EmailUtil.java b/src/main/java/edu/bjtu/groupone/backend/utils/EmailUtil.java
new file mode 100644
index 0000000..3ff7822
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/utils/EmailUtil.java
@@ -0,0 +1,39 @@
+package edu.bjtu.groupone.backend.utils;
+
+import jakarta.mail.MessagingException;
+import jakarta.mail.internet.MimeMessage;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.mail.javamail.JavaMailSender;
+import org.springframework.mail.javamail.MimeMessageHelper;
+import org.springframework.stereotype.Component;
+
+@Component
+@Slf4j
+public class EmailUtil {
+ @Autowired
+ private JavaMailSender mailSender;
+
+ @Value("${spring.mail.username}") // 确保从配置读取
+ private String from;
+
+ public void sendVerificationEmail(String to, String code) {
+ try {
+ MimeMessage message = mailSender.createMimeMessage();
+ MimeMessageHelper helper = new MimeMessageHelper(message);
+
+ // 关键修改：使用配置的授权邮箱作为发件人
+ helper.setFrom(from);
+ helper.setTo(to);
+ helper.setSubject("[PT SITE] 验证码通知");
+ helper.setText("您的验证码是：" + code + "，5分钟内有效");
+
+ mailSender.send(message);
+ log.info("邮件发送成功：{}", to);
+ } catch (MessagingException e) {
+ log.error("邮件发送失败", e);
+ throw new RuntimeException("邮件服务异常");
+ }
+ }
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/utils/GetTokenUserId.java b/src/main/java/edu/bjtu/groupone/backend/utils/GetTokenUserId.java
new file mode 100644
index 0000000..b798f05
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/utils/GetTokenUserId.java
@@ -0,0 +1,18 @@
+package edu.bjtu.groupone.backend.utils;
+
+import io.jsonwebtoken.Claims;
+import jakarta.servlet.http.HttpServletRequest;
+
+public class GetTokenUserId {
+
+ public static String getUserId(HttpServletRequest request) {
+ String token = request.getHeader("Authorization");
+ if (token == null || !token.startsWith("Bearer ")) {
+ return null;
+ }
+ // 解析 JWT Token，获取用户 ID
+ String jwt = token.substring(7); // 去掉 'Bearer ' 前缀
+ Claims claims = JwtUtils.parseJwt(jwt); // 从 JWT 中获取用户 ID
+ return claims.get("id").toString();
+ }
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/utils/JwtUtils.java b/src/main/java/edu/bjtu/groupone/backend/utils/JwtUtils.java
new file mode 100644
index 0000000..65e917a
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/utils/JwtUtils.java
@@ -0,0 +1,34 @@
+package edu.bjtu.groupone.backend.utils;
+
+import io.jsonwebtoken.Claims;
+import io.jsonwebtoken.Jwts;
+
+import java.util.Date;
+import java.util.Map;
+
+public class JwtUtils {
+
+ // 生成JWT// 生成JWT
+ public static String generateJwt(Map<String,Object> claims) {
+
+ String jwt = Jwts.builder().
+ addClaims(claims)
+ .signWith(io.jsonwebtoken.SignatureAlgorithm.HS256, "Bjtu")
+ .setExpiration(new Date(System.currentTimeMillis() + 43200000L))
+ .compact();
+
+
+ return jwt;
+
+
+ }
+
+ public static Claims parseJwt(String jwt) {
+
+ return Jwts.parser()
+ .setSigningKey("Bjtu")
+ .parseClaimsJws(jwt)
+ .getBody();
+ }
+
+}
diff --git a/src/main/java/edu/bjtu/groupone/backend/utils/OpenAiUtil.java b/src/main/java/edu/bjtu/groupone/backend/utils/OpenAiUtil.java
new file mode 100644
index 0000000..1dbf943
--- /dev/null
+++ b/src/main/java/edu/bjtu/groupone/backend/utils/OpenAiUtil.java
@@ -0,0 +1,35 @@
+package edu.bjtu.groupone.backend.utils;
+
+import com.theokanning.openai.OpenAiService;
+import com.theokanning.openai.completion.CompletionChoice;
+import com.theokanning.openai.completion.CompletionRequest;
+import jakarta.annotation.PostConstruct;
+import org.springframework.beans.factory.annotation.Value;
+
+import java.time.Duration;
+import java.util.HashMap;
+import java.util.List;
+
+
+public class OpenAiUtil {
+ @Value("${openai.secret_key}")
+ private String token;
+
+ private OpenAiService service;
+
+ @PostConstruct
+ public void init(){
+ service= new OpenAiService(token, Duration.ofSeconds(60L));
+ }
+ public List<CompletionChoice> sendComplete(String prompt) {
+ CompletionRequest completionRequest = CompletionRequest.builder()
+ .model("text-davinci-003") //采用最强模型，达芬奇模型3
+ .maxTokens(1500)
+ .prompt(prompt)
+ .user("testing")
+ .logitBias(new HashMap<>())
+ .build();
+
+ return service.createCompletion(completionRequest).getChoices();
+ }
+}
\ No newline at end of file
diff --git a/src/main/resources/application.properties b/src/main/resources/application.properties
index 3ca17a4..db34ebe 100644
--- a/src/main/resources/application.properties
+++ b/src/main/resources/application.properties
@@ -1 +1,19 @@
-spring.application.name=backend
+# ??MySQL??
+spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
+spring.datasource.url=jdbc:mysql://localhost:3306/groupone_db\
+?useSSL=false\
+&serverTimezone=Asia/Shanghai\
+&characterEncoding=utf8\
+&allowPublicKeyRetrieval=true
+
+spring.datasource.username=root
+spring.datasource.password=Rfw@2935
+
+# ???????????
+spring.datasource.hikari.maximum-pool-size=10
+spring.datasource.hikari.minimum-idle=5
+spring.datasource.hikari.idle-timeout=600000
+spring.datasource.hikari.connection-timeout=30000
+
+# ???? MyBatis ?????
+mybatis.configuration.map-underscore-to-camel-case=true
\ No newline at end of file
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
new file mode 100644
index 0000000..6cf60f3
--- /dev/null
+++ b/src/main/resources/application.yml
@@ -0,0 +1,60 @@
+spring:
+ application:
+ name: PtSite
+ mail:
+ host: smtp.qq.com
+ port: 465
+ username: 583398241@qq.com
+ password: feojrgbupvmxbbgi
+ protocol: smtps
+ properties:
+ mail.smtp.auth: true
+ mail.smtp.ssl.enable: true
+ mail.smtp.ssl.trust: smtp.qq.com
+ mail.smtp.socketFactory.class: javax.net.ssl.SSLSocketFactory
+
+ datasource:
+ driver-class-name: com.mysql.cj.jdbc.Driver
+ url: jdbc:mysql://localhost:3306/groupone_db?useSSL=false&serverTimezone=Asia/Shanghai&characterEncoding=utf8
+ username: root
+ password: "Rfw@2935"
+ hikari:
+ maximum-pool-size: 10
+ minimum-idle: 5
+ idle-timeout: 600000
+ connection-timeout: 30000
+
+ elasticsearch:
+ rest:
+ uris: http://192.168.74.129:9200
+
+ servlet:
+ multipart:
+ enabled: true
+ max-file-size: 1024MB
+ max-request-size: 2048MB
+
+ # Redis配置（当前已注释）
+ # data:
+ # redis:
+ # host: 192.168.74.129
+ # port: 6379
+ # database: 0
+ # lettuce:
+ # pool:
+ # max-active: 8
+ # max-idle: 8
+ # min-idle: 0
+
+mybatis:
+ configuration:
+ log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
+ map-underscore-to-camel-case: true
+
+server:
+ port: 8080
+
+ark:
+ api:
+ key: 5c648105-9e49-4a78-a1b2-ee8fcc7c02ad
+
diff --git a/src/main/resources/schema.sql b/src/main/resources/schema.sql
new file mode 100644
index 0000000..8bb0c85
--- /dev/null
+++ b/src/main/resources/schema.sql
@@ -0,0 +1,45 @@
+CREATE TABLE `user` (
+ `user_id` INT PRIMARY KEY AUTO_INCREMENT,
+ `username` VARCHAR(255) NOT NULL UNIQUE,
+ `email` VARCHAR(255) NOT NULL UNIQUE,
+ `password` VARCHAR(255) NOT NULL,
+ `address` VARCHAR(255),
+ `role` VARCHAR(50) NOT NULL DEFAULT 'user',
+ `profile_pic` VARCHAR(255),
+ `registration_date` DATETIME NOT NULL, -- 推荐用时间类型（而非字符串）
+ `identification_number` VARCHAR(18), -- 身份证号建议用 VARCHAR(18)
+ `avatar` VARCHAR(255),
+ `isfollowed` BOOLEAN NOT NULL DEFAULT FALSE,
+ INDEX `idx_user_id` (`user_id`) -- 可选：主键通常自带索引
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
+
+CREATE TABLE invite_code (
+ code VARCHAR(20) PRIMARY KEY,
+ max_uses INT NOT NULL,
+ remaining_uses INT NOT NULL,
+ expiry_time DATETIME NOT NULL,
+ created_by INT NOT NULL,
+ created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+ FOREIGN KEY (created_by) REFERENCES user(user_id)
+);
+
+CREATE TABLE user_follow (
+ follower_id INT NOT NULL,
+ followed_id INT NOT NULL,
+ created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+ PRIMARY KEY (follower_id, followed_id),
+ FOREIGN KEY (follower_id) REFERENCES user(user_id),
+ FOREIGN KEY (followed_id) REFERENCES user(user_id)
+);
+
+INSERT INTO `user` (
+ username, email, password, registration_date, identification_number, role
+) VALUES (
+ 'admin', 'admin@example.com', 'admin123', NOW(), 87654321, 'admin'
+ );
+
+INSERT INTO invite_code (
+ code, max_uses, remaining_uses, expiry_time, created_by
+) VALUES
+ ('WELCOME2025', 5, 5, '2026-01-01 00:00:00', 1),
+ ('INVITEONLY', 3, 3, '2025-12-31 23:59:59', 1);
\ No newline at end of file
diff --git a/src/main/resources/static/home.html b/src/main/resources/static/home.html
new file mode 100644
index 0000000..635856a
--- /dev/null
+++ b/src/main/resources/static/home.html
@@ -0,0 +1,24 @@
+<!-- src/main/resources/static/home.html -->
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head><meta charset="UTF-8"><title>首页</title></head>
+<body>
+<h1>欢迎访问受保护页面</h1>
+<div id="userInfo"></div>
+<script>
+ (async ()=>{
+ const token = localStorage.getItem('token');
+ if (!token) return window.location.href='login.html';
+ const res = await fetch('/api/me', {
+ headers: { 'token': token }
+ });
+ const json = await res.json();
+ if (json.code!==0) {
+ return window.location.href='login.html';
+ }
+ document.getElementById('userInfo')
+ .innerText = '当前用户：' + json.data.username;
+ })();
+</script>
+</body>
+</html>
diff --git a/src/main/resources/static/login.html b/src/main/resources/static/login.html
new file mode 100644
index 0000000..993941d
--- /dev/null
+++ b/src/main/resources/static/login.html
@@ -0,0 +1,38 @@
+<!-- src/main/resources/static/login.html -->
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+ <meta charset="UTF-8">
+ <title>登录</title>
+</head>
+<body>
+<h2>用户登录</h2>
+<form id="loginForm">
+ <label>身份证号：</label>
+ <input type="number" name="identificationNumber" required>
+ <label>密码：</label>
+ <input type="password" name="password" required>
+ <button type="submit">登录</button>
+</form>
+<script>
+ document.getElementById('loginForm')
+ .addEventListener('submit', async e => {
+ e.preventDefault();
+ const form = new URLSearchParams(new FormData(e.target));
+ const res = await fetch('/login', {
+ method: 'POST',
+ body: form
+ });
+ const json = await res.json();
+ if (res.ok && json.code === 0) {
+ // 保存 token
+ localStorage.setItem('token', json.data);
+ // 跳转到受保护的页面
+ window.location.href = 'home.html';
+ } else {
+ alert(json.msg);
+ }
+ });
+</script>
+</body>
+</html>
diff --git a/src/main/resources/static/register.html b/src/main/resources/static/register.html
new file mode 100644
index 0000000..1e43e90
--- /dev/null
+++ b/src/main/resources/static/register.html
@@ -0,0 +1,54 @@
+<!-- src/main/resources/static/register.html -->
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+ <meta charset="UTF-8">
+ <title>注册</title>
+</head>
+<body>
+<h2>用户注册</h2>
+<form id="regForm">
+ <label>用户名：</label>
+ <input type="text" name="username" required>
+ <label>邮箱：</label>
+ <input type="email" name="email" required>
+ <button type="button" id="sendCode">发送验证码</button>
+ <label>邮箱验证码：</label>
+ <input type="text" name="verificationCode" required>
+ <label>密码：</label>
+ <input type="password" name="password" required>
+ <label>身份证号（8 位数字）：</label>
+ <input type="number" name="identificationNumber" required>
+ <label>邀请码：</label>
+ <input type="text" name="inviteCode" required>
+ <button type="submit">注册</button>
+</form>
+<script>
+ document.getElementById('sendCode')
+ .addEventListener('click', async () => {
+ const form = new URLSearchParams();
+ form.set('email', document.querySelector('[name=email]').value);
+ form.set('inviteCode', document.querySelector('[name=inviteCode]').value);
+ const res = await fetch('/sendVerification', {
+ method: 'POST',
+ body: form
+ });
+ alert((await res.json()).msg);
+ });
+ document.getElementById('regForm')
+ .addEventListener('submit', async e => {
+ e.preventDefault();
+ const form = new URLSearchParams(new FormData(e.target));
+ const res = await fetch('/register', {
+ method: 'POST',
+ body: form
+ });
+ const json = await res.json();
+ alert(json.msg);
+ if (res.ok && json.code===0) {
+ window.location.href='login.html';
+ }
+ });
+</script>
+</body>
+</html>