🎨 feat: 缓存优化

spcookie · Jan 24, 2024 · 8ad92a1 · 8ad92a1
1 parent af0859c
commit 8ad92a1
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 2 deletions.
diff --git a/tokisaki-server/src/main/kotlin/io/micro/server/auth/domain/repository/IAuthRepository.kt b/tokisaki-server/src/main/kotlin/io/micro/server/auth/domain/repository/IAuthRepository.kt
@@ -38,6 +38,8 @@ interface IAuthRepository {
 
     fun findAuthorityByExample(authorityDO: AuthorityDO): Uni<List<AuthorityDO>>
 
+    fun findAuthorityCacheByExample(authorityDO: AuthorityDO): Uni<List<AuthorityDO>>
+
     fun saveAuthority(authorityDO: AuthorityDO): Uni<AuthorityDO>
 
 }
diff --git a/tokisaki-server/src/main/kotlin/io/micro/server/auth/domain/service/AuthService.kt b/tokisaki-server/src/main/kotlin/io/micro/server/auth/domain/service/AuthService.kt
@@ -23,6 +23,8 @@ interface AuthService {
 
     fun getAuthorityByCode(code: String): Uni<AuthorityDO>
 
+    fun getAuthorityCacheByCode(code: String): Uni<AuthorityDO>
+
     fun addAuthority(authorityDO: AuthorityDO): Uni<AuthorityDO>
 
 }
diff --git a/tokisaki-server/src/main/kotlin/io/micro/server/auth/domain/service/impl/AuthServiceImpl.kt b/tokisaki-server/src/main/kotlin/io/micro/server/auth/domain/service/impl/AuthServiceImpl.kt
@@ -70,6 +70,18 @@ class AuthServiceImpl(private val authRepository: IAuthRepository) : AuthService
             }
     }
 
+    @WithSession
+    override fun getAuthorityCacheByCode(code: String): Uni<AuthorityDO> {
+        return authRepository.findAuthorityCacheByExample(AuthorityDO().also { it.value = code })
+            .map {
+                if (it.isNotEmpty()) {
+                    it.first()
+                } else {
+                    null
+                }
+            }
+    }
+
     @WithTransaction
     override fun addAuthority(authorityDO: AuthorityDO): Uni<AuthorityDO> {
         return authRepository.findAuthorityByExample(AuthorityDO().also { it.value = authorityDO.value })

diff --git a/tokisaki-server/src/main/kotlin/io/micro/server/auth/infra/repository/AuthRepository.kt b/tokisaki-server/src/main/kotlin/io/micro/server/auth/infra/repository/AuthRepository.kt
@@ -9,6 +9,7 @@ import io.micro.server.auth.domain.repository.IAuthRepository
 import io.micro.server.auth.infra.converter.AuthConverter
 import io.micro.server.auth.infra.dao.impl.AuthorityDAO
 import io.micro.server.auth.infra.dao.impl.UserDAO
+import io.quarkus.cache.CacheResult
 import io.quarkus.panache.common.Page
 import io.smallrye.mutiny.Uni
 import jakarta.enterprise.context.ApplicationScoped
@@ -94,6 +95,11 @@ class AuthRepository(
             .map { it.converter(authConverter::authorityEntity2authorityDO) }
     }
 
+    @CacheResult(cacheName = "authority")
+    override fun findAuthorityCacheByExample(authorityDO: AuthorityDO): Uni<List<AuthorityDO>> {
+        return findAuthorityByExample(authorityDO)
+    }
+
     override fun saveAuthority(authorityDO: AuthorityDO): Uni<AuthorityDO> {
         val entity = authConverter.authorityDO2authorityEntity(authorityDO).also { it.id = null }
         return authorityDAO.persist(entity).map(authConverter::authorityEntity2authorityDO)

diff --git a/...aki-server/src/main/kotlin/io/micro/server/robot/domain/model/entity/FeatureFunctionDO.kt b/...aki-server/src/main/kotlin/io/micro/server/robot/domain/model/entity/FeatureFunctionDO.kt
@@ -37,6 +37,10 @@ class FeatureFunctionDO {
 
     var memberId: Long? = null
 
+    var isMenu: Boolean = false
+
+    var isUndefined: Boolean = false
+
     companion object {
         const val BRACES = "{}"
     }

diff --git a/tokisaki-server/src/main/kotlin/io/micro/server/robot/domain/model/entity/RobotDO.kt b/tokisaki-server/src/main/kotlin/io/micro/server/robot/domain/model/entity/RobotDO.kt
@@ -66,7 +66,7 @@ class RobotDO : BaseDomainEntity() {
                         it.args += args
                     }
             } else {
-                null
+                FeatureFunctionDO().also { it.isUndefined = true }
             }
         } else {
             null

diff --git a/...rver/src/main/kotlin/io/micro/server/robot/domain/service/impl/RobotManagerServiceImpl.kt b/...rver/src/main/kotlin/io/micro/server/robot/domain/service/impl/RobotManagerServiceImpl.kt
@@ -280,7 +280,7 @@ class RobotManagerServiceImpl(
             val featureFunction = getFeatureFunction(event, latestRobot)
             if (featureFunction != null) {
                 val authority = sessionFactory.withSession {
-                    authService.getAuthorityByCode(featureFunction.code!!)
+                    authService.getAuthorityCacheByCode(featureFunction.code!!)
                 }.runSubscriptionOn { vertxContext.runOnContext(it) }.awaitSuspending()
                 if (authority != null && authority.enabled == true) {
                     val switch = sessionFactory.withSession {