Hibernate Shards 数据的水平、垂直切割（二）- Hibernate Shards基本演示

Hibernate Shards 数据的水平、垂直切割（二）- Hibernate Shards基本演示

准备
1. 以 Hibernate Shards 数据的水平、垂直切割（一）- Hibernate测试环境的项目为基础
2. Hibernate Shards使用了 commons logging，下载个项目最新的release版本。这里用的版本为1.1.1
3. 在 Hibernate Shards 数据的水平、垂直切割（一）- Hibernate测试环境中，我们在mysql中建立了一个hbshards数据库，为了测试shards再建立一个hbshards2的数据库，contact表的结构一样
4. 将Hibernate Shards发布包中的hibernate-shards.jar放到lib目录
    项目引用添加hibernate-shards.jar
    将commons logging发布包中的commons-logging-1.1.1.jar放到lib目录
    CLASSPATH中添加hibernate-shards.jar、commons-logging-1.1.1.jar

Hibernate Shards测试项目
出于结构上的简单考虑，定义一个ShardableEntity的接口，让ContactEntity实现这个接口:

public interface ShardableEntity {
    public String getIdentifier();
}
public class ContactEntity implements ShardableEntity {
    public String getIdentifier(){
        return this._id;
    }
    //other code omitted is the same with previous post
}

映射文件ContactEntity.hbm.xml与上一篇文章中的一样
因为我们需要将contact的数据以切片的形式存入2个数据库中，因此需要配置这2个数据库的信息。Hibernate Shards通过定义2个hibernate的配置文件解决这个问题:
shard0.hibernate.cfg.xml:

      com.mysql.jdbc.Driver
      jdbc:mysql://localhost/hbshards
      root
      dev
      10
      true
      org.hibernate.dialect.MySQLInnoDBDialect
      validate
      0
      false
      

shard1.hibernate.cfg.xml:

      com.mysql.jdbc.Driver
      jdbc:mysql://localhost/hbshards2
      root
      dev
      org.hibernate.dialect.MySQLInnoDBDialect
      1
      false

其中shard0.hibernate.cfg.xml作为主要的配置文件，针对每个shard创建的SessionFactory对象，除了数据库连接信息的配置之外，都来自shard0.hibernate.cfg.xml
针对其他每个shard创建的SessionFactory，数据库连接信息从相应的配置文件（例如shard1.hibernate.cfg.xml）中读取，读取的内容包括connection.url、connection.username、connection.password、connection.datasource。为了hibernate加载其他shard的配置文件时不抛异常，在其他shard的配置文件中我们还是把必要的配置属性写上去了
hibernate.connection.shard_id属性为每个shard定义一个id，必须有一个shard_id为0的shard，其他的shard_id可以定义为任意整数

我们假定contact的id都是由数字组成的一个8位长度的字符串
对contact的切割策略简单定义为: contact id第一个字符为0-4的存入shard0中，为5-9的存入shard1中
为了实现这一策略，我们需要实现hibernate shards的2个接口
ShardResolutionStrategy接口的实现:

import java.util.List;
import java.util.ArrayList;
import org.hibernate.shards.strategy.resolution.ShardResolutionStrategy;
import org.hibernate.shards.strategy.selection.ShardResolutionStrategyData;
import org.hibernate.shards.ShardId;
/*
 * a simple ShardResolutionStrategy implementation for our ContactEntity
 */
public class MyShardResolutionStrategy implements ShardResolutionStrategy {
    private List _shardIds;
    public MyShardResolutionStrategy(List shardIds){
        this._shardIds = shardIds;
    }
    public List selectShardIdsFromShardResolutionStrategyData(
            ShardResolutionStrategyData arg0){
        List ids = new ArrayList();
        String id = (String)arg0.getId();
        if(id==null || id.isEmpty()) ids.add(this._shardIds.get(0));
        else{
            //our shard selection is identified by the 
            //first char(number) in contact id
            //0-4 => shards0, 5-9 => shards1
            Integer i = new Integer(id.substring(0, 1));
            ids.add(this._shardIds.get(i/5));
        }
        return ids;
    }
}

