log_rotation.hpp

Go to the documentation of this file.

/*
 * Copyright 2025 Jinwoo Sung
 *
 * 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.
 */
 
#pragma once
 
#include <atomic>
#include <chrono>
#include <filesystem>
#include <mutex>
#include <string>
#include <vector>
 
#include "unilink/base/visibility.hpp"
 
namespace unilink {
namespace diagnostics {
 
struct LogRotationConfig {
  size_t max_file_size_bytes = 10 * 1024 * 1024;    // 10MB default
  size_t max_files = 10;                            // Keep 10 files max
  bool enable_compression = false;                  // Future feature
  std::string file_pattern = "{name}.{index}.log";  // File naming pattern
 
  LogRotationConfig() = default;
 
  LogRotationConfig(size_t max_size, size_t max_count) : max_file_size_bytes(max_size), max_files(max_count) {}
};
 
class UNILINK_API LogRotation {
 public:
  explicit LogRotation(const LogRotationConfig& config = LogRotationConfig{});
 
  bool should_rotate(const std::string& filepath) const;
 
  std::string rotate(const std::string& filepath);
 
  void cleanup_old_files(const std::string& base_filepath);
 
  std::string get_next_file_path(const std::string& base_filepath) const;
 
  void update_config(const LogRotationConfig& config);
 
  const LogRotationConfig& get_config() const { return config_; }
 
  static size_t get_file_size(const std::string& filepath);
 
  static std::vector<std::string> get_log_files(const std::string& base_filepath);
 
 private:
  LogRotationConfig config_;
  mutable std::mutex mutex_;
 
  std::string get_base_filename(const std::string& filepath) const;
 
  std::string get_directory(const std::string& filepath) const;
 
  int get_file_index(const std::string& filename) const;
 
  std::string generate_filename(const std::string& base_name, int index) const;
 
  void sort_files_by_time(std::vector<std::string>& files) const;
};
 
}  // namespace diagnostics
}  // namespace unilink