about summary refs log tree commit diff homepage
path: root/lib/Core/PTreeWriter.h
blob: 12709116c997ffb83ed0c701bace692a81299a83 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
//===-- PTreeWriter.h -------------------------------------------*- C++ -*-===//
//
//                     The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#pragma once

#include <sqlite3.h>

#include <cstdint>
#include <string>

namespace klee {
class AnnotatedPTreeNode;

/// @brief Writes process tree nodes into an SQLite database
class PTreeWriter {
  friend class PersistentPTree;

  ::sqlite3 *db{nullptr};
  ::sqlite3_stmt *insertStmt{nullptr};
  ::sqlite3_stmt *transactionBeginStmt{nullptr};
  ::sqlite3_stmt *transactionCommitStmt{nullptr};
  std::uint32_t batch{0};
  bool flushed{true};

  /// Writes nodes in batches
  void batchCommit(bool force = false);

public:
  explicit PTreeWriter(const std::string &dbPath);
  ~PTreeWriter();
  PTreeWriter(const PTreeWriter &other) = delete;
  PTreeWriter(PTreeWriter &&other) noexcept = delete;
  PTreeWriter &operator=(const PTreeWriter &other) = delete;
  PTreeWriter &operator=(PTreeWriter &&other) noexcept = delete;

  /// Write new node into database
  void write(const AnnotatedPTreeNode &node);
};

} // namespace klee