Examples with LogBuffer org.neo4j.kernel.impl.transaction.xaframework.LogBuffer used on opensource projects

Example 1 with LogBuffer

use of org.neo4j.kernel.impl.transaction.xaframework.LogBuffer in project graphdb by neo4j-attic.

the class SlaveTxIdGenerator method generate.

public long generate(final XaDataSource dataSource, final int identifier) {
    try {
        final int eventIdentifier = txManager.getEventIdentifier();
        Response<Long> response = broker.getMaster().first().commitSingleResourceTransaction(receiver.getSlaveContext(eventIdentifier), dataSource.getName(), new TxExtractor() {

            public void extract(LogBuffer buffer) {
                try {
                    dataSource.getPreparedTransaction(identifier, buffer);
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }

            public ReadableByteChannel extract() {
                try {
                    return dataSource.getPreparedTransaction(identifier);
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
        });
        return receiver.receive(response);
    } catch (ZooKeeperException e) {
        receiver.newMaster(null, e);
        throw e;
    } catch (ComException e) {
        receiver.newMaster(null, e);
        throw e;
    }
}

Example 2 with LogBuffer

use of org.neo4j.kernel.impl.transaction.xaframework.LogBuffer in project graphdb by neo4j-attic.

the class MasterUtil method packResponse.

public static <T> Response<T> packResponse(GraphDatabaseService graphDb, SlaveContext context, T response, Predicate<Long> filter) {
    List<Triplet<String, Long, TxExtractor>> stream = new ArrayList<Triplet<String, Long, TxExtractor>>();
    Set<String> resourceNames = new HashSet<String>();
    XaDataSourceManager dsManager = ((AbstractGraphDatabase) graphDb).getConfig().getTxModule().getXaDataSourceManager();
    for (Pair<String, Long> txEntry : context.lastAppliedTransactions()) {
        String resourceName = txEntry.first();
        final XaDataSource dataSource = dsManager.getXaDataSource(resourceName);
        if (dataSource == null) {
            throw new RuntimeException("No data source '" + resourceName + "' found");
        }
        resourceNames.add(resourceName);
        long masterLastTx = dataSource.getLastCommittedTxId();
        for (long txId = txEntry.other() + 1; txId <= masterLastTx; txId++) {
            if (filter.accept(txId)) {
                final long tx = txId;
                TxExtractor extractor = new TxExtractor() {

                    @Override
                    public ReadableByteChannel extract() {
                        try {
                            return dataSource.getCommittedTransaction(tx);
                        } catch (IOException e) {
                            throw new RuntimeException(e);
                        }
                    }

                    @Override
                    public void extract(LogBuffer buffer) {
                        try {
                            dataSource.getCommittedTransaction(tx, buffer);
                        } catch (IOException e) {
                            throw new RuntimeException(e);
                        }
                    }
                };
                stream.add(Triplet.of(resourceName, txId, extractor));
            }
        }
    }
    StoreId storeId = ((NeoStoreXaDataSource) dsManager.getXaDataSource(Config.DEFAULT_DATA_SOURCE_NAME)).getStoreId();
    return new Response<T>(response, storeId, TransactionStream.create(resourceNames, stream));
}