blob: ab39e51710e962cf0c5ea1001cf3a1e7a1a9bf0d [file] [log] [blame]
// Copyright 2019 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.
#ifndef SRC_STORAGE_FSHOST_METRICS_COBALT_H_
#define SRC_STORAGE_FSHOST_METRICS_COBALT_H_
#include <lib/async/dispatcher.h>
#include <zircon/compiler.h>
#include <condition_variable>
#include <memory>
#include <mutex>
#include <thread>
#include <unordered_map>
#include <utility>
#include <cobalt-client/cpp/collector.h>
#include <cobalt-client/cpp/counter.h>
#include "src/lib/storage/vfs/cpp/metrics/events.h"
#include "src/storage/fshost/metrics.h"
namespace fshost {
// This class is an adapter for the cobalt-client interface, which is specialized for
// fshost metrics.
//
// This class is thread-compatible.
class FsHostMetricsCobalt : public FsHostMetrics {
public:
explicit FsHostMetricsCobalt(std::unique_ptr<cobalt_client::Collector> collector);
~FsHostMetricsCobalt() override;
// Returns a pointer to the underlying |cobalt_client::Collector| instance.
cobalt_client::Collector* mutable_collector() { return collector_.get(); }
void LogMinfsCorruption() override;
// Repeatedly attempt to flush to cobalt until success.
//
// Retries every 10 seconds.
// The retry is done async.
void Flush() override;
private:
// Sleep duration between two successive attempts to flush metrics.
static constexpr std::chrono::nanoseconds kSleepDuration = std::chrono::seconds(10);
void Run();
std::unique_ptr<cobalt_client::Collector> collector_;
std::unordered_map<fs_metrics::Event, std::unique_ptr<cobalt_client::Counter>> counters_;
std::mutex mutex_;
// A way to wakeup sleeping |thread_| when object is getting destroyed.
std::condition_variable_any condition_;
// True if destructor is called.
bool shut_down_ __TA_GUARDED(mutex_) = false;
// True if |thread_| should try to flush metrics.
bool flush_ __TA_GUARDED(mutex_) = false;
// Thread which periodically flushes metrics.
std::thread thread_;
};
} // namespace fshost
#endif // SRC_STORAGE_FSHOST_METRICS_COBALT_H_