ShardSelectionStrategy接口的实现:

import java.util.List;
import org.hibernate.shards.ShardId;
import org.hibernate.shards.strategy.selection.ShardSelectionStrategy;
/*
 * a simple ShardSelectionStrategy implementation for our ContactEntity
 */
public class MyShardSelectionStrategy implements ShardSelectionStrategy {
       private List _shardIds;
       public MyShardSelectionStrategy(List shardIds){
           this._shardIds=shardIds;
       }
       public ShardId selectShardIdForNewObject(Object obj) {
           if(obj instanceof ShardableEntity) {
               String id = ((ShardableEntity)obj).getIdentifier();
               if(id==null || id.isEmpty()) return this._shardIds.get(0);
               Integer i = new Integer(id.substring(0, 1));
               //our shard selection is identified by the 
               //first char(number) in contact id
               //0-4 => shards0, 5-9 => shards1
               return this._shardIds.get(i/5);
           }
           //for non-shardable entities we just use shard0
           return this._shardIds.get(0);
    }
}

接下来就是怎么使用hibernate shards的测试代码了:

import java.util.Iterator;
import java.util.List;
import java.util.ArrayList;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.shards.*;
import org.hibernate.shards.cfg.*;
import org.hibernate.shards.strategy.*;
import org.hibernate.shards.strategy.access.*;
import org.hibernate.shards.strategy.resolution.*;
import org.hibernate.shards.strategy.selection.*;


