Files
scylladb/core/fstream.cc
Avi Kivity 7f8d88371a Add LICENSE, NOTICE, and copyright headers to all source files.
The two files imported from the OSv project retain their original licenses.
2015-02-19 16:52:34 +02:00

53 lines
1.9 KiB
C++

/*
* This file is open source software, licensed to you under the terms
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. 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.
*/
/*
* Copyright (C) 2015 Cloudius Systems, Ltd.
*/
#include "fstream.hh"
#include <new>
#include <malloc.h>
#include <algorithm>
future<temporary_buffer<char>>
file_data_source_impl::get() {
// must align allocation for dma
auto p = ::memalign(std::min<size_t>(_buffer_size, 4096), _buffer_size);
if (!p) {
throw std::bad_alloc();
}
auto q = static_cast<char*>(p);
temporary_buffer<char> buf(q, _buffer_size, make_free_deleter(p));
auto old_pos = _pos;
// dma_read needs to be aligned. It doesn't have to be page-aligned,
// though, and we could get away with something much smaller. However, if
// we start reading in things outside page boundaries, we will end up with
// various pages around, some of them with overlapping ranges. Those would
// be very challenging to cache.
old_pos &= ~4095;
auto front = _pos - old_pos;
_pos += _buffer_size - front;
return _file.dma_read(old_pos, q, _buffer_size).then(
[buf = std::move(buf), front] (size_t size) mutable {
buf.trim(size);
buf.trim_front(front);
return make_ready_future<temporary_buffer<char>>(std::move(buf));
});
}