/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); 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. */ /* * Modified by Cloudius Systems * Copyright 2015 Cloudius Systems */ #include #include // for boost::crc_32_type #include #include #include #include #include "core/align.hh" #include "core/reactor.hh" #include "core/scollectd.hh" #include "core/future-util.hh" #include "core/file.hh" #include "core/rwlock.hh" #include "core/fstream.hh" #include "net/byteorder.hh" #include "commitlog.hh" #include "db/config.hh" #include "utils/data_input.hh" class crc32: public boost::crc_32_type { public: template void process(T t) { auto v = net::hton(t); this->process_bytes(&v, sizeof(T)); } }; db::commitlog::config::config(const db::config& cfg) : commit_log_location(cfg.commitlog_directory()) , commitlog_total_space_in_mb(cfg.commitlog_total_space_in_mb()) , commitlog_segment_size_in_mb(cfg.commitlog_segment_size_in_mb()) , commitlog_sync_period_in_ms(cfg.commitlog_sync_batch_window_in_ms()) , mode(cfg.commitlog_sync() == "batch" ? sync_mode::BATCH : sync_mode::PERIODIC) {} class db::commitlog::descriptor { public: static const std::string SEPARATOR; static const std::string FILENAME_PREFIX; static const std::string FILENAME_EXTENSION; descriptor(descriptor&&) = default; descriptor(const descriptor&) = default; // TODO : version management descriptor(uint64_t i, uint32_t v = 1) : id(i), ver(v) { } descriptor(std::pair p) : descriptor(p.first, p.second) { } descriptor(sstring filename) : descriptor([filename]() { std::smatch m; // match both legacy and new version of commitlogs Ex: CommitLog-12345.log and CommitLog-4-12345.log. std::regex rx(FILENAME_PREFIX + "((\\d+)(" + SEPARATOR + "\\d+)?)" + FILENAME_EXTENSION); if (!std::regex_match(std::string(filename), m, rx)) { throw std::runtime_error("Cannot parse the version of the file: " + filename); } if (m[3].length() == 0) { // CMH. Can most likely ignore this throw std::domain_error("Commitlog segment is too old to open; upgrade to 1.2.5+ first"); } uint64_t id = std::stoull(m[3].str().substr(1)); uint32_t ver = std::stoul(m[2].str()); return std::make_pair(id, ver); }()) { } sstring filename() const { return FILENAME_PREFIX + std::to_string(ver) + SEPARATOR + std::to_string(id) + FILENAME_EXTENSION; } const uint64_t id; const uint32_t ver; }; const std::string db::commitlog::descriptor::SEPARATOR("-"); const std::string db::commitlog::descriptor::FILENAME_PREFIX( "CommitLog" + SEPARATOR); const std::string db::commitlog::descriptor::FILENAME_EXTENSION(".log"); class db::commitlog::segment_manager { public: config cfg; const uint64_t max_size; const uint64_t max_mutation_size; semaphore _new_segment_semaphore; std::vector _regs; // TODO: verify that we're ok with not-so-great granularity using clock_type = lowres_clock; using time_point = clock_type::time_point; using sseg_ptr = lw_shared_ptr; struct stats { uint64_t cycle_count = 0; uint64_t flush_count = 0; uint64_t allocation_count = 0; uint64_t bytes_written = 0; uint64_t bytes_slack = 0; uint64_t segments_created = 0; uint64_t segments_destroyed = 0; }; stats totals; segment_manager(config cfg) : cfg(cfg), max_size( std::max(cfg.commitlog_segment_size_in_mb, 1) * 1024 * 1024), max_mutation_size(max_size >> 1) { assert(max_size > 0); if (cfg.commit_log_location.empty()) { cfg.commit_log_location = "/tmp/urchin/commitlog/" + std::to_string(engine().cpu_id()); } _regs = create_counters(); } uint64_t next_id() { return ++_ids; } future<> process(directory_entry de) { if (de.type && de.type == directory_entry_type::regular) { descriptor d(de.name); _ids = std::max(_ids, d.id); } return make_ready_future<>(); } future<> init(); future new_segment(); future active_segment(); future<> clear(); std::vector create_counters(); void discard_unused_segments(); void discard_completed_segments(const cf_id_type& id, const replay_position& pos); void sync(); void arm() { _timer.arm(std::chrono::milliseconds(cfg.commitlog_sync_period_in_ms)); } private: uint64_t _ids = 0; std::vector _segments; timer _timer; }; /* * A single commit log file on disk. Manages creation of the file and writing mutations to disk, * as well as tracking the last mutation position of any "dirty" CFs covered by the segment file. Segment * files are initially allocated to a fixed size and can grow to accomidate a larger value if necessary. */ class db::commitlog::segment: public enable_lw_shared_from_this { segment_manager* _segment_manager; descriptor _desc; file _file; uint64_t _file_pos = 0; uint64_t _flush_pos = 0; uint64_t _buf_pos = 0; bool _closed = false; using buffer_type = temporary_buffer; using sseg_ptr = segment_manager::sseg_ptr; using clock_type = segment_manager::clock_type; using time_point = segment_manager::time_point; buffer_type _buffer; rwlock _dwrite; // used as a barrier between write & flush std::unordered_map _cf_dirty; time_point _sync_time; public: // The commit log entry overhead in bytes (int: length + int: head checksum + int: tail checksum) static const constexpr size_t entry_overhead_size = 3 * sizeof(uint32_t); static const constexpr size_t segment_overhead_size = 2 * sizeof(uint32_t); static const constexpr size_t descriptor_header_size = 4 * sizeof(uint32_t); // The commit log (chained) sync marker/header size in bytes (int: length + int: checksum [segmentId, position]) static const constexpr size_t sync_marker_size = 2 * sizeof(uint32_t); static const constexpr size_t alignment = 4096; // TODO : tune initial / default size static const constexpr size_t default_size = 8 * alignment; segment(segment_manager* m, const descriptor& d, file && f) : _segment_manager(m), _desc(std::move(d)), _file(std::move(f)) { ++_segment_manager->totals.segments_created; } ~segment() { ++_segment_manager->totals.segments_destroyed; ::unlink( (_segment_manager->cfg.commit_log_location + "/" + _desc.filename()).c_str()); } bool must_sync() { if (_segment_manager->cfg.mode == sync_mode::BATCH) { return true; } auto now = clock_type::now(); auto ms = std::chrono::duration_cast( now - _sync_time).count(); return _segment_manager->cfg.commitlog_sync_period_in_ms < uint64_t(ms); } /** * Finalize this segment and get a new one */ future finish_and_get_new() { _closed = true; return sync().then([](auto seg) { return seg->_segment_manager->active_segment(); }); } future sync() { if (position() <= _flush_pos) { return make_ready_future(shared_from_this()); } return cycle().then([](auto seg) { return seg->flush(); }); } future flush(uint64_t pos = 0) { auto me = shared_from_this(); if (pos == 0) { pos = _file_pos; } if (pos != 0 && pos <= _flush_pos) { return make_ready_future(std::move(me)); } // Make sure all disk writes are done. // This is not 100% neccesary, we really only need the ones below our flush pos, // but since we pretty much assume that task ordering will make this the case anyway... return _dwrite.write_lock().then( [this, me = std::move(me), pos]() mutable { _dwrite.write_unlock(); // release it already. pos = std::max(pos, _file_pos); if (pos <= _flush_pos) { return make_ready_future(std::move(me)); } _sync_time = clock_type::now(); return _file.flush().then([this, pos, me = std::move(me)]() { _flush_pos = std::max(pos, _flush_pos); ++_segment_manager->totals.flush_count; return make_ready_future(std::move(me)); }); }); } /** * Send any buffer contents to disk and get a new tmp buffer */ future cycle(size_t s = 0) { auto size = clear_buffer_slack(); auto buf = std::move(_buffer); auto off = _file_pos; _file_pos += size; _buf_pos = 0; // if we need new buffer, get one. // TODO: keep a queue of available buffers? if (s > 0) { auto overhead = segment_overhead_size; if (_file_pos == 0) { overhead += descriptor_header_size; } auto k = std::max(align_up(s + overhead, alignment), default_size); auto a = ::memalign(alignment, k); _buffer = buffer_type(reinterpret_cast(a), k, make_free_deleter(a)); _buf_pos = overhead; auto * p = reinterpret_cast(_buffer.get_write()); std::fill(p, p + overhead, 0); } auto me = shared_from_this(); if (size == 0) { return make_ready_future(std::move(me)); } auto * p = buf.get_write(); assert(std::count(p, p + 2 * sizeof(uint32_t), 0) == 2 * sizeof(uint32_t)); data_output out(p, p + buf.size()); auto header_size = 0; if (off == 0) { // first block. write file header. out.write(_desc.ver); out.write(_desc.id); crc32 crc; crc.process(_desc.ver); crc.process(_desc.id & 0xffffffff); crc.process(_desc.id >> 32); out.write(crc.checksum()); header_size = descriptor_header_size; } // write chunk header crc32 crc; crc.process(_desc.id & 0xffffffff); crc.process(_desc.id >> 32); crc.process(uint32_t(off + header_size)); out.write(uint32_t(_file_pos)); out.write(crc.checksum()); // acquire read lock return _dwrite.read_lock().then( [this, size, off, buf = std::move(buf), me = std::move(me)]() mutable { auto p = buf.get(); return _file.dma_write(off, p, size).then([this, size, buf = std::move(buf), me = std::move(me)](size_t written) mutable { assert(written == size); // we are not equipped to deal with partial writes. _segment_manager->totals.bytes_written += written; ++_segment_manager->totals.cycle_count; return make_ready_future(std::move(me)); }).finally([me, this]() { _dwrite.read_unlock(); // release }); }); } /** * Add a "mutation" to the segment. */ future allocate(const cf_id_type& id, size_t size, serializer_func func) { const auto s = size + entry_overhead_size; // total size if (s > _segment_manager->max_mutation_size) { return make_exception_future( std::invalid_argument( "Mutation of " + std::to_string(s) + " bytes is too large for the maxiumum size of " + std::to_string(_segment_manager->max_mutation_size))); } // would we make the file too big? if (position() + s > _segment_manager->max_size) { // do this in next segment instead. return finish_and_get_new().then( [id, size, func = std::move(func)](auto new_seg) { return new_seg->allocate(id, size, func); }); } // enough data? if (s > (_buffer.size() - _buf_pos)) { // TODO: iff we have to many writes running, maybe we should // wait for this? cycle(s); } replay_position rp(_desc.id, position()); auto pos = _buf_pos; _buf_pos += s; _cf_dirty[id] = rp.pos; auto * p = _buffer.get_write() + pos; auto * e = _buffer.get_write() + pos + s - sizeof(uint32_t); data_output out(p, e); crc32 crc; out.write(uint32_t(s)); crc.process(uint32_t(s)); out.write(crc.checksum()); // actual data func(out); crc.process_bytes(p + 2 * sizeof(uint32_t), size); out = data_output(e, sizeof(uint32_t)); out.write(crc.checksum()); ++_segment_manager->totals.allocation_count; // finally, check if we're required to sync. if (must_sync()) { return sync().then([rp](auto seg) { return make_ready_future(rp); }); } return make_ready_future(rp); } position_type position() const { return _file_pos + _buf_pos; } size_t size_on_disk() const { return _file_pos; } // ensures no more of this segment is writeable, by allocating any unused section at the end and marking it discarded // a.k.a. zero the tail. size_t clear_buffer_slack() { auto size = align_up(_buf_pos, alignment); std::fill(_buffer.get_write() + _buf_pos, _buffer.get_write() + size, 0); _segment_manager->totals.bytes_slack += (size - _buf_pos); return size; } void mark_clean(const cf_id_type& id, position_type pos) { auto i = _cf_dirty.find(id); if (i != _cf_dirty.end() && i->second <= pos) { _cf_dirty.erase(i); } } void mark_clean(const cf_id_type& id, const replay_position& pos) { if (pos.id == _desc.id) { mark_clean(id, pos.pos); } else if (pos.id > _desc.id) { mark_clean(id, std::numeric_limits::max()); } } void mark_clean() { _cf_dirty.clear(); } bool is_still_allocating() const { return !_closed && position() < _segment_manager->max_size; } bool is_clean() { return _cf_dirty.empty(); } bool is_unused() { return !is_still_allocating() && is_clean(); } bool contains(const replay_position& pos) { return pos.id == _desc.id; } }; const size_t db::commitlog::segment::default_size; future<> db::commitlog::segment_manager::init() { struct helper { file _file; subscription _list; helper(segment_manager * m, file && f) : _file(std::move(f)), _list( _file.list_directory( std::bind(&segment_manager::process, m, std::placeholders::_1))) { } future<> done() { return _list.done(); } }; return engine().open_directory(cfg.commit_log_location).then([this](auto dir) { // keep sub alive... auto h = make_lw_shared(this, std::move(dir)); return h->done().then([this, h]() { return this->active_segment().then([this, h](auto) { // nothing really. just keeping sub alive if (cfg.mode != sync_mode::BATCH) { _timer.set_callback(std::bind(&segment_manager::sync, this)); this->arm(); } }); }); }); } std::vector db::commitlog::segment_manager::create_counters() { using scollectd::add_polled_metric; using scollectd::make_typed; using scollectd::type_instance_id; using scollectd::per_cpu_plugin_instance; using scollectd::data_type; std::vector regs = { add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "queue_length", "segments") , make_typed(data_type::GAUGE , std::bind(&decltype(_segments)::size, &_segments)) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "queue_length", "allocating_segments") , make_typed(data_type::GAUGE , [this]() { return std::count_if(_segments.begin(), _segments.end(), [](const sseg_ptr & s) { return s->is_still_allocating(); }); }) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "queue_length", "unused_segments") , make_typed(data_type::GAUGE , [this]() { return std::count_if(_segments.begin(), _segments.end(), [](const sseg_ptr & s) { return s->is_unused(); }); }) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "total_operations", "alloc") , make_typed(data_type::DERIVE, totals.allocation_count) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "total_operations", "cycle") , make_typed(data_type::DERIVE, totals.cycle_count) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "total_operations", "flush") , make_typed(data_type::DERIVE, totals.flush_count) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "total_bytes", "written") , make_typed(data_type::GAUGE, totals.bytes_written) ), add_polled_metric(type_instance_id("commitlog" , per_cpu_plugin_instance, "total_bytes", "slack") , make_typed(data_type::GAUGE, totals.bytes_slack) ), }; return regs; } future db::commitlog::segment_manager::new_segment() { descriptor d(next_id()); return engine().open_file_dma(cfg.commit_log_location + "/" + d.filename(), open_flags::wo|open_flags::create).then( [this, d](file f) { if (cfg.commitlog_total_space_in_mb != 0) { auto i = _segments.rbegin(); auto e = _segments.rend(); size_t s = 0, n = 0; while (i != e) { auto& seg = *i; s += seg->size_on_disk(); if (!seg->is_still_allocating() && s >= cfg.commitlog_total_space_in_mb) { seg->mark_clean(); ++n; } ++i; } if (n > 0) { discard_unused_segments(); } } _segments.emplace_back(make_lw_shared(this, d, std::move(f))); return make_ready_future(_segments.back()); }); } future db::commitlog::segment_manager::active_segment() { if (_segments.empty() || !_segments.back()->is_still_allocating()) { return _new_segment_semaphore.wait().then([this]() { if (_segments.empty() || !_segments.back()->is_still_allocating()) { return new_segment(); } return make_ready_future(_segments.back()); }).finally([this]() { _new_segment_semaphore.signal(); }); } return make_ready_future(_segments.back()); } /** * go through all segments, clear id up to pos. if segment becomes clean and unused by this, * it is discarded. */ void db::commitlog::segment_manager::discard_completed_segments( const cf_id_type& id, const replay_position& pos) { for (auto&s : _segments) { s->mark_clean(id, pos); } discard_unused_segments(); } void db::commitlog::segment_manager::discard_unused_segments() { auto i = std::remove_if(_segments.begin(), _segments.end(), [=](auto& s) { return s->is_unused(); }); if (i != _segments.end()) { _segments.erase(i, _segments.end()); } } /* * Sync all segments, then clear them out. To ensure all ops are done. * (Assumes you have barriered adding ops!) */ future<> db::commitlog::segment_manager::clear() { return do_until([this]() {return _segments.empty();}, [this]() { auto s = _segments.front(); _segments.erase(_segments.begin()); return s->sync().then([](sseg_ptr) { }); }); } /** * Called by timer in periodic mode. */ void db::commitlog::segment_manager::sync() { for (auto& s : _segments) { if (s->must_sync()) { s->sync(); // we do not care about waiting... } } arm(); } /** * Add mutation. */ future db::commitlog::add(const cf_id_type& id, size_t size, serializer_func func) { return _segment_manager->active_segment().then([=](auto s) { return s->allocate(id, size, std::move(func)); }); } db::commitlog::commitlog(config cfg) : _segment_manager(new segment_manager(std::move(cfg))) { } db::commitlog::commitlog(commitlog&& v) : _segment_manager(std::move(v._segment_manager)) { } db::commitlog::~commitlog() { } future db::commitlog::create_commitlog(config cfg) { commitlog c(std::move(cfg)); auto f = c._segment_manager->init(); return f.then([c = std::move(c)]() mutable { return make_ready_future(std::move(c)); }); } void db::commitlog::discard_completed_segments(const cf_id_type& id, const replay_position& pos) { _segment_manager->discard_completed_segments(id, pos); } size_t db::commitlog::max_record_size() const { return _segment_manager->max_mutation_size - segment::entry_overhead_size; } future<> db::commitlog::clear() { return _segment_manager->clear(); } const db::commitlog::config& db::commitlog::active_config() const { return _segment_manager->cfg; } future>> db::commitlog::read_log_file(const sstring& filename, commit_load_reader_func next) { return engine().open_file_dma(filename, open_flags::ro).then([next = std::move(next)](file f) { return read_log_file(std::move(f), std::move(next)); }); } subscription> db::commitlog::read_log_file(file f, commit_load_reader_func next) { struct work { stream> s; input_stream fin; input_stream r; lw_shared_ptr f; uint64_t id = 0; size_t pos = 0; size_t next = 0; bool eof = false; bool header = true; }; auto w = make_lw_shared(); w->f = make_lw_shared(std::move(f)); w->fin = make_file_input_stream(w->f); auto ret = w->s.listen(std::move(next)); w->s.started().then([w] { return w->fin.read_exactly(segment::descriptor_header_size).then([w](temporary_buffer buf) { // Will throw if we got eof data_input in(buf); auto ver = in.read(); auto id = in.read(); auto checksum = in.read(); crc32 crc; crc.process(ver); crc.process(id & 0xffffffff); crc.process(id >> 32); auto cs = crc.checksum(); if (cs != checksum) { throw std::runtime_error("Checksum error in file header"); } w->id = id; w->next = 0; w->pos = buf.size(); auto eofcond = [w] { return w->eof; }; return do_until(eofcond, [w] { assert(w->pos == w->next || w->next == 0); return w->fin.read_exactly(segment::segment_overhead_size).then([w](temporary_buffer buf) { if (buf.size() == 0) { w->eof = true; return make_ready_future<>(); } data_input in(buf); auto next = in.read(); auto checksum = in.read(); crc32 crc; crc.process(w->id & 0xffffffff); crc.process(w->id >> 32); crc.process(w->pos); auto cs = crc.checksum(); if (cs != checksum) { throw std::runtime_error("Checksum error in chunk header"); } w->pos += 8; w->next = next; auto eoccond = [w] { return w->pos == w->next; }; return do_until(eoccond, [w] { static const constexpr size_t entry_header_size = segment::entry_overhead_size - sizeof(uint32_t); return w->fin.read_exactly(entry_header_size).then([w](temporary_buffer buf) { data_input in(buf); auto size = in.read(); auto checksum = in.read(); if (size == 0) { // special urchin case: zero padding due to dma blocks auto slack = w->next - entry_header_size - w->pos; return w->fin.read_exactly(slack).then([w, slack](temporary_buffer buf) { // should eof be an error here? w->pos += slack + entry_header_size; }); } if (size < 3 * sizeof(uint32_t)) { throw std::runtime_error("Invalid entry size"); } return w->fin.read_exactly(size - entry_header_size).then([w, size, checksum](temporary_buffer buf) { data_input in(buf); auto data_size = size - segment::entry_overhead_size; in.skip(data_size); auto checksum = in.read(); crc32 crc; crc.process(size); crc.process_bytes(buf.get(), data_size); if (crc.checksum() != checksum) { throw std::runtime_error("Checksum error in data entry"); } w->pos += size; return w->s.produce(buf.share(0, data_size)); }); }); }); }); }); }); }).then([w] { w->s.close(); }); return ret; }