public class Main {
    public static void main(String[] args) {
        HibernateShardsTest(args);
    }
    private static SessionFactory createSessionFactory() {
        //加载主配置文件，为每个shard创建SessionFactory对象时将
        //以他作为原型
        Configuration prototypeCfg = new Configuration()
            .configure("shard0.hibernate.cfg.xml");
        //每个shard的配置文件
        List shardCfgs = new ArrayList();
        shardCfgs.add(buildShardConfig("shard0.hibernate.cfg.xml"));
        shardCfgs.add(buildShardConfig("shard1.hibernate.cfg.xml"));
        //数据切片策略的工厂对象
        ShardStrategyFactory strategyFactory = buildShardStrategyFactory();
        ShardedConfiguration shardedConfig = new ShardedConfiguration(
            prototypeCfg, shardCfgs, strategyFactory);
        //返回一个ShardedSessionFactory对象
       return shardedConfig.buildShardedSessionFactory();
   }
    private static ShardStrategyFactory buildShardStrategyFactory() {
       ShardStrategyFactory factory = new ShardStrategyFactory() {
           //测试用的自定义数据切片策略的工厂类
           public ShardStrategy newShardStrategy(List shardIds) {
               ShardSelectionStrategy ss = new MyShardSelectionStrategy(shardIds);
               ShardResolutionStrategy rs = new MyShardResolutionStrategy(shardIds);
               ShardAccessStrategy as = new SequentialShardAccessStrategy();
               return new ShardStrategyImpl(ss, rs, as);
           }
       };
       return factory;
    }
    private static ShardConfiguration buildShardConfig(String configFile) {
        Configuration config = new Configuration().configure(configFile);
        return new ConfigurationToShardConfigurationAdapter(config);
    }
    private static void HibernateShardsTest(String[] args){
        String loginId = "[email protected]";
        String password = "123";
        if(args!=null && args.length==2){
            loginId = args[0];
            password = args[1];
        }
        SessionFactory factory = null;
        try{
            factory = createSessionFactory();
            ShardsTestCreate(factory);
            ShardsTestLogin(factory, loginId, password);
            ShardsTestDelete(factory);
        }catch(Exception e){
            System.out.println(e.getMessage());
            e.printStackTrace();
        }finally{
            if(factory!=null) factory.close();
        }
    }
    private static void ShardsTestCreate(SessionFactory factory){
        Session session = null;
        Transaction transaction = null;
        System.out.println("===Create Contacts===");
        try{
            session = factory.openSession();
            transaction = session.beginTransaction();            
            session.save(new ContactEntity("01111111","[email protected]"
                    , "123", "Richie", "[email protected]"));
            session.save(new ContactEntity("91111111","[email protected]"
                    , "123", "AAA", "[email protected]"));
            session.save(new ContactEntity("81111111","[email protected]"
                    , "123", "BBB", "[email protected]"));
            session.save(new ContactEntity("31111111","[email protected]"
                    , "123", "CCC", "[email protected]"));
            transaction.commit();
        }catch(Exception e){
            if(transaction!=null) transaction.rollback();
            System.out.println(e.getMessage());
            e.printStackTrace();
        }finally{
            if(session!=null) session.close();
        }
    }
    private static void ShardsTestLogin(SessionFactory factory
            , String loginId, String password){
        Session session = null;
        ContactEntity c = null;
        System.out.println("\n===Login Test===");
        try{
            session = factory.openSession();
            List contacts = session.createQuery("from ContactEntity where LoginId=:loginId")
                .setString("loginId", loginId)
                .list();
            if(contacts.isEmpty())
                System.out.println("Contact \"" + loginId + "\" not found!");
            else{
                c = (ContactEntity)contacts.get(0);
                if(c.getPassword().equals(password))
                    System.out.println("Contact \"" + loginId + "\" login successful");
                else 
                    System.out.println("Password is incorrect (should be: "
                        + c.getPassword() + ", but is: " + password + ")");
            }
            System.out.println("\n===Get Contact by Id===");
            c = (ContactEntity)session.get(ContactEntity.class, "81111111");
            System.out.println(c.toString());
            c = (ContactEntity)session.get(ContactEntity.class, "31111111");
            System.out.println(c.toString());
        }catch(Exception e){
            System.out.println(e.getMessage());
            e.printStackTrace();
        }finally{
            if(session!=null) session.close();
        }
    }
    private static void ShardsTestDelete(SessionFactory factory){
        Session session = null;
        Transaction transaction = null;
        System.out.println("\n===Delete Contacts===");
        try{
            session = factory.openSession();
            transaction = session.beginTransaction();
            List contacts = session.createQuery("from ContactEntity").list();
            Iterator it = contacts.iterator();
            while(it.hasNext()){
                session.delete(it.next());
            }
            transaction.commit();
        }catch(Exception e){
            if(transaction!=null) transaction.rollback();
            System.out.println(e.getMessage());
            e.printStackTrace();
        }finally{
            if(session!=null) session.close();
        }
    }
}

项目结构图:
    

运行测试
用Eclipse编译class文件，将生成的所有class文件以及相关的配置、映射文件 拷贝到lib目录中，运行Main.class结果如下:
D:\Work\research\Java\Hibernate-Test\lib>java Main
===Create Contacts===
Hibernate: insert into CONTACT (EMAIL, NAME, LOGIN_ID, PASSWORD, ID) values (?,?, ?, ?, ?)
Hibernate: insert into CONTACT (EMAIL, NAME, LOGIN_ID, PASSWORD, ID) values (?,?, ?, ?, ?)
Hibernate: insert into CONTACT (EMAIL, NAME, LOGIN_ID, PASSWORD, ID) values (?,?, ?, ?, ?)
Hibernate: insert into CONTACT (EMAIL, NAME, LOGIN_ID, PASSWORD, ID) values (?,?, ?, ?, ?)

===Login Test===
Hibernate: select contactent0_.ID as ID0_, contactent0_.EMAIL as EMAIL0_, contactent0_.NAME as NAME0_, contactent0_.LOGIN_ID as LOGIN4_0_, contactent0_.PASSWORD as PASSWORD0_ from CONTACT contactent0_ where contactent0_.LOGIN_ID=?
Hibernate: select contactent0_.ID as ID0_, contactent0_.EMAIL as EMAIL0_, contactent0_.NAME as NAME0_, contactent0_.LOGIN_ID as LOGIN4_0_, contactent0_.PASSWORD as PASSWORD0_ from CONTACT contactent0_ where contactent0_.LOGIN_ID=?
Contact "[email protected]" login successful

