use of org.apache.distributedlog.DLSN in project bookkeeper by apache.
the class DLOutputStream method write.
private synchronized void write(ByteBuf buf) throws IOException {
Throwable cause = exceptionUpdater.get(this);
if (null != cause) {
if (cause instanceof IOException) {
throw (IOException) cause;
} else {
throw new UnexpectedException("Encountered unknown issue", cause);
}
}
writePos += buf.readableBytes();
LogRecord record = new LogRecord(writePos, buf);
writer.write(record).whenComplete(new FutureEventListener<DLSN>() {
@Override
public void onSuccess(DLSN value) {
synchronized (syncPos) {
syncPos[0] = record.getTransactionId();
}
}
@Override
public void onFailure(Throwable cause) {
exceptionUpdater.compareAndSet(DLOutputStream.this, null, cause);
}
});
}
use of org.apache.distributedlog.DLSN in project bookkeeper by apache.
the class ZKSubscriptionStateStore method getLastCommitPositionFromZK.
CompletableFuture<DLSN> getLastCommitPositionFromZK() {
final CompletableFuture<DLSN> result = new CompletableFuture<DLSN>();
try {
logger.debug("Reading last commit position from path {}", zkPath);
zooKeeperClient.get().getData(zkPath, false, new AsyncCallback.DataCallback() {
@Override
public void processResult(int rc, String path, Object ctx, byte[] data, Stat stat) {
logger.debug("Read last commit position from path {}: rc = {}", zkPath, rc);
if (KeeperException.Code.NONODE.intValue() == rc) {
result.complete(DLSN.NonInclusiveLowerBound);
} else if (KeeperException.Code.OK.intValue() != rc) {
result.completeExceptionally(KeeperException.create(KeeperException.Code.get(rc), path));
} else {
try {
DLSN dlsn = DLSN.deserialize(new String(data, Charsets.UTF_8));
result.complete(dlsn);
} catch (Exception t) {
logger.warn("Invalid last commit position found from path {}", zkPath, t);
// invalid dlsn recorded in subscription state store
result.complete(DLSN.NonInclusiveLowerBound);
}
}
}
}, null);
} catch (ZooKeeperClient.ZooKeeperConnectionException zkce) {
result.completeExceptionally(zkce);
} catch (InterruptedException ie) {
Thread.currentThread().interrupt();
result.completeExceptionally(new DLInterruptedException("getLastCommitPosition was interrupted", ie));
}
return result;
}
use of org.apache.distributedlog.DLSN in project bookkeeper by apache.
the class MVCCStoreFactoryImplTest method setup.
@Before
public void setup() throws IOException {
this.namespace = mock(Namespace.class);
DistributedLogManager dlm = mock(DistributedLogManager.class);
when(dlm.asyncClose()).thenReturn(FutureUtils.Void());
when(namespace.openLog(anyString())).thenReturn(dlm);
AsyncLogWriter logWriter = mock(AsyncLogWriter.class);
when(dlm.openAsyncLogWriter()).thenReturn(FutureUtils.value(logWriter));
when(logWriter.getLastTxId()).thenReturn(-1L);
DLSN dlsn = new DLSN(0L, 0L, 0L);
when(logWriter.write(any(LogRecord.class))).thenReturn(FutureUtils.value(dlsn));
when(logWriter.asyncClose()).thenReturn(FutureUtils.Void());
AsyncLogReader logReader = mock(AsyncLogReader.class);
when(dlm.openAsyncLogReader(anyLong())).thenReturn(FutureUtils.value(logReader));
when(logReader.asyncClose()).thenReturn(FutureUtils.Void());
LogRecordWithDLSN record = new LogRecordWithDLSN(dlsn, 0L, NOP_CMD.toByteArray(), 0L);
when(logReader.readNext()).thenReturn(FutureUtils.value(record));
int numDirs = 3;
this.storeDirs = new File[numDirs];
for (int i = 0; i < numDirs; i++) {
storeDirs[i] = testDir.newFolder("test-" + i);
}
this.resources = StorageResources.create(StorageResourcesSpec.builder().numCheckpointThreads(3).numIOReadThreads(3).numIOWriteThreads(3).build());
this.factory = new MVCCStoreFactoryImpl(() -> namespace, storeDirs, resources, false);
}
use of org.apache.distributedlog.DLSN in project bookkeeper by apache.
the class MVCCAsyncStoreTestBase method mockNamespace.
private static Namespace mockNamespace() throws Exception {
Namespace namespace = mock(Namespace.class);
DistributedLogManager dlm = mock(DistributedLogManager.class);
when(dlm.asyncClose()).thenReturn(FutureUtils.Void());
when(namespace.openLog(anyString())).thenReturn(dlm);
AsyncLogWriter logWriter = mock(AsyncLogWriter.class);
when(dlm.openAsyncLogWriter()).thenReturn(FutureUtils.value(logWriter));
when(logWriter.getLastTxId()).thenReturn(-1L);
DLSN dlsn = new DLSN(0L, 0L, 0L);
when(logWriter.write(any(LogRecord.class))).thenReturn(FutureUtils.value(dlsn));
when(logWriter.asyncClose()).thenReturn(FutureUtils.Void());
AsyncLogReader logReader = mock(AsyncLogReader.class);
when(dlm.openAsyncLogReader(anyLong())).thenReturn(FutureUtils.value(logReader));
when(logReader.asyncClose()).thenReturn(FutureUtils.Void());
LogRecordWithDLSN record = new LogRecordWithDLSN(dlsn, 0L, NOP_CMD.toByteArray(), 0L);
when(logReader.readNext()).thenReturn(FutureUtils.value(record));
return namespace;
}
Aggregations