为LogDB添加本地线程变量做记录

zen-api/src/main/kotlin/com/gxzc/zen/logging/MQLogHelper.kt

@@ -0,0 +1,44 @@
+package com.gxzc.zen.logging
+
+import com.gxzc.zen.logging.model.MQLog
+
+/**
+ * MQLog 帮助类 (即时性) <br>
+ * 实现手动配置Ignore,remark等
+ * @author NorthLan
+ * @date 2018/9/2
+ * @url https://noahlan.com
+ */
+object MQLogHelper {
+    private val LOCAL_LOG = ThreadLocal<MQLog>()
+
+    private fun setLocalLog(log: MQLog) {
+        LOCAL_LOG.set(log)
+    }
+
+    /**
+     * 获取MQLog参数
+     */
+    fun getLocalLog(): MQLog? {
+        return LOCAL_LOG.get()
+    }
+
+    /**
+     * 清除本地变量(线程局部变量)
+     */
+    fun clearLog() {
+        LOCAL_LOG.remove()
+    }
+
+    /**
+     * 开始配置Log信息
+     */
+    fun configLog(ignore: Boolean = false, remark: String? = null) {
+        val log = MQLog(ignore, remark)
+        setLocalLog(log)
+    }
+
+    fun isConfigurable(): Boolean {
+        return getLocalLog() != null
+    }
+}

zen-api/src/main/kotlin/com/gxzc/zen/logging/aop/LogDBInterceptor.kt

@@ -2,6 +2,7 @@ package com.gxzc.zen.logging.aop
 
 import com.alibaba.dubbo.common.logger.LoggerFactory
 import com.baomidou.mybatisplus.toolkit.PluginUtils
+import com.gxzc.zen.logging.MQLogHelper
 import com.gxzc.zen.logging.constants.LogConstants
 import com.gxzc.zen.logging.util.MQLogUtil
 import net.sf.jsqlparser.parser.CCJSqlParserUtil
@@ -31,7 +32,7 @@ import java.util.concurrent.ConcurrentHashMap
  */
 @Intercepts(Signature(type = StatementHandler::class, method = "prepare", args = [Connection::class, Integer::class]))
 @Component
