// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SQL_DATABASE_MEMORY_DUMP_PROVIDER_H_
#define SQL_DATABASE_MEMORY_DUMP_PROVIDER_H_

#include <string>

#include "base/memory/raw_ptr.h"
#include "base/synchronization/lock.h"
#include "base/thread_annotations.h"
#include "base/trace_event/memory_dump_provider.h"

struct sqlite3;

namespace base::trace_event {
struct MemoryDumpArgs;
class ProcessMemoryDump;
}  // namespace base::trace_event

namespace sql {

class DatabaseMemoryDumpProvider
    : public base::trace_event::MemoryDumpProvider {
 public:
  DatabaseMemoryDumpProvider(sqlite3* db, const std::string& name);

  DatabaseMemoryDumpProvider(const DatabaseMemoryDumpProvider&) = delete;
  DatabaseMemoryDumpProvider& operator=(const DatabaseMemoryDumpProvider&) =
      delete;

  ~DatabaseMemoryDumpProvider() override;

  void ResetDatabase();

  // base::trace_event::MemoryDumpProvider implementation.
  bool OnMemoryDump(
      const base::trace_event::MemoryDumpArgs& args,
      base::trace_event::ProcessMemoryDump* process_memory_dump) override;

  // Reports memory usage into provided memory dump with the given |dump_name|.
  // Called by sql::Database when its owner asks it to report memory usage.
  bool ReportMemoryUsage(base::trace_event::ProcessMemoryDump* pmd,
                         const std::string& dump_name);

 private:
  struct MemoryUsageResult {
    bool is_valid = false;
    int cache_size = 0;
    int schema_size = 0;
    int statement_size = 0;
  };
  MemoryUsageResult GetDbMemoryUsage();

  std::string FormatDumpName() const;

  base::Lock lock_;
  raw_ptr<sqlite3> db_ GUARDED_BY_CONTEXT(lock_);  // not owned.
  const std::string connection_name_;
};

}  // namespace sql

#endif  // SQL_DATABASE_MEMORY_DUMP_PROVIDER_H_