Logger.h

Go to the documentation of this file.

/*
 * Copyright 2022 Stéphane Caron
 *
 * Licensed 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.
 *
 * This file incorporates work covered by the following copyright and
 * permission notice:
 *
 *     Logger class of mc_rtc
 *     Copyright 2015-2019 CNRS-UM LIRMM, CNRS-AIST JRL
 *     License: BSD-2-Clause (see licenses/LICENSE-mc_rtc)
 */
 
#pragma once
 
#include <palimpsest/Dictionary.h>
 
#include <fstream>
#include <string>
#include <thread>
#include <utility>
#include <vector>
 
#include "mpacklog/CircularBuffer.h"
 
namespace mpacklog {
 
using palimpsest::Dictionary;
 
constexpr size_t kBufferSize = 1024;
 
class Logger {
 public:
  explicit Logger(const std::string &path, bool append = false);
 
  ~Logger();
 
  bool put(char *data, size_t size);
 
  inline bool put(const Dictionary &dict) {
    size_t size = dict.serialize(serialization_buffer_);
    return Logger::put(serialization_buffer_.data(), size);
  }
 
  size_t last_size() const { return last_size_; }
 
 private:
  void flush_buffer();
 
 private:
  std::ofstream file_;
 
  std::thread thread_;
 
  bool keep_going_ = true;
 
  CircularBuffer<std::pair<char *, size_t>, kBufferSize> circular_buffer_;
 
  std::pair<char *, size_t> pop_;
 
  std::vector<char> serialization_buffer_;
 
  size_t last_size_ = 0;
};
 
}  // namespace mpacklog