注解方式实现读写分离

注解方式实现读写分离

       在web应用开发过程中，数据库的压力会越来越大，那么就避免不了，要提供一个比较可行的方案，数据库的读写分离。

   读写分离的机制：在从库读取数据，在主库写数据。

1.自定义数据库数据源类型注解：

@Target( ElementType.METHOD)
@Retention( RetentionPolicy.RUNTIME)
@Inherited
public @interface DBProfile
{
String dbName() default "";
}

2.在方法层面上调用：

@DBProfile(dbName = Constants.SLAVE)

3.用ThreadLocal，定义数据源的类型（主，从）

public class DataSourceContextHolder
{
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
/**
* @Description: 设置数据源类型
* @param dataSourceType  数据库类型
* @return void
* @throws
*/
public static void setDataSourceType(String dataSourceType)
{
contextHolder.set(dataSourceType);
}

/**
* @Description: 获取数据源类型
* @param
* @return String
* @throws
*/
public static String getDataSourceType()
{
return contextHolder.get();
}

/**
* @Description: 清除数据源类型
* @param
* @return void
* @throws
*/
public static void clearDataSourceType()
{
contextHolder.remove();
}
}

4.定义动态数据源类

public class DynamicDataSource extends AbstractRoutingDataSource
{
@Override
protected Object determineCurrentLookupKey()
{
return DataSourceContextHolder.getDataSourceType();
}
}

5.在xml中配置，

DynamicDataSource

<bean id="dynamicDataSource" class="com.xx.xx.business.service.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry value-ref="dataSource"   key="Master"/>
<entry value-ref="dataSource"  key="Slave"/>
<!-- <entry value-ref="businessDataSourceOther"  key="Slave"/> -->
</map>
</property>
<!--默认数据源-->
<property name="defaultTargetDataSource" ref="dataSource"/>
</bean>
<bean id="dataSourceMaster" class="dbcp.xxx">

<bean id="dataSourceSlaver" class="dbcp.xxx">

6.定义切面

@Component
@Aspect
public class DynamicDataSourceAspect
{

@Pointcut( "@annotation(com.xx.xx.xx.common.aspect.DBProfile)" )
public void DynamicDataSource()
{

}

@Before( "DynamicDataSource()" )
public void executeChangeDataSource( JoinPoint point )
{
MethodSignature msig = ( MethodSignature )point.getSignature();
Method method = msig.getMethod();
DBProfile dbProfile = ( DBProfile )method.getAnnotation( DBProfile.class );

if( dbProfile != null )
{
DataSourceContextHolder.setDataSourceType( dbProfile.dbName() );
}
}

@After( "DynamicDataSource()" )
public void executeClear( JoinPoint point )
{
DataSourceContextHolder.clearDataSourceType();
}

}