aboutsummaryrefslogtreecommitdiffstats
path: root/vespalib/src/vespa/vespalib/util/invokeserviceimpl.h
blob: ef0755e76146e67bfb1a8fb5003be19504201414 (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include "invokeservice.h"
#include "time.h"
#include <mutex>
#include <vector>
#include <thread>
#include <condition_variable>

namespace vespalib {

/**
 * An invoke service what will invoke the given function with at specified frequency.
 */
class InvokeServiceImpl : public InvokeService {
public:
    InvokeServiceImpl(duration napTime);
    InvokeServiceImpl(const InvokeServiceImpl &) = delete;
    InvokeServiceImpl & operator=(const InvokeServiceImpl &) = delete;
    ~InvokeServiceImpl() override;
    std::unique_ptr<IDestructorCallback> registerInvoke(InvokeFunc func) override;
    const std::atomic<steady_time> & nowRef() const { return _now; }
private:
    using IdAndFunc = std::pair<uint64_t, InvokeFunc>;
    class Registration;
    void unregister(uint64_t id);
    void runLoop();
    duration                       _naptime;
    std::atomic<steady_time>       _now;
    std::mutex                     _lock;
    std::condition_variable        _cond;
    uint64_t                       _currId;
    bool                           _closed;
    std::vector<IdAndFunc>         _toInvoke;
    std::unique_ptr<std::thread>   _thread;
};

}