blob: 23c6e1e559bb93eafbbf24912e89f5293f001f8e (
plain) (
blame)
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "attributefilebufferwriter.h"
#include <vespa/vespalib/data/databuffer.h>
namespace search {
AttributeFileBufferWriter::
AttributeFileBufferWriter(IAttributeFileWriter &fileWriter)
: BufferWriter(),
_buf(),
_bytesWritten(0),
_incompleteBuffers(0),
_fileWriter(fileWriter)
{
_buf = _fileWriter.allocBuf(BUFFER_SIZE);
assert(_buf->getFreeLen() >= BUFFER_SIZE);
setup(_buf->getFree(), BUFFER_SIZE);
}
AttributeFileBufferWriter::~AttributeFileBufferWriter()
{
assert(usedLen() == 0);
}
void
AttributeFileBufferWriter::flush()
{
assert(_incompleteBuffers == 0); // all previous buffers must have been full
size_t nowLen = usedLen();
if (nowLen != BUFFER_SIZE) {
// buffer is not full, only allowed for last buffer
++_incompleteBuffers;
}
if (nowLen == 0) {
return; // empty buffer
}
assert(_buf->getDataLen() == 0);
onFlush(nowLen);
assert(_buf->getFreeLen() >= BUFFER_SIZE);
setup(_buf->getFree(), BUFFER_SIZE);
_bytesWritten += nowLen;
}
} // namespace search
|