From 719a386f87e069af35658844bde4fab218ed20e0 Mon Sep 17 00:00:00 2001
From: Jesse-Ma <24167796@qq.com>
Date: Tue, 7 Mar 2023 17:56:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=97=B6=E9=97=B4=E7=BB=9F=E8=AE=A1=20?=
 =?UTF-8?q?=E5=92=8C=20=E5=86=85=E5=AD=98=E7=BC=93=E5=AD=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                       |  4 ++
 .../java/com/flagnote/note/aop/AopConfig.java | 42 +++++++++++++++++++
 .../java/com/flagnote/note/aop/CountTime.java | 12 ++++++
 .../com/flagnote/note/cache/NoteCache.java    | 27 ++++++++++++
 .../flagnote/note/schedule/NoteSchedule.java  |  4 +-
 .../note/service/NoteServiceImpl.java         | 19 +++++++--
 src/main/resources/application.yml            |  8 ++++
 7 files changed, 112 insertions(+), 4 deletions(-)
 create mode 100644 src/main/java/com/flagnote/note/aop/AopConfig.java
 create mode 100644 src/main/java/com/flagnote/note/aop/CountTime.java
 create mode 100644 src/main/java/com/flagnote/note/cache/NoteCache.java

diff --git a/pom.xml b/pom.xml
index fb93dba..0034e30 100644
--- a/pom.xml
+++ b/pom.xml
@@ -34,6 +34,10 @@
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-web</artifactId>
 		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-aop</artifactId>
+		</dependency>
 		<!-- 
 		<dependency>
 			<groupId>org.springframework.cloud</groupId>
diff --git a/src/main/java/com/flagnote/note/aop/AopConfig.java b/src/main/java/com/flagnote/note/aop/AopConfig.java
new file mode 100644
index 0000000..96559b8
--- /dev/null
+++ b/src/main/java/com/flagnote/note/aop/AopConfig.java
@@ -0,0 +1,42 @@
+package com.flagnote.note.aop;
+
+import java.util.concurrent.TimeUnit;
+
+import org.aspectj.lang.JoinPoint;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.annotation.AfterReturning;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.aspectj.lang.annotation.Before;
+import org.aspectj.lang.annotation.Pointcut;
+import org.springframework.stereotype.Component;
+import org.springframework.util.StopWatch;
+
+import lombok.extern.slf4j.Slf4j;
+
+@Aspect
+@Component
+@Slf4j
+public class AopConfig {
+    @Pointcut("@annotation(com.flagnote.note.aop.CountTime)")
+    public void logTime() {
+
+    }
+    
+    private ThreadLocal<StopWatch> stopWatchLocal = new ThreadLocal<>();
+    
+    @Before("logTime()")
+    public void before() {
+    	StopWatch stopWatch = new StopWatch();
+    	stopWatchLocal.set(stopWatch);
+    	stopWatch.start();
+    }
+    
+    @AfterReturning(value = "logTime()", returning = "val")
+    public void afterReturning(JoinPoint joinPoint, Object val) {
+    	StopWatch stopWatch = stopWatchLocal.get();
+    	stopWatch.stop();
+        log.info("method:"+ joinPoint.getSignature().getName()+">>> " + stopWatch.getLastTaskTimeMillis() + " ms.");
+    }
+
+}
\ No newline at end of file
diff --git a/src/main/java/com/flagnote/note/aop/CountTime.java b/src/main/java/com/flagnote/note/aop/CountTime.java
new file mode 100644
index 0000000..70dc500
--- /dev/null
+++ b/src/main/java/com/flagnote/note/aop/CountTime.java
@@ -0,0 +1,12 @@
+package com.flagnote.note.aop;
+
+import java.lang.annotation.Retention;
+import java.lang.annotation.Target;
+
+import static java.lang.annotation.ElementType.METHOD;
+import static java.lang.annotation.RetentionPolicy.RUNTIME;
+
+@Retention(RUNTIME)
+@Target(METHOD)
+public @interface CountTime {
+}
\ No newline at end of file
diff --git a/src/main/java/com/flagnote/note/cache/NoteCache.java b/src/main/java/com/flagnote/note/cache/NoteCache.java
new file mode 100644
index 0000000..e23d381
--- /dev/null
+++ b/src/main/java/com/flagnote/note/cache/NoteCache.java
@@ -0,0 +1,27 @@
+package com.flagnote.note.cache;
+
+import com.flagnote.note.entity.Note;
+
+import cn.hutool.cache.CacheUtil;
+import cn.hutool.cache.impl.LFUCache;
+
+public class NoteCache {
+
+	private static final LFUCache<String, Note> CACHE = CacheUtil.newLFUCache(256,30*60*1000);
+
+	public static Note getNote(String mixKey) {
+		return CACHE.get(mixKey);
+	}
+
+	public static void save(Note note) {
+		CACHE.put(note.getId(), note);
+	}
+
+	public static void deleteNote(String mixKey) {
+		Note note = CACHE.get(mixKey);
+		if(null!=note) {
+			note.setState(0);
+			note.setTextBytes(null);
+		}
+	}
+}
diff --git a/src/main/java/com/flagnote/note/schedule/NoteSchedule.java b/src/main/java/com/flagnote/note/schedule/NoteSchedule.java
index 4be2755..3acc1b8 100644
--- a/src/main/java/com/flagnote/note/schedule/NoteSchedule.java
+++ b/src/main/java/com/flagnote/note/schedule/NoteSchedule.java
@@ -12,6 +12,7 @@ import org.springframework.data.domain.Pageable;
 import org.springframework.scheduling.annotation.EnableScheduling;
 import org.springframework.scheduling.annotation.Scheduled;
 
