-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
ChunkedTransaction.java
112 lines (100 loc) · 3.39 KB
/
ChunkedTransaction.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
* Copyright (c) 2002-2018 "Neo4j,"
* Neo4j Sweden AB [http://neo4j.com]
*
* This file is part of Neo4j Enterprise Edition. The included source
* code can be redistributed and/or modified under the terms of the
* GNU AFFERO GENERAL PUBLIC LICENSE Version 3
* (http://www.fsf.org/licensing/licenses/agpl-3.0.html) with the
* Commons Clause, as found in the associated LICENSE.txt file.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* Neo4j object code can be licensed independently from the source
* under separate terms from the AGPL. Inquiries can be directed to:
* licensing@neo4j.com
*
* More information is also available at:
* https://neo4j.com/licensing/
*/
package org.neo4j.causalclustering.core.state.machines.tx;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.stream.ChunkedInput;
import io.netty.util.ReferenceCountUtil;
import java.util.LinkedList;
import java.util.Queue;
import org.neo4j.causalclustering.helper.ErrorHandler;
import org.neo4j.causalclustering.messaging.ChunkingNetworkChannel;
import org.neo4j.kernel.impl.transaction.TransactionRepresentation;
class ChunkedTransaction implements ChunkedInput<ByteBuf>
{
private static final int CHUNK_SIZE = 32 * 1024;
private final ReplicatedTransactionFactory.TransactionRepresentationWriter txWriter;
private ChunkingNetworkChannel channel;
private Queue<ByteBuf> chunks = new LinkedList<>();
ChunkedTransaction( TransactionRepresentation tx )
{
txWriter = ReplicatedTransactionFactory.transactionalRepresentationWriter( tx );
}
@Override
public boolean isEndOfInput()
{
return channel != null && channel.closed() && chunks.isEmpty();
}
@Override
public void close()
{
try ( ErrorHandler errorHandler = new ErrorHandler( "Closing ChunkedTransaction" ) )
{
if ( channel != null )
{
errorHandler.execute( () -> channel.close() );
}
chunks.forEach( byteBuf -> errorHandler.execute( () -> ReferenceCountUtil.release( byteBuf ) ) );
}
}
@Override
public ByteBuf readChunk( ChannelHandlerContext ctx ) throws Exception
{
return readChunk( ctx.alloc() );
}
@Override
public ByteBuf readChunk( ByteBufAllocator allocator ) throws Exception
{
if ( isEndOfInput() )
{
return null;
}
if ( channel == null )
{
// Ensure that the written buffers does not overflow the allocators chunk size.
channel = new ChunkingNetworkChannel( allocator, CHUNK_SIZE, chunks );
}
// write to chunks if empty and there is more to write
while ( txWriter.canWrite() && chunks.isEmpty() )
{
txWriter.write( channel );
}
// nothing more to write, close the channel to get the potential last buffer
if ( chunks.isEmpty() )
{
channel.close();
}
return chunks.poll();
}
@Override
public long length()
{
return -1;
}
@Override
public long progress()
{
return 0;
}
}