blob: ea79d957e66a974e1016d4ada53e924658357141 (
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
47
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "i_scheduled_executor.h"
#include <vespa/vespalib/stllike/hash_map.h>
#include <mutex>
#include <vector>
class FNET_Transport;
namespace proton {
class TimerTask;
/**
* ScheduledExecutor is a class capable of running Tasks at a regular
* interval. The timer can be reset to clear all tasks currently being
* scheduled.
*/
class ScheduledExecutor : public IScheduledExecutor
{
private:
using TaskList = vespalib::hash_map<uint64_t, std::unique_ptr<TimerTask>>;
FNET_Transport & _transport;
std::mutex _lock;
uint64_t _nextKey;
TaskList _taskList;
bool cancel(uint64_t key);
class Registration;
public:
/**
* Create a new timer, capable of scheduling tasks at fixed intervals.
*/
ScheduledExecutor(FNET_Transport & transport);
/**
* Destroys this timer, finishing the current task executing and then
* finishing.
*/
~ScheduledExecutor() override;
[[nodiscard]] Handle scheduleAtFixedRate(std::unique_ptr<Executor::Task> task, duration delay, duration interval) override;
};
}
|