+import com.flagnote.note.cache.NoteCache;
 import com.flagnote.note.entity.Note;
 import com.flagnote.note.repository.NoteMongoRepository;
 
@@ -22,7 +23,7 @@ public class NoteSchedule {
 	@Autowired
 	private NoteMongoRepository noteMongoRepository;
 
-	@Scheduled(cron = "0/10 * * * * ?")
+	@Scheduled(cron = "*/30 * * * * ?")
     public void updateExpiredNote() {
 		
 		while(true) {
@@ -34,6 +35,7 @@ public class NoteSchedule {
 				for (Note note : noteList) {
 					note.setState(0);
 					note.setTextBytes(null);
+					NoteCache.deleteNote(note.getId());
 				}
 				noteMongoRepository.saveAll(noteList);
 			}else {
diff --git a/src/main/java/com/flagnote/note/service/NoteServiceImpl.java b/src/main/java/com/flagnote/note/service/NoteServiceImpl.java
index 98ec6b8..2433f8b 100644
--- a/src/main/java/com/flagnote/note/service/NoteServiceImpl.java
+++ b/src/main/java/com/flagnote/note/service/NoteServiceImpl.java
@@ -9,6 +9,8 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.Assert;
 
+import com.flagnote.note.aop.CountTime;
+import com.flagnote.note.cache.NoteCache;
 import com.flagnote.note.entity.Note;
 import com.flagnote.note.repository.NoteMongoRepository;
 import com.flagnote.note.utils.BizKeyUtils;
@@ -26,12 +28,16 @@ public class NoteServiceImpl implements NoteService {
 	private NoteMongoRepository noteMongoRepository;
 
 	@Override
+	@CountTime
 	public Note getNote(String key) {
 
 		String mixKey = BizKeyUtils.mixKey(key);
 
-		Note note = noteMongoRepository.findById(mixKey).orElse(null);
-
+		Note note = NoteCache.getNote(mixKey);
+		if (null == note) {
+			note = noteMongoRepository.findById(mixKey).orElse(null);
+		}
+		
 		if (null == note) {
 			return null;
 		}
@@ -52,6 +58,7 @@ public class NoteServiceImpl implements NoteService {
 	}
 
 	@Override
+	@CountTime
 	public String getNoteText(String key) {
 		Note note = this.getNote(key);
 		if (null == note || ArrayUtil.isEmpty(note.getTextBytes())) {
@@ -61,6 +68,7 @@ public class NoteServiceImpl implements NoteService {
 	}
 
 	@Override
+	@CountTime
 	public void saveNote(Note note) {
 		String mixKey = BizKeyUtils.mixKey(note.getKey());
 
@@ -78,6 +86,8 @@ public class NoteServiceImpl implements NoteService {
 
 		if (null == exists) {
 			noteMongoRepository.save(note);
+			
+			NoteCache.save(note);
 		} else {
 			if (!exists.getMd5().equals(note.getMd5())) {
 				throw new RuntimeException("E:800001");
@@ -88,8 +98,11 @@ public class NoteServiceImpl implements NoteService {
 	@Override
 	public void deleteNote(String key) {
 		String mixKey = BizKeyUtils.mixKey(key);
+		
+		NoteCache.deleteNote(mixKey);
+		
 		Note note = noteMongoRepository.findById(mixKey).orElse(null);
-		if (null != note && ArrayUtil.isNotEmpty(note.getTextBytes())) {
+		if (null != note) {
 			note.setState(0);
 			note.setTextBytes(null);
 			noteMongoRepository.save(note);
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 29f3a22..51ac2fc 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -40,3 +40,11 @@ management:
       exposure:
         include: shutdown,health
         
+logging:
+  level:
+    org:
+      springframework:
+        data:
+          mongodb:
+            core:
+              MongoTemplate: DEBUG
\ No newline at end of file