从零开始学 Java - Spring AOP 实现主从读写分离

KathrinTodd 3年前
   <h2><strong>深刻讨论为什么要读写分离?</strong></h2>    <p>为了服务器承载更多的用户?提升了网站的响应速度?分摊数据库服务器的压力?就是为了双机热备又不想浪费备份服务器?上面这些回答,我认为都不是错误的,但也都不是完全正确的。「读写分离」并不是多么神奇的东西,也带不来多么大的性能提升,也许更多的作用的就是数据安全的备份吧。</p>    <p>从一个库到读写分离,从理论上对服务器压力来说是会带来一倍的性能提升,但你仔细思考一下,你的应用服务器真的很需要这一倍的提升么?那倒不如你去试着在服务器使用一下缓存系统,如 Memcached、Redis 这些分布式缓存,那性能可能是几十倍的提升。而且,在服务器硬件异常强悍及性能廉价的今天,完全更没必要了,所以,在今天,我认为它更多的职责就是为了数据安全而设计的,同时又提升了一些性能,这样也挺好。</p>    <p>可能我们更应该称之为 主从分离 。</p>    <h2><strong>利用 AOP 实现读写分离</strong></h2>    <p>读写分离方式很简单,就是在你读数据是去连接 <strong>从库</strong> ,在你写数据的时候去连接 <strong>主库</strong> ,具体代码实现当然就是连接时候去操作了,这没什么难度,在代码里写就是了。可是,有追求的程序猿都是不是这么解决问题的呢!</p>    <p>其实通过上篇的 <a href="/misc/goto?guid=4959714705839857745" rel="nofollow,noindex">从零开始学 Java - Spring AOP 拦截器的基本实现</a> 我们知道 AOP 可以实现在方法开始执行前后插入执行我们想要的代码,那这样,我们是不是可以在 <strong>执行数据库操作前根据业务来动态切换数据源</strong> 呢?</p>    <p>思考一下这个方式理论上好像是可行的,这种方式首先不需要在业务代码中去做切换,二是可能以后我们不需要读写分离了,把 AOP 切换的代码去掉就行了,三是可能就是拓展性好了。</p>    <h2><strong>等不了了,开始撸代码</strong></h2>    <p>你可能想深入的了解的话,我这里给你几个程序里用到的关键字 enum(枚举) 、 annotation(自定义注解) 、 JoinPoint(注入点) 、 AbstractRoutingDataSource(数据源接口子类) ,你理解了这些就知道了,其实你并不需要深入某些深层的东西,了解下即可。</p>    <p><strong>一、建立 JdbcContextHolder.java 类</strong></p>    <pre>  <code class="language-java">public class JdbcContextHolder {        private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();        public static void setJdbcType(String jdbcType) {          contextHolder.set(jdbcType);      }        public static void setSlave() {          setJdbcType("slave");      }        public static void setMaster() {          clearJdbcType();      }        public static String getJdbcType() {          return (String) contextHolder.get();      }        public static void clearJdbcType() {          contextHolder.remove();      }  }</code></pre>    <p>这个类的作用就是用来 <strong>设置、获取数据源连接</strong></p>    <p><strong>二、新建 DynamicDataSource.java 类,继承于 AbstractRoutingDataSource</strong></p>    <pre>  <code class="language-java">import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;    import cn.mayongfa.common.JdbcContextHolder;    public class DynamicDataSource extends AbstractRoutingDataSource {        @Override      protected Object determineCurrentLookupKey() {          // 获取当前数据源连接          return JdbcContextHolder.getJdbcType();      }  }</code></pre>    <p>通过研究,我们知道 determineCurrentLookupKey 方法是获取相关数据源连接的,所以重写 determineCurrentLookupKey 方法就可以啦,然后我们去通过刚刚我们建立的 JdbcContextHolder 类去获取。那怎么设置呢?</p>    <p><strong>三、建立数据源 DataSourceType.java 枚举类</strong></p>    <pre>  <code class="language-java">public enum DataSourceType {        //主库      Master("master"),        //从库      Slave("slave");        private DataSourceType(String name) {          this.name = name;      }        private String name;        public String getName() {          return name;      }      public void setName(String name) {          this.name = name;      }  }</code></pre>    <p>这个枚举类的作用其实就是为了设置数据源而生的,它的目的就是让设置数据源时更方便,如丝般顺滑。</p>    <p><strong>四、新建 DataSource.java Annotation(自定义注解)类</strong></p>    <pre>  <code class="language-java">import java.lang.annotation.Documented;  import java.lang.annotation.ElementType;  import java.lang.annotation.Retention;  import java.lang.annotation.RetentionPolicy;  import java.lang.annotation.Target;    @Retention(RetentionPolicy.RUNTIME)    @Target(ElementType.METHOD)  @Documented  public @interface DataSource {          DataSourceType value() default DataSourceType.Master;    }</code></pre>    <p>自定义注解的意义不再过多讨论,一句话来说就是可以让你 <strong>在类或方法名上以打标签的形式让该方法变得不一样</strong> 。具体怎么「不一样」,这个在于你。</p>    <p><strong>五、新建 DataSourceChoose.java 数据库切换类</strong></p>    <pre>  <code class="language-java">import java.lang.reflect.Method;    import org.aspectj.lang.JoinPoint;  import org.aspectj.lang.reflect.MethodSignature;    import cn.mayongfa.common.JdbcContextHolder;    public class DataSourceChoose {    //方法执行前  public void before(JoinPoint point){      Object target = point.getTarget();            String method = point.getSignature().getName();            Class<?>[] classz = target.getClass().getInterfaces();            MethodSignature methodSignature = (MethodSignature)point.getSignature();          Class<?>[] parameterTypes = methodSignature.getMethod().getParameterTypes();          try {              Method m = classz[0].getMethod(method, parameterTypes);                if (m!=null && m.isAnnotationPresent(DataSource.class)) {                    DataSource data = m.getAnnotation(DataSource.class);                    JdbcContextHolder.clearJdbcType();                  JdbcContextHolder.setJdbcType(data.value().getName());              }            } catch (Exception e) {                // TODO: handle exception            }   }  }</code></pre>    <p>这个其实是一个拦截器类,主要作用就是拦截那些方法名上有 @DataSource 这个自定义注解的,完了根据获取注解的 value() 值,来做相应的数据源切换。</p>    <p> </p>    <p> </p>    <p>来自:http://blog.mayongfa.cn/87.html</p>    <p> </p>