blob: 9633c6825d86791d140ddb9714caa14ea684d141 [file] [log] [blame]
// Copyright 2022 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <fuchsia/diagnostics/cpp/fidl.h>
#include <fuchsia/logger/cpp/fidl.h>
#include <lib/async/dispatcher.h>
#include <lib/fpromise/result.h>
#include <lib/sys/cpp/service_directory.h>
#include <memory>
#include <string>
#include <vector>
#include "src/lib/backoff/backoff.h"
#include "src/lib/fxl/memory/weak_ptr.h"
namespace forensics::feedback_data {
// Receives logs emitted by the system.
class LogSink {
// A message is a FIDL LogMessage or an error string.
using MessageOr = ::fpromise::result<fuchsia::logger::LogMessage, std::string>;
// Adds |message| to the sink.
// Returns false if the write fails though callers are not expected to take action on failure.
virtual bool Add(MessageOr message) = 0;
// Notifies the sink the log source was interrupted and the messages it received in the past may
// be sent again.
virtual void NotifyInterruption() = 0;
// Returns true if the sink is safe to use after an interruption has occurred.
virtual bool SafeAfterInterruption() const = 0;
// Receives log messages from the system's logging service and dispatches them to a sink.
class LogSource {
LogSource(async_dispatcher_t* dispatcher, std::shared_ptr<sys::ServiceDirectory> services,
LogSink* sink, std::unique_ptr<backoff::Backoff> backoff = nullptr);
// Starts log collection.
// Note: a check-fail will occur if |sink_| cannot safely receive messages after being interrupted
// and the log stream has stopped due to a disconnection or call to Stop.
void Start();
// Stops log collection and notfies |sink_| collection was interrupted.
void Stop();
void OnDisconnect();
void GetNext();
async_dispatcher_t* dispatcher_;
std::shared_ptr<sys::ServiceDirectory> services_;
LogSink* sink_;
fuchsia::diagnostics::ArchiveAccessorPtr archive_accessor_;
fuchsia::diagnostics::BatchIteratorPtr batch_iterator_;
bool is_stopped_{false};
std::unique_ptr<backoff::Backoff> backoff_;
fxl::WeakPtrFactory<LogSource> ptr_factory_{this};
} // namespace forensics::feedback_data