|
| 1 | +#include "rpc_scheme_base.h" |
| 2 | +#include "service_view.h" |
| 3 | + |
| 4 | +#include <ydb/core/grpc_services/base/base.h> |
| 5 | +#include <ydb/core/tx/schemeshard/schemeshard.h> |
| 6 | +#include <ydb/core/ydb_convert/ydb_convert.h> |
| 7 | +#include <ydb/library/actors/core/actor.h> |
| 8 | +#include <ydb/library/actors/core/hfunc.h> |
| 9 | +#include <ydb/public/api/protos/draft/ydb_view.pb.h> |
| 10 | + |
| 11 | +namespace NKikimr::NGRpcService { |
| 12 | + |
| 13 | +using namespace Ydb; |
| 14 | + |
| 15 | +using TEvDescribeView = TGrpcRequestOperationCall<View::DescribeViewRequest, View::DescribeViewResponse>; |
| 16 | + |
| 17 | +class TDescribeViewRPC : public TRpcSchemeRequestActor<TDescribeViewRPC, TEvDescribeView> { |
| 18 | + using TBase = TRpcSchemeRequestActor<TDescribeViewRPC, TEvDescribeView>; |
| 19 | + |
| 20 | +public: |
| 21 | + using TBase::TBase; |
| 22 | + |
| 23 | + void Bootstrap() { |
| 24 | + DescribeScheme(); |
| 25 | + } |
| 26 | + |
| 27 | + void PassAway() override { |
| 28 | + TBase::PassAway(); |
| 29 | + } |
| 30 | + |
| 31 | +private: |
| 32 | + void DescribeScheme() { |
| 33 | + auto ev = std::make_unique<TEvTxUserProxy::TEvNavigate>(); |
| 34 | + SetAuthToken(ev, *Request_); |
| 35 | + SetDatabase(ev.get(), *Request_); |
| 36 | + ev->Record.MutableDescribePath()->SetPath(GetProtoRequest()->path()); |
| 37 | + |
| 38 | + Send(MakeTxProxyID(), ev.release()); |
| 39 | + Become(&TDescribeViewRPC::StateDescribeScheme); |
| 40 | + } |
| 41 | + |
| 42 | + STATEFN(StateDescribeScheme) { |
| 43 | + switch (ev->GetTypeRewrite()) { |
| 44 | + HFunc(NSchemeShard::TEvSchemeShard::TEvDescribeSchemeResult, Handle); |
| 45 | + default: |
| 46 | + return TBase::StateWork(ev); |
| 47 | + } |
| 48 | + } |
| 49 | + |
| 50 | + void Handle(NSchemeShard::TEvSchemeShard::TEvDescribeSchemeResult::TPtr& ev, const TActorContext& ctx) { |
| 51 | + const auto& record = ev->Get()->GetRecord(); |
| 52 | + const auto& desc = record.GetPathDescription(); |
| 53 | + |
| 54 | + if (record.HasReason()) { |
| 55 | + Request_->RaiseIssue(NYql::TIssue(record.GetReason())); |
| 56 | + } |
| 57 | + |
| 58 | + switch (record.GetStatus()) { |
| 59 | + case NKikimrScheme::StatusSuccess: |
| 60 | + if (desc.GetSelf().GetPathType() != NKikimrSchemeOp::EPathTypeView) { |
| 61 | + auto message = TStringBuilder() << "Expected a view, but got: " << desc.GetSelf().GetPathType(); |
| 62 | + Request_->RaiseIssue(NYql::TIssue(message)); |
| 63 | + return Reply(StatusIds::SCHEME_ERROR, ctx); |
| 64 | + } |
| 65 | + |
| 66 | + ConvertDirectoryEntry(desc.GetSelf(), Result_.mutable_self(), true); |
| 67 | + Result_.set_query_text(desc.GetViewDescription().GetQueryText()); |
| 68 | + |
| 69 | + return ReplyWithResult(StatusIds::SUCCESS, Result_, ctx); |
| 70 | + |
| 71 | + case NKikimrScheme::StatusPathDoesNotExist: |
| 72 | + case NKikimrScheme::StatusSchemeError: |
| 73 | + return Reply(StatusIds::SCHEME_ERROR, ctx); |
| 74 | + |
| 75 | + case NKikimrScheme::StatusAccessDenied: |
| 76 | + return Reply(StatusIds::UNAUTHORIZED, ctx); |
| 77 | + |
| 78 | + case NKikimrScheme::StatusNotAvailable: |
| 79 | + return Reply(StatusIds::UNAVAILABLE, ctx); |
| 80 | + |
| 81 | + default: |
| 82 | + return Reply(StatusIds::GENERIC_ERROR, ctx); |
| 83 | + } |
| 84 | + } |
| 85 | + |
| 86 | +private: |
| 87 | + View::DescribeViewResult Result_; |
| 88 | +}; |
| 89 | + |
| 90 | +void DoDescribeView(std::unique_ptr<IRequestOpCtx> p, const IFacilityProvider& f) { |
| 91 | + f.RegisterActor(new TDescribeViewRPC(p.release())); |
| 92 | +} |
| 93 | + |
| 94 | +} |
0 commit comments