===Get Contact by Id===
Hibernate: select contactent0_.ID as ID0_0_, contactent0_.EMAIL as EMAIL0_0_, contactent0_.NAME as NAME0_0_, contactent0_.LOGIN_ID as LOGIN4_0_0_, contactent0_.PASSWORD as PASSWORD0_0_ from CONTACT contactent0_ where contactent0_.ID=?
{ Id="81111111", LoginId="[email protected]", Name="BBB", EMail="[email protected]" }
Hibernate: select contactent0_.ID as ID0_0_, contactent0_.EMAIL as EMAIL0_0_, contactent0_.NAME as NAME0_0_, contactent0_.LOGIN_ID as LOGIN4_0_0_, contactent0_.PASSWORD as PASSWORD0_0_ from CONTACT contactent0_ where contactent0_.ID=?
{ Id="31111111", LoginId="[email protected]", Name="CCC", EMail="[email protected]" }

===Delete Contacts===
Hibernate: select contactent0_.ID as ID0_, contactent0_.EMAIL as EMAIL0_, contactent0_.NAME as NAME0_, contactent0_.LOGIN_ID as LOGIN4_0_, contactent0_.PASSWORD as PASSWORD0_ from CONTACT contactent0_
Hibernate: select contactent0_.ID as ID0_, contactent0_.EMAIL as EMAIL0_, contactent0_.NAME as NAME0_, contactent0_.LOGIN_ID as LOGIN4_0_, contactent0_.PASSWORD as PASSWORD0_ from CONTACT contactent0_
Hibernate: delete from CONTACT where ID=?
Hibernate: delete from CONTACT where ID=?
Hibernate: delete from CONTACT where ID=?
Hibernate: delete from CONTACT where ID=?

1. 通过id加载实体的时候，hibernate shards使用ShardSelectionStrategy来决定应当从哪个shard加载数据
2. 新增数据的时候，hibernate shards使用ShardResolutionStrategy来决定应当将数据insert到哪个shard中
    这里为什么不能使用ShardSelectionStrategy呢？因为与id的生成机制相关，比如说可以让数据库来生成id（hibernate中的native方式），例如自增id，在shard的应用场景下可以简单的给每个shard一个起止范围，只是在insert数据的时候必须通过特定的算法决定将数据insert到哪个shard中，这样的情况下ShardSelectionStrategy是无法运用的。而对于assigned等类似的id生成机制，还是可以运用ShardSelectionStrategy策略的
3. 执行hql或者Criteria查询的时候，则使用ShardAccessStrategy从shard中查询数据
    上面示例中的buildShardStrategyFactory方法中，我们使用了hibernate shards项目提供的SequentialShardAccessStrategy策略，这个策略在所有shards中逐个执行查询语句，然后对各个结果进行合并

基于上面几点，我们可以对上面测试运行后的输出做出解释了:
两个使用了hql的地方，都产生了两次sql查询，这是因为使用了SequentialShardAccessStrategy缘故，他从我们定义的两个shard中分别执行sql查询，合并结果集返回给调用者

把测试代码中的删除语句注释掉，运行测试后查询数据库，可以确认数据都是正确的insert在相应的shard中的

总结
上面演示了hibernate shards的基本用法，总体来看使用上非常简单，只需要针对实施shard的实体使用hibernate shards提供的ShardedSessionFactory即可
使用hibernate shards也存在一些限制，但大部分都是使用shards后的一些设计要求，或者是目前的it技术背景下对shards设计上的约束，在后面一篇中再详细整理一下这方面内容
文中使用的测试项目源文件在这里: 下载。引用到的jar文件需要从相关项目网站去下载