Compare commits
6 Commits
60df001385
...
feature-Ki
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a6259875f2 | ||
| 34cad7457b | |||
| 61c4c9d442 | |||
|
|
717c0a0507 | ||
|
|
413bafa275 | ||
|
|
2ce8116126 |
6
pom.xml
6
pom.xml
@@ -159,6 +159,12 @@
|
||||
<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>
|
||||
|
||||
@@ -19,6 +19,14 @@ public enum ResultCode {
|
||||
FORBIDDEN(403, "无权限"),
|
||||
/** 数据不存在 */
|
||||
NOT_FOUND(404, "数据不存在"),
|
||||
/** 验证码已发送 */
|
||||
CODE_SENT(200, "验证码已发送"),
|
||||
/** 验证码错误 */
|
||||
CODE_ERROR(400, "验证码错误"),
|
||||
/** 验证码已过期 */
|
||||
CODE_EXPIRED(400, "验证码已过期"),
|
||||
/** 验证码发送失败 */
|
||||
CODE_SEND_FAIL(500, "验证码发送失败"),
|
||||
/** 系统异常 */
|
||||
SYSTEM_ERROR(500, "系统异常"),
|
||||
/** 操作失败 */
|
||||
|
||||
@@ -149,3 +149,4 @@ public class SmsUtil {
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -20,9 +20,13 @@ public class RedisConfig {
|
||||
// 设置redis的连接工厂对象
|
||||
redisTemplate.setConnectionFactory(redisConnectionFactory);
|
||||
|
||||
// key 使用 String
|
||||
// 设置redis key的序列化器
|
||||
redisTemplate.setKeySerializer(new StringRedisSerializer());
|
||||
// 设置value的序列化器
|
||||
redisTemplate.setValueSerializer(new GenericJackson2JsonRedisSerializer());
|
||||
// 设置hash类型的key和value的序列化器
|
||||
redisTemplate.setHashKeySerializer(new StringRedisSerializer());
|
||||
redisTemplate.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());
|
||||
|
||||
return redisTemplate;
|
||||
}
|
||||
|
||||
@@ -61,7 +61,7 @@ public class PostController {
|
||||
@PostMapping("/deletePost")
|
||||
public Result<String> deleteById(@RequestBody List<Long> postIds){
|
||||
int deletedCount = postService.deletePostById(postIds);
|
||||
return Result.success(ResultCode.SUCCESS_DELETE, deletedCount > 0 ? "成功删除" : "删除失败,该动态不存在", null);
|
||||
return Result.success(ResultCode.SUCCESS_DELETE, "成功删除" + deletedCount + "条动态", null);
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -71,8 +71,7 @@ public class TokenInterceptor implements HandlerInterceptor {
|
||||
String userId = JwtUtil.getSubjectFromToken(token);
|
||||
|
||||
// 从Redis获取存储的token进行比对
|
||||
Object redisTokenObj = redisTemplate.opsForValue()
|
||||
.get("login:token:" + userId);
|
||||
Object redisTokenObj = redisTemplate.opsForValue().get("login:token:" + userId);
|
||||
String redisToken = redisTokenObj != null ? redisTokenObj.toString() : null;
|
||||
|
||||
// 验证Redis中的token是否存在且匹配
|
||||
|
||||
@@ -18,4 +18,11 @@ public interface CommentsMapper {
|
||||
// 根据动态ID查询评论列表
|
||||
@Select("SELECT * FROM comments WHERE post_id = #{post_id} ORDER BY created_at DESC")
|
||||
List<Comments> getCommentsByPostId(@Param("post_id") Long post_id);
|
||||
|
||||
/**
|
||||
* 根据动态ID批量删除评论
|
||||
* @param postIds
|
||||
* @return
|
||||
*/
|
||||
int deleteCommentsByPostIds(@Param("postIds") List<Long> postIds);
|
||||
}
|
||||
|
||||
@@ -12,4 +12,12 @@ public interface PostFavoriteMapper {
|
||||
List<Long> selectUserIDByPostID(@Param("postId") Long postId);
|
||||
int addPostFavorite(PostFavorite postFavorite);
|
||||
int deletePostFavorite(@Param("postId") Long postId);
|
||||
|
||||
/**
|
||||
* 批量删除收藏
|
||||
* @param postIds
|
||||
* @return
|
||||
*/
|
||||
int deleteFavoritesByPostIds(@Param("postIds") List<Long> postIds);
|
||||
|
||||
}
|
||||
|
||||
@@ -4,6 +4,8 @@ import com.bao.dating.pojo.entity.PostLike;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Mapper
|
||||
public interface PostLikeMapper {
|
||||
/**
|
||||
@@ -31,4 +33,12 @@ public interface PostLikeMapper {
|
||||
* @return
|
||||
*/
|
||||
int deleteByPostIdAndUserId(@Param("postId") Long postId, @Param("userId") Long userId);
|
||||
|
||||
/**
|
||||
* 批量删除点赞记录
|
||||
*
|
||||
* @param postIds
|
||||
* @return
|
||||
*/
|
||||
int deleteLikesByPostIds(@Param("postIds") List<Long> postIds);
|
||||
}
|
||||
|
||||
@@ -21,11 +21,11 @@ public interface PostMapper {
|
||||
void insert(Post post);
|
||||
|
||||
/**
|
||||
* 根据ID删除动态
|
||||
* 根据ID修改动态状态
|
||||
*
|
||||
* @param postIds 动态ID
|
||||
*/
|
||||
int deletePostByIds(List<Long> postIds);
|
||||
int updatePublicById(@Param("postIds") List<Long> postIds, @Param("userId") Long userId);
|
||||
|
||||
/**
|
||||
* 根据ID查询动态
|
||||
|
||||
@@ -27,7 +27,7 @@ public interface PostService {
|
||||
Post createPost(PostRequestDTO postRequestDTO);
|
||||
|
||||
/**
|
||||
* 批量删除动态
|
||||
* 批量删除动态(将动态状态改为已删除)
|
||||
* @param postIds 动态ID
|
||||
* @return 删除的动态对象
|
||||
*/
|
||||
|
||||
@@ -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);
|
||||
}
|
||||
|
||||
@@ -4,17 +4,22 @@ 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.CommentsMapper;
|
||||
import com.bao.dating.mapper.PostFavoriteMapper;
|
||||
import com.bao.dating.mapper.PostLikeMapper;
|
||||
import com.bao.dating.mapper.PostMapper;
|
||||
import com.bao.dating.pojo.dto.PostRequestDTO;
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
import com.bao.dating.pojo.vo.PostEditVO;
|
||||
import com.bao.dating.service.PostService;
|
||||
import com.bao.dating.common.aliyun.AliOssUtil;
|
||||
import com.bao.dating.service.UserService;
|
||||
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.util.CollectionUtils;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.io.IOException;
|
||||
@@ -43,6 +48,15 @@ public class PostServiceImpl implements PostService {
|
||||
@Autowired
|
||||
private PostMapper postMapper;
|
||||
|
||||
@Autowired
|
||||
private PostLikeMapper postLikeMapper;
|
||||
|
||||
@Autowired
|
||||
private PostFavoriteMapper postFavoriteMapper;
|
||||
|
||||
@Autowired
|
||||
private CommentsMapper commentsMapper;
|
||||
|
||||
/**
|
||||
* 上传媒体文件
|
||||
* @param files 媒体文件数组
|
||||
@@ -177,7 +191,7 @@ public class PostServiceImpl implements PostService {
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量删除动态
|
||||
* 批量删除动态(将动态状态改为已删除)
|
||||
*
|
||||
* @param postIds 动态ID
|
||||
* @return 删除的动态对象
|
||||
@@ -188,19 +202,21 @@ public class PostServiceImpl implements PostService {
|
||||
// 判断用户权限
|
||||
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("无权限删除此动态");
|
||||
}
|
||||
if (CollectionUtils.isEmpty(postIds)) {
|
||||
return 0;
|
||||
}
|
||||
// 批量删除动态
|
||||
return postMapper.deletePostByIds(postIds);
|
||||
int affected = postMapper.updatePublicById(postIds, userId);
|
||||
|
||||
if (affected == 0) {
|
||||
throw new RuntimeException("未删除任何动态,可能无权限或动态不存在");
|
||||
}
|
||||
|
||||
// 删除动态下的评论、点赞、收藏
|
||||
commentsMapper.deleteCommentsByPostIds(postIds);
|
||||
postLikeMapper.deleteLikesByPostIds(postIds);
|
||||
postFavoriteMapper.deleteFavoritesByPostIds(postIds);
|
||||
|
||||
return affected;
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -13,8 +13,8 @@ spring:
|
||||
password: JoyeeServe2025
|
||||
driver-class-name: com.mysql.cj.jdbc.Driver
|
||||
redis:
|
||||
host: 127.0.0.1
|
||||
port: 6379
|
||||
host: 110.42.41.177
|
||||
port: 6389
|
||||
password: ""
|
||||
database: 0
|
||||
timeout: 10000
|
||||
|
||||
15
src/main/resources/com/bao/dating/mapper/CommentsMapper.xml
Normal file
15
src/main/resources/com/bao/dating/mapper/CommentsMapper.xml
Normal file
@@ -0,0 +1,15 @@
|
||||
<?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.CommentsMapper">
|
||||
|
||||
<!-- 批量删除动态下的所有评论 -->
|
||||
<delete id="deleteCommentsByPostIds">
|
||||
DELETE FROM comments
|
||||
WHERE post_id IN
|
||||
<foreach collection="postIds" item="postId" open="(" close=")" separator=",">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
|
||||
</mapper>
|
||||
@@ -14,4 +14,14 @@
|
||||
<select id="selectUserIDByPostID" resultType="java.lang.Long">
|
||||
SELECT user_id FROM post_favorite WHERE post_id = #{postId}
|
||||
</select>
|
||||
|
||||
<!--批量删除动态收藏-->
|
||||
<delete id="deleteFavoritesByPostIds">
|
||||
DELETE FROM post_favorite
|
||||
WHERE post_id IN
|
||||
<foreach collection="postIds" item="postId" open="(" close=")" separator=",">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
|
||||
</mapper>
|
||||
@@ -14,4 +14,14 @@
|
||||
<delete id="deleteByPostIdAndUserId">
|
||||
delete from dating.post_like where post_id = #{postId} and user_id = #{userId}
|
||||
</delete>
|
||||
|
||||
<!--批量删除点赞记录-->
|
||||
<delete id="deleteLikesByPostIds">
|
||||
DELETE FROM post_like
|
||||
WHERE post_id IN
|
||||
<foreach collection="postIds" item="postId" open="(" close=")" separator=",">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
|
||||
</mapper>
|
||||
@@ -28,25 +28,19 @@
|
||||
#{isPublic}, 0, 0, #{createdAt}, #{updatedAt})
|
||||
</insert>
|
||||
|
||||
<!--动态删除-->
|
||||
<delete id="deletePostByIds">
|
||||
DELETE FROM post WHERE post_id IN
|
||||
<!--修改动态状态-->
|
||||
<update id="updatePublicById">
|
||||
UPDATE post
|
||||
<set>
|
||||
is_public = 3,
|
||||
updated_at = NOW()
|
||||
</set>
|
||||
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>
|
||||
AND user_id = #{userId}
|
||||
</update>
|
||||
|
||||
<!--动态查询-->
|
||||
<resultMap id="PostResultMap" type="com.bao.dating.pojo.entity.Post">
|
||||
|
||||
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("=====================================");
|
||||
}
|
||||
}
|
||||
|
||||
@@ -75,3 +75,4 @@ public class EmailAndSmsTest {
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user