aboutsummaryrefslogtreecommitdiff
path: root/src/zenserver/proxy/httpproxystats.h
blob: 76ac7c87501197ea14500340022e07a31c408b29 (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
// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include <zenhttp/httpserver.h>

#include <memory>
#include <vector>

namespace zen {

class TcpProxyService;

class HttpProxyStatsService : public HttpService, public IHttpStatsProvider
{
public:
	HttpProxyStatsService(const std::vector<std::unique_ptr<TcpProxyService>>& ProxyServices,
						  IHttpStatsService&								   StatsService,
						  std::string										   DefaultRecordDir);
	~HttpProxyStatsService();

	virtual const char* BaseUri() const override;
	virtual void		HandleRequest(HttpServerRequest& Request) override;

	// IHttpStatsProvider
	virtual void	 HandleStatsRequest(HttpServerRequest& Request) override;
	virtual CbObject CollectStats() override;

private:
	void HandleRecordStart(HttpServerRequest& Request);
	void HandleRecordStop(HttpServerRequest& Request);
	void HandleRecordStatus(HttpServerRequest& Request);

	const std::vector<std::unique_ptr<TcpProxyService>>& m_ProxyServices;
	IHttpStatsService&									 m_StatsService;
	std::string											 m_DefaultRecordDir;
};

}  // namespace zen