Search in sources :

Example 6 with ReferenceFileRegion

use of com.ctrip.xpipe.netty.filechannel.ReferenceFileRegion in project x-pipe by ctripcorp.

the class DefaultReplicationStoreTest method testReadWhileDestroy.

@Test
public void testReadWhileDestroy() throws Exception {
    store = new DefaultReplicationStore(baseDir, new DefaultKeeperConfig(), randomKeeperRunid(), createkeeperMonitor());
    store.getMetaStore().becomeActive();
    int dataLen = 1000;
    RdbStore rdbStore = store.beginRdb(randomKeeperRunid(), -1, new LenEofType(dataLen));
    rdbStore.writeRdb(Unpooled.wrappedBuffer(randomString(dataLen).getBytes()));
    rdbStore.endRdb();
    CountDownLatch latch = new CountDownLatch(2);
    AtomicBoolean result = new AtomicBoolean(true);
    executors.execute(new AbstractExceptionLogTask() {

        @Override
        protected void doRun() throws Exception {
            try {
                sleep(2);
                store.close();
                store.destroy();
            } finally {
                latch.countDown();
            }
        }
    });
    executors.execute(new AbstractExceptionLogTask() {

        @Override
        protected void doRun() throws Exception {
            try {
                store.fullSyncIfPossible(new FullSyncListener() {

                    @Override
                    public ChannelFuture onCommand(ReferenceFileRegion referenceFileRegion) {
                        return null;
                    }

                    @Override
                    public void beforeCommand() {
                    }

                    @Override
                    public void setRdbFileInfo(EofType eofType, long rdbFileKeeperOffset) {
                    }

                    @Override
                    public void onFileData(ReferenceFileRegion referenceFileRegion) throws IOException {
                        sleep(10);
                    }

                    @Override
                    public boolean isOpen() {
                        return true;
                    }

                    @Override
                    public void exception(Exception e) {
                        logger.info("[exception][fail]" + e.getMessage());
                        result.set(false);
                    }

                    @Override
                    public void beforeFileData() {
                    }
                });
            } catch (Exception e) {
                logger.info("[exception][fail]" + e.getMessage());
                result.set(false);
            } finally {
                latch.countDown();
            }
        }
    });
    latch.await(100, TimeUnit.MILLISECONDS);
    Assert.assertFalse(result.get());
}
Also used : RdbStore(com.ctrip.xpipe.redis.core.store.RdbStore) LenEofType(com.ctrip.xpipe.redis.core.protocal.protocal.LenEofType) EofType(com.ctrip.xpipe.redis.core.protocal.protocal.EofType) LenEofType(com.ctrip.xpipe.redis.core.protocal.protocal.LenEofType) CountDownLatch(java.util.concurrent.CountDownLatch) IOException(java.io.IOException) ReferenceFileRegion(com.ctrip.xpipe.netty.filechannel.ReferenceFileRegion) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) FullSyncListener(com.ctrip.xpipe.redis.core.store.FullSyncListener) DefaultKeeperConfig(com.ctrip.xpipe.redis.keeper.config.DefaultKeeperConfig) AbstractExceptionLogTask(com.ctrip.xpipe.concurrent.AbstractExceptionLogTask) Test(org.junit.Test) AbstractRedisKeeperTest(com.ctrip.xpipe.redis.keeper.AbstractRedisKeeperTest)

Example 7 with ReferenceFileRegion

use of com.ctrip.xpipe.netty.filechannel.ReferenceFileRegion in project x-pipe by ctripcorp.

the class AbstractRedisKeeperTest method readRdbFileTilEnd.

protected String readRdbFileTilEnd(RdbStore rdbStore) throws IOException, InterruptedException {
    final ByteArrayWritableByteChannel bachannel = new ByteArrayWritableByteChannel();
    final CountDownLatch latch = new CountDownLatch(1);
    rdbStore.readRdbFile(new RdbFileListener() {

        @Override
        public void setRdbFileInfo(EofType eofType, long rdbFileOffset) {
        }

        @Override
        public void onFileData(ReferenceFileRegion referenceFileRegion) throws IOException {
            if (referenceFileRegion == null) {
                latch.countDown();
                return;
            }
            referenceFileRegion.transferTo(bachannel, 0L);
        }

        @Override
        public boolean isOpen() {
            return true;
        }

        @Override
        public void exception(Exception e) {
            latch.countDown();
        }

        @Override
        public void beforeFileData() {
        }
    });
    latch.await(5, TimeUnit.SECONDS);
    return new String(bachannel.getResult());
}
Also used : ByteArrayWritableByteChannel(com.ctrip.xpipe.payload.ByteArrayWritableByteChannel) EofType(com.ctrip.xpipe.redis.core.protocal.protocal.EofType) IOException(java.io.IOException) CountDownLatch(java.util.concurrent.CountDownLatch) IOException(java.io.IOException) ReferenceFileRegion(com.ctrip.xpipe.netty.filechannel.ReferenceFileRegion)

Aggregations

ReferenceFileRegion (com.ctrip.xpipe.netty.filechannel.ReferenceFileRegion)7 IOException (java.io.IOException)7 AbstractExceptionLogTask (com.ctrip.xpipe.concurrent.AbstractExceptionLogTask)2 EofType (com.ctrip.xpipe.redis.core.protocal.protocal.EofType)2 CommandsListener (com.ctrip.xpipe.redis.core.store.CommandsListener)2 AbstractRedisKeeperTest (com.ctrip.xpipe.redis.keeper.AbstractRedisKeeperTest)2 ChannelFuture (io.netty.channel.ChannelFuture)2 CountDownLatch (java.util.concurrent.CountDownLatch)2 Semaphore (java.util.concurrent.Semaphore)2 Test (org.junit.Test)2 ByteArrayWritableByteChannel (com.ctrip.xpipe.payload.ByteArrayWritableByteChannel)1 LenEofType (com.ctrip.xpipe.redis.core.protocal.protocal.LenEofType)1 CommandReader (com.ctrip.xpipe.redis.core.store.CommandReader)1 FullSyncListener (com.ctrip.xpipe.redis.core.store.FullSyncListener)1 RdbStore (com.ctrip.xpipe.redis.core.store.RdbStore)1 DefaultKeeperConfig (com.ctrip.xpipe.redis.keeper.config.DefaultKeeperConfig)1 ChannelFutureListener (io.netty.channel.ChannelFutureListener)1 ByteArrayOutputStream (java.io.ByteArrayOutputStream)1 ExecutionException (java.util.concurrent.ExecutionException)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1