diff options
| author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
|---|---|---|
| committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
| commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
| tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/monlib/service/monservice.cpp | |
| download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz | |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/monlib/service/monservice.cpp')
| -rw-r--r-- | library/cpp/monlib/service/monservice.cpp | 129 |
1 files changed, 129 insertions, 0 deletions
diff --git a/library/cpp/monlib/service/monservice.cpp b/library/cpp/monlib/service/monservice.cpp new file mode 100644 index 0000000000..d1b9cda1d2 --- /dev/null +++ b/library/cpp/monlib/service/monservice.cpp @@ -0,0 +1,129 @@ +#include "monservice.h" + +#include <library/cpp/malloc/api/malloc.h> +#include <library/cpp/string_utils/base64/base64.h> +#include <library/cpp/svnversion/svnversion.h> + +#include <util/generic/map.h> +#include <util/generic/ptr.h> +#include <util/system/hostname.h> + +#include <google/protobuf/text_format.h> + +using namespace NMonitoring; + +TMonService2::TMonService2(ui16 port, const TString& host, ui32 threads, const TString& title, THolder<IAuthProvider> auth) + : TMonService2(HttpServerOptions(port, host, threads), title, std::move(auth)) +{ +} + +TMonService2::TMonService2(const THttpServerOptions& options, const TString& title, THolder<IAuthProvider> auth) + : NMonitoring::TMtHttpServer(options, std::bind(&TMonService2::ServeRequest, this, std::placeholders::_1, std::placeholders::_2)) + , Title(title) + , IndexMonPage(new TIndexMonPage("", Title)) + , AuthProvider_{std::move(auth)} +{ + Y_VERIFY(!!title); + time_t t = time(nullptr); + ctime_r(&t, StartTime); +} + +TMonService2::TMonService2(const THttpServerOptions& options, TSimpleSharedPtr<IThreadPool> pool, const TString& title, THolder<IAuthProvider> auth) + : NMonitoring::TMtHttpServer(options, std::bind(&TMonService2::ServeRequest, this, std::placeholders::_1, std::placeholders::_2), std::move(pool)) + , Title(title) + , IndexMonPage(new TIndexMonPage("", Title)) + , AuthProvider_{std::move(auth)} +{ + Y_VERIFY(!!title); + time_t t = time(nullptr); + ctime_r(&t, StartTime); +} + +TMonService2::TMonService2(ui16 port, ui32 threads, const TString& title, THolder<IAuthProvider> auth) + : TMonService2(port, TString(), threads, title, std::move(auth)) +{ +} + +TMonService2::TMonService2(ui16 port, const TString& title, THolder<IAuthProvider> auth) + : TMonService2(port, TString(), 0, title, std::move(auth)) +{ +} + +void TMonService2::OutputIndex(IOutputStream& out) { + IndexMonPage->OutputIndex(out, true); +} + +void TMonService2::OutputIndexPage(IOutputStream& out) { + out << HTTPOKHTML; + out << "<html>\n"; + IndexMonPage->OutputHead(out); + OutputIndexBody(out); + out << "</html>\n"; +} + +void TMonService2::OutputIndexBody(IOutputStream& out) { + out << "<body>\n"; + + // part of common navbar + out << "<ol class='breadcrumb'>\n"; + out << "<li class='active'>" << Title << "</li>\n"; + out << "</ol>\n"; + + out << "<div class='container'>\n" + << "<h2>" << Title << "</h2>\n"; + OutputIndex(out); + out + << "<div>\n" + << "</body>\n"; +} + +void TMonService2::ServeRequest(IOutputStream& out, const NMonitoring::IHttpRequest& request) { + TString path = request.GetPath(); + Y_VERIFY(path.StartsWith('/')); + + if (AuthProvider_) { + const auto authResult = AuthProvider_->Check(request); + switch (authResult.Status) { + case TAuthResult::EStatus::NoCredentials: + out << HTTPUNAUTHORIZED; + return; + case TAuthResult::EStatus::Denied: + out << HTTPFORBIDDEN; + return; + case TAuthResult::EStatus::Ok: + break; + } + } + + if (path == "/") { + OutputIndexPage(out); + } else { + TMonService2HttpRequest monService2HttpRequest( + &out, &request, this, IndexMonPage.Get(), path, nullptr); + IndexMonPage->Output(monService2HttpRequest); + } +} + +void TMonService2::Register(IMonPage* page) { + IndexMonPage->Register(page); +} + +void TMonService2::Register(TMonPagePtr page) { + IndexMonPage->Register(std::move(page)); +} + +TIndexMonPage* TMonService2::RegisterIndexPage(const TString& path, const TString& title) { + return IndexMonPage->RegisterIndexPage(path, title); +} + +IMonPage* TMonService2::FindPage(const TString& relativePath) { + return IndexMonPage->FindPage(relativePath); +} + +TIndexMonPage* TMonService2::FindIndexPage(const TString& relativePath) { + return IndexMonPage->FindIndexPage(relativePath); +} + +void TMonService2::SortPages() { + IndexMonPage->SortPages(); +} |