Compare commits
44 Commits
feature-Em
...
feature-ip
| Author | SHA1 | Date | |
|---|---|---|---|
| 5624e598ec | |||
| eb2b0cb999 | |||
| 30db5ad706 | |||
|
|
5151ccfacc | ||
|
|
ab2bde0517 | ||
|
|
c9cb410819 | ||
|
|
07a6b86d7d | ||
|
|
1cf9ba0aa9 | ||
|
|
6a68fefbd4 | ||
|
|
da66c3b048 | ||
|
|
a9730a8414 | ||
|
|
0da6cde202 | ||
|
|
59838e1f5b | ||
|
|
7601fbb3c0 | ||
|
|
170ce83fe1 | ||
|
|
aa16c30d57 | ||
|
|
8fdbad8bd5 | ||
|
|
a9598ba795 | ||
|
|
439b6789fa | ||
|
|
9c7afc4470 | ||
|
|
e92cd31c27 | ||
|
|
bd12c599b0 | ||
|
|
b5a15a3f01 | ||
|
|
039443dc0f | ||
| ae0cca5437 | |||
| c329eaef79 | |||
|
|
55c7c9a03a | ||
|
|
a6716e32b6 | ||
|
|
8506fbb7e4 | ||
|
|
64bfb08257 | ||
|
|
71dbb6fc82 | ||
|
|
b6ac177148 | ||
|
|
df463f2a52 | ||
|
|
34663e8e84 | ||
|
|
176e133a57 | ||
|
|
fd3a38efb5 | ||
|
|
7d99b30d73 | ||
| d5ec858b06 | |||
| 7331553528 | |||
| 6a51eb48e4 | |||
| 2fe875a6ae | |||
|
|
c96af708df | ||
| fcdba51570 | |||
|
|
370e81f3c6 |
62
pom.xml
62
pom.xml
@@ -19,6 +19,13 @@
|
||||
<artifactId>spring-boot-starter</artifactId>
|
||||
</dependency>
|
||||
|
||||
|
||||
<dependency>
|
||||
<groupId>org.mybatis</groupId>
|
||||
<artifactId>mybatis</artifactId>
|
||||
<version>3.5.10</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-web</artifactId>
|
||||
@@ -64,6 +71,19 @@
|
||||
<version>3.12.0</version>
|
||||
</dependency>
|
||||
|
||||
<!-- OkHttp(用于调用API) -->
|
||||
<dependency>
|
||||
<groupId>com.squareup.okhttp3</groupId>
|
||||
<artifactId>okhttp</artifactId>
|
||||
<version>4.12.0</version>
|
||||
</dependency>
|
||||
|
||||
<!-- AOP起步依赖 -->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-aop</artifactId>
|
||||
</dependency>
|
||||
|
||||
<!-- 阿里云相关依赖 -->
|
||||
<dependency>
|
||||
<groupId>com.aliyun.oss</groupId>
|
||||
@@ -103,6 +123,44 @@
|
||||
<version>0.2.8</version>
|
||||
</dependency>
|
||||
|
||||
<!-- JWT 相关依赖 -->
|
||||
<dependency>
|
||||
<groupId>io.jsonwebtoken</groupId>
|
||||
<artifactId>jjwt-api</artifactId>
|
||||
<version>0.11.5</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>io.jsonwebtoken</groupId>
|
||||
<artifactId>jjwt-impl</artifactId>
|
||||
<version>0.11.5</version>
|
||||
<scope>runtime</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>io.jsonwebtoken</groupId>
|
||||
<artifactId>jjwt-jackson</artifactId>
|
||||
<version>0.11.5</version>
|
||||
<scope>runtime</scope>
|
||||
</dependency>
|
||||
|
||||
<!-- 阿里云短信服务SDK -->
|
||||
<dependency>
|
||||
<groupId>com.aliyun</groupId>
|
||||
<artifactId>dysmsapi20170525</artifactId>
|
||||
<version>3.0.0</version>
|
||||
</dependency>
|
||||
|
||||
<!-- Spring Mail 邮件发送 -->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-mail</artifactId>
|
||||
</dependency>
|
||||
|
||||
<!-- Redis 依赖 -->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-data-redis</artifactId>
|
||||
</dependency>
|
||||
|
||||
</dependencies>
|
||||
<dependencyManagement>
|
||||
<dependencies>
|
||||
@@ -123,8 +181,8 @@
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<version>3.8.1</version>
|
||||
<configuration>
|
||||
<source>1.8</source>
|
||||
<target>1.8</target>
|
||||
<source>9</source>
|
||||
<target>9</target>
|
||||
<encoding>UTF-8</encoding>
|
||||
</configuration>
|
||||
</plugin>
|
||||
|
||||
31
src/main/java/com/bao/dating/aspect/RecordTimeAspect.java
Normal file
31
src/main/java/com/bao/dating/aspect/RecordTimeAspect.java
Normal file
@@ -0,0 +1,31 @@
|
||||
package com.bao.dating.aspect;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.aspectj.lang.ProceedingJoinPoint;
|
||||
import org.aspectj.lang.annotation.Around;
|
||||
import org.aspectj.lang.annotation.Aspect;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
/**
|
||||
* 记录方法运行耗时
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@Aspect
|
||||
@Component
|
||||
public class RecordTimeAspect {
|
||||
@Around("execution(* com.bao.dating.service.impl.*.*(..))")
|
||||
public Object recordTime(ProceedingJoinPoint pjp) throws Throwable {
|
||||
//1. 记录方法运行的开始时间
|
||||
long begin = System.currentTimeMillis();
|
||||
|
||||
//2. 执行原始的方法
|
||||
Object result = pjp.proceed();
|
||||
|
||||
//3. 记录方法运行的结束时间, 记录耗时
|
||||
long end = System.currentTimeMillis();
|
||||
log.info("方法 {} 执行耗时: {}ms", pjp.getSignature() ,end-begin);
|
||||
return result;
|
||||
}
|
||||
|
||||
}
|
||||
16
src/main/java/com/bao/dating/common/Post.java
Normal file
16
src/main/java/com/bao/dating/common/Post.java
Normal file
@@ -0,0 +1,16 @@
|
||||
package com.bao.dating.common;
|
||||
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
/**
|
||||
* 动态表
|
||||
*/
|
||||
@Data
|
||||
public class Post {
|
||||
private Long post_id; // 动态ID
|
||||
private String content; // 动态内容
|
||||
private Long user_id; // 发布人ID
|
||||
private LocalDateTime created_at; // 创建时间
|
||||
}
|
||||
@@ -1,14 +1,36 @@
|
||||
package com.bao.dating.common;
|
||||
|
||||
/**
|
||||
* 响应状态码枚举
|
||||
* @author KilLze
|
||||
*/
|
||||
public enum ResultCode {
|
||||
/** 成功 */
|
||||
SUCCESS(200, "成功"),
|
||||
/** 请求已成功处理 */
|
||||
SUCCESS_REVIEW(201, "请求已成功处理"),
|
||||
/** 删除成功 */
|
||||
SUCCESS_DELETE(204, "删除成功"),
|
||||
/** 参数错误 */
|
||||
PARAM_ERROR(400, "参数错误"),
|
||||
/** 未登录或 Token 失效 */
|
||||
UNAUTHORIZED(401, "未登录或 Token 失效"),
|
||||
/** 无权限 */
|
||||
FORBIDDEN(403, "无权限"),
|
||||
/** 数据不存在 */
|
||||
NOT_FOUND(404, "数据不存在"),
|
||||
SYSTEM_ERROR(500, "系统异常");
|
||||
/** 验证码已发送 */
|
||||
CODE_SENT(200, "验证码已发送"),
|
||||
/** 验证码错误 */
|
||||
CODE_ERROR(400, "验证码错误"),
|
||||
/** 验证码已过期 */
|
||||
CODE_EXPIRED(400, "验证码已过期"),
|
||||
/** 验证码发送失败 */
|
||||
CODE_SEND_FAIL(500, "验证码发送失败"),
|
||||
/** 系统异常 */
|
||||
SYSTEM_ERROR(500, "系统异常"),
|
||||
/** 操作失败 */
|
||||
FAIL(500, "操作失败");
|
||||
|
||||
private final int code;
|
||||
private final String msg;
|
||||
@@ -25,4 +47,4 @@ public enum ResultCode {
|
||||
public String msg() {
|
||||
return msg;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -4,7 +4,6 @@ import com.aliyun.oss.ClientException;
|
||||
import com.aliyun.oss.OSS;
|
||||
import com.aliyun.oss.OSSClientBuilder;
|
||||
import com.aliyun.oss.OSSException;
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Data;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
@@ -12,6 +11,10 @@ import org.springframework.stereotype.Component;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
|
||||
/**
|
||||
* 阿里云OSS工具类
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
@Slf4j
|
||||
@Component
|
||||
@@ -30,7 +33,6 @@ public class AliOssUtil {
|
||||
* @return 完整的文件访问URL
|
||||
*/
|
||||
public String upload(byte[] bytes, String objectName) {
|
||||
|
||||
// 创建OSSClient实例。
|
||||
OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
|
||||
|
||||
|
||||
152
src/main/java/com/bao/dating/common/aliyun/SmsUtil.java
Normal file
152
src/main/java/com/bao/dating/common/aliyun/SmsUtil.java
Normal file
@@ -0,0 +1,152 @@
|
||||
package com.bao.dating.common.aliyun;
|
||||
|
||||
import com.aliyun.dysmsapi20170525.Client;
|
||||
import com.aliyun.dysmsapi20170525.models.SendSmsRequest;
|
||||
import com.aliyun.dysmsapi20170525.models.SendSmsResponse;
|
||||
import com.aliyun.teaopenapi.models.Config;
|
||||
import lombok.Data;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 阿里云短信服务工具类
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
@Slf4j
|
||||
@Component
|
||||
@ConfigurationProperties(prefix = "aliyun.sms")
|
||||
public class SmsUtil {
|
||||
|
||||
/**
|
||||
* 访问密钥ID
|
||||
*/
|
||||
private String accessKeyId;
|
||||
|
||||
/**
|
||||
* 访问密钥Secret
|
||||
*/
|
||||
private String accessKeySecret;
|
||||
|
||||
/**
|
||||
* 短信服务区域节点
|
||||
*/
|
||||
private String regionId = "cn-hangzhou";
|
||||
|
||||
/**
|
||||
* 默认签名名称
|
||||
*/
|
||||
private String signName;
|
||||
|
||||
/**
|
||||
* 默认模板代码
|
||||
*/
|
||||
private String templateCode;
|
||||
|
||||
/**
|
||||
* 创建短信客户端
|
||||
* @return Client对象
|
||||
* @throws Exception 创建失败时抛出异常
|
||||
*/
|
||||
private Client createClient() throws Exception {
|
||||
Config config = new Config()
|
||||
.setAccessKeyId(accessKeyId)
|
||||
.setAccessKeySecret(accessKeySecret)
|
||||
.setEndpoint("dysmsapi.aliyuncs.com");
|
||||
return new Client(config);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送短信
|
||||
* @param phoneNumber 手机号码
|
||||
* @param templateCode 模板代码(如果为空则使用默认模板代码)
|
||||
* @param templateParam 模板参数(JSON格式字符串,如:{"code":"123456"})
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendSms(String phoneNumber, String templateCode, String templateParam) {
|
||||
try {
|
||||
Client client = createClient();
|
||||
SendSmsRequest sendSmsRequest = new SendSmsRequest()
|
||||
.setPhoneNumbers(phoneNumber)
|
||||
.setSignName(signName)
|
||||
.setTemplateCode(templateCode != null ? templateCode : this.templateCode)
|
||||
.setTemplateParam(templateParam);
|
||||
|
||||
SendSmsResponse response = client.sendSms(sendSmsRequest);
|
||||
|
||||
if ("OK".equals(response.getBody().getCode())) {
|
||||
log.info("短信发送成功,手机号:{},请求ID:{}", phoneNumber, response.getBody().getRequestId());
|
||||
return true;
|
||||
} else {
|
||||
log.error("短信发送失败,手机号:{},错误码:{},错误信息:{}",
|
||||
phoneNumber, response.getBody().getCode(), response.getBody().getMessage());
|
||||
return false;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.error("发送短信异常,手机号:{},异常信息:{}", phoneNumber, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送短信(使用默认模板代码)
|
||||
* @param phoneNumber 手机号码
|
||||
* @param templateParam 模板参数(JSON格式字符串)
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendSms(String phoneNumber, String templateParam) {
|
||||
return sendSms(phoneNumber, null, templateParam);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送验证码短信
|
||||
* @param phoneNumber 手机号码
|
||||
* @param code 验证码
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendVerificationCode(String phoneNumber, String code) {
|
||||
String templateParam = String.format("{\"code\":\"%s\"}", code);
|
||||
return sendSms(phoneNumber, templateParam);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送验证码短信(使用指定模板代码)
|
||||
* @param phoneNumber 手机号码
|
||||
* @param code 验证码
|
||||
* @param templateCode 模板代码
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendVerificationCode(String phoneNumber, String code, String templateCode) {
|
||||
String templateParam = String.format("{\"code\":\"%s\"}", code);
|
||||
return sendSms(phoneNumber, templateCode, templateParam);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送短信(使用Map参数)
|
||||
* @param phoneNumber 手机号码
|
||||
* @param templateCode 模板代码
|
||||
* @param params 模板参数Map
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendSmsWithParams(String phoneNumber, String templateCode, Map<String, String> params) {
|
||||
StringBuilder jsonBuilder = new StringBuilder("{");
|
||||
boolean first = true;
|
||||
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||
if (!first) {
|
||||
jsonBuilder.append(",");
|
||||
}
|
||||
jsonBuilder.append("\"").append(entry.getKey()).append("\":\"")
|
||||
.append(entry.getValue()).append("\"");
|
||||
first = false;
|
||||
}
|
||||
jsonBuilder.append("}");
|
||||
return sendSms(phoneNumber, templateCode, jsonBuilder.toString());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,14 @@
|
||||
package com.bao.dating.common.ip2location;
|
||||
|
||||
import lombok.Data;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
@Data
|
||||
@Component
|
||||
@ConfigurationProperties(prefix = "ip2location.api")
|
||||
public class Ip2LocationConfig {
|
||||
private String key;
|
||||
private String url;
|
||||
private int timeout;
|
||||
}
|
||||
11
src/main/java/com/bao/dating/common/result/AliOssResult.java
Normal file
11
src/main/java/com/bao/dating/common/result/AliOssResult.java
Normal file
@@ -0,0 +1,11 @@
|
||||
package com.bao.dating.common.result;
|
||||
|
||||
/**
|
||||
* 阿里云 OSS 文件上传结果
|
||||
* @author KilLze
|
||||
*/
|
||||
public class AliOssResult {
|
||||
public static final String IMAGE = "image";
|
||||
public static final String VIDEO = "video";
|
||||
|
||||
}
|
||||
15
src/main/java/com/bao/dating/common/result/FileResult.java
Normal file
15
src/main/java/com/bao/dating/common/result/FileResult.java
Normal file
@@ -0,0 +1,15 @@
|
||||
package com.bao.dating.common.result;
|
||||
|
||||
/**
|
||||
* 文件上传结果
|
||||
* @author KilLze
|
||||
*/
|
||||
public class FileResult {
|
||||
public static final String JPG = "jpg";
|
||||
public static final String JPEG = "jpeg";
|
||||
public static final String PNG = "png";
|
||||
public static final String GIF = "gif";
|
||||
public static final String MP4 = "mp4";
|
||||
public static final String AVI = "avi";
|
||||
public static final String MOV = "mov";
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
package com.bao.dating.common.result;
|
||||
|
||||
/**
|
||||
* 阿里云敏感内容审核结果
|
||||
* @author KilLze
|
||||
*/
|
||||
public class GreenAuditResult {
|
||||
public static final String PASS = "pass";
|
||||
public static final String REVIEW = "review";
|
||||
public static final String BLOCK = "block";
|
||||
}
|
||||
35
src/main/java/com/bao/dating/config/WebConfig.java
Normal file
35
src/main/java/com/bao/dating/config/WebConfig.java
Normal file
@@ -0,0 +1,35 @@
|
||||
package com.bao.dating.config;
|
||||
|
||||
|
||||
import com.bao.dating.interceptor.TokenInterceptor;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
|
||||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
|
||||
|
||||
/**
|
||||
* WebMvc配置类
|
||||
* @author KilLze
|
||||
*/
|
||||
@Configuration
|
||||
public class WebConfig implements WebMvcConfigurer {
|
||||
|
||||
@Autowired
|
||||
private TokenInterceptor tokenInterceptor;
|
||||
|
||||
/**
|
||||
* 添加拦截器到Spring MVC配置中
|
||||
* @param registry 拦截器注册中心
|
||||
*/
|
||||
@Override
|
||||
public void addInterceptors(InterceptorRegistry registry) {
|
||||
//注册自定义拦截器对象
|
||||
registry.addInterceptor(tokenInterceptor)
|
||||
// 拦截所有请求
|
||||
.addPathPatterns("/**")
|
||||
// 忽略的接口
|
||||
.excludePathPatterns(
|
||||
"/user/login","/user/register","/user/emailLogin","/ip/location"
|
||||
);
|
||||
}
|
||||
}
|
||||
33
src/main/java/com/bao/dating/context/UserContext.java
Normal file
33
src/main/java/com/bao/dating/context/UserContext.java
Normal file
@@ -0,0 +1,33 @@
|
||||
package com.bao.dating.context;
|
||||
|
||||
/**
|
||||
* 用户上下文类,用于保存当前线程的用户ID
|
||||
* @author lenovo
|
||||
*/
|
||||
public class UserContext {
|
||||
|
||||
private static final ThreadLocal<Long> USER_HOLDER = new ThreadLocal<>();
|
||||
|
||||
/**
|
||||
* 设置当前线程的用户ID
|
||||
* @param userId 用户ID
|
||||
*/
|
||||
public static void setUserId(Long userId) {
|
||||
USER_HOLDER.set(userId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取当前线程的用户ID
|
||||
* @return 当前用户ID,如果未设置则返回null
|
||||
*/
|
||||
public static Long getUserId() {
|
||||
return USER_HOLDER.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* 清除当前线程的用户ID
|
||||
*/
|
||||
public static void clear() {
|
||||
USER_HOLDER.remove();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,52 @@
|
||||
package com.bao.dating.controller;
|
||||
|
||||
import com.bao.dating.pojo.entity.Comments;
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.service.CommentsService;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.util.List;
|
||||
|
||||
@RestController
|
||||
@RequestMapping("/comments")
|
||||
@CrossOrigin
|
||||
public class CommentController {
|
||||
@Resource
|
||||
private CommentsService commentsService;
|
||||
|
||||
/**
|
||||
* 添加评论
|
||||
* @param comment 评论对象(含content、userId、dynamicId)
|
||||
*/
|
||||
@PostMapping("/add")
|
||||
public Result<String> addComment(@RequestBody Comments comment) {
|
||||
boolean success = commentsService.addComment(comment);
|
||||
return success ?
|
||||
Result.success(ResultCode.SUCCESS, "评论成功") :
|
||||
Result.error(ResultCode.FAIL, "评论失败");
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除评论
|
||||
* @param user_id 评论ID
|
||||
*/
|
||||
@DeleteMapping("/delete/{user_id}")
|
||||
public Result<String> deleteComment(@PathVariable Long user_id) {
|
||||
boolean success = commentsService.deleteComment(user_id);
|
||||
return success ?
|
||||
Result.success(ResultCode.SUCCESS, "删除成功") :
|
||||
Result.error(ResultCode.FAIL, "删除失败");
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据动态ID查询评论列表
|
||||
* @param post_id 动态ID
|
||||
*/
|
||||
@GetMapping("/list/{post_id}")
|
||||
public Result<List<Comments>> getCommentList(@PathVariable Long post_id) {
|
||||
List<Comments> commentList = commentsService.getCommentByPostId(post_id);
|
||||
return Result.success(ResultCode.SUCCESS, commentList);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
package com.bao.dating.controller;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.pojo.vo.IpLocationVO;
|
||||
import com.bao.dating.service.Ip2LocationClientService;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RequestParam;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
@RestController
|
||||
@RequestMapping("/ip")
|
||||
public class IpLocationController {
|
||||
@Autowired
|
||||
private Ip2LocationClientService ip2LocationClientService;
|
||||
/**
|
||||
* 前端访问接口,获取IP地址位置信息
|
||||
* @param ip 可选参数,要查询的IP地址
|
||||
* @return IP位置信息(JSON格式)
|
||||
*/
|
||||
@GetMapping("/location")
|
||||
public Result<?> getIpLocation(@RequestParam(required = false) String ip) {
|
||||
if (ip.isEmpty()){
|
||||
return Result.error(ResultCode.PARAM_ERROR);
|
||||
}
|
||||
try {
|
||||
// 调用工具类获取API响应
|
||||
IpLocationVO ipLocationVo = ip2LocationClientService.getIpLocation(ip);
|
||||
return Result.success(ResultCode.SUCCESS,ipLocationVo);
|
||||
} catch (Exception e) {
|
||||
// 异常处理:返回错误信息(实际项目建议封装统一响应格式)
|
||||
return Result.error(ResultCode.SYSTEM_ERROR,e.getMessage());
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -13,6 +13,11 @@ import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 动态接口
|
||||
*
|
||||
* @author KilLze
|
||||
*/
|
||||
@RestController
|
||||
@RequestMapping("/posts")
|
||||
public class PostController {
|
||||
@@ -28,32 +33,31 @@ public class PostController {
|
||||
@PostMapping(value = "/upload", consumes = "multipart/form-data")
|
||||
public Result<List<String>> uploadMedia(@RequestParam("files") MultipartFile[] files) {
|
||||
List<String> fileUrls = postService.uploadMedia(files);
|
||||
return Result.success(ResultCode.SUCCESS_REVIEW, "文件上传成功", fileUrls);
|
||||
return Result.success(ResultCode.SUCCESS, "文件上传成功", fileUrls);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发布动态接口 - JSON格式请求
|
||||
* @param postDTO 动态信息
|
||||
* @param userId 用户ID
|
||||
* @return 发布的动态对象
|
||||
*/
|
||||
@PostMapping(consumes = "application/json")
|
||||
public Result<Post> createPostJson(@RequestBody PostRequestDTO postDTO, @RequestParam Long userId) {
|
||||
@PostMapping( "/createPost")
|
||||
public Result<Post> createPostJson(@RequestBody PostRequestDTO postDTO) {
|
||||
// 调用 Service 层处理发布动态业务逻辑
|
||||
Post result = postService.createPost(userId, postDTO);
|
||||
return Result.success(ResultCode.SUCCESS_REVIEW, "动态发布成功,等待审核。", result);
|
||||
Post result = postService.createPost(postDTO);
|
||||
return Result.success(ResultCode.SUCCESS, "动态发布成功,等待审核。", result);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除动态
|
||||
* 批量删除动态
|
||||
*
|
||||
* @param postId 动态ID
|
||||
* @param postIds 动态ID
|
||||
* @return 删除结果
|
||||
*/
|
||||
@DeleteMapping("/{postId}")
|
||||
public Result<String> deleteById(@PathVariable Long postId){
|
||||
postService.deletePostById(postId);
|
||||
return Result.success(ResultCode.SUCCESS_DELETE, "动态删除成功", null);
|
||||
@PostMapping("/deletePost")
|
||||
public Result<String> deleteById(@RequestBody List<Long> postIds){
|
||||
int deletedCount = postService.deletePostById(postIds);
|
||||
return Result.success(ResultCode.SUCCESS_DELETE, deletedCount > 0 ? "成功删除" : "删除失败,该动态不存在", null);
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -61,7 +65,7 @@ public class PostController {
|
||||
* @param postId 动态ID
|
||||
* @return 动态对象
|
||||
*/
|
||||
@GetMapping("/{postId}")
|
||||
@PostMapping("/{postId}")
|
||||
public Result<PostEditVO> getPostById(@PathVariable Long postId) {
|
||||
PostEditVO postEditVO = postService.getPostForEdit(postId);
|
||||
return Result.success(ResultCode.SUCCESS,"查询成功", postEditVO);
|
||||
@@ -73,9 +77,9 @@ public class PostController {
|
||||
* @param postRequestDTO 动态信息
|
||||
* @return 更新后的动态对象
|
||||
*/
|
||||
@PutMapping("/{postId}")
|
||||
@PostMapping("/{postId}/updatePost")
|
||||
public Result<PostEditVO> updatePost(@PathVariable Long postId, @RequestBody PostRequestDTO postRequestDTO) {
|
||||
PostEditVO result = postService.updatePost(postId, postRequestDTO);
|
||||
return Result.success(ResultCode.SUCCESS_REVIEW, "动态更新成功", result);
|
||||
return Result.success(ResultCode.SUCCESS, "动态更新成功", result);
|
||||
}
|
||||
}
|
||||
@@ -2,12 +2,13 @@ package com.bao.dating.controller;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
import com.bao.dating.pojo.entity.User;
|
||||
import com.bao.dating.service.PostFavoriteService;
|
||||
import com.bao.dating.service.PostService;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
@RestController
|
||||
@@ -15,6 +16,13 @@ import java.util.Map;
|
||||
public class PostFavoriteController {
|
||||
@Autowired
|
||||
private PostFavoriteService postFavoriteService;
|
||||
|
||||
/**
|
||||
* 收藏
|
||||
* @param postId 动态ID
|
||||
* @param user 当前登录用户对象
|
||||
* @return 结果
|
||||
*/
|
||||
@PostMapping("/{post_id}/favorites")
|
||||
public Result<Map<String,Long>> addPostFavorite(@PathVariable("post_id")Long postId, User user){
|
||||
if (user == null){
|
||||
@@ -23,6 +31,13 @@ public class PostFavoriteController {
|
||||
Long userId = user.getUserId();
|
||||
return postFavoriteService.postFavorite(userId,postId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 取消收藏
|
||||
* @param postId 动态id
|
||||
* @param user 登录用户
|
||||
* @return 结果
|
||||
*/
|
||||
@DeleteMapping("/{post_id}/favorites")
|
||||
public Result<?> deletePostFavorite(@PathVariable("post_id")Long postId, User user){
|
||||
if (user == null){
|
||||
@@ -31,4 +46,21 @@ public class PostFavoriteController {
|
||||
Long userId = user.getUserId();
|
||||
return postFavoriteService.deletePostFavorite(userId, postId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 展示所有收藏
|
||||
* @param user 当前登录用户
|
||||
* @return 结果
|
||||
*/
|
||||
@GetMapping("/showFavorites")
|
||||
public Result<List<Post>> showPostFavorites(@RequestBody User user){
|
||||
//校验参数
|
||||
if (user == null){
|
||||
return Result.error(ResultCode.PARAM_ERROR);
|
||||
}
|
||||
Long userId = user.getUserId();
|
||||
List<Post> posts = postFavoriteService.selectAllFavorites(userId);
|
||||
return Result.success(ResultCode.SUCCESS,posts);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
121
src/main/java/com/bao/dating/controller/UserController.java
Normal file
121
src/main/java/com/bao/dating/controller/UserController.java
Normal file
@@ -0,0 +1,121 @@
|
||||
package com.bao.dating.controller;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.context.UserContext;
|
||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
||||
import com.bao.dating.pojo.dto.UserLoginDTO;
|
||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||
import com.bao.dating.service.UserService;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
|
||||
/**
|
||||
* 用户接口
|
||||
*
|
||||
* @author KilLze
|
||||
*/
|
||||
@RestController
|
||||
@RequestMapping("/user")
|
||||
public class UserController {
|
||||
|
||||
@Autowired
|
||||
private UserService userService;
|
||||
|
||||
/**
|
||||
* 登录
|
||||
* @param userLoginDTO 登录参数
|
||||
*/
|
||||
@PostMapping("/login")
|
||||
public Result<UserLoginVO> login(@RequestBody UserLoginDTO userLoginDTO) {
|
||||
UserLoginVO userloginVO = userService.userLogin(userLoginDTO);
|
||||
return Result.success(ResultCode.SUCCESS, "登录成功", userloginVO);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户信息
|
||||
* @return 用户信息
|
||||
*/
|
||||
@GetMapping("/info")
|
||||
public Result<UserInfoVO> getUserInfo() {
|
||||
Long userId = UserContext.getUserId();
|
||||
UserInfoVO userInfoVO = userService.getUserInfo(userId);
|
||||
return Result.success(ResultCode.SUCCESS, "获取用户信息成功", userInfoVO);
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传头像接口
|
||||
* @param file 头像文件
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
@PostMapping(value = "/info/uploadAvatar", consumes = "multipart/form-data")
|
||||
public Result<String> uploadAvatar(@RequestParam("file") MultipartFile file) {
|
||||
String fileUrl = userService.uploadAvatar(file);
|
||||
return Result.success(ResultCode.SUCCESS, "头像上传成功", fileUrl);
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传背景接口
|
||||
* @param file 背景文件
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
@PostMapping(value = "/info/uploadBackground", consumes = "multipart/form-data")
|
||||
public Result<String> uploadBackground(@RequestParam("file") MultipartFile file) {
|
||||
String fileUrl = userService.uploadBackground(file);
|
||||
return Result.success(ResultCode.SUCCESS, "背景上传成功", fileUrl);
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新用户信息
|
||||
* @param userInfoUpdateDTO 用户信息更新参数
|
||||
* @return 更新后的用户信息
|
||||
*/
|
||||
@PostMapping("/info/update")
|
||||
public Result<UserInfoVO> userInfoUpdate(@RequestBody UserInfoUpdateDTO userInfoUpdateDTO) {
|
||||
Long userId = UserContext.getUserId();
|
||||
userInfoUpdateDTO.setUserId(userId);
|
||||
UserInfoVO userInfoVO =userService.updateUserInfo(userInfoUpdateDTO);
|
||||
return Result.success(ResultCode.SUCCESS, "用户信息更新成功", userInfoVO);
|
||||
}
|
||||
|
||||
/**
|
||||
* 用户注册
|
||||
* @param userName 用户名称
|
||||
* @param userPassword 用户密码
|
||||
* @return 返回
|
||||
*/
|
||||
@PostMapping("/register")
|
||||
public Result<?> userRegister(@RequestParam String userName , @RequestParam String userPassword){
|
||||
//校验参数是否为空
|
||||
if (userName.isEmpty() || userPassword.isEmpty()){
|
||||
return Result.error(ResultCode.PARAM_ERROR);
|
||||
}
|
||||
if (!userService.registerUser(userName,userPassword)){
|
||||
return Result.error(ResultCode.FAIL);
|
||||
}
|
||||
return Result.success(ResultCode.SUCCESS,"用户注册成功",null);
|
||||
}
|
||||
|
||||
/**
|
||||
* 通过邮箱登录
|
||||
* @param email 邮箱
|
||||
* @param code 验证码
|
||||
* @return 用户信息
|
||||
*/
|
||||
@PostMapping("/emailLogin")
|
||||
public Result<UserLoginVO> emailLogin(@RequestParam String email , @RequestParam String code){
|
||||
//校验参数是否为空
|
||||
if (email.isEmpty() || code.isEmpty()){
|
||||
return Result.error(ResultCode.PARAM_ERROR);
|
||||
}
|
||||
UserLoginVO userLoginVO = userService.emailLogin(email, code);
|
||||
if (userLoginVO == null){
|
||||
return Result.error(ResultCode.FAIL,"请先注册用户或添加邮箱");
|
||||
}
|
||||
return Result.success(ResultCode.SUCCESS,"用户登录成功",userLoginVO);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,90 @@
|
||||
package com.bao.dating.controller;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.service.VerificationCodeService;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.util.StringUtils;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
/**
|
||||
* 验证码控制器
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@RestController
|
||||
@RequestMapping("/api/verification")
|
||||
public class VerificationCodeController {
|
||||
|
||||
@Autowired
|
||||
private VerificationCodeService verificationCodeService;
|
||||
|
||||
/**
|
||||
* 发送邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @return 结果
|
||||
*/
|
||||
@PostMapping("/send-email-code")
|
||||
public Result<String> sendEmailCode(@RequestParam String email) {
|
||||
// 参数校验
|
||||
if (!StringUtils.hasText(email)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "邮箱地址不能为空");
|
||||
}
|
||||
|
||||
// 简单的邮箱格式校验
|
||||
if (!isValidEmail(email)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "邮箱格式不正确");
|
||||
}
|
||||
|
||||
// 发送验证码
|
||||
boolean success = verificationCodeService.sendEmailCode(email);
|
||||
|
||||
if (success) {
|
||||
return Result.success(ResultCode.CODE_SENT, "验证码已发送到您的邮箱,请查收");
|
||||
} else {
|
||||
return Result.error(ResultCode.CODE_SEND_FAIL, "验证码发送失败,请稍后重试");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 结果
|
||||
*/
|
||||
@PostMapping("/verify-email-code")
|
||||
public Result<String> verifyEmailCode(@RequestParam String email, @RequestParam String code) {
|
||||
// 参数校验
|
||||
if (!StringUtils.hasText(email)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "邮箱地址不能为空");
|
||||
}
|
||||
|
||||
if (!StringUtils.hasText(code)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "验证码不能为空");
|
||||
}
|
||||
|
||||
// 验证验证码
|
||||
boolean success = verificationCodeService.verifyEmailCode(email, code);
|
||||
|
||||
if (success) {
|
||||
return Result.success(ResultCode.SUCCESS, "验证码验证成功");
|
||||
} else {
|
||||
return Result.error(ResultCode.CODE_ERROR, "验证码错误或已过期");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 简单的邮箱格式校验
|
||||
* @param email 邮箱地址
|
||||
* @return 是否有效
|
||||
*/
|
||||
private boolean isValidEmail(String email) {
|
||||
if (email == null || email.trim().isEmpty()) {
|
||||
return false;
|
||||
}
|
||||
// 简单的邮箱格式校验:包含@和.
|
||||
return email.contains("@") && email.contains(".") && email.length() > 5;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,4 +0,0 @@
|
||||
package com.bao.dating.controller;
|
||||
|
||||
public class text {
|
||||
}
|
||||
109
src/main/java/com/bao/dating/handler/GlobalExceptionHandler.java
Normal file
109
src/main/java/com/bao/dating/handler/GlobalExceptionHandler.java
Normal file
@@ -0,0 +1,109 @@
|
||||
package com.bao.dating.handler;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.dao.DuplicateKeyException;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.http.converter.HttpMessageNotReadableException;
|
||||
import org.springframework.web.HttpRequestMethodNotSupportedException;
|
||||
import org.springframework.web.bind.MethodArgumentNotValidException;
|
||||
import org.springframework.web.bind.MissingServletRequestParameterException;
|
||||
import org.springframework.web.bind.annotation.ExceptionHandler;
|
||||
import org.springframework.web.bind.annotation.RestControllerAdvice;
|
||||
import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;
|
||||
import org.springframework.web.servlet.NoHandlerFoundException;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
|
||||
/**
|
||||
* 全局异常处理器
|
||||
* 统一处理控制器层抛出的异常
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@RestControllerAdvice
|
||||
public class GlobalExceptionHandler {
|
||||
|
||||
/**
|
||||
* 处理参数验证失败异常
|
||||
*/
|
||||
@ExceptionHandler(MethodArgumentNotValidException.class)
|
||||
public Result<String> handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
|
||||
log.error("参数验证失败: {}", e.getMessage());
|
||||
return Result.error(ResultCode.PARAM_ERROR, e.getBindingResult().getFieldError().getDefaultMessage());
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理请求参数缺失异常
|
||||
*/
|
||||
@ExceptionHandler(MissingServletRequestParameterException.class)
|
||||
public Result<String> handleMissingServletRequestParameterException(MissingServletRequestParameterException e) {
|
||||
log.error("请求参数缺失: 参数名={}, 参数类型={}", e.getParameterName(), e.getParameterType());
|
||||
return Result.error(ResultCode.PARAM_ERROR, "缺少必需的请求参数: " + e.getParameterName());
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理请求参数类型不匹配异常
|
||||
*/
|
||||
@ExceptionHandler(MethodArgumentTypeMismatchException.class)
|
||||
public Result<String> handleMethodArgumentTypeMismatchException(MethodArgumentTypeMismatchException e) {
|
||||
log.error("请求参数类型不匹配: {}", e.getMessage());
|
||||
return Result.error(ResultCode.PARAM_ERROR, "请求参数类型错误: " + e.getName());
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理请求体缺失或格式错误异常
|
||||
*/
|
||||
@ExceptionHandler(HttpMessageNotReadableException.class)
|
||||
public Result<String> handleHttpMessageNotReadableException(HttpMessageNotReadableException e) {
|
||||
log.error("请求体格式错误: {}", e.getMessage());
|
||||
return Result.error(ResultCode.PARAM_ERROR, "请求体格式错误或缺失");
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理不支持的HTTP请求方法异常
|
||||
*/
|
||||
@ExceptionHandler(HttpRequestMethodNotSupportedException.class)
|
||||
public Result<String> handleHttpRequestMethodNotSupportedException(HttpRequestMethodNotSupportedException e) {
|
||||
log.error("不支持的HTTP请求方法: {}", e.getMethod());
|
||||
return Result.error(ResultCode.PARAM_ERROR, "不支持的请求方法: " + e.getMethod());
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理404异常
|
||||
*/
|
||||
@ExceptionHandler(NoHandlerFoundException.class)
|
||||
public Result<String> handleNoHandlerFoundException(HttpServletRequest request, NoHandlerFoundException e) {
|
||||
log.error("请求的接口不存在: {} {}", request.getMethod(), request.getRequestURI());
|
||||
return Result.error(ResultCode.NOT_FOUND, "请求的接口不存在");
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理数据库唯一约束违反异常
|
||||
*/
|
||||
@ExceptionHandler(DuplicateKeyException.class)
|
||||
public Result<String> handleDuplicateKeyException(DuplicateKeyException e) {
|
||||
log.error("数据库唯一约束违反: {}", e.getMessage());
|
||||
return Result.error(ResultCode.FAIL, "数据已存在,操作失败");
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理运行时异常
|
||||
*/
|
||||
@ExceptionHandler(RuntimeException.class)
|
||||
public Result<String> handleRuntimeException(RuntimeException e) {
|
||||
log.error("运行时异常: ", e);
|
||||
return Result.error(ResultCode.SYSTEM_ERROR, e.getMessage());
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理通用异常
|
||||
*/
|
||||
@ExceptionHandler(Exception.class)
|
||||
public Result<String> handleException(HttpServletRequest request, Exception e) {
|
||||
log.error("系统异常 [{} {}]: ", request.getMethod(), request.getRequestURI(), e);
|
||||
return Result.error(ResultCode.SYSTEM_ERROR, "系统内部错误");
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,69 @@
|
||||
package com.bao.dating.handler;
|
||||
|
||||
import com.fasterxml.jackson.core.JsonProcessingException;
|
||||
import com.fasterxml.jackson.core.type.TypeReference;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import org.apache.ibatis.type.JdbcType;
|
||||
import org.apache.ibatis.type.MappedJdbcTypes;
|
||||
import org.apache.ibatis.type.MappedTypes;
|
||||
import org.apache.ibatis.type.TypeHandler;
|
||||
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.util.List;
|
||||
|
||||
|
||||
/**
|
||||
* List类型转换成JSON类型
|
||||
* @author KilLze
|
||||
*/
|
||||
@MappedJdbcTypes(JdbcType.VARCHAR)
|
||||
@MappedTypes(List.class)
|
||||
public class ListToJsonTypeHandler implements TypeHandler<List<String>> {
|
||||
|
||||
private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();
|
||||
|
||||
@Override
|
||||
public void setParameter(PreparedStatement ps, int i, List<String> parameter, JdbcType jdbcType) throws SQLException {
|
||||
if (parameter == null || parameter.isEmpty()) {
|
||||
ps.setNull(i, java.sql.Types.VARCHAR); // 或者 Types.JSON 如果数据库支持
|
||||
return;
|
||||
}
|
||||
try {
|
||||
String json = OBJECT_MAPPER.writeValueAsString(parameter);
|
||||
ps.setString(i, json);
|
||||
} catch (JsonProcessingException e) {
|
||||
throw new SQLException("Error converting list to JSON", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> getResult(ResultSet rs, String columnName) throws SQLException {
|
||||
String json = rs.getString(columnName);
|
||||
return convertJsonToList(json);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> getResult(ResultSet rs, int columnIndex) throws SQLException {
|
||||
String json = rs.getString(columnIndex);
|
||||
return convertJsonToList(json);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> getResult(java.sql.CallableStatement cs, int columnIndex) throws SQLException {
|
||||
String json = cs.getString(columnIndex);
|
||||
return convertJsonToList(json);
|
||||
}
|
||||
|
||||
private List<String> convertJsonToList(String json) throws SQLException {
|
||||
if (json == null || json.isEmpty()) {
|
||||
return null;
|
||||
}
|
||||
try {
|
||||
return OBJECT_MAPPER.readValue(json, new TypeReference<List<String>>() {});
|
||||
} catch (JsonProcessingException e) {
|
||||
throw new SQLException("Error converting JSON to list", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -6,10 +6,7 @@ import org.apache.ibatis.type.MappedJdbcTypes;
|
||||
import org.apache.ibatis.type.MappedTypes;
|
||||
import org.apache.ibatis.type.TypeHandler;
|
||||
|
||||
import java.sql.CallableStatement;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.*;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
@@ -22,6 +19,11 @@ import java.util.List;
|
||||
public class ListToVarcharTypeHandler implements TypeHandler<List<String>> {
|
||||
@Override
|
||||
public void setParameter(PreparedStatement preparedStatement, int i, List<String> strings, JdbcType jdbcType) throws SQLException {
|
||||
// 允许 null
|
||||
if (strings == null || strings.isEmpty()) {
|
||||
preparedStatement.setNull(i, Types.VARCHAR);
|
||||
return;
|
||||
}
|
||||
// 遍历List类型的入参,拼装为String类型,使用Statement对象插入数据库
|
||||
StringBuffer sb = new StringBuffer();
|
||||
for (int j = 0; j < strings.size(); j++) {
|
||||
|
||||
@@ -0,0 +1,79 @@
|
||||
package com.bao.dating.interceptor;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
|
||||
import com.bao.dating.context.UserContext;
|
||||
import com.bao.dating.util.JwtUtil;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.web.method.HandlerMethod;
|
||||
import org.springframework.web.servlet.HandlerInterceptor;
|
||||
|
||||
/**
|
||||
* HttpToken拦截器类
|
||||
* 用于拦截请求并验证JWT token的有效性,同时从token中解析用户信息
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@Component
|
||||
public class TokenInterceptor implements HandlerInterceptor {
|
||||
/**
|
||||
* 在请求处理之前进行拦截
|
||||
* 从请求头或URL参数中获取token,验证其有效性,并将用户ID保存到ThreadLocal中
|
||||
* @param request HTTP请求对象
|
||||
* @param response HTTP响应对象
|
||||
* @param handler 处理器
|
||||
* @return 验证通过返回true,否则返回false
|
||||
* @throws Exception 异常
|
||||
*/
|
||||
@Override
|
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
|
||||
|
||||
//判断当前拦截到的是Controller的方法还是其他资源
|
||||
if (!(handler instanceof HandlerMethod)) {
|
||||
//当前拦截到的不是动态方法,直接放行
|
||||
return true;
|
||||
}
|
||||
// 从 header 获取 token
|
||||
String token = request.getHeader("token");
|
||||
|
||||
try {
|
||||
log.info("jwt校验: {}", token);
|
||||
|
||||
// 验证 token 是否有效(包括是否过期)
|
||||
if (!JwtUtil.validateToken(token)) {
|
||||
log.error("Token无效或已过期");
|
||||
response.setStatus(401);
|
||||
return false;
|
||||
}
|
||||
|
||||
// 解析 token
|
||||
String userId = JwtUtil.getSubjectFromToken(token);
|
||||
log.info("用户: {}", userId);
|
||||
|
||||
// 保存 userId 到 ThreadLocal
|
||||
UserContext.setUserId(Long.valueOf(userId));
|
||||
return true;
|
||||
} catch (Exception e) {
|
||||
log.error("Token 校验失败: {}", e.getMessage());
|
||||
response.setStatus(401);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 在请求完成之后执行清理工作
|
||||
* 清除保存在ThreadLocal中的用户ID,防止内存泄漏
|
||||
* @param request HTTP请求对象
|
||||
* @param response HTTP响应对象
|
||||
* @param handler 处理器
|
||||
* @param ex 异常对象
|
||||
* @throws Exception 异常
|
||||
*/
|
||||
@Override
|
||||
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
|
||||
UserContext.clear();
|
||||
}
|
||||
|
||||
}
|
||||
21
src/main/java/com/bao/dating/mapper/CommentsMapper.java
Normal file
21
src/main/java/com/bao/dating/mapper/CommentsMapper.java
Normal file
@@ -0,0 +1,21 @@
|
||||
package com.bao.dating.mapper;
|
||||
|
||||
|
||||
import com.bao.dating.pojo.entity.Comments;
|
||||
import org.apache.ibatis.annotations.*;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface CommentsMapper {
|
||||
// 添加评论
|
||||
@Insert("INSERT INTO comments(content, user_id, post_id, created_at) VALUES(#{content}, #{user_id}, #{post_id}, #{created_at})")
|
||||
int addComment(Comments comments);
|
||||
|
||||
// 删除评论
|
||||
@Delete("DELETE FROM comments WHERE user_id = #{user_id}")
|
||||
int deleteComments(@Param("user_id") Long user_id);
|
||||
|
||||
// 根据动态ID查询评论列表
|
||||
@Select("SELECT * FROM comments WHERE post_id = #{post_id} ORDER BY created_at DESC")
|
||||
List<Comments> getCommentsByPostId(@Param("post_id") Long post_id);
|
||||
}
|
||||
@@ -1,5 +1,6 @@
|
||||
package com.bao.dating.mapper;
|
||||
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
import com.bao.dating.pojo.entity.PostFavorite;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
@@ -10,6 +11,10 @@ import java.util.List;
|
||||
public interface PostFavoriteMapper {
|
||||
//查询当前已收藏所有用户
|
||||
List<Long> selectUserIDByPostID(@Param("postId") Long postId);
|
||||
//添加收藏
|
||||
int addPostFavorite(PostFavorite postFavorite);
|
||||
//删除收藏
|
||||
int deletePostFavorite(@Param("postId") Long postId);
|
||||
//查询用户所有收藏
|
||||
List<Post> showAllFavorites(@Param("userid")Long userid);
|
||||
}
|
||||
|
||||
@@ -4,43 +4,49 @@ import com.bao.dating.pojo.entity.Post;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 动态Mapper
|
||||
*
|
||||
* @author KilLze lanyangyang-yzx
|
||||
*/
|
||||
@Mapper
|
||||
public interface PostMapper {
|
||||
/**
|
||||
* 插入动态
|
||||
*
|
||||
* @param post
|
||||
* @param post 动态
|
||||
*/
|
||||
void insert(Post post);
|
||||
|
||||
/**
|
||||
* 根据ID删除动态
|
||||
*
|
||||
* @param postId 动态ID
|
||||
* @param postIds 动态ID
|
||||
*/
|
||||
void deletePostById(@Param("postId") Long postId);
|
||||
int deletePostByIds(List<Long> postIds);
|
||||
|
||||
/**
|
||||
* 根据ID查询动态
|
||||
*
|
||||
* @param postId
|
||||
* @return
|
||||
* @param postId 动态ID
|
||||
* @return 动态
|
||||
*/
|
||||
Post selectById(@Param("postId") Long postId);
|
||||
Post selectById(Long postId);
|
||||
|
||||
/**
|
||||
* 根据ID更新动态
|
||||
*
|
||||
* @param post
|
||||
* @return
|
||||
* @param post 动态
|
||||
*/
|
||||
void updateById(Post post);
|
||||
|
||||
/**
|
||||
* 查询点赞数
|
||||
*
|
||||
* @param postId
|
||||
* @return
|
||||
* @param postId 动态ID
|
||||
* @return 点赞数
|
||||
*/
|
||||
int selectLikeCount(Long postId);
|
||||
|
||||
@@ -70,8 +76,8 @@ public interface PostMapper {
|
||||
/**
|
||||
* 查询点赞数
|
||||
*
|
||||
* @param postId
|
||||
* @return
|
||||
* @param postId 动态ID
|
||||
* @return 点赞数
|
||||
*/
|
||||
int selectFavoriteCount(Long postId);
|
||||
|
||||
|
||||
58
src/main/java/com/bao/dating/mapper/UserMapper.java
Normal file
58
src/main/java/com/bao/dating/mapper/UserMapper.java
Normal file
@@ -0,0 +1,58 @@
|
||||
package com.bao.dating.mapper;
|
||||
|
||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
||||
import com.bao.dating.pojo.entity.User;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
|
||||
/**
|
||||
* 用户Mapper
|
||||
* @author KilLze
|
||||
*/
|
||||
@Mapper
|
||||
public interface UserMapper {
|
||||
|
||||
/**
|
||||
* 根据用户名查询用户
|
||||
*
|
||||
* @param username 用户名
|
||||
* @return 用户
|
||||
*/
|
||||
User getByUsername(String username);
|
||||
|
||||
/**
|
||||
* 根据用户id查询用户信息
|
||||
*
|
||||
* @param userid 用户id
|
||||
* @return 用户
|
||||
*/
|
||||
User selectByUserId(Long userid);
|
||||
|
||||
/**
|
||||
* 更新用户信息
|
||||
* @param userInfoUpdateDTO 用户信息更新参数
|
||||
*/
|
||||
void updateUserInfoByUserId(UserInfoUpdateDTO userInfoUpdateDTO);
|
||||
|
||||
/**
|
||||
* 添加用户
|
||||
* @param user 用户对象
|
||||
* @return 受影响行数
|
||||
*/
|
||||
int saveUser(User user);
|
||||
|
||||
/**
|
||||
* 查询最大用户id
|
||||
* @return 用户id
|
||||
*/
|
||||
Long selectMaxId();
|
||||
|
||||
/**
|
||||
* 根据邮箱查询用户
|
||||
* @param email 用户邮箱
|
||||
* @return 用户信息
|
||||
*/
|
||||
User selectByUserEmailUser(@Param("userEmail") String email);
|
||||
|
||||
|
||||
}
|
||||
@@ -2,13 +2,15 @@ package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 动态数据传输对象
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class PostRequestDTO {
|
||||
public class PostRequestDTO implements Serializable{
|
||||
private String content;
|
||||
private List<String> mediaOssKeys;
|
||||
private List<String> tags;
|
||||
|
||||
25
src/main/java/com/bao/dating/pojo/dto/UserInfoUpdateDTO.java
Normal file
25
src/main/java/com/bao/dating/pojo/dto/UserInfoUpdateDTO.java
Normal file
@@ -0,0 +1,25 @@
|
||||
package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.time.LocalDate;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 用户信息更新数据传输对象
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class UserInfoUpdateDTO {
|
||||
private Long userId;
|
||||
private String userName;
|
||||
private String nickname;
|
||||
private String avatarUrl;
|
||||
private String backgroundUrl;
|
||||
private Integer gender;
|
||||
private LocalDate birthday;
|
||||
private List<String> hobbies;
|
||||
private String signature;
|
||||
private LocalDateTime updatedAt;
|
||||
}
|
||||
15
src/main/java/com/bao/dating/pojo/dto/UserLoginDTO.java
Normal file
15
src/main/java/com/bao/dating/pojo/dto/UserLoginDTO.java
Normal file
@@ -0,0 +1,15 @@
|
||||
package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
|
||||
/**
|
||||
* 用户登录数据传输对象
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class UserLoginDTO implements Serializable {
|
||||
private String username;
|
||||
private String password;
|
||||
}
|
||||
20
src/main/java/com/bao/dating/pojo/entity/Comments.java
Normal file
20
src/main/java/com/bao/dating/pojo/entity/Comments.java
Normal file
@@ -0,0 +1,20 @@
|
||||
package com.bao.dating.pojo.entity;
|
||||
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDateTime;
|
||||
|
||||
|
||||
/**
|
||||
* 评论表
|
||||
*/
|
||||
@Data
|
||||
public class Comments implements Serializable {
|
||||
private Long comment_id; // 评论ID
|
||||
private String content; // 评论内容
|
||||
private Long user_id; // 评论人ID
|
||||
private Long post_id; // 关联动态ID
|
||||
private LocalDateTime created_at; // 评论时间
|
||||
}
|
||||
@@ -2,6 +2,7 @@ package com.bao.dating.pojo.entity;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
@@ -10,7 +11,7 @@ import java.util.List;
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class Post {
|
||||
public class Post implements Serializable {
|
||||
|
||||
private Long postId;
|
||||
|
||||
|
||||
@@ -2,6 +2,7 @@ package com.bao.dating.pojo.entity;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDateTime;
|
||||
|
||||
/**
|
||||
@@ -9,7 +10,7 @@ import java.time.LocalDateTime;
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class PostFavorite {
|
||||
public class PostFavorite implements Serializable {
|
||||
|
||||
private Long favoriteId;
|
||||
|
||||
|
||||
@@ -2,6 +2,7 @@ package com.bao.dating.pojo.entity;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDateTime;
|
||||
|
||||
/**
|
||||
@@ -9,7 +10,7 @@ import java.time.LocalDateTime;
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class PostLike {
|
||||
public class PostLike implements Serializable {
|
||||
private Long likeId;
|
||||
|
||||
private Long userId;
|
||||
|
||||
@@ -2,15 +2,17 @@ package com.bao.dating.pojo.entity;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDate;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 用户表
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class User {
|
||||
public class User implements Serializable {
|
||||
|
||||
private Long userId;
|
||||
|
||||
@@ -30,11 +32,15 @@ public class User {
|
||||
|
||||
private LocalDate birthday;
|
||||
|
||||
private String hobbies;
|
||||
private List<String> hobbies;
|
||||
|
||||
private String signature;
|
||||
|
||||
private LocalDateTime createdAt;
|
||||
|
||||
private LocalDateTime updatedAt;
|
||||
|
||||
private String userEmail;
|
||||
|
||||
private String userPhone;
|
||||
}
|
||||
|
||||
30
src/main/java/com/bao/dating/pojo/vo/IpLocationVO.java
Normal file
30
src/main/java/com/bao/dating/pojo/vo/IpLocationVO.java
Normal file
@@ -0,0 +1,30 @@
|
||||
package com.bao.dating.pojo.vo;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
|
||||
@Data
|
||||
@AllArgsConstructor
|
||||
@NoArgsConstructor
|
||||
@JsonIgnoreProperties(ignoreUnknown = true)
|
||||
public class IpLocationVO {
|
||||
// 国家
|
||||
@JsonProperty("country_name")
|
||||
private String countryName;
|
||||
// 省份/地区
|
||||
@JsonProperty("region_name")
|
||||
private String regionName;
|
||||
// 城市
|
||||
@JsonProperty("city_name")
|
||||
private String cityName;
|
||||
// 纬度
|
||||
private String latitude;
|
||||
// 经度
|
||||
private String longitude;
|
||||
// ISP运营商
|
||||
@JsonProperty("isp")
|
||||
private String isp;
|
||||
}
|
||||
@@ -2,14 +2,16 @@ package com.bao.dating.pojo.vo;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 修改内容查询返回数据
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class PostEditVO {
|
||||
public class PostEditVO implements Serializable {
|
||||
private Long postId;
|
||||
private String content;
|
||||
private List<String> mediaOssKeys;
|
||||
|
||||
27
src/main/java/com/bao/dating/pojo/vo/UserInfoVO.java
Normal file
27
src/main/java/com/bao/dating/pojo/vo/UserInfoVO.java
Normal file
@@ -0,0 +1,27 @@
|
||||
package com.bao.dating.pojo.vo;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.time.LocalDate;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 用户信息VO
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class UserInfoVO implements Serializable {
|
||||
private Long userId;
|
||||
private String userName;
|
||||
private String nickname;
|
||||
private String avatarUrl;
|
||||
private String backgroundUrl;
|
||||
private Integer gender;
|
||||
private LocalDate birthday;
|
||||
private List<String> hobbies;
|
||||
private String signature;
|
||||
private LocalDateTime updatedAt;
|
||||
private LocalDateTime createdAt;
|
||||
}
|
||||
15
src/main/java/com/bao/dating/pojo/vo/UserLoginVO.java
Normal file
15
src/main/java/com/bao/dating/pojo/vo/UserLoginVO.java
Normal file
@@ -0,0 +1,15 @@
|
||||
package com.bao.dating.pojo.vo;
|
||||
import lombok.Data;
|
||||
|
||||
import java.io.Serializable;
|
||||
|
||||
/**
|
||||
* 登录返回数据
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class UserLoginVO implements Serializable {
|
||||
private Long userId;
|
||||
private String nickname;
|
||||
private String token;
|
||||
}
|
||||
20
src/main/java/com/bao/dating/service/CommentsService.java
Normal file
20
src/main/java/com/bao/dating/service/CommentsService.java
Normal file
@@ -0,0 +1,20 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
|
||||
import com.bao.dating.pojo.entity.Comments;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface CommentsService{
|
||||
// 添加评论
|
||||
boolean addComment(Comments comments);
|
||||
|
||||
// 删除评论
|
||||
boolean deleteComment(Long user_id);
|
||||
|
||||
// 根据动态ID查询评论
|
||||
List<Comments> getCommentByPostId(Long post_id);
|
||||
|
||||
|
||||
|
||||
}
|
||||
@@ -0,0 +1,7 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
import com.bao.dating.pojo.vo.IpLocationVO;
|
||||
|
||||
public interface Ip2LocationClientService {
|
||||
IpLocationVO getIpLocation(String ip)throws Exception;
|
||||
}
|
||||
@@ -1,10 +1,13 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
public interface PostFavoriteService {
|
||||
Result<Map<String,Long>> postFavorite(Long userid,Long postId);
|
||||
Result<?> deletePostFavorite(Long userid,Long postId);
|
||||
List<Post> selectAllFavorites(Long userid);
|
||||
}
|
||||
|
||||
@@ -7,6 +7,10 @@ import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 动态服务
|
||||
* @author bao KilLze lanyangyang-yzx yang
|
||||
*/
|
||||
public interface PostService {
|
||||
/**
|
||||
* 上传媒体文件
|
||||
@@ -17,24 +21,23 @@ public interface PostService {
|
||||
|
||||
/**
|
||||
* 创建动态
|
||||
* @param userId 用户ID
|
||||
* @param postRequestDTO 动态数据传输对象
|
||||
* @return 创建的动态对象
|
||||
*/
|
||||
Post createPost(Long userId, PostRequestDTO postRequestDTO);
|
||||
Post createPost(PostRequestDTO postRequestDTO);
|
||||
|
||||
/**
|
||||
* 删除动态
|
||||
* @param postId 动态ID
|
||||
* 批量删除动态
|
||||
* @param postIds 动态ID
|
||||
* @return 删除的动态对象
|
||||
*/
|
||||
void deletePostById(Long postId);
|
||||
int deletePostById(List<Long> postIds);
|
||||
|
||||
void deletePostById(Integer postId);
|
||||
|
||||
/**
|
||||
* 查询动态详情(用于编辑)
|
||||
* @param postId 动态ID
|
||||
* @return 动态详情
|
||||
*/
|
||||
PostEditVO getPostForEdit(Long postId);
|
||||
|
||||
|
||||
64
src/main/java/com/bao/dating/service/UserService.java
Normal file
64
src/main/java/com/bao/dating/service/UserService.java
Normal file
@@ -0,0 +1,64 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
||||
import com.bao.dating.pojo.dto.UserLoginDTO;
|
||||
import com.bao.dating.pojo.entity.User;
|
||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
/**
|
||||
* 用户服务接口
|
||||
* @author KilLze
|
||||
*/
|
||||
public interface UserService {
|
||||
/**
|
||||
* 登录
|
||||
* @param userLoginDTO 登录参数
|
||||
* @return 登录结果
|
||||
*/
|
||||
UserLoginVO userLogin(UserLoginDTO userLoginDTO);
|
||||
|
||||
/**
|
||||
* 查询个人信息
|
||||
* @param userId 动态ID
|
||||
* @return 个人信息
|
||||
*/
|
||||
UserInfoVO getUserInfo(Long userId);
|
||||
|
||||
/**
|
||||
* 上传头像
|
||||
* @param file 头像文件
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
String uploadAvatar(MultipartFile file);
|
||||
|
||||
/**
|
||||
* 上传背景
|
||||
* @param file 背景文件
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
String uploadBackground(MultipartFile file);
|
||||
|
||||
/**
|
||||
* 更新用户信息
|
||||
* @param userInfoUpdateDTO 用户信息
|
||||
* @return 更新后的用户信息
|
||||
*/
|
||||
UserInfoVO updateUserInfo(UserInfoUpdateDTO userInfoUpdateDTO);
|
||||
|
||||
/**
|
||||
* 用户注册
|
||||
* @param userName 用户民称
|
||||
* @return 用户信息
|
||||
*/
|
||||
Boolean registerUser(String userName,String userPassword);
|
||||
|
||||
/**
|
||||
* 邮箱登录
|
||||
* @param email 邮箱
|
||||
* @param code 验证码
|
||||
* @return
|
||||
*/
|
||||
UserLoginVO emailLogin(String email , String code);
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
/**
|
||||
* 验证码服务接口
|
||||
* @author KilLze
|
||||
*/
|
||||
public interface VerificationCodeService {
|
||||
|
||||
/**
|
||||
* 发送邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
boolean sendEmailCode(String email);
|
||||
|
||||
/**
|
||||
* 验证邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 是否验证成功
|
||||
*/
|
||||
boolean verifyEmailCode(String email, String code);
|
||||
|
||||
/**
|
||||
* 生成验证码
|
||||
* @param length 验证码长度(默认6位)
|
||||
* @return 验证码字符串
|
||||
*/
|
||||
String generateCode(int length);
|
||||
}
|
||||
|
||||
@@ -0,0 +1,39 @@
|
||||
package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.pojo.entity.Comments;
|
||||
|
||||
import com.bao.dating.mapper.CommentsMapper;
|
||||
|
||||
|
||||
import com.bao.dating.service.CommentsService;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
@Service
|
||||
public class CommentsServiceImpl implements CommentsService {
|
||||
|
||||
|
||||
|
||||
@Resource
|
||||
private CommentsMapper commentsMapper;
|
||||
|
||||
@Override
|
||||
public boolean addComment(Comments comments) {
|
||||
// 设置创建时间
|
||||
comments.setCreated_at(LocalDateTime.now());
|
||||
return commentsMapper.addComment(comments) > 0;
|
||||
}
|
||||
@Override
|
||||
public boolean deleteComment(Long user_id) {
|
||||
return commentsMapper.deleteComments(user_id) > 0;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<Comments> getCommentByPostId(Long post_id) {
|
||||
return commentsMapper.getCommentsByPostId(post_id);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,67 @@
|
||||
package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.common.ip2location.Ip2LocationConfig;
|
||||
import com.bao.dating.pojo.vo.IpLocationVO;
|
||||
import com.bao.dating.service.Ip2LocationClientService;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import org.apache.http.client.config.RequestConfig;
|
||||
import org.apache.http.client.methods.CloseableHttpResponse;
|
||||
import org.apache.http.client.methods.HttpGet;
|
||||
import org.apache.http.client.utils.URIBuilder;
|
||||
import org.apache.http.impl.client.CloseableHttpClient;
|
||||
import org.apache.http.impl.client.HttpClients;
|
||||
import org.apache.http.util.EntityUtils;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.net.URI;
|
||||
|
||||
@Service
|
||||
public class Ip2LocationClientServiceImpl implements Ip2LocationClientService {
|
||||
@Autowired
|
||||
private Ip2LocationConfig ip2LocationConfig;
|
||||
|
||||
// Jackson的ObjectMapper,用于JSON解析
|
||||
private final ObjectMapper objectMapper = new ObjectMapper();
|
||||
|
||||
/**
|
||||
* 调用API并只返回核心位置信息
|
||||
* @param ip 要查询的IP地址
|
||||
* @return 精简的位置信息实体类
|
||||
* @throws Exception 异常
|
||||
*/
|
||||
@Override
|
||||
public IpLocationVO getIpLocation(String ip) throws Exception {
|
||||
// 1. 构建请求URL
|
||||
URIBuilder uriBuilder = new URIBuilder(ip2LocationConfig.getUrl());
|
||||
uriBuilder.addParameter("key", ip2LocationConfig.getKey());
|
||||
if (ip != null && !ip.trim().isEmpty()) {
|
||||
uriBuilder.addParameter("ip", ip);
|
||||
}
|
||||
URI uri = uriBuilder.build();
|
||||
|
||||
// 2. 配置超时时间(逻辑和之前一致)
|
||||
RequestConfig requestConfig = RequestConfig.custom()
|
||||
.setConnectTimeout(ip2LocationConfig.getTimeout())
|
||||
.setSocketTimeout(ip2LocationConfig.getTimeout())
|
||||
.build();
|
||||
|
||||
// 3. 发送请求并解析响应
|
||||
try (CloseableHttpClient httpClient = HttpClients.createDefault()) {
|
||||
HttpGet httpGet = new HttpGet(uri);
|
||||
httpGet.setConfig(requestConfig);
|
||||
|
||||
try (CloseableHttpResponse response = httpClient.execute(httpGet)) {
|
||||
if (response.getStatusLine().getStatusCode() == 200) {
|
||||
String jsonStr = EntityUtils.toString(response.getEntity(), "UTF-8");
|
||||
// 核心:将完整JSON解析为只包含位置信息的VO类
|
||||
return objectMapper.readValue(jsonStr, IpLocationVO.class);
|
||||
} else {
|
||||
throw new RuntimeException("调用API失败,状态码:" + response.getStatusLine().getStatusCode());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@@ -4,6 +4,7 @@ import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.mapper.PostFavoriteMapper;
|
||||
import com.bao.dating.mapper.PostMapper;
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
import com.bao.dating.pojo.entity.PostFavorite;
|
||||
import com.bao.dating.service.PostFavoriteService;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
@@ -58,4 +59,18 @@ public class PostFavoriteServiceImpl implements PostFavoriteService {
|
||||
postMapper.decreaseFavoriteCount(postId);
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 展示所有收藏
|
||||
* @param userid 用户id
|
||||
* @return 查询到的结果
|
||||
*/
|
||||
|
||||
@Override
|
||||
public List<Post> selectAllFavorites(Long userid) {
|
||||
if (userid == null){
|
||||
return null;
|
||||
}
|
||||
return postFavoriteMapper.showAllFavorites(userid);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -2,6 +2,8 @@ package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.common.aliyun.GreenImageScan;
|
||||
import com.bao.dating.common.aliyun.GreenTextScan;
|
||||
import com.bao.dating.common.result.GreenAuditResult;
|
||||
import com.bao.dating.context.UserContext;
|
||||
import com.bao.dating.mapper.PostMapper;
|
||||
import com.bao.dating.pojo.dto.PostRequestDTO;
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
@@ -12,19 +14,19 @@ import com.bao.dating.util.FileUtil;
|
||||
import org.springframework.beans.BeanUtils;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.time.LocalDate;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.UUID;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.*;
|
||||
|
||||
/**
|
||||
* 动态服务实现类
|
||||
*
|
||||
* @author KilLze
|
||||
* @author KilLze yang
|
||||
*/
|
||||
@Service
|
||||
public class PostServiceImpl implements PostService {
|
||||
@@ -48,46 +50,70 @@ public class PostServiceImpl implements PostService {
|
||||
*/
|
||||
@Override
|
||||
public List<String> uploadMedia(MultipartFile[] files) {
|
||||
// 如果没有文件,则返回空列表
|
||||
if (files == null || files.length == 0) {
|
||||
return Collections.emptyList();
|
||||
}
|
||||
// 创建媒体文件列表
|
||||
List<String> mediaUrls = new ArrayList<>();
|
||||
if (files != null && files.length > 0) {
|
||||
|
||||
for (MultipartFile file : files) {
|
||||
if (!file.isEmpty()) {
|
||||
// 跳过空文件
|
||||
if (file == null || file.isEmpty()) {
|
||||
continue;
|
||||
}
|
||||
// 获取文件名并跳过空文件
|
||||
String originalFilename = file.getOriginalFilename();
|
||||
if (originalFilename == null) {
|
||||
continue;
|
||||
}
|
||||
// 校验文件类型
|
||||
String fileType = FileUtil.getFileType(originalFilename);
|
||||
if (!"image".equals(fileType) && !"video".equals(fileType)) {
|
||||
throw new RuntimeException("不支持的文件类型:" + originalFilename);
|
||||
}
|
||||
|
||||
// 创建目录
|
||||
String dir = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy/MM"));
|
||||
// 获取文件扩展名
|
||||
String extension = FileUtil.getFileExtension(originalFilename);
|
||||
// 生成唯一文件名
|
||||
String newFileName = UUID.randomUUID().toString().replace("-", "") + "." + extension;
|
||||
// 获取用户ID
|
||||
Long userId = UserContext.getUserId();
|
||||
// 创建文件名
|
||||
String fileName = "post/" + userId + "/" + dir + "/" + newFileName;
|
||||
|
||||
try {
|
||||
// 根据文件扩展名判断文件类型
|
||||
String fileType = FileUtil.getFileType(file.getOriginalFilename());
|
||||
// 生成唯一文件名
|
||||
String fileName = UUID.randomUUID().toString() + "." + FileUtil.getFileExtension(file.getOriginalFilename());
|
||||
// 获取文件字节数据
|
||||
byte[] fileBytes = file.getBytes();
|
||||
// 根据文件类型处理
|
||||
String ossUrl = "";
|
||||
if ("image".equals(fileType) || "video".equals(fileType)) {
|
||||
// 上传图片或视频
|
||||
ossUrl = ossUtil.upload(fileBytes, fileName);
|
||||
// 上传图片或视频
|
||||
String ossUrl = ossUtil.upload(fileBytes, fileName);
|
||||
if (ossUrl == null || ossUrl.isEmpty()) {
|
||||
throw new RuntimeException("文件上传失败:" + originalFilename);
|
||||
}
|
||||
// 添加上传后的 URL
|
||||
mediaUrls.add(ossUrl);
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
// 统一异常处理
|
||||
throw new RuntimeException("上传媒体文件失败:" + originalFilename, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return mediaUrls;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建动态
|
||||
*
|
||||
* @param userId 用户ID
|
||||
* @param postRequestDTO 动态数据传输对象
|
||||
* @return 创建的动态对象
|
||||
*/
|
||||
@Override
|
||||
public Post createPost(Long userId, PostRequestDTO postRequestDTO) {
|
||||
public Post createPost(PostRequestDTO postRequestDTO) {
|
||||
|
||||
// 创建动态对象
|
||||
Post post = new Post();
|
||||
Long userId = UserContext.getUserId();
|
||||
post.setUserId(userId);
|
||||
post.setContent(postRequestDTO.getContent());
|
||||
post.setTags(postRequestDTO.getTags());
|
||||
@@ -106,9 +132,6 @@ public class PostServiceImpl implements PostService {
|
||||
}
|
||||
|
||||
String textSuggestion = (String) textResult.get("suggestion");
|
||||
if ("block".equals(textSuggestion)) {
|
||||
throw new RuntimeException("动态内容违规,禁止发布");
|
||||
}
|
||||
|
||||
// 2. 图片审核(如果有)
|
||||
if (postRequestDTO.getMediaOssKeys() != null && !postRequestDTO.getMediaOssKeys().isEmpty()) {
|
||||
@@ -122,10 +145,10 @@ public class PostServiceImpl implements PostService {
|
||||
String imageSuggestion = (String) imageResult.get("suggestion");
|
||||
|
||||
// 根据审核结果设置状态
|
||||
if ("block".equals(textSuggestion) || "block".equals(imageSuggestion)) {
|
||||
if (GreenAuditResult.BLOCK.equals(textSuggestion) || GreenAuditResult.BLOCK.equals(imageSuggestion)) {
|
||||
// 审核未通过,允许用户修改
|
||||
post.setIsPublic(2);
|
||||
} else if ("review".equals(textSuggestion) || "review".equals(imageSuggestion)) {
|
||||
} else if (GreenAuditResult.REVIEW.equals(textSuggestion) || GreenAuditResult.REVIEW.equals(imageSuggestion)) {
|
||||
// 待审核,需人工审核
|
||||
post.setIsPublic(1);
|
||||
} else {
|
||||
@@ -134,10 +157,10 @@ public class PostServiceImpl implements PostService {
|
||||
}
|
||||
} else {
|
||||
// 只有文本内容的情况
|
||||
if ("block".equals(textSuggestion)) {
|
||||
if (GreenAuditResult.BLOCK.equals(textSuggestion)) {
|
||||
// 审核未通过,允许用户修改
|
||||
post.setIsPublic(2);
|
||||
} else if ("review".equals(textSuggestion)) {
|
||||
} else if (GreenAuditResult.REVIEW.equals(textSuggestion)) {
|
||||
// 待审核,需人工审核
|
||||
post.setIsPublic(1);
|
||||
} else {
|
||||
@@ -153,29 +176,50 @@ public class PostServiceImpl implements PostService {
|
||||
return post;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deletePostById(Long postId) {
|
||||
postMapper.deletePostById(postId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除动态
|
||||
* 批量删除动态
|
||||
*
|
||||
* @param postId 动态ID
|
||||
* @param postIds 动态ID
|
||||
* @return 删除的动态对象
|
||||
*/
|
||||
@Override
|
||||
public void deletePostById(Integer postId) {
|
||||
postMapper.deletePostById(postId);
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public int deletePostById(List<Long> postIds) {
|
||||
// 判断用户权限
|
||||
Long userId = UserContext.getUserId();
|
||||
|
||||
// 遍历所有要删除的帖子ID,验证权限
|
||||
for (Long postId : postIds) {
|
||||
Post post = postMapper.selectById(postId);
|
||||
if (post == null) {
|
||||
throw new RuntimeException("动态不存在");
|
||||
}
|
||||
// 验证用户权限
|
||||
if (post.getUserId() == null || !post.getUserId().equals(userId)) {
|
||||
throw new RuntimeException("无权限删除此动态");
|
||||
}
|
||||
}
|
||||
// 批量删除动态
|
||||
return postMapper.deletePostByIds(postIds);
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询动态详情(用于编辑)
|
||||
*
|
||||
* @param postId 动态ID
|
||||
* @return 动态详情
|
||||
*/
|
||||
@Override
|
||||
public PostEditVO getPostForEdit(Long postId) {
|
||||
|
||||
Post post = postMapper.selectById(postId);
|
||||
if (post == null) {
|
||||
throw new RuntimeException("动态不存在");
|
||||
}
|
||||
// 判断用户权限
|
||||
Long userId = UserContext.getUserId();
|
||||
if (post.getUserId() == null || !post.getUserId().equals(userId)){
|
||||
throw new RuntimeException("无权限查看此动态");
|
||||
}
|
||||
PostEditVO postEditVO = new PostEditVO();
|
||||
BeanUtils.copyProperties(post, postEditVO);
|
||||
return postEditVO;
|
||||
@@ -185,7 +229,7 @@ public class PostServiceImpl implements PostService {
|
||||
* 修改动态
|
||||
* @param postId 动态ID
|
||||
* @param postRequestDTO 修改的动态数据传输对象
|
||||
* @return
|
||||
* @return 修改的动态对象
|
||||
*/
|
||||
@Override
|
||||
public PostEditVO updatePost(Long postId, PostRequestDTO postRequestDTO) {
|
||||
@@ -194,10 +238,14 @@ public class PostServiceImpl implements PostService {
|
||||
if (post == null) {
|
||||
throw new RuntimeException("动态不存在");
|
||||
}
|
||||
post.setContent(postRequestDTO.getContent());
|
||||
if (postRequestDTO.getMediaOssKeys() != null && !postRequestDTO.getMediaOssKeys().isEmpty()) {
|
||||
post.setMediaOssKeys(postRequestDTO.getMediaOssKeys());
|
||||
// 判断用户权限
|
||||
Long userId = UserContext.getUserId();
|
||||
if (post.getUserId() == null || !post.getUserId().equals(userId)){
|
||||
throw new RuntimeException("无权限修改此动态");
|
||||
}
|
||||
post.setContent(postRequestDTO.getContent());
|
||||
// 如果请求中的mediaOssKeys不为null(即使是空列表),则更新为新值
|
||||
post.setMediaOssKeys(postRequestDTO.getMediaOssKeys());
|
||||
|
||||
// 1. 文本内容审核
|
||||
Map textResult;
|
||||
@@ -209,11 +257,11 @@ public class PostServiceImpl implements PostService {
|
||||
// 文本审核结果
|
||||
String textSuggestion = (String) textResult.get("suggestion");
|
||||
|
||||
// 2. 图片审核(如果有)
|
||||
if (postRequestDTO.getMediaOssKeys() != null && !postRequestDTO.getMediaOssKeys().isEmpty()) {
|
||||
// 2. 图片审核(如果有媒体文件)
|
||||
if (post.getMediaOssKeys() != null && !post.getMediaOssKeys().isEmpty()) {
|
||||
Map imageResult;
|
||||
try {
|
||||
imageResult = greenImageScan.imageScan(postRequestDTO.getMediaOssKeys());
|
||||
imageResult = greenImageScan.imageScan(post.getMediaOssKeys());
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
@@ -221,10 +269,10 @@ public class PostServiceImpl implements PostService {
|
||||
String imageSuggestion = (String) imageResult.get("suggestion");
|
||||
|
||||
// 根据审核结果设置状态
|
||||
if ("block".equals(textSuggestion) || "block".equals(imageSuggestion)) {
|
||||
if (GreenAuditResult.BLOCK.equals(textSuggestion) || GreenAuditResult.BLOCK.equals(imageSuggestion)) {
|
||||
// 审核未通过,允许用户修改
|
||||
post.setIsPublic(2);
|
||||
} else if ("review".equals(textSuggestion) || "review".equals(imageSuggestion)) {
|
||||
} else if (GreenAuditResult.REVIEW.equals(textSuggestion) || GreenAuditResult.REVIEW.equals(imageSuggestion)) {
|
||||
// 待审核,需人工审核
|
||||
post.setIsPublic(1);
|
||||
} else {
|
||||
@@ -233,10 +281,10 @@ public class PostServiceImpl implements PostService {
|
||||
}
|
||||
} else {
|
||||
// 只有文本内容的情况
|
||||
if ("block".equals(textSuggestion)) {
|
||||
if (GreenAuditResult.BLOCK.equals(textSuggestion)) {
|
||||
// 审核未通过,允许用户修改
|
||||
post.setIsPublic(2);
|
||||
} else if ("review".equals(textSuggestion)) {
|
||||
} else if (GreenAuditResult.REVIEW.equals(textSuggestion)) {
|
||||
// 待审核,需人工审核
|
||||
post.setIsPublic(1);
|
||||
} else {
|
||||
|
||||
339
src/main/java/com/bao/dating/service/impl/UserServiceImpl.java
Normal file
339
src/main/java/com/bao/dating/service/impl/UserServiceImpl.java
Normal file
@@ -0,0 +1,339 @@
|
||||
package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.common.aliyun.AliOssUtil;
|
||||
import com.bao.dating.common.aliyun.GreenImageScan;
|
||||
import com.bao.dating.common.aliyun.GreenTextScan;
|
||||
import com.bao.dating.common.result.AliOssResult;
|
||||
import com.bao.dating.common.result.GreenAuditResult;
|
||||
import com.bao.dating.context.UserContext;
|
||||
import com.bao.dating.mapper.UserMapper;
|
||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
||||
import com.bao.dating.pojo.dto.UserLoginDTO;
|
||||
import com.bao.dating.pojo.entity.User;
|
||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||
import com.bao.dating.service.UserService;
|
||||
import com.bao.dating.service.VerificationCodeService;
|
||||
import com.bao.dating.util.FileUtil;
|
||||
import com.bao.dating.util.JwtUtil;
|
||||
import com.bao.dating.util.MD5Util;
|
||||
import org.springframework.beans.BeanUtils;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.UUID;
|
||||
|
||||
/**
|
||||
* 用户服务实现类
|
||||
*
|
||||
* @author KilLze
|
||||
*/
|
||||
@Service
|
||||
public class UserServiceImpl implements UserService {
|
||||
|
||||
@Autowired
|
||||
private AliOssUtil ossUtil;
|
||||
|
||||
@Autowired
|
||||
private GreenTextScan greenTextScan;
|
||||
|
||||
@Autowired
|
||||
private GreenImageScan greenImageScan;
|
||||
|
||||
@Autowired
|
||||
private UserMapper userMapper;
|
||||
|
||||
@Autowired
|
||||
private VerificationCodeService verificationCodeService;
|
||||
|
||||
/**
|
||||
* 用户登录
|
||||
*
|
||||
* @param userLoginDTO 登录参数
|
||||
* @return 登录信息
|
||||
*/
|
||||
@Override
|
||||
public UserLoginVO userLogin(UserLoginDTO userLoginDTO) {
|
||||
// 参数校验
|
||||
if (userLoginDTO == null || userLoginDTO.getUsername() == null || userLoginDTO.getPassword() == null) {
|
||||
throw new RuntimeException("用户名或密码不能为空");
|
||||
}
|
||||
// 查询用户
|
||||
User user = userMapper.getByUsername(userLoginDTO.getUsername());
|
||||
if (user == null) {
|
||||
throw new RuntimeException("用户不存在");
|
||||
}
|
||||
// 密码校验
|
||||
boolean match = MD5Util.verifyWithSalt(
|
||||
userLoginDTO.getPassword(),
|
||||
user.getSalt(),
|
||||
user.getPasswordHash()
|
||||
);
|
||||
if (!match) {
|
||||
throw new RuntimeException("密码错误");
|
||||
}
|
||||
// 生成token
|
||||
String token = JwtUtil.generateToken(String.valueOf(user.getUserId()));
|
||||
// 封装返回
|
||||
UserLoginVO userLoginVO = new UserLoginVO();
|
||||
userLoginVO.setUserId(user.getUserId());
|
||||
userLoginVO.setNickname(user.getNickname());
|
||||
userLoginVO.setToken(token);
|
||||
return userLoginVO;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户信息
|
||||
*
|
||||
* @param userId 用户ID
|
||||
* @return 用户信息
|
||||
*/
|
||||
@Override
|
||||
public UserInfoVO getUserInfo(Long userId) {
|
||||
User user = userMapper.selectByUserId(userId);
|
||||
if (user == null) {
|
||||
throw new RuntimeException("用户不存在");
|
||||
}
|
||||
UserInfoVO userInfoVO = new UserInfoVO();
|
||||
BeanUtils.copyProperties(user, userInfoVO);
|
||||
return userInfoVO;
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传头像接口
|
||||
*
|
||||
* @param file 头像文件
|
||||
* @return 上传后的文件URL
|
||||
*/
|
||||
@Override
|
||||
public String uploadAvatar(MultipartFile file) {
|
||||
// 参数校验
|
||||
if (file == null || file.isEmpty()) {
|
||||
throw new RuntimeException("图片不存在");
|
||||
}
|
||||
|
||||
String originalFilename = file.getOriginalFilename();
|
||||
if (originalFilename == null) {
|
||||
throw new RuntimeException("文件名非法");
|
||||
}
|
||||
|
||||
String fileType = FileUtil.getFileType(originalFilename);
|
||||
if (!AliOssResult.IMAGE.equals(fileType)) {
|
||||
throw new RuntimeException("仅支持图片上传");
|
||||
}
|
||||
|
||||
//生成 OSS 路径
|
||||
String extension = FileUtil.getFileExtension(originalFilename);
|
||||
String fileName = UUID.randomUUID().toString().replace("-", "") + "." + extension;
|
||||
Long userId = UserContext.getUserId();
|
||||
String objectKey = "user/" + userId + "/avatar/" + fileName;
|
||||
|
||||
try {
|
||||
byte[] fileBytes = file.getBytes();
|
||||
String ossUrl = ossUtil.upload(fileBytes, objectKey);
|
||||
|
||||
if (ossUrl == null || ossUrl.isEmpty()) {
|
||||
throw new RuntimeException("图片上传失败");
|
||||
}
|
||||
|
||||
return ossUrl;
|
||||
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("上传图片失败", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传背景图片
|
||||
*
|
||||
* @param file 背景图片文件
|
||||
* @return 上传后的文件URL
|
||||
*/
|
||||
@Override
|
||||
public String uploadBackground(MultipartFile file) {
|
||||
// 参数校验
|
||||
if (file == null || file.isEmpty()) {
|
||||
throw new RuntimeException("图片不存在");
|
||||
}
|
||||
|
||||
String originalFilename = file.getOriginalFilename();
|
||||
if (originalFilename == null) {
|
||||
throw new RuntimeException("文件名非法");
|
||||
}
|
||||
|
||||
String fileType = FileUtil.getFileType(originalFilename);
|
||||
if (!AliOssResult.IMAGE.equals(fileType)) {
|
||||
throw new RuntimeException("仅支持图片上传");
|
||||
}
|
||||
|
||||
//生成 OSS 路径
|
||||
String extension = FileUtil.getFileExtension(originalFilename);
|
||||
String fileName = UUID.randomUUID().toString().replace("-", "") + "." + extension;
|
||||
Long userId = UserContext.getUserId();
|
||||
String objectKey = "user/" + userId + "/background/" + fileName;
|
||||
|
||||
try {
|
||||
byte[] fileBytes = file.getBytes();
|
||||
String ossUrl = ossUtil.upload(fileBytes, objectKey);
|
||||
|
||||
if (ossUrl == null || ossUrl.isEmpty()) {
|
||||
throw new RuntimeException("图片上传失败");
|
||||
}
|
||||
|
||||
return ossUrl;
|
||||
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("上传图片失败", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新用户信息
|
||||
*
|
||||
* @param userInfoUpdateDTO 用户信息更新参数
|
||||
*/
|
||||
@Override
|
||||
public UserInfoVO updateUserInfo(UserInfoUpdateDTO userInfoUpdateDTO) {
|
||||
Long userId = userInfoUpdateDTO.getUserId();
|
||||
User user = userMapper.selectByUserId(userId);
|
||||
if (user == null) {
|
||||
throw new RuntimeException("用户不存在");
|
||||
}
|
||||
|
||||
// 将需要审核的内容合并成一个文本,用于减少调用次数
|
||||
StringBuilder textBuilder = new StringBuilder();
|
||||
|
||||
if (userInfoUpdateDTO.getNickname() != null && !userInfoUpdateDTO.getNickname().isEmpty()) {
|
||||
textBuilder.append(userInfoUpdateDTO.getNickname()).append(" ");
|
||||
}
|
||||
if (userInfoUpdateDTO.getHobbies() != null && !userInfoUpdateDTO.getHobbies().isEmpty()) {
|
||||
// 将爱好列表转换为字符串,用空格分隔
|
||||
String hobbiesStr = String.join(" ", userInfoUpdateDTO.getHobbies());
|
||||
textBuilder.append(hobbiesStr).append(" ");
|
||||
}
|
||||
if (userInfoUpdateDTO.getSignature() != null && !userInfoUpdateDTO.getSignature().isEmpty()) {
|
||||
textBuilder.append(userInfoUpdateDTO.getSignature()).append(" ");
|
||||
}
|
||||
// 文本审核
|
||||
if (textBuilder.length() > 0) {
|
||||
Map textResult;
|
||||
try {
|
||||
textResult = greenTextScan.greeTextScan(textBuilder.toString());
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("用户信息文本审核失败");
|
||||
}
|
||||
|
||||
String suggestion = (String) textResult.get("suggestion");
|
||||
|
||||
if (GreenAuditResult.BLOCK.equals(suggestion)) {
|
||||
throw new RuntimeException("用户信息包含违规内容,修改失败");
|
||||
}
|
||||
if (GreenAuditResult.REVIEW.equals(suggestion)) {
|
||||
throw new RuntimeException("用户信息需要人工审核,暂无法修改");
|
||||
}
|
||||
}
|
||||
|
||||
// 图片审核
|
||||
List<String> imageKeys = new ArrayList<>();
|
||||
if (userInfoUpdateDTO.getAvatarUrl() != null && !userInfoUpdateDTO.getAvatarUrl().isEmpty()) {
|
||||
imageKeys.add(userInfoUpdateDTO.getAvatarUrl());
|
||||
}
|
||||
if (userInfoUpdateDTO.getBackgroundUrl() != null && !userInfoUpdateDTO.getBackgroundUrl().isEmpty()) {
|
||||
imageKeys.add(userInfoUpdateDTO.getBackgroundUrl());
|
||||
}
|
||||
if (!imageKeys.isEmpty()) {
|
||||
Map imageResult;
|
||||
try {
|
||||
imageResult = greenImageScan.imageScan(imageKeys);
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("用户图片审核失败");
|
||||
}
|
||||
|
||||
String suggestion = (String) imageResult.get("suggestion");
|
||||
|
||||
if (GreenAuditResult.BLOCK.equals(suggestion)) {
|
||||
throw new RuntimeException("头像或背景图不合规,修改失败");
|
||||
}
|
||||
if (GreenAuditResult.REVIEW.equals(suggestion)) {
|
||||
throw new RuntimeException("头像或背景图需要人工审核,暂无法修改");
|
||||
}
|
||||
}
|
||||
// 默认昵称兜底
|
||||
if (userInfoUpdateDTO.getNickname() == null || userInfoUpdateDTO.getNickname().trim().isEmpty()) {
|
||||
userInfoUpdateDTO.setNickname(user.getUserName());
|
||||
}
|
||||
userInfoUpdateDTO.setUpdatedAt(LocalDateTime.now());
|
||||
|
||||
// 更新数据库
|
||||
userMapper.updateUserInfoByUserId(userInfoUpdateDTO);
|
||||
|
||||
// 封装返回结果
|
||||
User updatedUser = userMapper.selectByUserId(userInfoUpdateDTO.getUserId());
|
||||
UserInfoVO userInfoVO = new UserInfoVO();
|
||||
BeanUtils.copyProperties(updatedUser, userInfoVO);
|
||||
return userInfoVO;
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询用户
|
||||
* @param userName 用户民称
|
||||
* @return
|
||||
*/
|
||||
@Override
|
||||
public Boolean registerUser(String userName,String userPassword) {
|
||||
//校验参数是否为空
|
||||
if (userName.isEmpty() || userPassword.isEmpty()){
|
||||
return false;
|
||||
}
|
||||
//产看数据库是否存在已注册用户
|
||||
User user = userMapper.getByUsername(userName);
|
||||
if (user != null){
|
||||
return false;
|
||||
}
|
||||
//将用户数据存入苏数据库
|
||||
String salt = "lyy123";
|
||||
String passwordHash = MD5Util.encryptWithSalt(userPassword, salt);
|
||||
//查询最大用户id
|
||||
Long maxId = userMapper.selectMaxId();
|
||||
User saveUser = new User();
|
||||
saveUser.setUserId(maxId+1);
|
||||
saveUser.setUserName(userName);
|
||||
saveUser.setPasswordHash(passwordHash);
|
||||
saveUser.setSalt(salt);
|
||||
saveUser.setUpdatedAt(LocalDateTime.now());
|
||||
saveUser.setCreatedAt(LocalDateTime.now());
|
||||
int count = userMapper.saveUser(saveUser);
|
||||
return count > 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* 邮箱登录
|
||||
* @param email 邮箱
|
||||
* @param code 验证码
|
||||
* @return 脱敏用户信息
|
||||
*/
|
||||
@Override
|
||||
public UserLoginVO emailLogin(String email, String code) {
|
||||
User user = userMapper.selectByUserEmailUser(email);
|
||||
if (user == null)
|
||||
return null;
|
||||
boolean flag = verificationCodeService.verifyEmailCode(email, code);
|
||||
if (!flag)
|
||||
return null;
|
||||
// 生成token
|
||||
String token = JwtUtil.generateToken(String.valueOf(user.getUserId()));
|
||||
UserLoginVO userLoginVO = new UserLoginVO();
|
||||
userLoginVO.setUserId(user.getUserId());
|
||||
userLoginVO.setNickname(user.getNickname());
|
||||
userLoginVO.setToken(token);
|
||||
return userLoginVO;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,122 @@
|
||||
package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.service.VerificationCodeService;
|
||||
import com.bao.dating.util.EmailUtil;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.data.redis.core.StringRedisTemplate;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.Random;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* 验证码服务实现类
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@Service
|
||||
public class VerificationCodeServiceImpl implements VerificationCodeService {
|
||||
|
||||
@Autowired
|
||||
private EmailUtil emailUtil;
|
||||
|
||||
@Autowired
|
||||
private StringRedisTemplate redisTemplate;
|
||||
|
||||
/**
|
||||
* Redis中验证码的key前缀
|
||||
*/
|
||||
private static final String CODE_KEY_PREFIX = "email:code:";
|
||||
|
||||
/**
|
||||
* 验证码过期时间(分钟)
|
||||
*/
|
||||
private static final long CODE_EXPIRE_MINUTES = 10;
|
||||
|
||||
/**
|
||||
* 验证码长度
|
||||
*/
|
||||
private static final int CODE_LENGTH = 6;
|
||||
|
||||
/**
|
||||
* 发送邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
@Override
|
||||
public boolean sendEmailCode(String email) {
|
||||
try {
|
||||
// 生成验证码
|
||||
String code = generateCode(CODE_LENGTH);
|
||||
|
||||
// 存储到Redis,设置过期时间
|
||||
String key = CODE_KEY_PREFIX + email;
|
||||
redisTemplate.opsForValue().set(key, code, CODE_EXPIRE_MINUTES, TimeUnit.MINUTES);
|
||||
|
||||
// 发送邮件
|
||||
boolean sendResult = emailUtil.sendVerificationCode(email, code);
|
||||
|
||||
if (sendResult) {
|
||||
log.info("邮箱验证码发送成功,邮箱:{},验证码:{}", email, code);
|
||||
return true;
|
||||
} else {
|
||||
// 如果发送失败,删除Redis中的验证码
|
||||
redisTemplate.delete(key);
|
||||
log.error("邮箱验证码发送失败,邮箱:{}", email);
|
||||
return false;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.error("发送邮箱验证码异常,邮箱:{},异常信息:{}", email, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 是否验证成功
|
||||
*/
|
||||
@Override
|
||||
public boolean verifyEmailCode(String email, String code) {
|
||||
try {
|
||||
String key = CODE_KEY_PREFIX + email;
|
||||
String storedCode = redisTemplate.opsForValue().get(key);
|
||||
|
||||
if (storedCode == null) {
|
||||
log.warn("验证码不存在或已过期,邮箱:{}", email);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (storedCode.equals(code)) {
|
||||
// 验证成功后,删除验证码(防止重复使用)
|
||||
redisTemplate.delete(key);
|
||||
log.info("邮箱验证码验证成功,邮箱:{}", email);
|
||||
return true;
|
||||
} else {
|
||||
log.warn("邮箱验证码错误,邮箱:{},输入的验证码:{},正确的验证码:{}", email, code, storedCode);
|
||||
return false;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.error("验证邮箱验证码异常,邮箱:{},异常信息:{}", email, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成验证码
|
||||
* @param length 验证码长度
|
||||
* @return 验证码字符串
|
||||
*/
|
||||
@Override
|
||||
public String generateCode(int length) {
|
||||
Random random = new Random();
|
||||
StringBuilder code = new StringBuilder();
|
||||
for (int i = 0; i < length; i++) {
|
||||
code.append(random.nextInt(10)); // 生成0-9的随机数字
|
||||
}
|
||||
return code.toString();
|
||||
}
|
||||
}
|
||||
|
||||
178
src/main/java/com/bao/dating/util/EmailUtil.java
Normal file
178
src/main/java/com/bao/dating/util/EmailUtil.java
Normal file
@@ -0,0 +1,178 @@
|
||||
package com.bao.dating.util;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.mail.SimpleMailMessage;
|
||||
import org.springframework.mail.javamail.JavaMailSender;
|
||||
import org.springframework.mail.javamail.MimeMessageHelper;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import javax.mail.MessagingException;
|
||||
import javax.mail.internet.MimeMessage;
|
||||
import java.io.File;
|
||||
|
||||
/**
|
||||
* 邮箱工具类
|
||||
* 提供邮件发送功能(支持普通文本邮件和HTML邮件)
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@Component
|
||||
public class EmailUtil {
|
||||
|
||||
/**
|
||||
* 邮件发送器(由Spring自动注入)
|
||||
*/
|
||||
private final JavaMailSender mailSender;
|
||||
|
||||
/**
|
||||
* 发件人邮箱地址(从配置中读取)
|
||||
*/
|
||||
private final String from;
|
||||
|
||||
/**
|
||||
* 构造函数,用于注入JavaMailSender和配置
|
||||
* @param mailSender 邮件发送器
|
||||
* @param from 发件人邮箱地址
|
||||
*/
|
||||
public EmailUtil(JavaMailSender mailSender, @Value("${spring.mail.username}") String from) {
|
||||
this.mailSender = mailSender;
|
||||
this.from = from;
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送简单文本邮件
|
||||
* @param to 收件人邮箱地址
|
||||
* @param subject 邮件主题
|
||||
* @param text 邮件内容
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendSimpleMail(String to, String subject, String text) {
|
||||
try {
|
||||
SimpleMailMessage message = new SimpleMailMessage();
|
||||
message.setFrom(from);
|
||||
message.setTo(to);
|
||||
message.setSubject(subject);
|
||||
message.setText(text);
|
||||
mailSender.send(message);
|
||||
log.info("简单邮件发送成功,收件人:{},主题:{}", to, subject);
|
||||
return true;
|
||||
} catch (Exception e) {
|
||||
log.error("简单邮件发送失败,收件人:{},异常信息:{}", to, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送HTML格式邮件
|
||||
* @param to 收件人邮箱地址
|
||||
* @param subject 邮件主题
|
||||
* @param htmlContent HTML内容
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendHtmlMail(String to, String subject, String htmlContent) {
|
||||
try {
|
||||
MimeMessage message = mailSender.createMimeMessage();
|
||||
MimeMessageHelper helper = new MimeMessageHelper(message, true, "UTF-8");
|
||||
helper.setFrom(from);
|
||||
helper.setTo(to);
|
||||
helper.setSubject(subject);
|
||||
helper.setText(htmlContent, true);
|
||||
mailSender.send(message);
|
||||
log.info("HTML邮件发送成功,收件人:{},主题:{}", to, subject);
|
||||
return true;
|
||||
} catch (MessagingException e) {
|
||||
log.error("HTML邮件发送失败,收件人:{},异常信息:{}", to, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送带附件的邮件
|
||||
* @param to 收件人邮箱地址
|
||||
* @param subject 邮件主题
|
||||
* @param text 邮件内容
|
||||
* @param attachmentPath 附件文件路径
|
||||
* @param attachmentName 附件显示名称
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendMailWithAttachment(String to, String subject, String text,
|
||||
String attachmentPath, String attachmentName) {
|
||||
try {
|
||||
MimeMessage message = mailSender.createMimeMessage();
|
||||
MimeMessageHelper helper = new MimeMessageHelper(message, true, "UTF-8");
|
||||
helper.setFrom(from);
|
||||
helper.setTo(to);
|
||||
helper.setSubject(subject);
|
||||
helper.setText(text);
|
||||
|
||||
File file = new File(attachmentPath);
|
||||
if (file.exists()) {
|
||||
helper.addAttachment(attachmentName, file);
|
||||
} else {
|
||||
log.warn("附件文件不存在:{}", attachmentPath);
|
||||
return false;
|
||||
}
|
||||
|
||||
mailSender.send(message);
|
||||
log.info("带附件邮件发送成功,收件人:{},主题:{},附件:{}", to, subject, attachmentName);
|
||||
return true;
|
||||
} catch (MessagingException e) {
|
||||
log.error("带附件邮件发送失败,收件人:{},异常信息:{}", to, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送验证码邮件
|
||||
* @param to 收件人邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendVerificationCode(String to, String code) {
|
||||
String subject = "验证码";
|
||||
String htmlContent = String.format(
|
||||
"<html><body>" +
|
||||
"<h2>您的验证码</h2>" +
|
||||
"<p>验证码:<strong style='color: #1890ff; font-size: 24px;'>%s</strong></p>" +
|
||||
"<p>验证码有效期为10分钟,请勿泄露给他人。</p>" +
|
||||
"</body></html>", code);
|
||||
return sendHtmlMail(to, subject, htmlContent);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送验证码邮件(自定义主题)
|
||||
* @param to 收件人邮箱地址
|
||||
* @param code 验证码
|
||||
* @param subject 邮件主题
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
public boolean sendVerificationCode(String to, String code, String subject) {
|
||||
String htmlContent = String.format(
|
||||
"<html><body>" +
|
||||
"<h2>您的验证码</h2>" +
|
||||
"<p>验证码:<strong style='color: #1890ff; font-size: 24px;'>%s</strong></p>" +
|
||||
"<p>验证码有效期为10分钟,请勿泄露给他人。</p>" +
|
||||
"</body></html>", code);
|
||||
return sendHtmlMail(to, subject, htmlContent);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量发送邮件
|
||||
* @param toList 收件人邮箱地址数组
|
||||
* @param subject 邮件主题
|
||||
* @param text 邮件内容
|
||||
* @return 成功发送的数量
|
||||
*/
|
||||
public int sendBatchMail(String[] toList, String subject, String text) {
|
||||
int successCount = 0;
|
||||
for (String to : toList) {
|
||||
if (sendSimpleMail(to, subject, text)) {
|
||||
successCount++;
|
||||
}
|
||||
}
|
||||
log.info("批量邮件发送完成,总数:{},成功:{}", toList.length, successCount);
|
||||
return successCount;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
package com.bao.dating.util;
|
||||
|
||||
import com.bao.dating.common.result.FileResult;
|
||||
|
||||
/**
|
||||
* 文件工具类
|
||||
* @author KilLze
|
||||
@@ -13,9 +15,9 @@ public class FileUtil {
|
||||
public static String getFileType(String fileUrl) {
|
||||
String extension = getFileExtension(fileUrl);
|
||||
|
||||
if (extension.equals("jpg") || extension.equals("jpeg") || extension.equals("png") || extension.equals("gif")) {
|
||||
if (FileResult.JPG.equals(extension) || FileResult.JPEG.equals(extension) || FileResult.PNG.equals(extension) || FileResult.GIF.equals(extension)) {
|
||||
return "image";
|
||||
} else if (extension.equals("mp4") || extension.equals("avi") || extension.equals("mov")) {
|
||||
} else if (FileResult.MP4.equals(extension) || FileResult.AVI.equals(extension) || FileResult.MOV.equals(extension)) {
|
||||
return "video";
|
||||
}
|
||||
return "unknown";
|
||||
|
||||
204
src/main/java/com/bao/dating/util/JwtUtil.java
Normal file
204
src/main/java/com/bao/dating/util/JwtUtil.java
Normal file
@@ -0,0 +1,204 @@
|
||||
package com.bao.dating.util;
|
||||
|
||||
import io.jsonwebtoken.Claims;
|
||||
import io.jsonwebtoken.Jwts;
|
||||
import io.jsonwebtoken.SignatureAlgorithm;
|
||||
import io.jsonwebtoken.security.Keys;
|
||||
|
||||
import javax.crypto.SecretKey;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.Date;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* JWT工具类
|
||||
* 提供JWT Token的生成、解析和验证功能
|
||||
* @author KilLze
|
||||
*/
|
||||
public class JwtUtil {
|
||||
|
||||
/**
|
||||
* 默认密钥(建议从配置文件读取)
|
||||
*/
|
||||
private static final String DEFAULT_SECRET = "dating-application-secret-key-for-jwt-token-generation-2025";
|
||||
|
||||
/**
|
||||
* 默认过期时间(毫秒)- 7天
|
||||
*/
|
||||
private static final long DEFAULT_EXPIRATION = 7 * 24 * 60 * 60 * 1000L;
|
||||
|
||||
/**
|
||||
* 生成JWT Token
|
||||
* @param subject 主题(通常是用户ID或用户名)
|
||||
* @return JWT Token字符串
|
||||
*/
|
||||
public static String generateToken(String subject) {
|
||||
return generateToken(subject, DEFAULT_EXPIRATION);
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成JWT Token(自定义过期时间)
|
||||
* @param subject 主题(通常是用户ID或用户名)
|
||||
* @param expiration 过期时间(毫秒)
|
||||
* @return JWT Token字符串
|
||||
*/
|
||||
public static String generateToken(String subject, long expiration) {
|
||||
return generateToken(subject, expiration, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成JWT Token(带自定义claims)
|
||||
* @param subject 主题(通常是用户ID或用户名)
|
||||
* @param expiration 过期时间(毫秒)
|
||||
* @param claims 自定义claims
|
||||
* @return JWT Token字符串
|
||||
*/
|
||||
public static String generateToken(String subject, long expiration, Map<String, Object> claims) {
|
||||
Date now = new Date();
|
||||
Date expiryDate = new Date(now.getTime() + expiration);
|
||||
|
||||
SecretKey key = getSecretKey();
|
||||
|
||||
if (claims == null) {
|
||||
claims = new HashMap<>();
|
||||
}
|
||||
|
||||
return Jwts.builder()
|
||||
.setClaims(claims)
|
||||
.setSubject(subject)
|
||||
.setIssuedAt(now)
|
||||
.setExpiration(expiryDate)
|
||||
.signWith(key, SignatureAlgorithm.HS256)
|
||||
.compact();
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成JWT Token(使用自定义密钥)
|
||||
* @param subject 主题
|
||||
* @param expiration 过期时间(毫秒)
|
||||
* @param secret 自定义密钥
|
||||
* @return JWT Token字符串
|
||||
*/
|
||||
public static String generateTokenWithSecret(String subject, long expiration, String secret) {
|
||||
Date now = new Date();
|
||||
Date expiryDate = new Date(now.getTime() + expiration);
|
||||
|
||||
SecretKey key = getSecretKey(secret);
|
||||
|
||||
return Jwts.builder()
|
||||
.setSubject(subject)
|
||||
.setIssuedAt(now)
|
||||
.setExpiration(expiryDate)
|
||||
.signWith(key, SignatureAlgorithm.HS256)
|
||||
.compact();
|
||||
}
|
||||
|
||||
/**
|
||||
* 从Token中获取Claims
|
||||
* @param token JWT Token
|
||||
* @return Claims对象
|
||||
*/
|
||||
public static Claims getClaimsFromToken(String token) {
|
||||
return getClaimsFromToken(token, DEFAULT_SECRET);
|
||||
}
|
||||
|
||||
/**
|
||||
* 从Token中获取Claims(使用自定义密钥)
|
||||
* @param token JWT Token
|
||||
* @param secret 密钥
|
||||
* @return Claims对象
|
||||
*/
|
||||
public static Claims getClaimsFromToken(String token, String secret) {
|
||||
try {
|
||||
SecretKey key = getSecretKey(secret);
|
||||
return Jwts.parserBuilder()
|
||||
.setSigningKey(key)
|
||||
.build()
|
||||
.parseClaimsJws(token)
|
||||
.getBody();
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("解析JWT Token失败", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 从Token中获取主题(通常是用户ID)
|
||||
* @param token JWT Token
|
||||
* @return 主题字符串
|
||||
*/
|
||||
public static String getSubjectFromToken(String token) {
|
||||
Claims claims = getClaimsFromToken(token);
|
||||
return claims.getSubject();
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证Token是否有效
|
||||
* @param token JWT Token
|
||||
* @return 是否有效
|
||||
*/
|
||||
public static boolean validateToken(String token) {
|
||||
return validateToken(token, DEFAULT_SECRET);
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证Token是否有效(使用自定义密钥)
|
||||
* @param token JWT Token
|
||||
* @param secret 密钥
|
||||
* @return 是否有效
|
||||
*/
|
||||
public static boolean validateToken(String token, String secret) {
|
||||
try {
|
||||
Claims claims = getClaimsFromToken(token, secret);
|
||||
return !isTokenExpired(claims);
|
||||
} catch (Exception e) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查Token是否过期
|
||||
* @param claims Claims对象
|
||||
* @return 是否过期
|
||||
*/
|
||||
private static boolean isTokenExpired(Claims claims) {
|
||||
Date expiration = claims.getExpiration();
|
||||
return expiration.before(new Date());
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Token的过期时间
|
||||
* @param token JWT Token
|
||||
* @return 过期时间
|
||||
*/
|
||||
public static Date getExpirationDateFromToken(String token) {
|
||||
Claims claims = getClaimsFromToken(token);
|
||||
return claims.getExpiration();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取默认密钥的SecretKey对象
|
||||
* @return SecretKey对象
|
||||
*/
|
||||
private static SecretKey getSecretKey() {
|
||||
return getSecretKey(DEFAULT_SECRET);
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据字符串密钥生成SecretKey对象
|
||||
* @param secret 密钥字符串
|
||||
* @return SecretKey对象
|
||||
*/
|
||||
private static SecretKey getSecretKey(String secret) {
|
||||
// 确保密钥长度至少为256位(32字节)以支持HS256算法
|
||||
byte[] keyBytes = secret.getBytes(StandardCharsets.UTF_8);
|
||||
if (keyBytes.length < 32) {
|
||||
// 如果密钥太短,进行填充
|
||||
byte[] paddedKey = new byte[32];
|
||||
System.arraycopy(keyBytes, 0, paddedKey, 0, Math.min(keyBytes.length, 32));
|
||||
return Keys.hmacShaKeyFor(paddedKey);
|
||||
}
|
||||
return Keys.hmacShaKeyFor(keyBytes);
|
||||
}
|
||||
}
|
||||
|
||||
93
src/main/java/com/bao/dating/util/MD5Util.java
Normal file
93
src/main/java/com/bao/dating/util/MD5Util.java
Normal file
@@ -0,0 +1,93 @@
|
||||
package com.bao.dating.util;
|
||||
|
||||
import java.security.MessageDigest;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
|
||||
/**
|
||||
* MD5工具类
|
||||
* 提供MD5加密功能
|
||||
* @author KilLze
|
||||
*/
|
||||
public class MD5Util {
|
||||
|
||||
/**
|
||||
* 对字符串进行MD5加密
|
||||
* @param input 待加密的字符串
|
||||
* @return MD5加密后的32位小写字符串
|
||||
*/
|
||||
public static String encrypt(String input) {
|
||||
if (input == null || input.isEmpty()) {
|
||||
return null;
|
||||
}
|
||||
|
||||
try {
|
||||
MessageDigest md = MessageDigest.getInstance("MD5");
|
||||
byte[] digest = md.digest(input.getBytes());
|
||||
return bytesToHex(digest);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new RuntimeException("MD5算法不可用", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 对字符串进行MD5加密(带盐值)
|
||||
* @param input 待加密的字符串
|
||||
* @param salt 盐值
|
||||
* @return MD5加密后的32位小写字符串
|
||||
*/
|
||||
public static String encryptWithSalt(String input, String salt) {
|
||||
if (input == null || input.isEmpty()) {
|
||||
return null;
|
||||
}
|
||||
if (salt == null) {
|
||||
salt = "";
|
||||
}
|
||||
return encrypt(input + salt);
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证字符串与MD5值是否匹配
|
||||
* @param input 原始字符串
|
||||
* @param md5Hash MD5哈希值
|
||||
* @return 是否匹配
|
||||
*/
|
||||
public static boolean verify(String input, String md5Hash) {
|
||||
if (input == null || md5Hash == null) {
|
||||
return false;
|
||||
}
|
||||
return encrypt(input).equalsIgnoreCase(md5Hash);
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证字符串与MD5值是否匹配(带盐值)
|
||||
* @param input 原始字符串
|
||||
* @param salt 盐值
|
||||
* @param md5Hash MD5哈希值
|
||||
* @return 是否匹配
|
||||
*/
|
||||
public static boolean verifyWithSalt(String input, String salt, String md5Hash) {
|
||||
if (input == null || md5Hash == null) {
|
||||
return false;
|
||||
}
|
||||
if (salt == null) {
|
||||
salt = "";
|
||||
}
|
||||
return encryptWithSalt(input, salt).equalsIgnoreCase(md5Hash);
|
||||
}
|
||||
|
||||
/**
|
||||
* 将字节数组转换为十六进制字符串
|
||||
* @param bytes 字节数组
|
||||
* @return 十六进制字符串
|
||||
*/
|
||||
private static String bytesToHex(byte[] bytes) {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (byte b : bytes) {
|
||||
sb.append(String.format("%02x", b));
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -7,6 +7,38 @@ spring:
|
||||
username: root
|
||||
password: JoyeeServe2025
|
||||
driver-class-name: com.mysql.cj.jdbc.Driver
|
||||
redis:
|
||||
host: 110.42.41.177
|
||||
port: 6389
|
||||
password: ""
|
||||
database: 0
|
||||
timeout: 10000
|
||||
# 连接池配置(lettuce是Spring Boot默认Redis客户端,性能更优)
|
||||
lettuce:
|
||||
pool:
|
||||
max-active: 8 # 连接池最大连接数(默认8,可根据业务并发调整)
|
||||
max-wait: -1 # 连接池最大阻塞等待时间(毫秒,-1表示无限制)
|
||||
max-idle: 8 # 连接池最大空闲连接数(默认8)
|
||||
min-idle: 1 # 连接池最小空闲连接数(默认0,建议设置1-4,提高连接复用率)
|
||||
# 邮箱SMTP配置
|
||||
mail:
|
||||
host: smtp.163.com # QQ邮箱SMTP服务器地址
|
||||
port: 465 # SMTP端口
|
||||
username: 19271189822@163.com # 发件人邮箱
|
||||
password: CAwXh39PXajy3fyH # 邮箱授权码
|
||||
default-encoding: UTF-8
|
||||
properties:
|
||||
mail:
|
||||
smtp:
|
||||
auth: true
|
||||
starttls:
|
||||
enable: true
|
||||
required: true
|
||||
ssl:
|
||||
enable: true # 使用587端口时设为false,使用465端口时设为true
|
||||
connectiontimeout: 10000 # 连接超时时间(毫秒)
|
||||
timeout: 10000 # 读取超时时间(毫秒)
|
||||
writetimeout: 10000 # 写入超时时间(毫秒)
|
||||
|
||||
# MyBatis 配置
|
||||
mybatis:
|
||||
@@ -26,3 +58,18 @@ aliyun:
|
||||
accessKeyId: LTAI5t5vpcbCZwweNHEDDDaF
|
||||
secret: bBHBAPiCqGyVBHUv07348wsHXkKqrk
|
||||
scenes: antispam
|
||||
# 阿里云短信服务配置
|
||||
sms:
|
||||
access-key-id: LTAI5t5vpcbCZwweNHEDDDaF
|
||||
access-key-secret: bBHBAPiCqGyVBHUv07348wsHXkKqrk
|
||||
region-id: cn-hangzhou
|
||||
sign-name: 速通互联验证码
|
||||
template-code: 100001
|
||||
|
||||
# ip2location.io 相关配置
|
||||
ip2location:
|
||||
api:
|
||||
key: 95F4AB991174E296AFD5AD0EF927B2ED # ip2location.io API密钥
|
||||
url: https://api.ip2location.io/
|
||||
timeout: 5000 # 请求超时时间(毫秒)
|
||||
|
||||
|
||||
@@ -14,4 +14,8 @@
|
||||
<select id="selectUserIDByPostID" resultType="java.lang.Long">
|
||||
SELECT user_id FROM post_favorite WHERE post_id = #{postId}
|
||||
</select>
|
||||
<!-- 查询所有收藏动态 -->
|
||||
<select id="showAllFavorites" resultType="com.bao.dating.pojo.entity.Post">
|
||||
SELECT * FROM post WHERE post_id IN (SELECT post_id FROM post_favorite WHERE user_id = #{userid})
|
||||
</select>
|
||||
</mapper>
|
||||
@@ -29,8 +29,23 @@
|
||||
</insert>
|
||||
|
||||
<!--动态删除-->
|
||||
<delete id="deletePostById">
|
||||
DELETE FROM post WHERE post_id = #{postId}
|
||||
<delete id="deletePostByIds">
|
||||
DELETE FROM post WHERE post_id IN
|
||||
<foreach item="postId" index="index" collection="postIds" separator="," open="(" close=")">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
<!--删除收藏记录-->
|
||||
<delete id="1">
|
||||
DELETE FROM post_favorite WHERE post_id = #{postId}
|
||||
</delete>
|
||||
<!--删除点赞记录-->
|
||||
<delete id="2">
|
||||
DELETE FROM post_like WHERE post_id = #{postId}
|
||||
</delete>
|
||||
<!--动态评论删除-->
|
||||
<delete id="3">
|
||||
DELETE FROM comments WHERE post_id = #{postId}
|
||||
</delete>
|
||||
|
||||
<!--动态查询-->
|
||||
@@ -67,10 +82,10 @@
|
||||
<update id="updateById">
|
||||
UPDATE post
|
||||
<set>
|
||||
<if test="content != null">
|
||||
<if test="content != null and content != '' ">
|
||||
content = #{content},
|
||||
</if>
|
||||
<if test="tags != null">
|
||||
<if test="tags != null and tags != '' ">
|
||||
tags = #{tags, typeHandler=com.bao.dating.handler.ListToVarcharTypeHandler},
|
||||
</if>
|
||||
<if test="mediaOssKeys != null">
|
||||
|
||||
91
src/main/resources/com/bao/dating/mapper/UserMapper.xml
Normal file
91
src/main/resources/com/bao/dating/mapper/UserMapper.xml
Normal file
@@ -0,0 +1,91 @@
|
||||
<?xml version="1.0" encoding="UTF-8" ?>
|
||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
|
||||
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
||||
|
||||
<mapper namespace="com.bao.dating.mapper.UserMapper">
|
||||
<!-- 向数据库中添加用户 -->
|
||||
<insert id="saveUser">
|
||||
insert into user(user_id,user_name,password_hash,salt,created_at,updated_at) values (#{userId},#{userName},#{passwordHash},#{salt},#{createdAt},#{updatedAt})
|
||||
</insert>
|
||||
|
||||
<!--根据用户名查询用户-->
|
||||
<select id="getByUsername" resultType="com.bao.dating.pojo.entity.User">
|
||||
SELECT
|
||||
user_id,
|
||||
user_name,
|
||||
password_hash,
|
||||
salt,
|
||||
nickname
|
||||
FROM user WHERE user_name = #{userName}
|
||||
</select>
|
||||
|
||||
<!--根据用户id查询用户信息-->
|
||||
<resultMap id="UserResultMap" type="com.bao.dating.pojo.entity.User">
|
||||
<id property="userId" column="user_id"/>
|
||||
<result property="userName" column="user_name"/>
|
||||
<result property="nickname" column="nickname"/>
|
||||
<result property="avatarUrl" column="avatar_url"/>
|
||||
<result property="backgroundUrl" column="background_url"/>
|
||||
<result property="gender" column="gender"/>
|
||||
<result property="birthday" column="birthday"/>
|
||||
<result property="hobbies" column="hobbies" typeHandler="com.bao.dating.handler.ListToJsonTypeHandler"/>
|
||||
<result property="signature" column="signature"/>
|
||||
<result property="createdAt" column="created_at"/>
|
||||
<result property="updatedAt" column="updated_at"/>
|
||||
</resultMap>
|
||||
<select id="selectByUserId" resultMap="UserResultMap">
|
||||
SELECT
|
||||
user_id,
|
||||
user_name,
|
||||
nickname,
|
||||
avatar_url,
|
||||
background_url,
|
||||
gender,
|
||||
birthday,
|
||||
hobbies,
|
||||
signature,
|
||||
created_at,
|
||||
updated_at
|
||||
FROM user WHERE user_id = #{userId}
|
||||
</select>
|
||||
|
||||
<!-- 查询最大用户id -->
|
||||
<select id="selectMaxId" resultType="java.lang.Long">
|
||||
SELECT MAX(user_id) FROM user
|
||||
</select>
|
||||
|
||||
<!-- 根据邮箱查询用户信息 -->
|
||||
<select id="selectByUserEmailUser" resultType="com.bao.dating.pojo.entity.User">
|
||||
select * from user where user_email = #{userEmail}
|
||||
</select>
|
||||
|
||||
<!--根据ID更新动态-->
|
||||
<update id="updateUserInfoByUserId">
|
||||
UPDATE user
|
||||
<set>
|
||||
<if test="nickname != null">
|
||||
nickname = #{nickname},
|
||||
</if>
|
||||
<if test="avatarUrl != null">
|
||||
avatar_url = #{avatarUrl},
|
||||
</if>
|
||||
<if test="backgroundUrl != null">
|
||||
background_url = #{backgroundUrl},
|
||||
</if>
|
||||
<if test="gender != null">
|
||||
gender = #{gender},
|
||||
</if>
|
||||
<if test="birthday != null">
|
||||
birthday = #{birthday},
|
||||
</if>
|
||||
<if test="hobbies != null">
|
||||
hobbies = #{hobbies, typeHandler=com.bao.dating.handler.ListToJsonTypeHandler},
|
||||
</if>
|
||||
<if test="signature != null">
|
||||
signature = #{signature},
|
||||
</if>
|
||||
updated_at = #{updatedAt}
|
||||
</set>
|
||||
WHERE user_id = #{userId}
|
||||
</update>
|
||||
</mapper>
|
||||
304
src/test/java/com/bao/dating/common/aliyun/OssUploadTest.java
Normal file
304
src/test/java/com/bao/dating/common/aliyun/OssUploadTest.java
Normal file
@@ -0,0 +1,304 @@
|
||||
package com.bao.dating.common.aliyun;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.net.URL;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.UUID;
|
||||
|
||||
/**
|
||||
* OSS图片上传测试类
|
||||
* @author KilLze
|
||||
*/
|
||||
@SpringBootTest
|
||||
public class OssUploadTest {
|
||||
|
||||
@Autowired
|
||||
private AliOssUtil aliOssUtil;
|
||||
|
||||
/**
|
||||
* 测试上传本地图片文件
|
||||
*/
|
||||
@Test
|
||||
public void testUploadLocalImage() {
|
||||
try {
|
||||
// 本地图片文件路径(请修改为实际存在的图片路径)
|
||||
String localImagePath = "D:/image/image.jpg"; // 请修改为实际路径
|
||||
|
||||
File imageFile = new File(localImagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("图片文件不存在: " + localImagePath);
|
||||
System.out.println("请修改localImagePath为实际存在的图片路径");
|
||||
return;
|
||||
}
|
||||
|
||||
// 读取文件字节
|
||||
byte[] imageBytes = Files.readAllBytes(Paths.get(localImagePath));
|
||||
|
||||
// 生成OSS对象名称(使用时间戳和UUID确保唯一性)
|
||||
String objectName = generateObjectName("test", "jpg");
|
||||
|
||||
// 上传到OSS
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("========== 本地图片上传测试 ==========");
|
||||
System.out.println("本地文件路径: " + localImagePath);
|
||||
System.out.println("文件大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("OSS对象名称: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
System.out.println("=====================================");
|
||||
|
||||
} catch (IOException e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试从网络URL下载图片并上传到OSS
|
||||
*/
|
||||
@Test
|
||||
public void testUploadImageFromUrl() {
|
||||
try {
|
||||
// 网络图片URL(请修改为实际的图片URL)
|
||||
String imageUrl = "https://example.com/image.jpg"; // 请修改为实际URL
|
||||
|
||||
System.out.println("========== 网络图片上传测试 ==========");
|
||||
System.out.println("开始下载图片: " + imageUrl);
|
||||
|
||||
// 从URL下载图片
|
||||
URL url = new URL(imageUrl);
|
||||
try (InputStream inputStream = url.openStream()) {
|
||||
byte[] imageBytes = readAllBytes(inputStream);
|
||||
|
||||
// 从URL中提取文件扩展名
|
||||
String extension = getFileExtensionFromUrl(imageUrl);
|
||||
if (extension.isEmpty()) {
|
||||
extension = "jpg"; // 默认扩展名
|
||||
}
|
||||
|
||||
// 生成OSS对象名称
|
||||
String objectName = generateObjectName("download", extension);
|
||||
|
||||
// 上传到OSS
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("图片下载成功");
|
||||
System.out.println("文件大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("OSS对象名称: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
}
|
||||
|
||||
System.out.println("=====================================");
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试上传不同格式的图片
|
||||
*/
|
||||
@Test
|
||||
public void testUploadDifferentImageFormats() {
|
||||
String[] imagePaths = {
|
||||
"D:/test/image1.jpg", // 请修改为实际路径
|
||||
"D:/test/image2.png", // 请修改为实际路径
|
||||
"D:/test/image3.gif" // 请修改为实际路径
|
||||
};
|
||||
|
||||
System.out.println("========== 多格式图片上传测试 ==========");
|
||||
|
||||
for (String imagePath : imagePaths) {
|
||||
try {
|
||||
File imageFile = new File(imagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("跳过不存在的文件: " + imagePath);
|
||||
continue;
|
||||
}
|
||||
|
||||
// 读取文件
|
||||
byte[] imageBytes = Files.readAllBytes(Paths.get(imagePath));
|
||||
|
||||
// 获取文件扩展名
|
||||
String extension = getFileExtension(imagePath);
|
||||
|
||||
// 生成OSS对象名称
|
||||
String objectName = generateObjectName("format-test", extension);
|
||||
|
||||
// 上传
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("\n文件: " + imagePath);
|
||||
System.out.println("格式: " + extension);
|
||||
System.out.println("大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("URL: " + fileUrl);
|
||||
System.out.println("状态: 成功");
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("\n文件: " + imagePath);
|
||||
System.out.println("状态: 失败 - " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
System.out.println("\n=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试上传图片到指定目录
|
||||
*/
|
||||
@Test
|
||||
public void testUploadImageToDirectory() {
|
||||
try {
|
||||
// 本地图片文件路径
|
||||
String localImagePath = "D:/test/image.jpg"; // 请修改为实际路径
|
||||
|
||||
File imageFile = new File(localImagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("图片文件不存在: " + localImagePath);
|
||||
return;
|
||||
}
|
||||
|
||||
// 读取文件
|
||||
byte[] imageBytes = Files.readAllBytes(Paths.get(localImagePath));
|
||||
|
||||
// 上传到指定目录(例如:images/2025/12/24/xxx.jpg)
|
||||
String directory = "images/" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd"));
|
||||
String fileName = UUID.randomUUID().toString() + ".jpg";
|
||||
String objectName = directory + "/" + fileName;
|
||||
|
||||
// 上传
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("========== 目录上传测试 ==========");
|
||||
System.out.println("目录: " + directory);
|
||||
System.out.println("文件名: " + fileName);
|
||||
System.out.println("完整路径: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
System.out.println("=================================");
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试使用FileInputStream上传
|
||||
*/
|
||||
@Test
|
||||
public void testUploadWithFileInputStream() {
|
||||
try {
|
||||
// 本地图片文件路径
|
||||
String localImagePath = "D:/test/image.jpg"; // 请修改为实际路径
|
||||
|
||||
File imageFile = new File(localImagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("图片文件不存在: " + localImagePath);
|
||||
return;
|
||||
}
|
||||
|
||||
// 使用FileInputStream读取文件
|
||||
try (FileInputStream fis = new FileInputStream(imageFile)) {
|
||||
byte[] imageBytes = new byte[(int) imageFile.length()];
|
||||
fis.read(imageBytes);
|
||||
|
||||
// 生成OSS对象名称
|
||||
String objectName = generateObjectName("stream", "jpg");
|
||||
|
||||
// 上传
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("========== FileInputStream上传测试 ==========");
|
||||
System.out.println("文件路径: " + localImagePath);
|
||||
System.out.println("文件大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("OSS对象名称: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
System.out.println("===========================================");
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成OSS对象名称
|
||||
* @param prefix 前缀
|
||||
* @param extension 文件扩展名
|
||||
* @return 对象名称
|
||||
*/
|
||||
private String generateObjectName(String prefix, String extension) {
|
||||
// 格式: prefix/yyyyMMdd/HHmmss-uuid.extension
|
||||
String dateTime = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd/HHmmss"));
|
||||
String uuid = UUID.randomUUID().toString().substring(0, 8);
|
||||
return String.format("%s/%s-%s.%s", prefix, dateTime, uuid, extension);
|
||||
}
|
||||
|
||||
/**
|
||||
* 从文件路径获取扩展名
|
||||
* @param filePath 文件路径
|
||||
* @return 扩展名
|
||||
*/
|
||||
private String getFileExtension(String filePath) {
|
||||
int lastDot = filePath.lastIndexOf('.');
|
||||
if (lastDot > 0 && lastDot < filePath.length() - 1) {
|
||||
return filePath.substring(lastDot + 1).toLowerCase();
|
||||
}
|
||||
return "jpg"; // 默认扩展名
|
||||
}
|
||||
|
||||
/**
|
||||
* 从URL获取文件扩展名
|
||||
* @param url URL地址
|
||||
* @return 扩展名
|
||||
*/
|
||||
private String getFileExtensionFromUrl(String url) {
|
||||
try {
|
||||
// 移除查询参数
|
||||
String path = url.split("\\?")[0];
|
||||
int lastDot = path.lastIndexOf('.');
|
||||
int lastSlash = path.lastIndexOf('/');
|
||||
|
||||
if (lastDot > lastSlash && lastDot < path.length() - 1) {
|
||||
return path.substring(lastDot + 1).toLowerCase();
|
||||
}
|
||||
} catch (Exception e) {
|
||||
// 忽略异常
|
||||
}
|
||||
return "jpg"; // 默认扩展名
|
||||
}
|
||||
|
||||
/**
|
||||
* 从InputStream读取所有字节(兼容方法)
|
||||
* @param inputStream 输入流
|
||||
* @return 字节数组
|
||||
* @throws IOException IO异常
|
||||
*/
|
||||
private byte[] readAllBytes(InputStream inputStream) throws IOException {
|
||||
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
|
||||
byte[] data = new byte[8192]; // 8KB缓冲区
|
||||
int nRead;
|
||||
while ((nRead = inputStream.read(data, 0, data.length)) != -1) {
|
||||
buffer.write(data, 0, nRead);
|
||||
}
|
||||
return buffer.toByteArray();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,91 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
/**
|
||||
* 验证码服务测试类
|
||||
* @author KilLze
|
||||
*/
|
||||
@SpringBootTest
|
||||
public class VerificationCodeServiceTest {
|
||||
|
||||
@Autowired
|
||||
private VerificationCodeService verificationCodeService;
|
||||
|
||||
/**
|
||||
* 测试发送邮箱验证码
|
||||
*/
|
||||
@Test
|
||||
public void testSendEmailCode() {
|
||||
String email = "test@example.com"; // 请修改为实际邮箱地址
|
||||
|
||||
System.out.println("========== 发送邮箱验证码测试 ==========");
|
||||
System.out.println("邮箱地址: " + email);
|
||||
|
||||
boolean result = verificationCodeService.sendEmailCode(email);
|
||||
|
||||
System.out.println("发送结果: " + (result ? "成功" : "失败"));
|
||||
System.out.println("=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试验证邮箱验证码
|
||||
*/
|
||||
@Test
|
||||
public void testVerifyEmailCode() {
|
||||
String email = "test@example.com"; // 请修改为实际邮箱地址
|
||||
String code = "123456"; // 请修改为实际收到的验证码
|
||||
|
||||
System.out.println("========== 验证邮箱验证码测试 ==========");
|
||||
System.out.println("邮箱地址: " + email);
|
||||
System.out.println("验证码: " + code);
|
||||
|
||||
boolean result = verificationCodeService.verifyEmailCode(email, code);
|
||||
|
||||
System.out.println("验证结果: " + (result ? "成功" : "失败"));
|
||||
System.out.println("=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试完整流程:发送验证码 -> 验证验证码
|
||||
*/
|
||||
@Test
|
||||
public void testCompleteFlow() {
|
||||
String email = "test@example.com"; // 请修改为实际邮箱地址
|
||||
|
||||
System.out.println("========== 完整流程测试 ==========");
|
||||
System.out.println("邮箱地址: " + email);
|
||||
|
||||
// 1. 发送验证码
|
||||
System.out.println("\n1. 发送验证码...");
|
||||
boolean sendResult = verificationCodeService.sendEmailCode(email);
|
||||
System.out.println("发送结果: " + (sendResult ? "成功" : "失败"));
|
||||
|
||||
if (sendResult) {
|
||||
// 2. 等待用户输入验证码(这里模拟,实际应该从控制台或API获取)
|
||||
System.out.println("\n2. 请查看邮箱获取验证码,然后手动测试验证功能");
|
||||
System.out.println(" 使用 testVerifyEmailCode() 方法进行验证");
|
||||
}
|
||||
|
||||
System.out.println("\n=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试生成验证码
|
||||
*/
|
||||
@Test
|
||||
public void testGenerateCode() {
|
||||
System.out.println("========== 生成验证码测试 ==========");
|
||||
|
||||
// 测试不同长度的验证码
|
||||
for (int length = 4; length <= 8; length++) {
|
||||
String code = verificationCodeService.generateCode(length);
|
||||
System.out.println(length + "位验证码: " + code);
|
||||
}
|
||||
|
||||
System.out.println("=====================================");
|
||||
}
|
||||
}
|
||||
|
||||
78
src/test/java/com/bao/dating/util/EmailAndSmsTest.java
Normal file
78
src/test/java/com/bao/dating/util/EmailAndSmsTest.java
Normal file
@@ -0,0 +1,78 @@
|
||||
package com.bao.dating.util;
|
||||
|
||||
import com.bao.dating.common.aliyun.SmsUtil;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 邮箱和短信发送测试类
|
||||
* @author KilLze
|
||||
*/
|
||||
@SpringBootTest
|
||||
public class EmailAndSmsTest {
|
||||
|
||||
@Autowired
|
||||
private EmailUtil emailUtil;
|
||||
|
||||
@Autowired
|
||||
private SmsUtil smsUtil;
|
||||
|
||||
/**
|
||||
* 测试发送简单文本邮件
|
||||
*/
|
||||
@Test
|
||||
public void testSendSimpleMail() {
|
||||
String to = "n_1127@qq.com";
|
||||
String subject = "测试邮件 - 简单文本";
|
||||
String text = "这是一封测试邮件,用于测试简单文本邮件发送功能。\n\n如果您收到此邮件,说明邮件发送功能正常。";
|
||||
|
||||
boolean result = emailUtil.sendSimpleMail(to, subject, text);
|
||||
System.out.println("简单邮件发送结果: " + (result ? "成功" : "失败"));
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试发送HTML格式邮件
|
||||
*/
|
||||
@Test
|
||||
public void testSendHtmlMail() {
|
||||
String to = "n_1127@qq.com";
|
||||
String subject = "测试邮件 - HTML格式";
|
||||
String htmlContent = "<html><body>" +
|
||||
"<h2 style='color: #1890ff;'>欢迎使用我们的服务</h2>" +
|
||||
"<p>这是一封<strong>HTML格式</strong>的测试邮件。</p>" +
|
||||
"<p>邮件内容支持:</p>" +
|
||||
"<ul>" +
|
||||
"<li>HTML标签</li>" +
|
||||
"<li>样式设置</li>" +
|
||||
"<li>富文本内容</li>" +
|
||||
"</ul>" +
|
||||
"<p style='color: #999;'>感谢您的使用!</p>" +
|
||||
"</body></html>";
|
||||
|
||||
boolean result = emailUtil.sendHtmlMail(to, subject, htmlContent);
|
||||
System.out.println("HTML邮件发送结果: " + (result ? "成功" : "失败"));
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试发送验证码邮件
|
||||
*/
|
||||
@Test
|
||||
public void testSendVerificationCodeEmail() {
|
||||
String to = "n_1127@qq.com";
|
||||
String code = "123456";
|
||||
|
||||
boolean result = emailUtil.sendVerificationCode(to, code);
|
||||
System.out.println("验证码邮件发送结果: " + (result ? "成功" : "失败"));
|
||||
System.out.println("验证码: " + code);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user