nbcp.myoql.db.cache.RedisCacheAopService.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of ktmyoql Show documentation
Show all versions of ktmyoql Show documentation
kotlin orm -- mysql,mongo , just like ktorm
The newest version!
package nbcp.myoql.db.cache
import nbcp.base.comm.JsonMap
import nbcp.base.comm.StringKeyMap
import nbcp.base.comm.config
import nbcp.base.extend.*
import nbcp.base.extend.*
import nbcp.base.utils.ClassUtil
import nbcp.myoql.annotation.*
import nbcp.myoql.db.db
import org.aspectj.lang.ProceedingJoinPoint
import org.aspectj.lang.annotation.Around
import org.aspectj.lang.annotation.Aspect
import org.aspectj.lang.reflect.MethodSignature
import org.slf4j.LoggerFactory
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass
import org.springframework.data.redis.core.RedisTemplate
import org.springframework.scheduling.annotation.Scheduled
import org.springframework.scheduling.support.CronExpression
import org.springframework.stereotype.Component
import java.lang.reflect.Method
import java.time.Duration
import java.time.LocalDateTime
/**
* #数字表示参数名，如: #0 == userName
* #参数值 如：#userName
* ##methodName, ##returnType ##result
*/
@Aspect
@Component
@ConditionalOnClass(RedisTemplate::class)
open class RedisCacheAopService {
class LockJobModel(
var name: String = "",
var lastRedisTryLockAt: LocalDateTime? = null,
var lastRedisLockResult: Boolean = false,
var lastRedisLockErrorMsg: String = "",
var lastExecuteAt: LocalDateTime? = null,
var lastExecuteErrorMsg: String = "",
var lastRedisDeleteLockErrorMsg: String = ""
)
companion object {
private val logger = LoggerFactory.getLogger(this::class.java.declaringClass)
val jobMonitor = StringKeyMap()
fun getRequestParamFullUrl(request: Any): String {
val type = ClassUtil.loadClass("javax.servlet.http.HttpServletRequest")
val requestClass = request::class.java
if (type.isAssignableFrom(requestClass) == false) return "";
val getRequestURI = requestClass.getMethod("getRequestURI")
val getQueryString = requestClass.getMethod("getQueryString")
return getRequestURI.invoke(request).toString() + "?" + getQueryString.invoke(request).AsString()
}
}
//@annotation 表示拦截方法级别上的注解。
//@within 表示拦截类级别上的注解。
/**
* RedisCache
* 使用 scan 遍历key.
* insert破坏: 所有 join_tabls，主表
*/
@Around("@annotation(nbcp.myoql.annotation.FromRedisCache)")
fun cacheSelect(joinPoint: ProceedingJoinPoint): Any? {
val signature = joinPoint.signature as MethodSignature;
val method = signature.method
val cache = method.getAnnotationsByType(FromRedisCache::class.java).firstOrNull()
val args = joinPoint.args
if (cache == null || cache.getTableName().isEmpty() || cache.cacheSeconds <= 0) {
return joinPoint.proceed(args)
}
//如果有 HttpRequest,则添加Url
var hasHttpRequest = false;
val variableMap = JsonMap(
method.parameters
.filter {
if (it.type.AnySuperClass { it.name.startsWith("javax.servlet.") }) {
hasHttpRequest = true;
return@filter false;
}
return@filter true;
}
.mapIndexed { index, it -> it.name to args.get(index) }
);
var ext = "";
if (cache.sql.isEmpty()) {
ext = signature.declaringType.name + "." + method.name;
try {
if (hasHttpRequest) {
val httpContext = ClassUtil.loadClass("nbcp.base.mvc.HttpContext")
ext += getRequestParamFullUrl(httpContext.getMethod("getRequest").invoke(null));
}
} catch (e: Exception) {
logger.error("在Web环境下找不到 HttpContext.request，忽略缓存中的路径", e)
}
if (variableMap.any()) {
ext += ":" + variableMap.values.ToJson()
}
}
return cache
.resolveWithVariable(variableMap, ext)
.getJson(signature.returnType, {
return@getJson joinPoint.proceed(args)
});
}
/**
* MasterAlternateMap
*/
@Around("@annotation(nbcp.myoql.annotation.BrokeRedisCache)")
fun cacheBroke(joinPoint: ProceedingJoinPoint): Any? {
val signature = joinPoint.signature as MethodSignature;
val method = signature.method
val cache = method.getAnnotationsByType(BrokeRedisCache::class.java).firstOrNull()
val args = joinPoint.args
// if (cache != null && cache.table.HasValue) {
// brokeCache(method, args, cache);
// }
val ret = joinPoint.proceed(args)
if (cache != null && cache.getTableName().HasValue) {
brokeCache(method, args, cache);
}
return ret;
}
private fun brokeCache(method: Method, args: Array, cache: BrokeRedisCache) {
val variableMap = JsonMap(
method.parameters
.filter {
if (it.type.AnySuperClass { it.name == "javax.servlet.ServletRequest" }) {
return@filter false;
}
return@filter true;
}
.mapIndexed { index, it -> it.name to args.get(index) }
);
cache.resolveWithVariable(variableMap).brokeCache();
}
@Around("@annotation(org.springframework.scheduling.annotation.Scheduled)")
fun oneTask(joinPoint: ProceedingJoinPoint): Any? {
val signature = joinPoint.signature as MethodSignature;
var method = signature.method
val key = listOf(
config.applicationName,
signature.declaringType.name + "." + method.name
)
.filter { it.HasValue }
.joinToString(":")
val now = LocalDateTime.now();
var cacheTime = 0;
var scheduled = method.getAnnotationsByType(Scheduled::class.java).first()
if (scheduled.cron.HasValue) {
var cornExp = CronExpression.parse(scheduled.cron)
var timeSpan = cornExp.next(now)!! - now;
cacheTime = timeSpan.seconds.AsInt();
} else if (scheduled.fixedDelay > 0) {
cacheTime = (scheduled.fixedDelay / 1000).AsInt();
} else if (scheduled.fixedRate > 0) {
cacheTime = (scheduled.fixedRate / 1000).AsInt();
}
if (cacheTime < 1) {
cacheTime = 1;
}
var jobModel = LockJobModel(key, LocalDateTime.now());
var setted = false;
try {
//如果存在,则查看时间
var v = db.rerBase.taskLock(key).get();
if (v.HasValue) {
if (Duration.between(v.AsLocalDateTime(), now).seconds > 2 * cacheTime) {
logger.debug("Redis锁 ${key} 超时,将执行新的任务.")
try {
db.rerBase.taskLock(key).deleteKey();
} catch (e: Exception) {
logger.error(e.message, e);
jobModel.lastRedisDeleteLockErrorMsg = e.javaClass.name + " " + e.message.AsString()
}
}
}
setted = db.rerBase.taskLock(key)
.setIfAbsent(now.toNumberString());
} catch (e: Exception) {
logger.error(e.message, e);
jobModel.lastRedisLockErrorMsg = e.javaClass.name + " " + e.message.AsString()
return null;
} finally {
if (setted == false) {
logger.debug("未能获取到锁 ${key}")
}
jobModel.lastRedisLockResult = setted;
addJobMonitor(jobModel);
}
if (setted == false) {
return null;
}
jobModel.lastExecuteAt = LocalDateTime.now();
try {
return joinPoint.proceed(joinPoint.args)
} catch (e: Exception) {
logger.error(e.message, e);
jobModel.lastExecuteErrorMsg = e.javaClass.name + " " + e.message.AsString()
return null;
} finally {
try {
db.rerBase.taskLock(key).deleteKey();
} catch (e: Exception) {
logger.error(e.message, e);
jobModel.lastRedisDeleteLockErrorMsg = e.javaClass.name + " " + e.message.AsString()
}
addJobMonitor(jobModel);
}
}
private fun addJobMonitor(lockJobModel: LockJobModel) {
if (jobMonitor.contains(lockJobModel.name)) {
jobMonitor.remove(lockJobModel.name)
}
jobMonitor.put(lockJobModel.name, lockJobModel);
}
}