Commit d07827a3 authored by cherrylzhao's avatar cherrylzhao
Browse files

add SPIShardingHook for BaseShardingEngine.

parent 2d62fa10

sharding-core/sharding-core-entry/src/main/java/org/apache/shardingsphere/core/BaseShardingEngine.java

+19 −7
Original line number Diff line number Diff line
@@ -22,6 +22,7 @@ import org.apache.shardingsphere.api.hint.HintManager; 
import org.apache.shardingsphere.core.constant.DatabaseType;

import org.apache.shardingsphere.core.constant.properties.ShardingProperties;

import org.apache.shardingsphere.core.constant.properties.ShardingPropertiesConstant;

import org.apache.shardingsphere.core.hook.SPIShardingHook;

import org.apache.shardingsphere.core.metadata.ShardingMetaData;

import org.apache.shardingsphere.core.rewrite.SQLBuilder;

import org.apache.shardingsphere.core.rewrite.SQLRewriteEngine;
@@ -52,6 +53,8 @@ public abstract class BaseShardingEngine { 
    

    private final DatabaseType databaseType;

    

    private final SPIShardingHook shardingHook = new SPIShardingHook();

    

    /**

     * Shard.

     *
@@ -60,6 +63,8 @@ public abstract class BaseShardingEngine { 
     * @return SQL route result

     */

    public SQLRouteResult shard(final String sql, final List<Object> parameters) {

        shardingHook.start(sql);

        try {

            List<Object> clonedParameters = cloneParameters(parameters);

            SQLRouteResult result = route(sql, clonedParameters);

            result.getRouteUnits().addAll(HintManager.isDatabaseShardingOnly() ? convert(sql, clonedParameters, result) : rewriteAndConvert(sql, clonedParameters, result));
@@ -67,7 +72,14 @@ public abstract class BaseShardingEngine { 
                boolean showSimple = shardingProperties.getValue(ShardingPropertiesConstant.SQL_SIMPLE);

                SQLLogger.logSQL(sql, showSimple, result.getSqlStatement(), result.getRouteUnits());

            }

            shardingHook.finishSuccess(result, metaData.getTable());

            return result;

            // CHECKSTYLE:OFF

        } catch (final Exception ex) {

            // CHECKSTYLE:ON

            shardingHook.finishFailure(ex);

            throw ex;

        }

    }

    

    protected abstract List<Object> cloneParameters(List<Object> parameters);