Mysql读写分离的两种实现对比：Spring+JPA应用层实现 vs Amoeba中间件实现
前段时间看了篇文章，讲Youku网数据库架构的演变，如何从最开始的读写分离，再到垂直分区，最后到水平分片，一步一步慢慢成熟的。看完之后很有冲动抽出一个模型来把这几种技术都实现一下。
说干就干，首先是读写分离了，我使用的数据库是Mysql，主从数据复制用的是半同步机制(mysql版本必须 5.5以上)，具体配置，可以参照这篇文章： http://blog.csdn.net/changerlove/article/details/6167255， 要注意Windows环境下，mysql配置文件为my.ini, linux下才是my.cnf。
然后主从复制有两类做法，一类就是在应用层去动态切换数据源，最简单的做法就是根据事务类型来做Route，另外一类就是利用所谓的“中间件”，在应用层和数据源之间通过分析SQL来Route，目前可选的“中间件有Mysql自带的Mysql Proxy还有阿里开发的Amoeba，后者据说比Mysql Proxy更稳定，使用也更简单，所以“中间件”做法里我选了第二种来尝试，待会儿会分析Amoeba的优劣，这儿先讲第一种做法，在应用层实现读写分离。
在例子中有两台Mysql Server，一台Master 负责写，一台Slave负责读，对应的JPA 配置文件如下：
org.eclipse.persistence.jpa.PersistenceProvider
false
org.eclipse.persistence.jpa.PersistenceProvider
false
然后系统的事务控制，这个是关键，因为在事务初始化的时候要根据事务属性(R/W)来初始化对应的EntityManager，并开启事务，简单起见，例子里使用JPA的Entity Transaction，而不是JTA。
实现第一步，我们得自定义transactionManager来控制事务的开关，回滚以及资源的初始化等等，直接看代码吧：
package util.transaction;
import javax.persistence.EntityManager;
import javax.persistence.EntityTransaction;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionException;
import org.springframework.transaction.support.AbstractPlatformTransactionManager;
import org.springframework.transaction.support.DefaultTransactionStatus;
public class RWSTransactionManager extends AbstractPlatformTransactionManager {
private static final long serialVersionUID = -1369860968344021154L;
@Override
protected Object doGetTransaction() throws TransactionException {
RWSJPATransactionObject rwsTransaction = new RWSJPATransactionObject();
EntityManager entityManager = EntityManagerHolder.getInstance()
.initializeResource(definition.isReadOnly());
EntityTransaction transaction = entityManager.getTransaction();
rwsTransaction.setTransaction(transaction);
return rwsTransaction;
}
@Override
protected void doBegin(Object txObject, TransactionDefinition definition)
throws TransactionException {
RWSJPATransactionObject rwsTransaction = (RWSJPATransactionObject) txObject;
if (!definition.isReadOnly()) {
rwsTransaction.getTransaction().begin();
}
}
@Override
protected void doCommit(DefaultTransactionStatus transactionStatus)
throws TransactionException {
if (transactionStatus.isReadOnly()) {
return;
}
RWSJPATransactionObject rwsTransaction = (RWSJPATransactionObject) transactionStatus
.getTransaction();
rwsTransaction.getTransaction().commit();
}
@Override
protected void doRollback(DefaultTransactionStatus transactionStatus)
throws TransactionException {
if (transactionStatus.isReadOnly()) {
return;
}
RWSJPATransactionObject rwsTransaction = (RWSJPATransactionObject) transactionStatus
.getTransaction();
rwsTransaction.getTransaction().rollback();
}
private class RWSJPATransactionObject {
private EntityTransaction transaction;
public EntityTransaction getTransaction() {
return transaction;
}
public void setTransaction(EntityTransaction transaction) {
this.transaction = transaction;
}
public RWSJPATransactionObject() {
super();
}
}
}对应的 EntityManagerHolder 代码如下：
package util.transaction;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;
import org.apache.commons.lang.StringUtils;
public class EntityManagerHolder {
private Random random = new Random();
private String PERSISTENCE_UNIT_HEADER_MASTER = "MASTER_";
private String PERSISTENCE_UNIT_HEADER_SLAVE = "SLAVE_";
private final ThreadLocal THREAD_LOCAL = new ThreadLocal();
private final Map RESOURCE = new HashMap();
private static EntityManagerHolder emHolder;
private EntityManagerHolder() {
}
public static EntityManagerHolder getInstance() {
if (null == emHolder) {
emHolder = new EntityManagerHolder();
}
return emHolder;
}
EntityManager initializeResource(boolean isReadOnly) {
// get random persistence unit, dependents on master and slave
// quantities
// configured in persistence.xml
String puSuffix = "";
if (isReadOnly) {
puSuffix = StringUtils.leftPad(String.valueOf(random.nextInt(1)),
3, "0");
} else {
puSuffix = StringUtils.leftPad(String.valueOf(random.nextInt(1)),
3, "0");
}
String persistenceUnit = (isReadOnly ? PERSISTENCE_UNIT_HEADER_SLAVE
: PERSISTENCE_UNIT_HEADER_MASTER) + puSuffix;
return createEntityManager(persistenceUnit);
}
private EntityManager createEntityManager(String persistenceUnit) {
if (null == RESOURCE.get(persistenceUnit)) {
EntityManagerFactory entityManagerFactory = Persistence
.createEntityManagerFactory(persistenceUnit);
RESOURCE.put(persistenceUnit, entityManagerFactory);
}
EntityManager entityManager = THREAD_LOCAL.get();
if (null == entityManager || !entityManager.isOpen()) {
entityManager = RESOURCE.get(persistenceUnit).createEntityManager();
THREAD_LOCAL.set(entityManager);
}
return entityManager;
}
public EntityManager getEntityManager() {
EntityManager entityManager = THREAD_LOCAL.get();
if (null == entityManager) {
throw new IllegalArgumentException();
}
return entityManager;
}
public void closeEntityManager() {
EntityManager entityManager = THREAD_LOCAL.get();
THREAD_LOCAL.set(null);
if (null != entityManager)
entityManager.close();
}
} 可以看到，在getTransaction方法里我们简单new了一个自定义的Object，其实就是一个EntityTransaction holder，关键是在begin（..）方法里，首先会根据事务属性到EntityManagerHolder中进行资源初始化（主要是EMFactory和EM）,在initializeResource 方法里面，简单的写了种路由算法，就是Java随机数，生成对应PersistenceUnit Name的后缀，最后根据这个name去生成对应的EMFactory和EM，并且set到Threadlocal的变量中去，之后在Dao层可以直接通过getEntityManager 方法来获取EntityManager.初始化结束之后，对应的事务直接使用EM里面的EntityTransaction来begin，commit和rollback。
事务管理器写好了，我们先看一下相关的Spring配置：
最后需要在service上加上事务注解, 指明事务属性：
@Override
@Transactional(readOnly = false)
public void add(T object) {
getDao().push(object);
}
@Override
@Transactional(readOnly = true)
public List findByExample(T object) {
return getDao().findByExample(object);
}
测试类：
package test;
import service.EmployeeService;
import test.mainTest.BaseMainTest;
import domain.Employee;
public class TestRWS extends BaseMainTest {
public static void main(String[] args) {
EmployeeService employeeService = (EmployeeService) ctx
.getBean("employeeService");
Employee employee = new Employee();
employee.setFirstName("Yinkan");
employee.setLastName("Zhu");
employeeService.add(employee);
// employeeService.findByExample(employee);
}
}
当调用EmployeeService里面的Add方法时（事务属性为写事务），从Log里可以看到成功连接到Master Server，并且插入成功，从库里也同步过去了。
当调用EmployeeService里面的FindByExample方法的时候（事务属性为读事务），从Log里可以看到是从Slave Server读取数据：
到这里第一种方法，在应用层来实现读写分离已经结束了，可以看到，通过事务属性来进行数据源切换这种做法比较简单，但是从软件设计的角度来看，事务控制里面耦合了数据源切换的逻辑，下面一种做法，使用Amoeba直接在SQL level做Route，代码的耦合度大大降低，但是也带来了其他问题。
Amoeba的配置，请参照官方文档： http://docs.hexnova.com/amoeba/
Amoeba的原理如下图所示，就是一层代理，对APP来说可见的只有一个DataSource，具体的Masters和Slaves可以在Amoeba里配置。
对应应用层JPA的配置如下：
org.eclipse.persistence.jpa.PersistenceProvider
false
该PersistenceUnit里面用到的Mysql相关信息，配置在Amoeba的amobe.xml文件里，是一个对外的proxy：
8066
${clientConnectioneManager}
128
64
amoebaUser
123
${amoeba.home}/conf/access_list.conf
然后是事务，事务的话可以直接使用Spring自带的JpaTransactionManager来管理，Entitymanager可以在Service层用annotation来注入，具体代码我就不贴了，一切准备好之后先启动Amoeba，我们来跑一下test， 首先是读，发现log如下：
可以看到Database version： 。。mysql-amoeba-proxy。。 说明连接正确，在看结果读数据也的正常的。
接下来就是测试写数据： 跑了一下，报错了：
Internal Exception: java.sql.SQLException: ResultSet is from UPDATE. No Data.
Error Code: 0
Call: SELECT LAST_INSERT_ID()
Check一下生成的SQL，发现insert之后跑了句SELECT LAST_INSERT_ID()， 因为我程序里Entity的主键是Mysql的自增主键，每次插入都会生成这句话用来返回当前ID，而这个ID是跟着Connection走的，也就是说如果该connection里面没有更新，就拿不到值，可是我们的insert和这句select难道不在一个Connection里面？对的，忘了我们用的是“SQL路由器”吗？insert语句开启主库connection，select则走的是从库的connection，所以查询当然是空了，而天真的eclipselink查出resultSet之后直接.next()了，也没判断是否有值，于是就出现这个error code了，当然对于eclipselink，这种check只能说是nice to have的，正常情况下是不会出现这种情况的，connection在获取数据源的时候就已经有了，这里因为我们拿到的其实是个代理。
考虑到Amoeba的实现原理，即"SQL路由"，即使Mysql支持Oracle一样的主键生成策略，在这种情况下似乎也不能正常工作。所以得由应用层来生成主键，这又是Effort。。。
另外从Amoeba的文档里看到，他的另一个缺点是目前不支持事务，这个我的理解是跨connection的事务做不到原子性，特别是当规则配置多主情况下，事务无效。
这么看来Amoeba的使用场景确实有限，首先domain层最好不要使用JPA之类的ORM框架，纯JDBC开发更可靠；然后在数据库架构上，一主多从尽量原子化事务。
综合对比一下，应用层实现读写分离比较靠谱。