ARTEMIS-4543 Journal/NIO Sync is not scalling up with TimedBuffer
This commit is contained in:
parent
379dd4088f
commit
a1add09734
|
@ -29,6 +29,7 @@ import io.netty.buffer.ByteBuf;
|
||||||
import org.apache.activemq.artemis.api.core.ActiveMQBuffer;
|
import org.apache.activemq.artemis.api.core.ActiveMQBuffer;
|
||||||
import org.apache.activemq.artemis.api.core.ActiveMQBuffers;
|
import org.apache.activemq.artemis.api.core.ActiveMQBuffers;
|
||||||
import org.apache.activemq.artemis.api.core.ActiveMQException;
|
import org.apache.activemq.artemis.api.core.ActiveMQException;
|
||||||
|
import org.apache.activemq.artemis.api.core.ActiveMQExceptionType;
|
||||||
import org.apache.activemq.artemis.api.core.ActiveMQIOErrorException;
|
import org.apache.activemq.artemis.api.core.ActiveMQIOErrorException;
|
||||||
import org.apache.activemq.artemis.core.io.buffer.TimedBuffer;
|
import org.apache.activemq.artemis.core.io.buffer.TimedBuffer;
|
||||||
import org.apache.activemq.artemis.core.io.buffer.TimedBufferObserver;
|
import org.apache.activemq.artemis.core.io.buffer.TimedBufferObserver;
|
||||||
|
@ -271,6 +272,11 @@ public abstract class AbstractSequentialFile implements SequentialFile {
|
||||||
|
|
||||||
protected class LocalBufferObserver implements TimedBufferObserver {
|
protected class LocalBufferObserver implements TimedBufferObserver {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean supportSync() {
|
||||||
|
return factory.isSyncSupported();
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void flushBuffer(final ByteBuf byteBuf, final boolean requestedSync, final List<IOCallback> callbacks) {
|
public void flushBuffer(final ByteBuf byteBuf, final boolean requestedSync, final List<IOCallback> callbacks) {
|
||||||
final int bytes = byteBuf.readableBytes();
|
final int bytes = byteBuf.readableBytes();
|
||||||
|
@ -285,12 +291,29 @@ public abstract class AbstractSequentialFile implements SequentialFile {
|
||||||
ByteUtil.zeros(buffer, bytes, missingNonZeroedBytes);
|
ByteUtil.zeros(buffer, bytes, missingNonZeroedBytes);
|
||||||
}
|
}
|
||||||
buffer.flip();
|
buffer.flip();
|
||||||
writeDirect(buffer, requestedSync, new DelegateCallback(callbacks));
|
writeDirect(buffer, requestedSync, DelegateCallback.wrap(callbacks));
|
||||||
} else {
|
} else {
|
||||||
IOCallback.done(callbacks);
|
IOCallback.done(callbacks);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void checkSync(boolean syncRequested, List<IOCallback> callbacks) {
|
||||||
|
if (syncRequested) {
|
||||||
|
try {
|
||||||
|
sync();
|
||||||
|
} catch (Throwable e) {
|
||||||
|
logger.warn(e.getMessage(), e);
|
||||||
|
if (callbacks != null) {
|
||||||
|
callbacks.forEach(c -> c.onError(ActiveMQExceptionType.IO_ERROR.getCode(), e.getMessage()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (callbacks != null) {
|
||||||
|
callbacks.forEach(c -> c.done());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public int getRemainingBytes() {
|
public int getRemainingBytes() {
|
||||||
if (fileSize - position.get() > Integer.MAX_VALUE) {
|
if (fileSize - position.get() > Integer.MAX_VALUE) {
|
||||||
|
|
|
@ -26,12 +26,20 @@ public final class DelegateCallback implements IOCallback {
|
||||||
|
|
||||||
private final Collection<? extends IOCallback> delegates;
|
private final Collection<? extends IOCallback> delegates;
|
||||||
|
|
||||||
|
public static DelegateCallback wrap(final Collection<? extends IOCallback> delegates) {
|
||||||
|
if (delegates == null) {
|
||||||
|
return null;
|
||||||
|
} else {
|
||||||
|
return new DelegateCallback(delegates);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* It doesn't copy defensively the given {@code delegates}.
|
* It doesn't copy defensively the given {@code delegates}.
|
||||||
*
|
*
|
||||||
* @throws NullPointerException if {@code delegates} is {@code null}
|
* @throws NullPointerException if {@code delegates} is {@code null}
|
||||||
*/
|
*/
|
||||||
public DelegateCallback(final Collection<? extends IOCallback> delegates) {
|
private DelegateCallback(final Collection<? extends IOCallback> delegates) {
|
||||||
Objects.requireNonNull(delegates, "delegates cannot be null!");
|
Objects.requireNonNull(delegates, "delegates cannot be null!");
|
||||||
this.delegates = delegates;
|
this.delegates = delegates;
|
||||||
}
|
}
|
||||||
|
|
|
@ -46,6 +46,10 @@ public interface SequentialFileFactory {
|
||||||
|
|
||||||
int getMaxIO();
|
int getMaxIO();
|
||||||
|
|
||||||
|
default boolean isSyncSupported() {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Lists files that end with the given extension.
|
* Lists files that end with the given extension.
|
||||||
* <p>
|
* <p>
|
||||||
|
|
|
@ -76,6 +76,11 @@ public final class AIOSequentialFileFactory extends AbstractSequentialFileFactor
|
||||||
|
|
||||||
private static final String AIO_TEST_FILE = ".aio-test";
|
private static final String AIO_TEST_FILE = ".aio-test";
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isSyncSupported() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
public void beforeClose() {
|
public void beforeClose() {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -312,6 +312,9 @@ public final class TimedBuffer extends CriticalComponentImpl {
|
||||||
* @return {@code true} when are flushed any bytes, {@code false} otherwise
|
* @return {@code true} when are flushed any bytes, {@code false} otherwise
|
||||||
*/
|
*/
|
||||||
public boolean flushBatch() {
|
public boolean flushBatch() {
|
||||||
|
List<IOCallback> syncCallbackList = null;
|
||||||
|
boolean localUseSync = false;
|
||||||
|
TimedBufferObserver syncBufferObserver = null;
|
||||||
try (ArtemisCloseable measure = measureCritical(CRITICAL_PATH_FLUSH)) {
|
try (ArtemisCloseable measure = measureCritical(CRITICAL_PATH_FLUSH)) {
|
||||||
synchronized (this) {
|
synchronized (this) {
|
||||||
if (!started) {
|
if (!started) {
|
||||||
|
@ -325,7 +328,16 @@ public final class TimedBuffer extends CriticalComponentImpl {
|
||||||
bytesFlushed.addAndGet(pos);
|
bytesFlushed.addAndGet(pos);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (bufferObserver.supportSync()) {
|
||||||
|
// performing the sync away from the lock
|
||||||
|
// so other writes can be performed while that flush is happening
|
||||||
|
syncCallbackList = callbacks;
|
||||||
|
localUseSync = pendingSync;
|
||||||
|
syncBufferObserver = bufferObserver;
|
||||||
|
bufferObserver.flushBuffer(buffer.byteBuf(), false, null);
|
||||||
|
} else {
|
||||||
bufferObserver.flushBuffer(buffer.byteBuf(), pendingSync, callbacks);
|
bufferObserver.flushBuffer(buffer.byteBuf(), pendingSync, callbacks);
|
||||||
|
}
|
||||||
|
|
||||||
stopSpin();
|
stopSpin();
|
||||||
|
|
||||||
|
@ -345,6 +357,10 @@ public final class TimedBuffer extends CriticalComponentImpl {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
} finally {
|
||||||
|
if (syncBufferObserver != null) {
|
||||||
|
syncBufferObserver.checkSync(localUseSync, syncCallbackList);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -29,6 +29,13 @@ public interface TimedBufferObserver {
|
||||||
*/
|
*/
|
||||||
void flushBuffer(ByteBuf buffer, boolean syncRequested, List<IOCallback> callbacks);
|
void flushBuffer(ByteBuf buffer, boolean syncRequested, List<IOCallback> callbacks);
|
||||||
|
|
||||||
|
default void checkSync(boolean syncRequested, List<IOCallback> callbacks) {
|
||||||
|
}
|
||||||
|
|
||||||
|
default boolean supportSync() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Return the number of remaining bytes that still fit on the observer (file)
|
* Return the number of remaining bytes that still fit on the observer (file)
|
||||||
*/
|
*/
|
||||||
|
|
|
@ -18,6 +18,7 @@ package org.apache.activemq.artemis.core.io.mapped;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.lang.invoke.MethodHandles;
|
||||||
import java.nio.ByteBuffer;
|
import java.nio.ByteBuffer;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
@ -34,9 +35,14 @@ import org.apache.activemq.artemis.core.io.buffer.TimedBuffer;
|
||||||
import org.apache.activemq.artemis.core.io.buffer.TimedBufferObserver;
|
import org.apache.activemq.artemis.core.io.buffer.TimedBufferObserver;
|
||||||
import org.apache.activemq.artemis.core.journal.EncodingSupport;
|
import org.apache.activemq.artemis.core.journal.EncodingSupport;
|
||||||
import org.apache.activemq.artemis.core.journal.impl.SimpleWaitIOCallback;
|
import org.apache.activemq.artemis.core.journal.impl.SimpleWaitIOCallback;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
final class TimedSequentialFile implements SequentialFile {
|
final class TimedSequentialFile implements SequentialFile {
|
||||||
|
|
||||||
|
private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
|
||||||
|
|
||||||
|
|
||||||
private final SequentialFileFactory factory;
|
private final SequentialFileFactory factory;
|
||||||
private final SequentialFile sequentialFile;
|
private final SequentialFile sequentialFile;
|
||||||
private final LocalBufferObserver observer;
|
private final LocalBufferObserver observer;
|
||||||
|
@ -247,6 +253,11 @@ final class TimedSequentialFile implements SequentialFile {
|
||||||
|
|
||||||
private final class LocalBufferObserver implements TimedBufferObserver {
|
private final class LocalBufferObserver implements TimedBufferObserver {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean supportSync() {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void flushBuffer(final ByteBuf byteBuf, final boolean requestedSync, final List<IOCallback> callbacks) {
|
public void flushBuffer(final ByteBuf byteBuf, final boolean requestedSync, final List<IOCallback> callbacks) {
|
||||||
final int bytes = byteBuf.readableBytes();
|
final int bytes = byteBuf.readableBytes();
|
||||||
|
@ -266,8 +277,7 @@ final class TimedSequentialFile implements SequentialFile {
|
||||||
buffer.flip();
|
buffer.flip();
|
||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
blockingWriteDirect(buffer, requestedSync, releaseBuffer);
|
blockingWriteDirect(buffer, false, releaseBuffer);
|
||||||
IOCallback.done(callbacks);
|
|
||||||
} catch (Throwable t) {
|
} catch (Throwable t) {
|
||||||
final int code;
|
final int code;
|
||||||
if (t instanceof IOException) {
|
if (t instanceof IOException) {
|
||||||
|
@ -278,8 +288,19 @@ final class TimedSequentialFile implements SequentialFile {
|
||||||
}
|
}
|
||||||
IOCallback.onError(callbacks, code, t.getMessage());
|
IOCallback.onError(callbacks, code, t.getMessage());
|
||||||
}
|
}
|
||||||
} else {
|
}
|
||||||
IOCallback.done(callbacks);
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void checkSync(boolean syncRequested, List<IOCallback> callbacks) {
|
||||||
|
try {
|
||||||
|
sync();
|
||||||
|
} catch (Exception e) {
|
||||||
|
logger.warn(e.getMessage(), e);
|
||||||
|
} finally {
|
||||||
|
if (callbacks != null) {
|
||||||
|
callbacks.forEach(c -> c.done());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -387,7 +387,7 @@ public class NIOSequentialFile extends AbstractSequentialFile {
|
||||||
internalWrite(bytes, sync, null, releaseBuffer);
|
internalWrite(bytes, sync, null, releaseBuffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void internalWrite(final ByteBuffer bytes,
|
private synchronized void internalWrite(final ByteBuffer bytes,
|
||||||
final boolean sync,
|
final boolean sync,
|
||||||
final IOCallback callback,
|
final IOCallback callback,
|
||||||
boolean releaseBuffer) throws IOException, ActiveMQIOErrorException, InterruptedException {
|
boolean releaseBuffer) throws IOException, ActiveMQIOErrorException, InterruptedException {
|
||||||
|
@ -474,12 +474,14 @@ public class NIOSequentialFile extends AbstractSequentialFile {
|
||||||
//enable zero copy case
|
//enable zero copy case
|
||||||
if (byteBuf.nioBufferCount() == 1 && byteBuf.isDirect()) {
|
if (byteBuf.nioBufferCount() == 1 && byteBuf.isDirect()) {
|
||||||
final ByteBuffer buffer = byteBuf.internalNioBuffer(byteBuf.readerIndex(), bytes);
|
final ByteBuffer buffer = byteBuf.internalNioBuffer(byteBuf.readerIndex(), bytes);
|
||||||
final IOCallback callback = new DelegateCallback(callbacks);
|
final IOCallback callback = DelegateCallback.wrap(callbacks);
|
||||||
try {
|
try {
|
||||||
//no need to pool the buffer and don't care if the NIO buffer got modified
|
//no need to pool the buffer and don't care if the NIO buffer got modified
|
||||||
internalWrite(buffer, requestedSync, callback, false);
|
internalWrite(buffer, requestedSync, callback, false);
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
callback.onError(ActiveMQExceptionType.GENERIC_EXCEPTION.getCode(), e.getMessage());
|
if (callbacks != null) {
|
||||||
|
callbacks.forEach(c -> c.onError(ActiveMQExceptionType.GENERIC_EXCEPTION.getCode(), e.getMessage()));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
super.flushBuffer(byteBuf, requestedSync, callbacks);
|
super.flushBuffer(byteBuf, requestedSync, callbacks);
|
||||||
|
|
|
@ -81,6 +81,10 @@ public interface Journal extends ActiveMQComponent {
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
default File getHistoryFolder() {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
void appendAddRecord(long id, byte recordType, Persister persister, Object record, boolean sync) throws Exception;
|
void appendAddRecord(long id, byte recordType, Persister persister, Object record, boolean sync) throws Exception;
|
||||||
|
|
||||||
void appendAddRecord(long id,
|
void appendAddRecord(long id,
|
||||||
|
|
|
@ -239,6 +239,11 @@ public class JournalImpl extends JournalBase implements TestableJournal, Journal
|
||||||
return journalRetentionFolder != null;
|
return journalRetentionFolder != null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public File getHistoryFolder() {
|
||||||
|
return journalRetentionFolder;
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public JournalImpl setHistoryFolder(File historyFolder, long maxBytes, long period) throws Exception {
|
public JournalImpl setHistoryFolder(File historyFolder, long maxBytes, long period) throws Exception {
|
||||||
|
|
||||||
|
|
|
@ -25,11 +25,6 @@ import org.junit.Test;
|
||||||
|
|
||||||
public class DelegateCallbackTest {
|
public class DelegateCallbackTest {
|
||||||
|
|
||||||
@Test(expected = NullPointerException.class)
|
|
||||||
public void shouldFailWithNullDelegates() {
|
|
||||||
new DelegateCallback(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static final class CountingIOCallback implements IOCallback {
|
private static final class CountingIOCallback implements IOCallback {
|
||||||
|
|
||||||
long done = 0;
|
long done = 0;
|
||||||
|
@ -60,7 +55,7 @@ public class DelegateCallbackTest {
|
||||||
@Test
|
@Test
|
||||||
public void shouldCallDoneOnEachCallback() {
|
public void shouldCallDoneOnEachCallback() {
|
||||||
final CountingIOCallback countingIOCallback = new CountingIOCallback(false);
|
final CountingIOCallback countingIOCallback = new CountingIOCallback(false);
|
||||||
final DelegateCallback callback = new DelegateCallback(Arrays.asList(countingIOCallback, countingIOCallback));
|
final DelegateCallback callback = DelegateCallback.wrap(Arrays.asList(countingIOCallback, countingIOCallback));
|
||||||
callback.done();
|
callback.done();
|
||||||
Assert.assertEquals(2, countingIOCallback.done);
|
Assert.assertEquals(2, countingIOCallback.done);
|
||||||
Assert.assertEquals(0, countingIOCallback.onError);
|
Assert.assertEquals(0, countingIOCallback.onError);
|
||||||
|
@ -69,7 +64,7 @@ public class DelegateCallbackTest {
|
||||||
@Test
|
@Test
|
||||||
public void shouldCallOnErrorOnEachCallback() {
|
public void shouldCallOnErrorOnEachCallback() {
|
||||||
final CountingIOCallback countingIOCallback = new CountingIOCallback(false);
|
final CountingIOCallback countingIOCallback = new CountingIOCallback(false);
|
||||||
final DelegateCallback callback = new DelegateCallback(Arrays.asList(countingIOCallback, countingIOCallback));
|
final DelegateCallback callback = DelegateCallback.wrap(Arrays.asList(countingIOCallback, countingIOCallback));
|
||||||
callback.onError(0, "not a real error");
|
callback.onError(0, "not a real error");
|
||||||
Assert.assertEquals(0, countingIOCallback.done);
|
Assert.assertEquals(0, countingIOCallback.done);
|
||||||
Assert.assertEquals(2, countingIOCallback.onError);
|
Assert.assertEquals(2, countingIOCallback.onError);
|
||||||
|
@ -78,7 +73,7 @@ public class DelegateCallbackTest {
|
||||||
@Test
|
@Test
|
||||||
public void shouldCallDoneOnEachCallbackWithExceptions() {
|
public void shouldCallDoneOnEachCallbackWithExceptions() {
|
||||||
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
||||||
final DelegateCallback callback = new DelegateCallback(Arrays.asList(countingIOCallback, countingIOCallback));
|
final DelegateCallback callback = DelegateCallback.wrap(Arrays.asList(countingIOCallback, countingIOCallback));
|
||||||
callback.done();
|
callback.done();
|
||||||
Assert.assertEquals(2, countingIOCallback.done);
|
Assert.assertEquals(2, countingIOCallback.done);
|
||||||
Assert.assertEquals(0, countingIOCallback.onError);
|
Assert.assertEquals(0, countingIOCallback.onError);
|
||||||
|
@ -87,7 +82,7 @@ public class DelegateCallbackTest {
|
||||||
@Test
|
@Test
|
||||||
public void shouldCallOnErrorOnEachCallbackWithExceptions() {
|
public void shouldCallOnErrorOnEachCallbackWithExceptions() {
|
||||||
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
||||||
final DelegateCallback callback = new DelegateCallback(Arrays.asList(countingIOCallback, countingIOCallback));
|
final DelegateCallback callback = DelegateCallback.wrap(Arrays.asList(countingIOCallback, countingIOCallback));
|
||||||
callback.onError(0, "not a real error");
|
callback.onError(0, "not a real error");
|
||||||
Assert.assertEquals(0, countingIOCallback.done);
|
Assert.assertEquals(0, countingIOCallback.done);
|
||||||
Assert.assertEquals(2, countingIOCallback.onError);
|
Assert.assertEquals(2, countingIOCallback.onError);
|
||||||
|
@ -97,7 +92,7 @@ public class DelegateCallbackTest {
|
||||||
public void shouldLogOnDoneForEachExceptions() throws Exception {
|
public void shouldLogOnDoneForEachExceptions() throws Exception {
|
||||||
try (AssertionLoggerHandler loggerHandler = new AssertionLoggerHandler()) {
|
try (AssertionLoggerHandler loggerHandler = new AssertionLoggerHandler()) {
|
||||||
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
||||||
final DelegateCallback callback = new DelegateCallback(Collections.singleton(countingIOCallback));
|
final DelegateCallback callback = DelegateCallback.wrap(Collections.singleton(countingIOCallback));
|
||||||
callback.done();
|
callback.done();
|
||||||
Assert.assertTrue(loggerHandler.findText("AMQ142024"));
|
Assert.assertTrue(loggerHandler.findText("AMQ142024"));
|
||||||
}
|
}
|
||||||
|
@ -107,7 +102,7 @@ public class DelegateCallbackTest {
|
||||||
public void shouldLogOnErrorForEachExceptions() throws Exception {
|
public void shouldLogOnErrorForEachExceptions() throws Exception {
|
||||||
try (AssertionLoggerHandler loggerHandler = new AssertionLoggerHandler()) {
|
try (AssertionLoggerHandler loggerHandler = new AssertionLoggerHandler()) {
|
||||||
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
final CountingIOCallback countingIOCallback = new CountingIOCallback(true);
|
||||||
final DelegateCallback callback = new DelegateCallback(Collections.singleton(countingIOCallback));
|
final DelegateCallback callback = DelegateCallback.wrap(Collections.singleton(countingIOCallback));
|
||||||
callback.onError(0, "not a real error");
|
callback.onError(0, "not a real error");
|
||||||
Assert.assertTrue(loggerHandler.findText("AMQ142025"));
|
Assert.assertTrue(loggerHandler.findText("AMQ142025"));
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,257 @@
|
||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You 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 org.apache.activemq.artemis.tests.unit.core.journal.impl;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.lang.invoke.MethodHandles;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.concurrent.CountDownLatch;
|
||||||
|
import java.util.concurrent.ExecutorService;
|
||||||
|
import java.util.concurrent.Executors;
|
||||||
|
import java.util.concurrent.ThreadFactory;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
|
||||||
|
import org.apache.activemq.artemis.core.io.SequentialFileFactory;
|
||||||
|
import org.apache.activemq.artemis.core.io.aio.AIOSequentialFileFactory;
|
||||||
|
import org.apache.activemq.artemis.core.io.mapped.MappedSequentialFileFactory;
|
||||||
|
import org.apache.activemq.artemis.core.io.nio.NIOSequentialFileFactory;
|
||||||
|
import org.apache.activemq.artemis.core.journal.IOCompletion;
|
||||||
|
import org.apache.activemq.artemis.core.journal.Journal;
|
||||||
|
import org.apache.activemq.artemis.core.journal.PreparedTransactionInfo;
|
||||||
|
import org.apache.activemq.artemis.core.journal.RecordInfo;
|
||||||
|
import org.apache.activemq.artemis.core.journal.impl.JournalImpl;
|
||||||
|
import org.apache.activemq.artemis.core.server.JournalType;
|
||||||
|
import org.apache.activemq.artemis.nativo.jlibaio.LibaioContext;
|
||||||
|
import org.apache.activemq.artemis.tests.util.ActiveMQTestBase;
|
||||||
|
import org.apache.activemq.artemis.utils.ExecutorFactory;
|
||||||
|
import org.apache.activemq.artemis.utils.SimpleIDGenerator;
|
||||||
|
import org.apache.activemq.artemis.utils.actors.OrderedExecutorFactory;
|
||||||
|
import org.apache.activemq.artemis.utils.collections.ConcurrentHashSet;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Assume;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
public class BatchCommitTest extends ActiveMQTestBase {
|
||||||
|
|
||||||
|
private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
|
||||||
|
|
||||||
|
private static final int FILE_SIZE = 10 * 1024 * 1024;
|
||||||
|
private static final int MIN_FILES = 10;
|
||||||
|
private static final int POOL_SIZE = 10;
|
||||||
|
private static final String FILE_PREFIX = "journal-test";
|
||||||
|
private static final String FILE_EXTENSION = "amq";
|
||||||
|
private static final int BUFFER_SIZE = 100 * 1024;
|
||||||
|
private static final int BUFFER_TIMEOUT = 10 * 1024;
|
||||||
|
|
||||||
|
private static final int MAX_AIO = 255;
|
||||||
|
|
||||||
|
private static final int OK = 100;
|
||||||
|
private static final int ERROR = 101;
|
||||||
|
|
||||||
|
private static final int RECORDS = 10000;
|
||||||
|
|
||||||
|
JournalImpl journal;
|
||||||
|
|
||||||
|
SequentialFileFactory journalFF;
|
||||||
|
|
||||||
|
SimpleIDGenerator idGenerator = new SimpleIDGenerator(1);
|
||||||
|
|
||||||
|
public Journal testRunNIO(String testFolder, boolean sync) throws Throwable {
|
||||||
|
return testRun(testFolder, JournalType.NIO, sync);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Journal testRunMapped(String testFolder, boolean sync) throws Throwable {
|
||||||
|
return testRun(testFolder, JournalType.MAPPED, sync);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public Journal testRunAIO(String testFolder, boolean sync) throws Throwable {
|
||||||
|
return testRun(testFolder, JournalType.ASYNCIO, sync);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public Journal testRun(String testFolder, JournalType journalType, boolean sync) throws Throwable {
|
||||||
|
|
||||||
|
OrderedExecutorFactory orderedExecutorFactory = getExecutorFactory();
|
||||||
|
setupJournal(journalType, testFolder, orderedExecutorFactory);
|
||||||
|
journal.start();
|
||||||
|
runAfter(journal::stop);
|
||||||
|
journal.loadInternalOnly();
|
||||||
|
|
||||||
|
CountDownLatch latch = new CountDownLatch(RECORDS);
|
||||||
|
|
||||||
|
ConcurrentHashSet<Long> existingRecords = new ConcurrentHashSet<>();
|
||||||
|
|
||||||
|
AtomicInteger errors = new AtomicInteger(0);
|
||||||
|
|
||||||
|
|
||||||
|
for (int i = 0; i < RECORDS; i++) {
|
||||||
|
long tx = idGenerator.generateID();
|
||||||
|
long id = idGenerator.generateID();
|
||||||
|
long upid = idGenerator.generateID();
|
||||||
|
existingRecords.add(tx);
|
||||||
|
IOCompletion completion = new IOCompletion() {
|
||||||
|
@Override
|
||||||
|
public void storeLineUp() {
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void done() {
|
||||||
|
if (!existingRecords.remove(tx)) {
|
||||||
|
errors.incrementAndGet();
|
||||||
|
logger.warn("Id {} was removed before", tx);
|
||||||
|
}
|
||||||
|
latch.countDown();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onError(int errorCode, String errorMessage) {
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
journal.appendAddRecordTransactional(tx, id, (byte) 1, ("add " + id).getBytes());
|
||||||
|
journal.appendUpdateRecordTransactional(tx, upid, (byte) 1, ("up " + upid).getBytes());
|
||||||
|
journal.appendCommitRecord(tx, sync, completion, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!latch.await(10, TimeUnit.SECONDS)) {
|
||||||
|
logger.warn("latch didn't finish, count={}", latch.getCount());
|
||||||
|
errors.incrementAndGet();
|
||||||
|
}
|
||||||
|
|
||||||
|
existingRecords.forEach(l -> logger.warn("id {} still in the list", l));
|
||||||
|
|
||||||
|
Assert.assertEquals(0, errors.get());
|
||||||
|
Assert.assertEquals(0, existingRecords.size());
|
||||||
|
|
||||||
|
return journal;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private OrderedExecutorFactory getExecutorFactory() {
|
||||||
|
ExecutorService service = Executors.newFixedThreadPool(10, new ThreadFactory() {
|
||||||
|
int counter = 0;
|
||||||
|
@Override
|
||||||
|
public Thread newThread(Runnable r) {
|
||||||
|
Thread t = new Thread(r);
|
||||||
|
t.setName("AsyncCommitTest" + (counter++));
|
||||||
|
return t;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
OrderedExecutorFactory orderedExecutorFactory = new OrderedExecutorFactory(service);
|
||||||
|
runAfter(service::shutdownNow);
|
||||||
|
return orderedExecutorFactory;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNIO() throws Exception {
|
||||||
|
internalTest(JournalType.NIO, "testRunNIO", true);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNIONoSync() throws Exception {
|
||||||
|
internalTest(JournalType.NIO, "testRunNIO", false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testMapped() throws Exception {
|
||||||
|
internalTest(JournalType.MAPPED, "testRunMapped", true);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testMappedNoSync() throws Exception {
|
||||||
|
internalTest(JournalType.MAPPED, "testRunMapped", false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testAIO() throws Exception {
|
||||||
|
Assume.assumeTrue(LibaioContext.isLoaded());
|
||||||
|
internalTest(JournalType.ASYNCIO, "testRunAIO", true);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testAIONoSync() throws Exception {
|
||||||
|
Assume.assumeTrue(LibaioContext.isLoaded());
|
||||||
|
internalTest(JournalType.ASYNCIO, "testRunAIO", false);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void proceedCall(String testName, String testFolder, boolean sync) throws Exception {
|
||||||
|
Method method = getClass().getMethod(testName, String.class, Boolean.TYPE);
|
||||||
|
Journal journal = (Journal)method.invoke(this, testFolder, sync);
|
||||||
|
journal.stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void internalTest(JournalType journalType, String testRunName, boolean sync) throws Exception {
|
||||||
|
proceedCall(testRunName, getTestDir(), sync);
|
||||||
|
|
||||||
|
OrderedExecutorFactory orderedExecutorFactory = getExecutorFactory();
|
||||||
|
setupJournal(journalType, getTestDir(), orderedExecutorFactory);
|
||||||
|
|
||||||
|
ArrayList<RecordInfo> commited = new ArrayList<>();
|
||||||
|
ArrayList<PreparedTransactionInfo> prepared = new ArrayList<>();
|
||||||
|
AtomicInteger failedTX = new AtomicInteger(0);
|
||||||
|
|
||||||
|
journal.start();
|
||||||
|
journal.load(commited, prepared, (id, records, toDelete) -> failedTX.incrementAndGet(), false);
|
||||||
|
runAfter(journal::stop);
|
||||||
|
|
||||||
|
commited.forEach(r -> {
|
||||||
|
String dataAsString = new String(r.data);
|
||||||
|
logger.debug("data={}, isUpdate={}, id={}", dataAsString, r.isUpdate, r.id);
|
||||||
|
if (r.isUpdate) {
|
||||||
|
Assert.assertEquals("up " + r.id, dataAsString);
|
||||||
|
} else {
|
||||||
|
Assert.assertEquals("add " + r.id, dataAsString);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
Assert.assertEquals(RECORDS * 2, commited.size());
|
||||||
|
Assert.assertEquals(0, failedTX.get());
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
public void setupJournal(JournalType journalType, String location, ExecutorFactory executorFactory) {
|
||||||
|
|
||||||
|
File locationFile = new File(location);
|
||||||
|
|
||||||
|
switch (journalType) {
|
||||||
|
|
||||||
|
case NIO:
|
||||||
|
journalFF = new NIOSequentialFileFactory(locationFile, true, BUFFER_SIZE, BUFFER_TIMEOUT, 1, true, null, null);
|
||||||
|
break;
|
||||||
|
case ASYNCIO:
|
||||||
|
journalFF = new AIOSequentialFileFactory(locationFile, BUFFER_SIZE, BUFFER_TIMEOUT, MAX_AIO, true, null, null);
|
||||||
|
break;
|
||||||
|
case MAPPED:
|
||||||
|
journalFF = new MappedSequentialFileFactory(locationFile, FILE_SIZE, true, BUFFER_SIZE, BUFFER_TIMEOUT, null);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw new IllegalStateException("invalid journal type " + journalType);
|
||||||
|
}
|
||||||
|
|
||||||
|
journal = new JournalImpl(executorFactory, FILE_SIZE, MIN_FILES, POOL_SIZE, 0, 0, 30_000, journalFF, FILE_PREFIX, FILE_EXTENSION, MAX_AIO, 1, null, 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -32,6 +32,7 @@ import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
import org.apache.activemq.artemis.cli.commands.tools.journal.DecodeJournal;
|
import org.apache.activemq.artemis.cli.commands.tools.journal.DecodeJournal;
|
||||||
import org.apache.activemq.artemis.cli.commands.tools.journal.EncodeJournal;
|
import org.apache.activemq.artemis.cli.commands.tools.journal.EncodeJournal;
|
||||||
import org.apache.activemq.artemis.core.io.SequentialFileFactory;
|
import org.apache.activemq.artemis.core.io.SequentialFileFactory;
|
||||||
|
import org.apache.activemq.artemis.core.io.nio.NIOSequentialFileFactory;
|
||||||
import org.apache.activemq.artemis.core.journal.EncodingSupport;
|
import org.apache.activemq.artemis.core.journal.EncodingSupport;
|
||||||
import org.apache.activemq.artemis.core.journal.PreparedTransactionInfo;
|
import org.apache.activemq.artemis.core.journal.PreparedTransactionInfo;
|
||||||
import org.apache.activemq.artemis.core.journal.RecordInfo;
|
import org.apache.activemq.artemis.core.journal.RecordInfo;
|
||||||
|
@ -199,7 +200,7 @@ public abstract class JournalImplTestBase extends ActiveMQTestBase {
|
||||||
|
|
||||||
if (suportsRetention()) {
|
if (suportsRetention()) {
|
||||||
// FakeSequentialFile won't support retention
|
// FakeSequentialFile won't support retention
|
||||||
File fileBackup = new File(getTestDir(), "backupFoler");
|
File fileBackup = new File(getTestDir(), "backupFolder");
|
||||||
fileBackup.mkdirs();
|
fileBackup.mkdirs();
|
||||||
((JournalImpl) journal).setHistoryFolder(fileBackup, -1, -1);
|
((JournalImpl) journal).setHistoryFolder(fileBackup, -1, -1);
|
||||||
}
|
}
|
||||||
|
@ -691,35 +692,37 @@ public abstract class JournalImplTestBase extends ActiveMQTestBase {
|
||||||
Assert.assertArrayEquals(expectedArray, actualArray);
|
Assert.assertArrayEquals(expectedArray, actualArray);
|
||||||
} catch (AssertionError e) {
|
} catch (AssertionError e) {
|
||||||
|
|
||||||
|
logger.warn(e.getMessage(), e);
|
||||||
|
|
||||||
HashSet<RecordInfo> hashActual = new HashSet<>();
|
HashSet<RecordInfo> hashActual = new HashSet<>();
|
||||||
hashActual.addAll(actual);
|
hashActual.addAll(actual);
|
||||||
|
|
||||||
HashSet<RecordInfo> hashExpected = new HashSet<>();
|
HashSet<RecordInfo> hashExpected = new HashSet<>();
|
||||||
hashExpected.addAll(expected);
|
hashExpected.addAll(expected);
|
||||||
|
|
||||||
logger.debug("#Summary **********************************************************************************************************************");
|
logger.warn("#Summary **********************************************************************************************************************");
|
||||||
for (RecordInfo r : hashActual) {
|
for (RecordInfo r : hashActual) {
|
||||||
if (!hashExpected.contains(r)) {
|
if (!hashExpected.contains(r)) {
|
||||||
logger.debug("Record {} was supposed to be removed and it exists", r);
|
logger.warn("Record {} was supposed to be removed and it exists", r);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (RecordInfo r : hashExpected) {
|
for (RecordInfo r : hashExpected) {
|
||||||
if (!hashActual.contains(r)) {
|
if (!hashActual.contains(r)) {
|
||||||
logger.debug("Record {} was not found on actual list", r);
|
logger.warn("Record {} was not found on actual list", r);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.debug("#expected **********************************************************************************************************************");
|
logger.warn("#expected **********************************************************************************************************************");
|
||||||
for (RecordInfo recordInfo : expected) {
|
for (RecordInfo recordInfo : expected) {
|
||||||
logger.debug("Record::{}", recordInfo);
|
logger.warn("Record::{}", recordInfo);
|
||||||
}
|
}
|
||||||
logger.debug("#actual ************************************************************************************************************************");
|
logger.warn("#actual ************************************************************************************************************************");
|
||||||
for (RecordInfo recordInfo : actual) {
|
for (RecordInfo recordInfo : actual) {
|
||||||
logger.debug("Record::{}", recordInfo);
|
logger.warn("Record::{}", recordInfo);
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.debug("#records ***********************************************************************************************************************");
|
logger.warn("#records ***********************************************************************************************************************");
|
||||||
|
|
||||||
try {
|
try {
|
||||||
describeJournal(journal.getFileFactory(), (JournalImpl) journal, journal.getFileFactory().getDirectory(), System.out);
|
describeJournal(journal.getFileFactory(), (JournalImpl) journal, journal.getFileFactory().getDirectory(), System.out);
|
||||||
|
@ -727,6 +730,20 @@ public abstract class JournalImplTestBase extends ActiveMQTestBase {
|
||||||
e2.printStackTrace();
|
e2.printStackTrace();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
logger.warn("#records on retention (history) folder ***********************************************************************************************************************");
|
||||||
|
|
||||||
|
try {
|
||||||
|
NIOSequentialFileFactory nioSequentialFileFactory = new NIOSequentialFileFactory(journal.getHistoryFolder(), 1);
|
||||||
|
|
||||||
|
|
||||||
|
JournalImpl backupJournal = new JournalImpl(journal.getFileSize(), journal.getMinFiles(), 10, 0, 0, nioSequentialFileFactory, "amq", "amq", 1);
|
||||||
|
|
||||||
|
describeJournal(nioSequentialFileFactory, backupJournal, journal.getHistoryFolder(), System.out);
|
||||||
|
|
||||||
|
} catch (Exception e2) {
|
||||||
|
e2.printStackTrace();
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -906,8 +906,6 @@ public abstract class JournalImplTestUnit extends JournalImplTestBase {
|
||||||
testReclaimTransactionalAdd(false);
|
testReclaimTransactionalAdd(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO commit and rollback, also transactional deletes
|
|
||||||
|
|
||||||
private void testReclaimTransactionalAdd(final boolean commit) throws Exception {
|
private void testReclaimTransactionalAdd(final boolean commit) throws Exception {
|
||||||
setup(10, 10 * 1024, true);
|
setup(10, 10 * 1024, true);
|
||||||
createJournal();
|
createJournal();
|
||||||
|
|
Loading…
Reference in New Issue