用户类 邀请码 商城
Change-Id: If3a9ef5c464386647ae3b876104cbf4c71265c4b
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);
+ }
+ }
+
+
}