12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- #pragma once
- #include "service.h"
- #include "auth.h"
- #include "mon_service_http_request.h"
- #include <library/cpp/monlib/service/pages/index_mon_page.h>
- #include <library/cpp/monlib/service/pages/mon_page.h>
- #include <util/system/progname.h>
- #include <functional>
- namespace NMonitoring {
- class TMonService2: public TMtHttpServer {
- protected:
- const TString Title;
- char StartTime[26];
- TIntrusivePtr<TIndexMonPage> IndexMonPage;
- THolder<IAuthProvider> AuthProvider_;
- public:
- static THttpServerOptions HttpServerOptions(ui16 port, const TString& host, ui32 threads) {
- THttpServerOptions opts(port);
- if (!host.empty()) {
- opts.SetHost(host);
- }
- opts.SetClientTimeout(TDuration::Minutes(1));
- opts.EnableCompression(true);
- opts.SetThreads(threads);
- opts.SetMaxConnections(std::max<ui32>(100, threads));
- opts.EnableRejectExcessConnections(true);
- return opts;
- }
- static THttpServerOptions HttpServerOptions(ui16 port, ui32 threads) {
- return HttpServerOptions(port, TString(), threads);
- }
- public:
- explicit TMonService2(ui16 port, const TString& title = GetProgramName(), THolder<IAuthProvider> auth = nullptr);
- explicit TMonService2(ui16 port, ui32 threads, const TString& title = GetProgramName(), THolder<IAuthProvider> auth = nullptr);
- explicit TMonService2(ui16 port, const TString& host, ui32 threads, const TString& title = GetProgramName(), THolder<IAuthProvider> auth = nullptr);
- explicit TMonService2(const THttpServerOptions& options, const TString& title = GetProgramName(), THolder<IAuthProvider> auth = nullptr);
- explicit TMonService2(const THttpServerOptions& options, TSimpleSharedPtr<IThreadPool> pool, const TString& title = GetProgramName(), THolder<IAuthProvider> auth = nullptr);
- ~TMonService2() override {
- }
- const char* GetStartTime() const {
- return StartTime;
- }
- const TString& GetTitle() const {
- return Title;
- }
- virtual void ServeRequest(IOutputStream& out, const NMonitoring::IHttpRequest& request);
- virtual void OutputIndex(IOutputStream& out);
- virtual void OutputIndexPage(IOutputStream& out);
- virtual void OutputIndexBody(IOutputStream& out);
- void Register(IMonPage* page);
- void Register(TMonPagePtr page);
- TIndexMonPage* RegisterIndexPage(const TString& path, const TString& title);
- IMonPage* FindPage(const TString& relativePath);
- TIndexMonPage* FindIndexPage(const TString& relativePath);
- void SortPages();
- };
- }
|