用户类 邀请码 商城
Change-Id: If3a9ef5c464386647ae3b876104cbf4c71265c4b
diff --git a/src/main/java/com/pt5/pthouduan/controller/InvitesController.java b/src/main/java/com/pt5/pthouduan/controller/InvitesController.java
index e3a8d8d..3eaf050 100644
--- a/src/main/java/com/pt5/pthouduan/controller/InvitesController.java
+++ b/src/main/java/com/pt5/pthouduan/controller/InvitesController.java
@@ -12,8 +12,8 @@
public class InvitesController {
@Autowired
private InviteService inviteService;
- @PostMapping("/sold")
- public Map<String, Object> soldinvite(@RequestParam String buyername) {
- return inviteService.setbuyername(buyername);
+ @GetMapping("/getuserinvite")
+ public Map<String, Object> getUserInfo(@RequestParam String username) {
+ return inviteService.getInvitesByUsername(username);
}
}
diff --git a/src/main/java/com/pt5/pthouduan/controller/UserController.java b/src/main/java/com/pt5/pthouduan/controller/UserController.java
index 9198708..4a23ea9 100644
--- a/src/main/java/com/pt5/pthouduan/controller/UserController.java
+++ b/src/main/java/com/pt5/pthouduan/controller/UserController.java
@@ -3,10 +3,20 @@
import com.pt5.pthouduan.entity.User;
import com.pt5.pthouduan.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.stereotype.Controller;
+import org.springframework.web.multipart.MultipartFile;
-import java.util.Map;
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardCopyOption;
+import java.text.SimpleDateFormat;
+import java.util.*;
/**
* <p>
@@ -21,6 +31,9 @@
public class UserController {
@Autowired
private UserService userService;
+ private String uploadDir="../var/www/avatars/"; // 配置文件上传目录，例如: /var/www/avatars/
+
+ private String accessPath="../avatars/";
@PostMapping("/register")
public Map<String, Object> register(@RequestBody User user,@RequestParam String code,@RequestParam String emailcode) {//code是邀请码,emailcode是验证码
@@ -45,9 +58,12 @@
}
@PostMapping("/changeimage")
- public Map<String, Object> changeimage(@RequestParam String username,
- @RequestParam String image) {
- return userService.changeImage(username,image);
+ public Map<String, Object> changeimage(@RequestParam String username, @RequestParam String image) {
+ Map<String, Object> response = new HashMap<>();
+ userService.changeImage(username,image);
+ response.put("success", true);
+ response.put("message", "头像更改成功");
+ return response;
}
@PostMapping("/changePassword")
@@ -77,4 +93,88 @@
return userService.DeleteUser(username);
}
+ @GetMapping("/alluser")
+ public Map<String, Object> alluser() {
+ return userService.getAllUser();
+ }
+
+ @GetMapping("/finduser")
+ public Map<String, Object> finduser(@RequestParam String keyword) {
+ return userService.findUser(keyword);
+ }
+
+ @GetMapping("/getDecoration")
+ public Map<String, Object> getDecoration(@RequestParam Long userid) {
+ return userService.getDecoration(userid);
+ }
+
+ @GetMapping("/getUserid")
+ public Map<String, Object> getuserid(@RequestParam String username) {
+ return userService.getuserid(username);
+ }
+
+ @PostMapping("/uploadimage")
+ public ResponseEntity<Map<String, Object>> uploadAvatar(@RequestParam("avatar") MultipartFile file) {
+ Map<String, Object> response = new HashMap<>();
+
+ // 1. 验证文件是否为空
+ if (file.isEmpty()) {
+ response.put("success", false);
+ response.put("message", "请选择要上传的文件");
+ return ResponseEntity.badRequest().body(response);
+ }
+
+ // 2. 验证文件类型
+ String contentType = file.getContentType();
+ if (!"image/jpeg".equals(contentType) &&
+ !"image/png".equals(contentType) &&
+ !"image/gif".equals(contentType)) {
+ response.put("success", false);
+ response.put("message", "只支持JPG/PNG/GIF格式的图片");
+ return ResponseEntity.badRequest().body(response);
+ }
+
+ // 3. 验证文件大小 (前端已验证，后端再次验证)
+ if (file.getSize() > 10 * 1024 * 1024) { // 10MB
+ response.put("success", false);
+ response.put("message", "图片大小不能超过10MB");
+ return ResponseEntity.badRequest().body(response);
+ }
+
+ try {
+ // 4. 创建上传目录(如果不存在)
+ File dir = new File(uploadDir);
+ if (!dir.exists()) {
+ dir.mkdirs();
+ }
+ System.out.println(dir.getAbsolutePath());
+ // 5. 生成唯一文件名 (日期+UUID+后缀)
+ String originalFilename = file.getOriginalFilename();
+ String fileExt = originalFilename.substring(originalFilename.lastIndexOf("."));
+ String newFilename = new SimpleDateFormat("yyyyMMdd").format(new Date()) +
+ "_" + UUID.randomUUID().toString().replace("-", "") +
+ fileExt.toLowerCase();
+
+ // 6. 保存文件
+ Path path = Paths.get(uploadDir, newFilename);
+ Files.copy(file.getInputStream(), path);
+
+ // 7. 返回访问URL
+ String fileUrl = accessPath + newFilename;
+
+ response.put("success", true);
+ response.put("url", fileUrl);
+ response.put("message", "头像上传成功");
+
+ return ResponseEntity.ok(response);
+
+ } catch (IOException e) {
+ e.printStackTrace();
+ response.put("success", false);
+ response.put("message", "文件上传失败: " + e.getMessage());
+ return ResponseEntity.status(500).body(response);
+ }
+ }
+
+
}
diff --git a/src/main/java/com/pt5/pthouduan/mapper/InvitesMapper.java b/src/main/java/com/pt5/pthouduan/mapper/InvitesMapper.java
index fddf6df..62eb977 100644
--- a/src/main/java/com/pt5/pthouduan/mapper/InvitesMapper.java
+++ b/src/main/java/com/pt5/pthouduan/mapper/InvitesMapper.java
@@ -9,18 +9,32 @@
import org.apache.ibatis.annotations.Update;
import org.springframework.stereotype.Repository;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * <p>
+ * Mapper 接口
+ * </p>
+ *
+ * @author ljx
+ * @since 2025-04-14
+ */
@Repository
public interface InvitesMapper extends BaseMapper<Invites> {
@Select("SELECT * FROM invite WHERE code = #{code}")
- Invites selectByCode(String code);
+ Map<String, Object> selectByCode(String code);
@Update("UPDATE invite SET isUsed = 1, username = #{username} WHERE code = #{code}")
int updateInviteByCode(@Param("code") String code, @Param("username") String username);
- @Select("SELECT * FROM invite WHERE buyername = '' LIMIT 1")
- Invites selectFirstEmptyBuyername();
+ @Select("SELECT * FROM invite WHERE (buyername = '' OR buyername IS NULL) LIMIT 1")
+ Map<String, Object> selectFirstEmptyBuyername();
@Update("UPDATE invite SET buyername = #{buyername} WHERE code = #{code}")
int updatebuyer(@Param("code") String code, @Param("buyername") String buyername);
+
+ @Select("SELECT * FROM invite WHERE buyername = #{username}")
+ List<Map<String, Object>> selectInvitesByUsername(@Param("username") String username);
}
diff --git a/src/main/java/com/pt5/pthouduan/mapper/UserMapper.java b/src/main/java/com/pt5/pthouduan/mapper/UserMapper.java
index c3bc827..8d0301a 100644
--- a/src/main/java/com/pt5/pthouduan/mapper/UserMapper.java
+++ b/src/main/java/com/pt5/pthouduan/mapper/UserMapper.java
@@ -1,9 +1,10 @@
package com.pt5.pthouduan.mapper;
-import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.pt5.pthouduan.entity.PeerInfo;
import com.pt5.pthouduan.entity.User;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.*;
+import org.apache.ibatis.type.JdbcType;
import org.springframework.stereotype.Repository;
import java.util.List;
@@ -22,37 +23,22 @@
@Insert("INSERT INTO user(username, password, email, passkey) VALUES(#{username}, #{password}, #{email}, #{passkey})")
int insert(User user);
- @Delete("DELETE FROM user_behavior WHERE user_id = (SELECT user_id FROM User WHERE username = #{username})")
+ @Delete("DELETE FROM user WHERE username = #{username})")
int deleteByUsername(@Param("username") String username);
- @Select("SELECT * FROM User WHERE username = #{username}")
- User selectByUsername(String username);
+ @Select("SELECT * FROM user WHERE username = #{username}")
+ Map<String, Object> selectByUsername(String username);
- @Select("SELECT * FROM User WHERE email = #{email}")
- User selectByEmail(String email);
+ @Select("SELECT username FROM user WHERE email = #{email}")
+ String selectByEmail(String email);
- @Select("SELECT credit FROM User WHERE username = #{username}")
+ @Select("SELECT credit FROM user WHERE username = #{username}")
int getcreditByUsername(String username);
- @Select("SELECT passkey FROM user WHERE userid = #{userid}")
- String selectPasskeyByUserid(Long userid);
-
- @Select("SELECT permission FROM user WHERE userid = #{userid}")
- int getpermissionByUserid(Long userid);
-
- @Select("SELECT username FROM user WHERE userid = #{userid}")
- String selectUsernameByUserid(Long userid);
-
- @Select("SELECT userid FROM user WHERE username = #{username}")
- String selectUseridByusername(String username);
-
@Update("UPDATE user SET credit = credit - #{price} WHERE username = #{username}")
int deductCreditByUsername(@Param("username") String username, @Param("price") int price);
- @Update("UPDATE user SET credit = credit + #{price} WHERE username = #{username}")
- int addCreditByUsername(@Param("username") String username, @Param("price") int price);
-
- @Update("UPDATE user SET upload = upload + #{upload} WHERE username = #{username}")
+ @Update("UPDATE user SET user_upload = user_upload + #{upload} WHERE username = #{username}")
int increaseUploadByUsername(@Param("username") String username, @Param("upload") Integer upload);
@Update("UPDATE user SET password = #{password} WHERE username = #{username}")
@@ -70,20 +56,7 @@
@Update("UPDATE user SET decoration = CONCAT(IFNULL(decoration, ''), ' ', #{newDecoration}) WHERE username = #{username}")
int appendUserDecoration(@Param("username") String username, @Param("newDecoration") String newDecoration);
- @Delete("DELETE FROM user WHERE username = #{username})")
- int deleteByUsername(@Param("username") String username);
-
- @Select("SELECT username, passkey, grade_id FROM user")
- List<Map<String, Object>> selectAllUsersBasicInfo();
-
- @Select("SELECT * FROM user")
- List<Map<String, Object>> selectAllUsers();
-
- @Select("SELECT * FROM user WHERE username REGEXP #{regexPattern}")
- List<Map<String, Object>> selectUserByFuzzyKeyword(@Param("regexPattern") String regexPattern);
-
boolean existsByPasskey(String passkey);
-
void incrementUserTraffic( @Param("info_hash") String infoHash,@Param("passkey") String passkey, @Param("user_upload") long uploaded, @Param("user_download") long downloaded);
String getUsernameByPasskey(String passkey);
@@ -92,4 +65,29 @@
@Select("SELECT passkey FROM user WHERE userid = #{userid}")
String selectPasskeyByUserid(Long userid);
+
+ @Select("SELECT username, passkey, grade_id FROM user")
+ List<Map<String, Object>> selectAllUsersBasicInfo();
+
+ @Select("SELECT username FROM user WHERE userid = #{userid}")
+ String selectUsernameByUserid(Long userid);
+
+ @Select("SELECT image,decoration,username,grade_id FROM user WHERE userid = #{userid}")
+ Map<String, Object> selectdecorationByUserid(Long userid);
+
+ @Select("SELECT userid FROM user WHERE username = #{username}")
+ String selectUseridByusername(String username);
+
+ @Select("SELECT * FROM user")
+ List<Map<String, Object>> selectAllUsers();
+
+ @Select("SELECT * FROM user WHERE username REGEXP #{regexPattern}")
+ List<Map<String, Object>> selectUserByFuzzyKeyword(@Param("regexPattern") String regexPattern);
+
+ @Update("UPDATE user SET credit = credit + #{price} WHERE username = #{username}")
+ int addCreditByUsername(@Param("username") String username, @Param("price") int price);
+
+ @Select("SELECT permission FROM user WHERE userid = #{userid}")
+ int getpermissionByUserid(Long userid);
+
}
diff --git a/src/main/java/com/pt5/pthouduan/service/InviteService.java b/src/main/java/com/pt5/pthouduan/service/InviteService.java
index f6ca8a6..a97929c 100644
--- a/src/main/java/com/pt5/pthouduan/service/InviteService.java
+++ b/src/main/java/com/pt5/pthouduan/service/InviteService.java
@@ -2,27 +2,28 @@
import com.pt5.pthouduan.entity.Invites;
import com.pt5.pthouduan.mapper.InvitesMapper;
+import com.pt5.pthouduan.mapper.UserMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.HashMap;
+import java.util.List;
import java.util.Map;
@Service
public class InviteService {
@Autowired
private InvitesMapper invitesMapper;
- //批量生成邀请码并存入数据库（6位）(这个直接在数据库更新)
- //邀请码卖出（set buyername）
- public Map<String, Object> setbuyername(String buyername){
+ //获取“我的”邀请码
+ public Map<String, Object> getInvitesByUsername(String username){
Map<String, Object> result = new HashMap<>();
- Invites invite=invitesMapper.selectFirstEmptyBuyername();
- invitesMapper.updatebuyer(invite.getCode(),buyername);
+ List<Map<String, Object>> invites = invitesMapper.selectInvitesByUsername(username);
result.put("success", true);
- result.put("invitecode", invite.getCode());
- result.put("message", "邀请码购买成功");
+ result.put("invites", invites);
+ result.put("message", "获取成功");
return result;
}
+
}
diff --git a/src/main/java/com/pt5/pthouduan/service/ShopService.java b/src/main/java/com/pt5/pthouduan/service/ShopService.java
index a85b5ef..84ab88f 100644
--- a/src/main/java/com/pt5/pthouduan/service/ShopService.java
+++ b/src/main/java/com/pt5/pthouduan/service/ShopService.java
@@ -65,10 +65,10 @@
}
//更新用户保种积分
userMapper.deductCreditByUsername(buyername,price);
- Invites invite=invitesMapper.selectFirstEmptyBuyername();
- invitesMapper.updatebuyer(invite.getCode(),buyername);
+ Map<String, Object> invite=invitesMapper.selectFirstEmptyBuyername();
+ invitesMapper.updatebuyer((String) invite.get("code"),buyername);
result.put("success", true);
- result.put("invitecode", invite.getCode());
+ result.put("invitecode", (String) invite.get("code"));
result.put("message", "邀请码购买成功");
return result;
}
diff --git a/src/main/java/com/pt5/pthouduan/service/UserService.java b/src/main/java/com/pt5/pthouduan/service/UserService.java
index 59865c2..affa9e8 100644
--- a/src/main/java/com/pt5/pthouduan/service/UserService.java
+++ b/src/main/java/com/pt5/pthouduan/service/UserService.java
@@ -2,6 +2,7 @@
import com.pt5.pthouduan.entity.Invites;
import com.pt5.pthouduan.entity.User;
import com.pt5.pthouduan.mapper.InvitesMapper;
+import lombok.extern.java.Log;
import org.springframework.beans.factory.annotation.Autowired;
import com.pt5.pthouduan.mapper.UserMapper;
import org.springframework.stereotype.Service;
@@ -10,8 +11,10 @@
import java.security.SecureRandom;
import java.util.HashMap;
+import java.util.List;
import java.util.Map;
import java.util.UUID;
+import java.util.stream.Collectors;
@Service
public class UserService {
@@ -53,20 +56,21 @@
return result;
}
- // 检查邮箱是否已存在
- if (userMapper.selectByEmail(user.getEmail()) != null) {
- result.put("success", false);
- result.put("message", "邮箱已注册");
- return result;
- }
+// // 检查邮箱是否已存在
+// if (userMapper.selectByEmail(user.getEmail()) != null) {
+// result.put("success", false);
+// result.put("message", "邮箱已注册");
+// return result;
+// }
//检查邀请码是否有效
- Invites invite= invitesMapper.selectByCode(code);
+ Map<String, Object> invite= invitesMapper.selectByCode(code);
if(invite==null){//邀请码不存在
result.put("success", false);
result.put("message","无效的邀请码");
return result;
}
- if(invite.getIsUsed()==1){//邀请码已被使用
+ Boolean isused = (Boolean) invite.get("isUsed");
+ if(isused){//邀请码已被使用
result.put("success", false);
result.put("message","邀请码已被使用");
return result;
@@ -93,7 +97,7 @@
public Map<String, Object> login(String username, String password) {
Map<String, Object> result = new HashMap<>();
- User user = userMapper.selectByUsername(username);//首先验证用户名
+ Map<String, Object> user = userMapper.selectByUsername(username);//首先验证用户名
if (user == null) {
result.put("success", false);
result.put("message", "用户不存在");
@@ -101,13 +105,12 @@
}
//验证密码
- if(!password.equals(user.getPassword())){
+ if(!password.equals(user.get("password"))){
result.put("success", false);
result.put("message","密码错误，请检查后重新输入");
return result;
}
// 登录成功，返回用户信息(密码置空)
- user.setPassword(null);
result.put("success", true);
result.put("message", "登录成功");
result.put("user", user);
@@ -117,20 +120,25 @@
public Map<String, Object> CalGrade(String username){
Map<String, Object> result = new HashMap<>();
//上传量+下载量0-500000 一级 510000-2000000 二级 2010000-3500000 三级 3510000-6500000 四级 6500000以上 五级
- User user = userMapper.selectByUsername(username);
- if(user.getUserUpload()+user.getUserDownload()<=500000*1024){
- user.setGradeId(1);
- }else if(user.getUserUpload()+user.getUserDownload()<=2000000*1024){
- user.setGradeId(2);
- }else if(user.getUserUpload()+user.getUserDownload()<= 3500000L *1024){
- user.setGradeId(3);
- }else if(user.getUserUpload()+user.getUserDownload()<= 6500000L *1024){
- user.setGradeId(4);
- }else{ user.setGradeId(5);}
- userMapper.updateGrade(username,user.getGradeId());
+ Map<String, Object> user = userMapper.selectByUsername(username);
+ Number downloadNum = (Number) user.get("user_download");
+ long download = downloadNum.longValue();
+ Number uploadNum = (Number) user.get("user_upload");
+ long upload = uploadNum.longValue();
+ int grade=0;
+ if(upload+download<=500000*1024){
+ grade=1;
+ }else if(upload+download<=2000000*1024){
+ grade=2;
+ }else if(upload+download<= 3500000L *1024){
+ grade=3;
+ }else if(upload+download<= 6500000L *1024){
+ grade=4;
+ }else{ grade=5;}
+ userMapper.updateGrade(username,grade);
result.put("success", true);
result.put("message", "等级更新成功");
- result.put("grade", user.getGradeId());
+ result.put("grade", grade);
return result;
}
//设置性别
@@ -152,7 +160,7 @@
//更改密码
public Map<String, Object> changePassword(String username,String oldpassword,String newpassword){
Map<String, Object> result = new HashMap<>();
- User user = userMapper.selectByUsername(username);//首先验证用户名
+ Map<String, Object> user = userMapper.selectByUsername(username);//首先验证用户名
if (user == null) {
result.put("success", false);
result.put("message", "用户不存在");
@@ -160,7 +168,7 @@
}
//验证密码
- if(!oldpassword.equals(user.getPassword())){
+ if(!oldpassword.equals(user.get("password"))){
result.put("success", false);
result.put("message","密码错误，请检查后重新输入");
return result;
@@ -174,15 +182,17 @@
//获取用户信息（前端）
public Map<String, Object> UserInfo(String username){
Map<String, Object> result = new HashMap<>();
- User user = userMapper.selectByUsername(username);
+ Map<String, Object> user = userMapper.selectByUsername(username);
result.put("success", true);
result.put("message", "用户信息获取成功");
result.put("user", user);
return result;
}
//直接创建用户
- public Map<String, Object> CreateUser(User user){
+ public Map<String, Object> CreateUser(User user){//用户名、密码、邮箱
Map<String, Object> result = new HashMap<>();
+ SecureRandom random = new SecureRandom();
+ user.setPasskey(String.valueOf(10000000 + random.nextInt(90000000)));
userMapper.insert(user);
result.put("success", true);
result.put("message", "用户创建成功");
@@ -196,5 +206,44 @@
result.put("message", "用户删除成功");
return result;
}
+ //获取所有用户
+ public Map<String, Object> getAllUser(){
+ Map<String, Object> result = new HashMap<>();
+ List<Map<String, Object>> users =userMapper.selectAllUsers();
+ result.put("data", users);
+ result.put("success", true);
+ result.put("message", "用户获取成功");
+ return result;
+ }
+ //模糊搜索用户名
+ public Map<String, Object> findUser(String keyword){
+ Map<String, Object> result = new HashMap<>();
+ String regexPattern = keyword.chars()
+ .mapToObj(c -> String.valueOf((char) c))
+ .collect(Collectors.joining(".*"));
+ List<Map<String, Object>> users = userMapper.selectUserByFuzzyKeyword(regexPattern);
+ result.put("data", users);
+ result.put("success", true);
+ result.put("message", "用户查找成功");
+ return result;
+ }
+ //根据用户id返回用户的头像和装饰
+ public Map<String, Object> getDecoration(Long userid){
+ Map<String, Object> result = new HashMap<>();
+ Map<String, Object> info = userMapper.selectdecorationByUserid(userid);
+ result.put("data", info);
+ result.put("success", true);
+ result.put("message", "获取信息成功");
+ return result;
+ }
+ //根据用户名返回用户id
+ public Map<String, Object> getuserid(String username){
+ Map<String, Object> result = new HashMap<>();
+ String userid = userMapper.selectUseridByusername(username);
+ result.put("data",userid);
+ result.put("success", true);
+ result.put("message", "获取信息成功");
+ return result;
+ }
}