summaryrefslogtreecommitdiffstats
path: root/src/zstd/contrib/pzstd/ErrorHolder.h
blob: 829651c5961e579d19eafb8062e2c63f80c14b86 (plain)
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
47
48
49
50
51
52
53
54
/*
 * Copyright (c) 2016-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under both the BSD-style license (found in the
 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
 * in the COPYING file in the root directory of this source tree).
 */
#pragma once

#include <atomic>
#include <cassert>
#include <stdexcept>
#include <string>

namespace pzstd {

// Coordinates graceful shutdown of the pzstd pipeline
class ErrorHolder {
  std::atomic<bool> error_;
  std::string message_;

 public:
  ErrorHolder() : error_(false) {}

  bool hasError() noexcept {
    return error_.load();
  }

  void setError(std::string message) noexcept {
    // Given multiple possibly concurrent calls, exactly one will ever succeed.
    bool expected = false;
    if (error_.compare_exchange_strong(expected, true)) {
      message_ = std::move(message);
    }
  }

  bool check(bool predicate, std::string message) noexcept {
    if (!predicate) {
      setError(std::move(message));
    }
    return !hasError();
  }

  std::string getError() noexcept {
    error_.store(false);
    return std::move(message_);
  }

  ~ErrorHolder() {
    assert(!hasError());
  }
};
}