完成对announce请求的处理,初步完成上传量和下载量的统计
Change-Id: Ief942d0364a11d945a10a5a58d7652a054bedaa4
diff --git a/src/main/java/com/example/g8backend/controller/AuthController.java b/src/main/java/com/example/g8backend/controller/AuthController.java
index e824b63..5561dcd 100644
--- a/src/main/java/com/example/g8backend/controller/AuthController.java
+++ b/src/main/java/com/example/g8backend/controller/AuthController.java
@@ -2,7 +2,9 @@
import com.example.g8backend.dto.UserRegisterDTO;
import com.example.g8backend.entity.User;
+import com.example.g8backend.entity.UserStats;
import com.example.g8backend.service.IUserService;
+import com.example.g8backend.service.IUserStatsService;
import com.example.g8backend.util.JwtUtil;
import com.example.g8backend.util.mailUtil;
import org.springframework.beans.factory.annotation.Autowired;
@@ -22,6 +24,8 @@
@Autowired
private IUserService userService;
@Autowired
+ private IUserStatsService userStatsService;
+ @Autowired
private mailUtil mailUtil;
@Autowired
private PasswordEncoder passwordEncoder;
@@ -68,6 +72,12 @@
user.setPasskey(UUID.randomUUID().toString().replace("-", ""));
userService.save(user);
+ // 保存用户统计用户的上传量与下载量
+ UserStats userStats = new UserStats();
+ userStats.setUserId(user.getUserId());
+ userStats.setPasskey(user.getPasskey());
+ userStatsService.save(userStats);
+
return ResponseEntity.ok("注册成功");
}