org.makersoft.shards.unit.session.ShardedSqlSessionTests.java Source code

Introduction

Here is the source code for org.makersoft.shards.unit.session.ShardedSqlSessionTests.java

Source

/*
 * @(#)ShardedSqlSessionTests.java 2012-8-7 ?3:55:58
 *
 * Copyright (c) 2011-2012 Makersoft.org all rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *
 */
package org.makersoft.shards.unit.session;

import java.io.Reader;
import java.util.ArrayList;
import java.util.List;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.RowBounds;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.makersoft.shards.ShardId;
import org.makersoft.shards.cfg.ShardConfiguration;
import org.makersoft.shards.cfg.impl.ShardConfigurationImpl;
import org.makersoft.shards.domain.User;
import org.makersoft.shards.session.impl.ShardedSqlSessionFactoryBuilder;
import org.makersoft.shards.strategy.ShardStrategy;
import org.makersoft.shards.strategy.ShardStrategyFactory;
import org.makersoft.shards.strategy.ShardStrategyImpl;
import org.makersoft.shards.strategy.UserShardStrategyFactory;
import org.makersoft.shards.strategy.access.ShardAccessStrategy;
import org.makersoft.shards.strategy.access.impl.SequentialShardAccessStrategy;
import org.makersoft.shards.strategy.reduce.ShardReduceStrategy;
import org.makersoft.shards.strategy.resolution.ShardResolutionStrategy;
import org.makersoft.shards.strategy.resolution.impl.AllShardsShardResolutionStrategy;
import org.makersoft.shards.strategy.selection.ShardSelectionStrategy;
import org.makersoft.shards.unit.BaseTest;

/**
 * Unit test for
 * <p>
 * ShardedSqlSession
 * </p>
 * 
 * @version 2012-8-7
 * @author Feng Kuok
 */
public class ShardedSqlSessionTests extends BaseTest {
    private static SqlSessionFactory factory;

    @BeforeClass
    public static void setup() throws Exception {
        List<ShardConfiguration> shardConfigs = new ArrayList<ShardConfiguration>();

        final String resource_0 = "mybatis/mybatis-config-0.xml";
        final String resource_1 = "mybatis/mybatis-config-1.xml";
        final String resource_2 = "mybatis/mybatis-config-2.xml";
        final Reader reader_0 = Resources.getResourceAsReader(resource_0);
        final Reader reader_1 = Resources.getResourceAsReader(resource_1);
        final Reader reader_2 = Resources.getResourceAsReader(resource_2);

        SqlSessionFactory sqlMapper_0 = new SqlSessionFactoryBuilder().build(reader_0);
        SqlSessionFactory sqlMapper_1 = new SqlSessionFactoryBuilder().build(reader_1);
        SqlSessionFactory sqlMapper_2 = new SqlSessionFactoryBuilder().build(reader_2);

        shardConfigs.add(new ShardConfigurationImpl(0,
                sqlMapper_0.getConfiguration().getEnvironment().getDataSource(), sqlMapper_0));
        shardConfigs.add(new ShardConfigurationImpl(1,
                sqlMapper_1.getConfiguration().getEnvironment().getDataSource(), sqlMapper_1));
        shardConfigs.add(new ShardConfigurationImpl(2,
                sqlMapper_2.getConfiguration().getEnvironment().getDataSource(), sqlMapper_2));

        factory = new ShardedSqlSessionFactoryBuilder().build(shardConfigs, new UserShardStrategyFactory());

        Assert.assertNotNull(factory);

        factory = new ShardedSqlSessionFactoryBuilder().build(shardConfigs, new ShardStrategyFactory() {

            @Override
            public ShardStrategy newShardStrategy(List<ShardId> shardIds) {
                ShardSelectionStrategy pss = new ShardSelectionStrategy() {

                    @Override
                    public ShardId selectShardIdForNewObject(String statement, Object obj) {
                        return null;
                    }
                };

                ShardResolutionStrategy prs = new AllShardsShardResolutionStrategy(shardIds);
                ShardAccessStrategy pas = new SequentialShardAccessStrategy();

                ShardReduceStrategy srs = new ShardReduceStrategy() {

                    @Override
                    public List<Object> reduce(String statement, Object parameter, RowBounds rowBounds,
                            List<Object> values) {
                        return values;
                    }
                };

                return new ShardStrategyImpl(pss, prs, pas, srs);
            }
        });
    }

    @Test
    public void testSelectAllUsers() throws Exception {
        SqlSession session = factory.openSession();
        try {
            List<User> users = session.selectList("org.makersoft.shards.mapper.UserMapper.findAll");
            Assert.assertEquals(0, users.size());
        } finally {
            session.close();
        }
    }
}