dev

Branches (7)

Tags (101)

Manage

Manage

dev

2.5.x

2.4.x

2.3.x

2.2.x

2.1.x

next

2.5.20

2.4.11client

2.4.8client

2.4.9

2.3.7

2.3.6

2.2.4client

2.2.7

2.2.6

2.1.20

2.2.3client

2.2.5

2.2.4

2.2.1client

2.2.2

2.1.15client

2.1.14client

2.1.18

2.1.13client

2.1.17

urbackup_backend
/
PipeThrottler.h

#pragma once

#include "Interface/PipeThrottler.h"
#include <memory>

class IMutex;

class PipeThrottler : public IPipeThrottler
{
public:
	PipeThrottler(size_t bps, bool percent_max, IPipeThrottlerUpdater* updater);
	PipeThrottler();
	~PipeThrottler(void);

	virtual bool addBytes(size_t new_bytes, bool wait);

	virtual void changeThrottleLimit(size_t bps, bool p_percent_max);

	virtual void changeThrottleUpdater(IPipeThrottlerUpdater* new_updater);

private:
	enum ThrottleState
	{
		ThrottleState_Probe,
		ThrottleState_Throttle
	};

	size_t throttle_bps;
	bool percent_max;
	int64 update_time_interval;
	size_t curr_bytes;
	int64 lastresettime;
	int64 lastupdatetime;
	std::unique_ptr<IPipeThrottlerUpdater> updater;
	ThrottleState throttle_state;
	int64 lastprobetime;
	float probe_bps;
	size_t throttle_percent;
	float last_probe_result;
	size_t probe_interval;

	IMutex *mutex;
};