storage/src/vespa/storage/storageutil/resumeguard.h


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

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once

namespace storage {

class ResumeGuard {
public:
    class Callback {
    public:
        virtual ~Callback() {};

        virtual void resume() = 0;
    };

    ResumeGuard()
        : _cb(nullptr)
    {}

    ResumeGuard(Callback& cb)
        : _cb(&cb) {};

    ResumeGuard(const ResumeGuard& other) {
        _cb = other._cb;
        const_cast<ResumeGuard&>(other)._cb = nullptr;
    }

    ~ResumeGuard() {
        if (_cb) {
            _cb->resume();
        }
    }

private:
    Callback* _cb;
};

}