-open class LogDBInterceptor : Interceptor {
+class LogDBInterceptor : Interceptor {
     companion object {
         private val logger = LoggerFactory.getLogger(LogDBInterceptor::class.java)
 
@@ -40,68 +41,97 @@ open class LogDBInterceptor : Interceptor {
     }
 
     override fun intercept(invocation: Invocation): Any {
-        logger.debug("操作数据库日志记录开始...")
-        val startTime = Date()
-        val statementHandler = PluginUtils.realTarget(invocation.target) as StatementHandler
-        val metaObject = SystemMetaObject.forObject(statementHandler)
-        // 获取SQL操作类型
-        val mappedStatement = metaObject.getValue("delegate.mappedStatement") as MappedStatement
-        val boundSql = metaObject.getValue("delegate.boundSql") as BoundSql
-        if (boundSql.sql.startsWith("show table")) {
-            return invocation.proceed()
-        }
-
-        // 消息构造
-        val type: Int
-        val sql = boundSql.sql
-
-
-        val statement = CCJSqlParserUtil.parse(sql)
-
-        var dbName: String? = null
-        var tableName: String? = null
-        val table: Table?
-
-        // 操作类型
-        when (mappedStatement.sqlCommandType) {
-            SqlCommandType.SELECT -> {
-                val plainSelect = (statement as Select).selectBody as PlainSelect
-                table = (plainSelect.fromItem as Table)
-                type = LogConstants.DB_TYPE_SELECT
+        // 判定是否需要执行Log记录
+        try {
+            var needLog = true
+            if (MQLogHelper.isConfigurable()) {
+                val mqLog = MQLogHelper.getLocalLog()!!
+                needLog = !mqLog.ignore
             }
-            SqlCommandType.DELETE -> {
-                table = (statement as Delete).table
-                type = LogConstants.DB_TYPE_DELETE
-            }
-            SqlCommandType.INSERT -> {
-                table = (statement as Insert).table
-                type = LogConstants.DB_TYPE_INSERT
+            // 若不需要记录Log信息则直接返回
+            if (!needLog) {
+                return invocation.proceed()
             }
-            SqlCommandType.UPDATE -> {
-                table = (statement as Update).tables[0]
-                type = LogConstants.DB_TYPE_UPDATE
+            logger.debug("操作数据库日志记录开始...")
+            val startTime = Date()
+            val statementHandler = PluginUtils.realTarget(invocation.target) as StatementHandler
+            val metaObject = SystemMetaObject.forObject(statementHandler)
+            // 获取SQL操作类型
+            val mappedStatement = metaObject.getValue("delegate.mappedStatement") as MappedStatement
+            val boundSql = metaObject.getValue("delegate.boundSql") as BoundSql
+            if (boundSql.sql.startsWith("show table")) {
+                return invocation.proceed()
             }
-            else -> {
-                // 直接跳过
+            // 消息构造
+            val type: Int
+            val sql = boundSql.sql
+
+            val statement = try {
+                CCJSqlParserUtil.parse(sql)
+            } catch (e: Throwable) {
+                // 解析出错直接交给后续处理,不再记录Log了
+                logger.error("sql解析出错,不再发送记录Log消息. cause ", e)
                 return invocation.proceed()
             }
-        }
 
-        if (table != null) {
-            val tableArr = table.name.split("\\.")
-            if (tableArr.size == 1) {
-                tableName = tableArr[0]
-            } else {
-                dbName = tableArr[0]
-                tableName = tableArr[1]
+            var dbName: String? = null
+            var tableName: String? = null
+            val table: Table?
+
+            // 操作类型
+            when (mappedStatement.sqlCommandType) {
+                SqlCommandType.SELECT -> {
+                    val plainSelect = (statement as Select).selectBody as PlainSelect
+                    table = (plainSelect.fromItem as Table)
+                    type = LogConstants.DB_TYPE_SELECT
+                }
+                SqlCommandType.DELETE -> {
+                    table = (statement as Delete).table
+                    type = LogConstants.DB_TYPE_DELETE
+                }
+                SqlCommandType.INSERT -> {
+                    table = (statement as Insert).table
+                    type = LogConstants.DB_TYPE_INSERT
+                }
+                SqlCommandType.UPDATE -> {
+                    table = (statement as Update).tables[0]
+                    type = LogConstants.DB_TYPE_UPDATE
+                }
+                else -> {
+                    // 直接跳过
+                    return invocation.proceed()
+                }
             }
+
+            if (table != null) {
+                val tableArr = table.name.split("\\.")
+                if (tableArr.size == 1) {
+                    tableName = tableArr[0]
+                } else {
+                    dbName = tableArr[0]
+                    tableName = tableArr[1]
+                }
+            }
+            // 执行
+            val result = invocation.proceed()
+
+            // 发消息
+            MQLogUtil.logDB(
+                    type,
+                    sql,
+                    tableName,
+                    getTableNameCN(tableName, invocation.args[0] as Connection, dbName),
+                    if (MQLogHelper.isConfigurable()) {
+                        MQLogHelper.getLocalLog()!!.remark
+                    } else {
+                        null
+                    },
+                    startTime)
+            return result
+        } finally {
+            MQLogHelper.clearLog()
         }
-        // 执行
-        val result = invocation.proceed()
 
-        // 发消息
-        MQLogUtil.logDB(type, sql, tableName, getTableNameCN(tableName, invocation.args[0] as Connection, dbName), startTime)
-        return result
     }
 
     override fun plugin(target: Any): Any {

zen-api/src/main/kotlin/com/gxzc/zen/logging/model/MQLog.kt

@@ -0,0 +1,12 @@
+package com.gxzc.zen.logging.model
+
+/**
+ * MQLog 数据类
+ * @author NorthLan
+ * @date 2018/9/2
+ * @url https://noahlan.com
+ */
+data class MQLog(
+        val ignore: Boolean = false, // 是否忽略此条Log信息
+        val remark: String? = null // 此条Log信息的remark
+)

zen-api/src/main/kotlin/com/gxzc/zen/logging/util/MQLogUtil.kt

@@ -64,7 +64,7 @@ object MQLogUtil {
     /**
      * 发送DB日志消息
      */
-    fun logDB(type: Int, sql: String?, tableId: String?, tableName: String?, startTime: Date, endTime: Date = Date()) {
+    fun logDB(type: Int, sql: String?, tableId: String?, tableName: String?, remark: String?, startTime: Date, endTime: Date = Date()) {
         val logDB = LogDB().apply {
             this.platformId = PlatformUtil.getPlatformId()
             this.operatorStartTime = startTime
@@ -75,6 +75,7 @@ object MQLogUtil {
             this.sql = sql
             this.tableId = tableId
             this.tableName = tableName
+            this.remark = remark
         }
         mqProducer.asyncSend(MessageBuilder.of(logDB).topic(MQConstants.TOPIC_LOGS).tag(MQConstants.TAGS_LOGS_DB).build(),
                 object : SendCallback {