新增RabbitMQ相关配置和功能，添加微信登录和图片处理接口
- 新增RabbitMQ配置类和消息监听器 - 实现微信扫码登录功能 - 添加图片上传和处理接口，支持外网图片转存 - 优化过滤器类，移除多余日志 - 新增白名单查询接口
This commit is contained in:
parent
50fb21b795
commit
3557e7b1c2
33
ForJdk17/src/main/java/cn/whaifree/interview/KS1/P1.java
Normal file
33
ForJdk17/src/main/java/cn/whaifree/interview/KS1/P1.java
Normal file
@ -0,0 +1,33 @@
|
||||
package cn.whaifree.interview.KS1;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/16 13:53
|
||||
* @注释
|
||||
*/
|
||||
public class P1 {
|
||||
@Test
|
||||
public void test() {
|
||||
int[] prices = {7, 1, 5, 3, 6, 4};
|
||||
int maxProfit = maxProfit(prices);
|
||||
System.out.println(maxProfit);
|
||||
}
|
||||
|
||||
public int maxProfit(int[] prices) {
|
||||
int maxProfit = 0;
|
||||
int left = 0;
|
||||
int right = 0;
|
||||
while (right < prices.length) {
|
||||
if (prices[right] < prices[left]) {
|
||||
left = right;
|
||||
}else {
|
||||
maxProfit = Math.max(maxProfit, prices[right] - prices[left]);
|
||||
}
|
||||
right++;
|
||||
}
|
||||
return maxProfit;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,49 @@
|
||||
package cn.whaifree.leetCode.Array;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.HashSet;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/15 16:37
|
||||
* @注释
|
||||
*/
|
||||
public class LCR119 {
|
||||
@Test
|
||||
public void test() {
|
||||
int[] nums = {100, 4, 200, 1, 3, 2};
|
||||
System.out.println(new Solution().longestConsecutive(nums));
|
||||
}
|
||||
|
||||
class Solution {
|
||||
public int longestConsecutive(int[] nums) {
|
||||
HashSet<Integer> set = new HashSet<>();
|
||||
for (int num : nums) {
|
||||
set.add(num);
|
||||
}
|
||||
|
||||
int max = 0;
|
||||
for (int num : nums) {
|
||||
if (set.contains(num- 1)) { // 1 2 3 4 只要保证set里没有0
|
||||
continue;
|
||||
}
|
||||
// int base = num;
|
||||
// int tmp = 0;
|
||||
// while (set.contains(base++)) {
|
||||
// tmp++;
|
||||
// }
|
||||
// max = Math.max(max, tmp);
|
||||
|
||||
int maxnum = num;
|
||||
while (set.contains(maxnum + 1)) {
|
||||
maxnum++;
|
||||
}
|
||||
max = Math.max(max, maxnum - num + 1);
|
||||
}
|
||||
return max;
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,61 @@
|
||||
package cn.whaifree.redo.redo_all_241016;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Comparator;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/16 12:10
|
||||
* @注释
|
||||
*/
|
||||
public class LCR074 {
|
||||
|
||||
@Test
|
||||
public void test() {
|
||||
|
||||
StringBuilder stringBuilder = new StringBuilder();
|
||||
int[][] intervals = {{1, 4}, {4, 6}};
|
||||
Solution solution = new Solution();
|
||||
int[][] result = solution.merge(intervals);
|
||||
for (int[] interval : result) {
|
||||
System.out.println(Arrays.toString(interval));
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
class Solution {
|
||||
/**
|
||||
* | |
|
||||
* | |
|
||||
* | |
|
||||
* | |
|
||||
*
|
||||
* @param intervals
|
||||
* @return
|
||||
*/
|
||||
public int[][] merge(int[][] intervals) {
|
||||
Arrays.sort(intervals, Comparator.comparingInt(o -> o[0]));
|
||||
|
||||
List<int[]> result = new ArrayList<>();
|
||||
for (int i = 1; i < intervals.length; i++) {
|
||||
int[] before = intervals[i - 1];
|
||||
int[] current = intervals[i];
|
||||
if (before[1] >= current[0]) {
|
||||
current[0] = Math.min(before[0], current[0]);
|
||||
current[1] = Math.max(before[1], current[1]);
|
||||
}else {
|
||||
result.add(before);
|
||||
}
|
||||
}
|
||||
result.add(intervals[intervals.length - 1]);
|
||||
return result.toArray(new int[result.size()][]);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
33
springDemo/.gitignore
vendored
Normal file
33
springDemo/.gitignore
vendored
Normal file
@ -0,0 +1,33 @@
|
||||
HELP.md
|
||||
target/
|
||||
!.mvn/wrapper/maven-wrapper.jar
|
||||
!**/src/main/**/target/
|
||||
!**/src/test/**/target/
|
||||
|
||||
### STS ###
|
||||
.apt_generated
|
||||
.classpath
|
||||
.factorypath
|
||||
.project
|
||||
.settings
|
||||
.springBeans
|
||||
.sts4-cache
|
||||
|
||||
### IntelliJ IDEA ###
|
||||
.idea
|
||||
*.iws
|
||||
*.iml
|
||||
*.ipr
|
||||
|
||||
### NetBeans ###
|
||||
/nbproject/private/
|
||||
/nbbuild/
|
||||
/dist/
|
||||
/nbdist/
|
||||
/.nb-gradle/
|
||||
build/
|
||||
!**/src/main/**/build/
|
||||
!**/src/test/**/build/
|
||||
|
||||
### VS Code ###
|
||||
.vscode/
|
||||
@ -37,18 +37,42 @@
|
||||
<version>4.4.0</version>
|
||||
</dependency>
|
||||
|
||||
<!-- <!– https://mvnrepository.com/artifact/org.redisson/redisson-spring-boot-starter –>-->
|
||||
<!-- <dependency>-->
|
||||
<!-- <groupId>org.redisson</groupId>-->
|
||||
<!-- <artifactId>redisson-spring-boot-starter</artifactId>-->
|
||||
<!-- <version>3.23.5</version>-->
|
||||
<!-- </dependency>-->
|
||||
<!-- <!– https://mvnrepository.com/artifact/org.redisson/redisson-spring-boot-starter –>-->
|
||||
<!-- <dependency>-->
|
||||
<!-- <groupId>org.redisson</groupId>-->
|
||||
<!-- <artifactId>redisson-spring-boot-starter</artifactId>-->
|
||||
<!-- <version>3.23.5</version>-->
|
||||
<!-- </dependency>-->
|
||||
<!--redisson-->
|
||||
<dependency>
|
||||
<groupId>org.redisson</groupId>
|
||||
<groupId>org.redisson</groupId>
|
||||
<artifactId>redisson</artifactId>
|
||||
<version>3.16.8</version>
|
||||
</dependency>
|
||||
<!-- rabbitMQ-->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-amqp</artifactId>
|
||||
</dependency>
|
||||
<!--minio-->
|
||||
<dependency>
|
||||
<groupId>io.minio</groupId>
|
||||
<artifactId>minio</artifactId>
|
||||
<version>8.4.3</version>
|
||||
</dependency>
|
||||
<!-- 其他依赖 -->
|
||||
<dependency>
|
||||
<groupId>com.fasterxml.jackson.dataformat</groupId>
|
||||
<artifactId>jackson-dataformat-xml</artifactId>
|
||||
<version>2.13.0</version> <!-- 根据实际情况选择最新稳定版本 -->
|
||||
</dependency>
|
||||
|
||||
|
||||
<dependency>
|
||||
<groupId>com.google.guava</groupId>
|
||||
<artifactId>guava</artifactId>
|
||||
<version>31.0.1-jre</version>
|
||||
</dependency>
|
||||
|
||||
<!--FastJson-->
|
||||
<dependency>
|
||||
@ -104,11 +128,11 @@
|
||||
<artifactId>spring-boot-starter-test</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<!-- <dependency>-->
|
||||
<!-- <groupId>org.springframework.amqp</groupId>-->
|
||||
<!-- <artifactId>spring-rabbit-test</artifactId>-->
|
||||
<!-- <scope>test</scope>-->
|
||||
<!-- </dependency>-->
|
||||
<!-- <dependency>-->
|
||||
<!-- <groupId>org.springframework.amqp</groupId>-->
|
||||
<!-- <artifactId>spring-rabbit-test</artifactId>-->
|
||||
<!-- <scope>test</scope>-->
|
||||
<!-- </dependency>-->
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
|
||||
@ -0,0 +1,23 @@
|
||||
package cn.whaifree.springdemo.controller;
|
||||
|
||||
import jakarta.annotation.Resource;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/15 21:29
|
||||
* @注释
|
||||
*/
|
||||
@RestController
|
||||
public class WhiteListController {
|
||||
|
||||
@Resource
|
||||
private RedisTemplate redisTemplate;
|
||||
@PostMapping("/queryIn")
|
||||
public boolean query(String userId) {
|
||||
return Boolean.TRUE.equals(redisTemplate.opsForSet().isMember("whiteList", userId));
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,196 @@
|
||||
package cn.whaifree.springdemo.controller.minio;
|
||||
|
||||
import cn.hutool.crypto.digest.MD5;
|
||||
import cn.hutool.http.HttpRequest;
|
||||
import cn.hutool.http.HttpResponse;
|
||||
import cn.hutool.http.HttpUtil;
|
||||
import com.google.common.cache.CacheBuilder;
|
||||
import com.google.common.cache.CacheLoader;
|
||||
import com.google.common.cache.LoadingCache;
|
||||
import io.minio.MinioClient;
|
||||
import io.minio.ObjectWriteResponse;
|
||||
import io.minio.PutObjectArgs;
|
||||
import jakarta.annotation.Resource;
|
||||
import lombok.Data;
|
||||
import lombok.Getter;
|
||||
import lombok.Setter;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import okhttp3.Headers;
|
||||
import org.springframework.beans.factory.InitializingBean;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.util.StreamUtils;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.InputStream;
|
||||
import java.net.URI;
|
||||
import java.util.concurrent.ExecutionException;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/16 9:38
|
||||
* @注释
|
||||
*/
|
||||
@RestController
|
||||
@Slf4j
|
||||
public class MinioController {
|
||||
@Resource
|
||||
private ImageUploader imageUploader;
|
||||
/**
|
||||
* 外网图片转存缓存
|
||||
*/
|
||||
private LoadingCache<String, String> imgReplaceCache = CacheBuilder.newBuilder().maximumSize(300).expireAfterWrite(5, TimeUnit.MINUTES).build(new CacheLoader<String, String>() {
|
||||
@Override
|
||||
public String load(String img) {
|
||||
try {
|
||||
InputStream stream = null;
|
||||
if (img.startsWith("http")) {
|
||||
// 下载变输入
|
||||
HttpRequest get = HttpUtil.createGet(img);
|
||||
get.header("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36");
|
||||
HttpResponse response = get.execute();
|
||||
stream = response.bodyStream();
|
||||
}else {
|
||||
return "";
|
||||
}
|
||||
URI uri = URI.create(img);
|
||||
String path = uri.getPath();
|
||||
int index = path.lastIndexOf(".");
|
||||
String fileType = null;
|
||||
if (index > 0) {
|
||||
// 从url中获取文件类型
|
||||
fileType = path.substring(index + 1);
|
||||
}
|
||||
return imageUploader.upload(stream, fileType);
|
||||
} catch (Exception e) {
|
||||
log.error("外网图片转存异常! img:{}", img, e);
|
||||
return "";
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
@PostMapping("/changeLink")
|
||||
public String changeLink(String url) throws ExecutionException {
|
||||
return imgReplaceCache.get(url);
|
||||
}
|
||||
}
|
||||
@Slf4j
|
||||
@ConditionalOnExpression(value = "#{'minio'.equals(environment.getProperty('image.oss.type'))}")
|
||||
@Component
|
||||
class MinioOssWrapper implements ImageUploader, InitializingBean {
|
||||
|
||||
private MinioClient minioClient;
|
||||
@Autowired
|
||||
@Setter
|
||||
@Getter
|
||||
private ImageProperties properties;
|
||||
|
||||
|
||||
@Override
|
||||
public String upload(InputStream inputStream, String fileType) {
|
||||
try {
|
||||
byte[] bytes = StreamUtils.copyToByteArray(inputStream);
|
||||
// 计算md5作为文件名，避免重复上传
|
||||
String fileName = MD5.create().digestHex(bytes);
|
||||
ByteArrayInputStream input = new ByteArrayInputStream(bytes);
|
||||
|
||||
fileName = fileName + "." + fileType;
|
||||
log.info("上传文件名：{}", fileName);
|
||||
|
||||
PutObjectArgs args = PutObjectArgs.builder()
|
||||
.bucket(properties.getOss().getBucket())
|
||||
.object(fileName)
|
||||
.stream(input, input.available(), -1)
|
||||
.contentType(fileType)
|
||||
.build();
|
||||
|
||||
ObjectWriteResponse response = minioClient.putObject(args);
|
||||
|
||||
// 获取response状态码
|
||||
Headers headers = response.headers();
|
||||
log.info(headers.toString());
|
||||
|
||||
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append(properties.getOss().getEndpoint()).append("/").append(response.bucket()).append("/").append(response.object());
|
||||
|
||||
return sb.toString();
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error(e.getMessage());
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean uploadIgnore(String fileUrl) {
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void afterPropertiesSet() throws Exception {
|
||||
// 创建OSSClient实例。
|
||||
log.info("init ossClient");
|
||||
minioClient = MinioClient.builder()
|
||||
.credentials(
|
||||
properties.getOss().getAk(),
|
||||
properties.getOss().getSk()
|
||||
)
|
||||
.endpoint(properties.getOss().getEndpoint())
|
||||
.build();
|
||||
}
|
||||
}
|
||||
|
||||
@Setter
|
||||
@Getter
|
||||
@Component
|
||||
@ConfigurationProperties(prefix = "image")
|
||||
class ImageProperties {
|
||||
private String absTmpPath; // 存储绝对路径
|
||||
private String webImgPath; // 存储相对路径
|
||||
private String tmpUploadPath; // 上传文件的临时存储目录
|
||||
private String cdnHost; // 访问图片的host
|
||||
private OssProperties oss;
|
||||
}
|
||||
|
||||
@Data
|
||||
class OssProperties {
|
||||
private String prefix; // 上传文件前缀路径
|
||||
private String type; // oss类型
|
||||
//下面几个是oss的配置参数
|
||||
private String endpoint;
|
||||
private String ak;
|
||||
private String sk;
|
||||
private String bucket;
|
||||
private String host;
|
||||
}
|
||||
|
||||
|
||||
interface ImageUploader {
|
||||
String DEFAULT_FILE_TYPE = "txt";
|
||||
// Set<MediaType> STATIC_IMG_TYPE = new HashSet<>(Arrays.asList(MediaType.ImagePng, MediaType.ImageJpg, MediaType.ImageWebp, MediaType.ImageGif));
|
||||
|
||||
/**
|
||||
* 文件上传
|
||||
*
|
||||
* @param input
|
||||
* @param fileType
|
||||
* @return
|
||||
*/
|
||||
String upload(InputStream input, String fileType);
|
||||
|
||||
/**
|
||||
* 判断外网图片是否依然需要处理
|
||||
*
|
||||
* @param fileUrl
|
||||
* @return true 表示忽略，不需要转存
|
||||
*/
|
||||
boolean uploadIgnore(String fileUrl);
|
||||
}
|
||||
@ -0,0 +1,106 @@
|
||||
package cn.whaifree.springdemo.controller.rabbitMqEvent;
|
||||
|
||||
import cn.hutool.extra.spring.SpringUtil;
|
||||
import cn.whaifree.springdemo.utils.ResVo;
|
||||
import lombok.Getter;
|
||||
import lombok.ToString;
|
||||
import org.springframework.context.ApplicationEvent;
|
||||
import org.springframework.context.ApplicationListener;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/15 19:12
|
||||
* @注释
|
||||
*/
|
||||
@RestController
|
||||
public class EventController {
|
||||
|
||||
/**
|
||||
* 根据不同的青桔
|
||||
*
|
||||
* @param msg
|
||||
* @return
|
||||
*/
|
||||
@PostMapping("/sendMsg")
|
||||
public ResVo sendMsgNotify(NotifyTypeEnum notifyType, String msg) {
|
||||
// 发送异步消息
|
||||
SpringUtil.publishEvent(new NotifyMsgEvent<>(this, notifyType, msg));
|
||||
return ResVo.success();
|
||||
}
|
||||
}
|
||||
|
||||
@Getter
|
||||
@ToString
|
||||
class NotifyMsgEvent<T> extends ApplicationEvent {
|
||||
private NotifyTypeEnum notifyType;
|
||||
private T content;
|
||||
public NotifyMsgEvent(Object source, NotifyTypeEnum notifyType, T content) {
|
||||
super(source);
|
||||
this.notifyType = notifyType;
|
||||
this.content = content;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Service
|
||||
class NotifyMsgListener<T> implements ApplicationListener<NotifyMsgEvent<T>> {
|
||||
@Override
|
||||
public void onApplicationEvent(NotifyMsgEvent<T> event) {
|
||||
System.out.println(event); // 获取到发送的消息,做下一步处理
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Getter
|
||||
enum NotifyTypeEnum {
|
||||
COMMENT(1, "评论"),
|
||||
REPLY(2, "回复"),
|
||||
PRAISE(3, "点赞"),
|
||||
COLLECT(4, "收藏"),
|
||||
FOLLOW(5, "关注消息"),
|
||||
SYSTEM(6, "系统消息"),
|
||||
DELETE_COMMENT(1, "删除评论"),
|
||||
DELETE_REPLY(2, "删除回复"),
|
||||
CANCEL_PRAISE(3, "取消点赞"),
|
||||
CANCEL_COLLECT(4, "取消收藏"),
|
||||
CANCEL_FOLLOW(5, "取消关注"),
|
||||
|
||||
// 注册、登录添加系统相关提示消息
|
||||
REGISTER(6, "用户注册"),
|
||||
LOGIN(6, "用户登录"),
|
||||
;
|
||||
|
||||
|
||||
private int type;
|
||||
private String msg;
|
||||
|
||||
private static Map<Integer, NotifyTypeEnum> mapper;
|
||||
|
||||
static {
|
||||
mapper = new HashMap<>();
|
||||
for (NotifyTypeEnum type : values()) {
|
||||
mapper.put(type.type, type);
|
||||
}
|
||||
}
|
||||
|
||||
NotifyTypeEnum(int type, String msg) {
|
||||
this.type = type;
|
||||
this.msg = msg;
|
||||
}
|
||||
|
||||
public static NotifyTypeEnum typeOf(int type) {
|
||||
return mapper.get(type);
|
||||
}
|
||||
|
||||
public static NotifyTypeEnum typeOf(String type) {
|
||||
return valueOf(type.toUpperCase().trim());
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,268 @@
|
||||
package cn.whaifree.springdemo.controller.rabbitMqEvent;
|
||||
|
||||
import cn.hutool.core.date.DateUtil;
|
||||
import cn.whaifree.springdemo.utils.ResVo;
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import com.rabbitmq.client.Channel;
|
||||
import jakarta.annotation.Resource;
|
||||
import lombok.Data;
|
||||
import lombok.experimental.Accessors;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.BooleanUtils;
|
||||
import org.springframework.amqp.core.*;
|
||||
import org.springframework.amqp.rabbit.annotation.RabbitListener;
|
||||
import org.springframework.amqp.rabbit.core.RabbitTemplate;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.dao.DataAccessException;
|
||||
import org.springframework.data.redis.connection.RedisConnection;
|
||||
import org.springframework.data.redis.core.RedisCallback;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.Set;
|
||||
import java.util.function.Function;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/15 19:20
|
||||
* @注释
|
||||
*/
|
||||
@RestController
|
||||
@Slf4j
|
||||
public class RabbitMQController {
|
||||
@Resource
|
||||
private RabbitTemplate rabbitTemplate;
|
||||
@Resource
|
||||
private RedisTemplate<String, Object> redisTemplate;
|
||||
|
||||
@PostMapping("/send")
|
||||
public String send(NotifyTypeEnum type, String msg) {
|
||||
rabbitTemplate.convertAndSend(RabbitMQConstants.EXCHANGE, RabbitMQConstants.ROUTER, new NotifyMsgEvent<>(null, type, msg));
|
||||
return "success";
|
||||
}
|
||||
|
||||
@RabbitListener(queues = RabbitMQConstants.QUEUE)
|
||||
void synBlogConsumer(Message msg , Channel channel) throws IOException {
|
||||
try {
|
||||
log.info("synBlogConsumer 接收到消息：{}", msg.toString());
|
||||
consumer(msg, "user1"); // 某个用户
|
||||
|
||||
channel.basicAck(msg.getMessageProperties().getDeliveryTag(), false);
|
||||
} catch (Exception e) {
|
||||
log.error("synBlogConsumer 接收消息失败：{}", e.getMessage());
|
||||
channel.basicNack(msg.getMessageProperties().getDeliveryTag(), false, true);
|
||||
}
|
||||
}
|
||||
|
||||
private final String oprKey = "oprLog";
|
||||
|
||||
/**
|
||||
* 消息重复消费？幂等性
|
||||
* @param message
|
||||
* @return
|
||||
*/
|
||||
public void consumer(Message message,String user) {
|
||||
// 转为NotifyMsgEvent
|
||||
NotifyMsgEvent<String> event = JSON.parseObject(message.getBody(), NotifyMsgEvent.class);
|
||||
|
||||
// 构造活跃度增加的通知
|
||||
}
|
||||
|
||||
|
||||
final String todayRankKey = DateUtil.format(DateUtil.date(), "yyyyMMdd");
|
||||
@PostMapping("/activity")
|
||||
public ResVo getRank(int k) {
|
||||
// 获取前k个
|
||||
Set<String> execute = redisTemplate.execute((RedisCallback<Set<String>>) connection -> {
|
||||
Set<String> collect = connection.zRange(todayRankKey.getBytes(), 0, k - 1).stream().map(new Function<byte[], String>() {
|
||||
@Override
|
||||
public String apply(byte[] bytes) {
|
||||
return new String(bytes);
|
||||
}
|
||||
}).collect(Collectors.toSet());
|
||||
return collect;
|
||||
});
|
||||
return ResVo.success(execute);
|
||||
}
|
||||
|
||||
public void incrDecrByActivity(ActivityScoreBo activityScore,String userId) {
|
||||
|
||||
if (userId == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
String field;
|
||||
int score = 0;
|
||||
if (activityScore.getPath() != null) { // 关于页面
|
||||
field = "path_" + activityScore.getPath();
|
||||
score = 1;
|
||||
} else if (activityScore.getArticleId() != null) { // 关于文章
|
||||
field = activityScore.getArticleId() + "_";
|
||||
if (activityScore.getPraise() != null) {
|
||||
field += "praise";
|
||||
score = BooleanUtils.isTrue(activityScore.getPraise()) ? 2 : -2;
|
||||
} else if (activityScore.getCollect() != null) {
|
||||
field += "collect";
|
||||
score = BooleanUtils.isTrue(activityScore.getCollect()) ? 2 : -2;
|
||||
} else if (activityScore.getRate() != null) {
|
||||
// 评论回复
|
||||
field += "rate";
|
||||
score = BooleanUtils.isTrue(activityScore.getRate()) ? 3 : -3;
|
||||
} else if (BooleanUtils.isTrue(activityScore.getPublishArticle())) {
|
||||
// 发布文章
|
||||
field += "publish";
|
||||
score += 10;
|
||||
}
|
||||
} else if (activityScore.getFollowedUserId() != null) { // 关于关注
|
||||
field = activityScore.getFollowedUserId() + "_follow";
|
||||
score = BooleanUtils.isTrue(activityScore.getFollow()) ? 2 : -2;
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
// 幂等性
|
||||
final String userActionKey = "ActivityCore:" + userId + DateUtil.format(DateUtil.date(), ":yyyyMMdd");
|
||||
|
||||
// {user:{action1,action2}}
|
||||
Integer opr = (Integer) redisTemplate.opsForHash().get(userActionKey, field);
|
||||
if (opr == null) { // 某个用户在之前是否做过field这个操作
|
||||
// 没有操作过
|
||||
// 加记录
|
||||
redisTemplate.opsForHash().put(userActionKey, field, score);
|
||||
redisTemplate.execute((RedisCallback<Object>) connection -> {
|
||||
connection.expire(userActionKey.getBytes(), 31 * 24 * 60 * 60); // 保存一个月
|
||||
return null;
|
||||
});
|
||||
|
||||
// 加分
|
||||
// 更新当天和当月的活跃度排行榜
|
||||
final String todayRankKey = DateUtil.format(DateUtil.date(), "yyyyMMdd");
|
||||
final String monthRankKey = DateUtil.format(DateUtil.date(), "yyyyMM");
|
||||
|
||||
|
||||
Double newAns = redisTemplate.execute(new RedisCallback<Double>() {
|
||||
@Override
|
||||
public Double doInRedis(RedisConnection connection) throws DataAccessException {
|
||||
return connection.zScore(todayRankKey.getBytes(), userId.getBytes());
|
||||
}
|
||||
});
|
||||
Object execute = redisTemplate.execute(new RedisCallback<Object>() {
|
||||
@Override
|
||||
public Object doInRedis(RedisConnection connection) throws DataAccessException {
|
||||
return connection.zScore(monthRankKey.getBytes(), userId.getBytes());
|
||||
}
|
||||
});
|
||||
|
||||
redisTemplate.execute((RedisCallback<Object>) connection -> {
|
||||
connection.expire(todayRankKey.getBytes(), 31 * 24 * 60 * 60); // 保存一个月
|
||||
return null;
|
||||
});
|
||||
redisTemplate.execute((RedisCallback<Object>) connection -> {
|
||||
connection.expire(monthRankKey.getBytes(), 31 * 24 * 60 * 60 * 12); // 保存一年
|
||||
return null;
|
||||
});
|
||||
} else if (opr > 0 && score < 0) {
|
||||
// 减分
|
||||
Long delete = redisTemplate.opsForHash().delete(userActionKey, field);
|
||||
if (delete == 1) {
|
||||
// 减分成功
|
||||
// 更新日、月排行榜
|
||||
final String todayRankKey = DateUtil.format(DateUtil.date(), "yyyyMMdd");
|
||||
final String monthRankKey = DateUtil.format(DateUtil.date(), "yyyyMM");
|
||||
|
||||
redisTemplate.opsForHash().increment(todayRankKey, userId, -score);
|
||||
redisTemplate.opsForHash().increment(monthRankKey, userId, -score);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Data
|
||||
@Accessors(chain = true)
|
||||
public class ActivityScoreBo {
|
||||
/**
|
||||
* 访问页面增加活跃度
|
||||
*/
|
||||
private String path;
|
||||
|
||||
/**
|
||||
* 目标文章
|
||||
*/
|
||||
private Long articleId;
|
||||
|
||||
/**
|
||||
* 评论增加活跃度
|
||||
*/
|
||||
private Boolean rate;
|
||||
|
||||
/**
|
||||
* 点赞增加活跃度
|
||||
*/
|
||||
private Boolean praise;
|
||||
|
||||
/**
|
||||
* 收藏增加活跃度
|
||||
*/
|
||||
private Boolean collect;
|
||||
|
||||
/**
|
||||
* 发布文章增加活跃度
|
||||
*/
|
||||
private Boolean publishArticle;
|
||||
|
||||
/**
|
||||
* 被关注的用户
|
||||
*/
|
||||
private Long followedUserId;
|
||||
|
||||
/**
|
||||
* 关注增加活跃度
|
||||
*/
|
||||
private Boolean follow;
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Configuration
|
||||
class RabbitMQConfig {
|
||||
@Bean
|
||||
Queue aqueue() {
|
||||
return QueueBuilder.durable(RabbitMQConstants.QUEUE)
|
||||
.ttl(1000).maxLength(5)
|
||||
.deadLetterExchange(RabbitMQConstants.EXCHANGE).deadLetterRoutingKey(RabbitMQConstants.FAIL_ROUTER)
|
||||
.build();
|
||||
}
|
||||
@Bean
|
||||
Queue failQueue() {
|
||||
return QueueBuilder.durable(RabbitMQConstants.FAIL_QUEUE)
|
||||
.build();
|
||||
}
|
||||
|
||||
@Bean
|
||||
Exchange commentExchange() {
|
||||
return ExchangeBuilder.directExchange(RabbitMQConstants.EXCHANGE).durable(true).build();
|
||||
}
|
||||
|
||||
@Bean
|
||||
Binding commentBinding() {
|
||||
return BindingBuilder.bind(aqueue()).to(commentExchange()).with(RabbitMQConstants.ROUTER).noargs();
|
||||
}
|
||||
}
|
||||
|
||||
class RabbitMQConstants{
|
||||
public static final String QUEUE = "queue";
|
||||
public static final String FAIL_QUEUE = "fail_queue";
|
||||
public static final String EXCHANGE = "exchange";
|
||||
public static final String ROUTER = "router";
|
||||
public static final String FAIL_ROUTER = "fail_router";
|
||||
}
|
||||
@ -0,0 +1,284 @@
|
||||
package cn.whaifree.springdemo.controller.wxQrLogin;
|
||||
|
||||
import cn.hutool.core.util.RandomUtil;
|
||||
import cn.hutool.jwt.JWT;
|
||||
import com.fasterxml.jackson.dataformat.xml.annotation.JacksonXmlProperty;
|
||||
import com.fasterxml.jackson.dataformat.xml.annotation.JacksonXmlRootElement;
|
||||
import com.google.common.cache.CacheBuilder;
|
||||
import com.google.common.cache.CacheLoader;
|
||||
import com.google.common.cache.LoadingCache;
|
||||
import jakarta.annotation.PostConstruct;
|
||||
import lombok.Data;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.springframework.beans.factory.annotation.Qualifier;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.Date;
|
||||
|
||||
/**
|
||||
* @version 1.0
|
||||
* @Author whai文海
|
||||
* @Date 2024/10/15 17:07
|
||||
* @注释
|
||||
*/
|
||||
@Slf4j
|
||||
@RestController
|
||||
@RequestMapping("/wxQrLogin")
|
||||
public class WxQrLoginController {
|
||||
|
||||
/**
|
||||
* 1. 客户端请求网站，/subscribe，响应SSEEmitter
|
||||
* 获取客户端deviceId
|
||||
* 生成验证码numCode
|
||||
* 存放Cache < deviceId,numCode >
|
||||
* 存放Cache < numCode, SSEEmitter>
|
||||
* SSE发送验证码给客户端
|
||||
*
|
||||
* 2. 客户端扫描公众号，发送numCode到公众号
|
||||
* wx会请求/callback接口，带有numCode
|
||||
* 使用numCode找到deviceId，再找到SSEEmitter，发送生成的最新Token
|
||||
*
|
||||
*/
|
||||
|
||||
|
||||
/**
|
||||
* sse的超时时间，默认15min
|
||||
*/
|
||||
private final static Long SSE_EXPIRE_TIME = 15 * 60 * 1000L;
|
||||
private final RedisTemplate redisTemplate;
|
||||
|
||||
|
||||
/**
|
||||
* key = 验证码, value = 长连接
|
||||
*/
|
||||
private LoadingCache<String, SseEmitter> verifyCodeCache;
|
||||
/**
|
||||
* key = 设备 value = 验证码
|
||||
*/
|
||||
private LoadingCache<String, String> deviceCodeCache;
|
||||
|
||||
public WxQrLoginController(@Qualifier("redisTemplate") RedisTemplate redisTemplate) {
|
||||
this.redisTemplate = redisTemplate;
|
||||
}
|
||||
|
||||
|
||||
@PostConstruct
|
||||
public void init() {
|
||||
// 验证码，SSE
|
||||
verifyCodeCache = CacheBuilder.newBuilder().build(new CacheLoader<String, SseEmitter>() {
|
||||
@Override
|
||||
public SseEmitter load(String key) throws Exception {
|
||||
// 如果缓存未命中，则抛出异常，提示缓存未命中
|
||||
throw new RuntimeException("no val: " + key);
|
||||
}
|
||||
});
|
||||
// 设备，验证码
|
||||
deviceCodeCache = CacheBuilder.newBuilder().build(new CacheLoader<String, String>() {
|
||||
@Override
|
||||
public String load(String key) throws Exception {
|
||||
// 生成id
|
||||
int cnt = 0;
|
||||
while (true) {
|
||||
String code = "deviceId#" + cnt++; // 可以是其他生成算法
|
||||
// 如果verifyCodeCache中已经有这个缓存，证明这个Code已经被使用了
|
||||
if (!verifyCodeCache.asMap().containsKey(code)) {
|
||||
return code;
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* deviceId code
|
||||
* code sse
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
@GetMapping(path = "subscribe", produces = {org.springframework.http.MediaType.TEXT_EVENT_STREAM_VALUE})
|
||||
public SseEmitter subscribe() throws IOException {
|
||||
String deviceId = String.valueOf(RandomUtil.randomInt(100)); // 随机生成一个UUID
|
||||
|
||||
String realCode = deviceCodeCache.getUnchecked(deviceId) ;
|
||||
// 生成验证码
|
||||
|
||||
|
||||
// fixme 设置15min的超时时间, 超时时间一旦设置不能修改；因此导致刷新验证码并不会增加连接的有效期
|
||||
SseEmitter sseEmitter = new SseEmitter(SSE_EXPIRE_TIME);
|
||||
SseEmitter oldSse = verifyCodeCache.getIfPresent(realCode); // 是否已经存在旧的半长连接
|
||||
|
||||
if (oldSse != null) {
|
||||
oldSse.complete(); // 旧的长连接
|
||||
}
|
||||
|
||||
|
||||
verifyCodeCache.put(realCode, sseEmitter);
|
||||
sseEmitter.onTimeout(() -> {
|
||||
log.info("sse 超时中断 --> {}", realCode);
|
||||
verifyCodeCache.invalidate(realCode);
|
||||
sseEmitter.complete();
|
||||
});
|
||||
sseEmitter.onError((e) -> {
|
||||
log.warn("sse error! --> {}", realCode, e);
|
||||
verifyCodeCache.invalidate(realCode);
|
||||
sseEmitter.complete();
|
||||
});
|
||||
// 若实际的验证码与前端显示的不同，则通知前端更新
|
||||
sseEmitter.send("initCode!");
|
||||
sseEmitter.send("init#" + realCode);
|
||||
return sseEmitter;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* fixme: 需要做防刷校验
|
||||
*
|
||||
*
|
||||
* 微信的响应返回
|
||||
* 本地测试访问:
|
||||
* curl -X POST 'http://localhost:8080/wx/callback'
|
||||
* -H 'content-type:application/xml' -d
|
||||
* '<xml>
|
||||
* <URL><![CDATA[https://hhui.top]]></URL>
|
||||
* <ToUserName><![CDATA[一灰灰blog]]></ToUserName>
|
||||
* <FromUserName><![CDATA[demoUser1234]]></FromUserName>
|
||||
* <CreateTime>1655700579</CreateTime>
|
||||
* <MsgType><![CDATA[text]]></MsgType>
|
||||
* <Content><![CDATA[login]]></Content>
|
||||
* <MsgId>11111111</MsgId>
|
||||
* </xml>' -i
|
||||
*
|
||||
* @param msg
|
||||
* @return 返回给微信，微信会给客户端
|
||||
*/
|
||||
@PostMapping(path = "callback",
|
||||
consumes = {"application/xml", "text/xml"},
|
||||
produces = "application/xml;charset=utf-8")
|
||||
public BaseWxMsgResVo callBack(@RequestBody WxTxtMsgReqVo msg) throws IOException {
|
||||
BaseWxMsgResVo res = new BaseWxMsgResVo();
|
||||
res.setToUserName(msg.getFromUserName());
|
||||
res.setFromUserName(msg.getToUserName());
|
||||
res.setCreateTime(System.currentTimeMillis());
|
||||
|
||||
String content = msg.getContent();
|
||||
if ("subscribe".equals(msg.getEvent()) || "scan".equalsIgnoreCase(msg.getEvent())) {
|
||||
String key = msg.getEventKey();
|
||||
if (StringUtils.isNotBlank(key) || key.startsWith("qrscene_")) {
|
||||
|
||||
// 带参数的二维码，扫描、关注事件拿到之后，直接登录，省却输入验证码这一步
|
||||
// fixme 带参数二维码需要 微信认证，个人公众号无权限
|
||||
String code = key.substring("qrscene_".length());
|
||||
|
||||
|
||||
// TODO sessionService.autoRegisterWxUserInfo(msg.getFromUserName());
|
||||
// 自动注册一个用户，获得用户ID
|
||||
|
||||
// 找到对应的SSE，实现登录
|
||||
SseEmitter sseEmitter = verifyCodeCache.getIfPresent(code);
|
||||
|
||||
// 生成Token
|
||||
String session = genSession(100L);
|
||||
// 登录成功，写入session
|
||||
sseEmitter.send(session);
|
||||
|
||||
// 设置cookie的路径
|
||||
sseEmitter.send("login#Session=" + session + ";path=/;"); // session告诉前端，跳转到/根目录
|
||||
|
||||
|
||||
return res;
|
||||
}
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
|
||||
|
||||
public String genSession(Long userId) {
|
||||
// 1.生成jwt格式的会话，内部持有有效期，用户信息
|
||||
String session = String.valueOf(userId);
|
||||
String token = JWT.create()
|
||||
.setIssuer("issuer")
|
||||
.setPayload("session", session)
|
||||
.setExpiresAt(new Date(System.currentTimeMillis() + 2592000000L)).sign();
|
||||
|
||||
// 2.使用jwt生成的token时，后端可以不存储这个session信息, 完全依赖jwt的信息
|
||||
// 但是需要考虑到用户登出，需要主动失效这个token，而jwt本身无状态，所以再这里的redis做一个简单的token -> userId的缓存，用于双重判定
|
||||
redisTemplate.opsForValue().set("UserId:Session:" + session, token);
|
||||
return token;
|
||||
}
|
||||
|
||||
|
||||
@Data
|
||||
@JacksonXmlRootElement(localName = "xml")
|
||||
public class WxTxtMsgReqVo {
|
||||
@JacksonXmlProperty(localName = "ToUserName")
|
||||
private String toUserName;
|
||||
@JacksonXmlProperty(localName = "FromUserName")
|
||||
private String fromUserName;
|
||||
@JacksonXmlProperty(localName = "CreateTime")
|
||||
private Long createTime;
|
||||
@JacksonXmlProperty(localName = "MsgType")
|
||||
private String msgType;
|
||||
@JacksonXmlProperty(localName = "Event")
|
||||
private String event;
|
||||
@JacksonXmlProperty(localName = "EventKey")
|
||||
private String eventKey;
|
||||
@JacksonXmlProperty(localName = "Ticket")
|
||||
private String ticket;
|
||||
@JacksonXmlProperty(localName = "Content")
|
||||
private String content;
|
||||
@JacksonXmlProperty(localName = "MsgId")
|
||||
private String msgId;
|
||||
@JacksonXmlProperty(localName = "MsgDataId")
|
||||
private String msgDataId;
|
||||
@JacksonXmlProperty(localName = "Idx")
|
||||
private String idx;
|
||||
}
|
||||
|
||||
|
||||
@Data
|
||||
@JacksonXmlRootElement(localName = "xml")
|
||||
public class BaseWxMsgResVo {
|
||||
|
||||
@JacksonXmlProperty(localName = "ToUserName")
|
||||
private String toUserName;
|
||||
@JacksonXmlProperty(localName = "FromUserName")
|
||||
private String fromUserName;
|
||||
@JacksonXmlProperty(localName = "CreateTime")
|
||||
private Long createTime;
|
||||
@JacksonXmlProperty(localName = "MsgType")
|
||||
private String msgType;
|
||||
}
|
||||
|
||||
|
||||
//
|
||||
// /**
|
||||
// * 初始化设备id
|
||||
// *
|
||||
// * @return
|
||||
// */
|
||||
// private String getOrInitDeviceId(HttpServletRequest request, HttpServletResponse response) {
|
||||
// String deviceId = request.getParameter("deviceId");
|
||||
// if (StringUtils.isNotBlank(deviceId) && !"null".equalsIgnoreCase(deviceId)) {
|
||||
// return deviceId;
|
||||
// }
|
||||
//
|
||||
// Cookie device = SessionUtil.findCookieByName(request, LoginOutService.USER_DEVICE_KEY);
|
||||
// if (device == null) {
|
||||
// deviceId = UUID.randomUUID().toString();
|
||||
// if (response != null) {
|
||||
// response.addCookie(SessionUtil.newCookie(LoginOutService.USER_DEVICE_KEY, deviceId));
|
||||
// }
|
||||
// return deviceId;
|
||||
// }
|
||||
// return device.getValue();
|
||||
// }
|
||||
|
||||
|
||||
|
||||
}
|
||||
@ -22,9 +22,9 @@ public class SelfFilter implements Filter {
|
||||
|
||||
@Override
|
||||
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
|
||||
System.out.println("SelfFilter doFilter");
|
||||
// System.out.println("SelfFilter doFilter");
|
||||
filterChain.doFilter(servletRequest, servletResponse);
|
||||
System.out.println("SelfFilter doFilter end");
|
||||
// System.out.println("SelfFilter doFilter end");
|
||||
}
|
||||
|
||||
@Override
|
||||
|
||||
@ -16,11 +16,24 @@ spring:
|
||||
redisson:
|
||||
file: classpath:redisson.yaml
|
||||
|
||||
|
||||
rabbitmq:
|
||||
host: localhost
|
||||
port: 5672
|
||||
username: whai
|
||||
password: whai
|
||||
publisher-confirm-type: correlated
|
||||
# 不可达到 返回给生产者
|
||||
# 当启用publisher-returns时，如果发送者发送的消息无法被消费者确认，消息会返回发送者。否则发送者是不知道的
|
||||
template:
|
||||
mandatory: true
|
||||
publisher-returns: true
|
||||
listener:
|
||||
simple:
|
||||
acknowledge-mode: manual
|
||||
direct:
|
||||
auto-startup: false
|
||||
acknowledge-mode: manual
|
||||
|
||||
|
||||
|
||||
|
||||
# springdoc-openapi项目配置
|
||||
@ -42,3 +55,10 @@ knife4j:
|
||||
language: zh_cn
|
||||
server:
|
||||
port: 8080
|
||||
image:
|
||||
oss:
|
||||
type: minio
|
||||
bucket: picgo
|
||||
ak: wOSfawBbzug2S3qz9u6W
|
||||
sk: CCxIopdXdBRNPloaFV7l8XplKpVLPzjSnMxlKcru
|
||||
endpoint: http://42.192.130.83:9000
|
||||
|
||||
Loading…
Reference in New Issue
Block a user