com.test.sharksharding.util.sequence.GetMysqlSequenceId2Test.java Source code

Introduction

Here is the source code for com.test.sharksharding.util.sequence.GetMysqlSequenceId2Test.java

Source

/*
 * Copyright 2015-2101 gaoxianglong
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.test.sharksharding.util.sequence;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;

import javax.annotation.Resource;
import javax.sql.DataSource;

import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.alibaba.druid.pool.DruidDataSource;
import com.sharksharding.util.sequence.SequenceIDManger;

/**
 * ?mysql?SequenceId,
 * 
 * @author gaoxianglong
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath*:datasource3-context.xml")
public class GetMysqlSequenceId2Test {
    @Resource
    private DataSource dataSource;

    /**
     * ????
     * 
     * @author gaoxianglong
     */
    public void init() {
        SequenceIDManger.init(dataSource);
    }

    /**
     * ?SequenceId
     * 
     * @author gaoxianglong
     */
    public @Test void getSequenceId() {
        System.out.println(SequenceIDManger.getSequenceId(100, 10, 100));
    }

    /**
     * ??SequenceId
     * 
     * @author gaoxianglong
     */
    public @Test void getSequenceId2() {
        for (int i = 0; i < 1000; i++) {
            System.out.println(SequenceIDManger.getSequenceId(100, 10, 100));
        }
    }

    /**
     * ???SequenceId
     * 
     * @author gaoxianglong
     */
    public @Test void getSequenceId3() {
        final CountDownLatch count = new CountDownLatch(2);
        final List<Long> id1 = new ArrayList<Long>();
        final List<Long> id2 = new ArrayList<Long>();
        final int size = 10000;
        new Thread() {
            public void run() {
                for (int i = 0; i < size; i++) {
                    id1.add(SequenceIDManger.getSequenceId(100, 10, 5000));
                }
                count.countDown();
            }
        }.start();
        new Thread() {
            public void run() {
                for (int i = 0; i < size; i++) {
                    id2.add(SequenceIDManger.getSequenceId(100, 10, 5000));
                }
                count.countDown();
            }
        }.start();
        try {
            count.await();
            Assert.assertFalse(id1.containsAll(id2));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}