Filled DatabaseName in SchemeCache requests

Author: n00bcracker

ydb/core/discovery/discovery.cpp

@@ -751,6 +751,7 @@ class TDiscoverer: public TActorBootstrapped<TDiscoverer> {
             << ": path# " << id);
 
         auto request = MakeHolder<NSchemeCache::TSchemeCacheNavigate>();
+        request->DatabaseName = CanonizePath(AppData()->DomainsInfo->GetDomain()->Name);
 
         auto& entry = request->ResultSet.emplace_back();
         entry.Operation = NSchemeCache::TSchemeCacheNavigate::OpPath;

ydb/core/graph/service/service_impl.cpp

@@ -57,6 +57,8 @@ class TGraphService : public TActor<TGraphService> {
 
         BLOG_D("ResolveDatabase " << Database);
         TAutoPtr<NSchemeCache::TSchemeCacheNavigate> request(new NSchemeCache::TSchemeCacheNavigate());
+        request->DatabaseName = Database;
+
         NSchemeCache::TSchemeCacheNavigate::TEntry entry;
         entry.Operation = NSchemeCache::TSchemeCacheNavigate::OpList;
         entry.SyncVersion = false;

ydb/core/grpc_services/rpc_kh_describe.cpp

@@ -106,6 +106,8 @@ class TKikhouseDescribeTableRPC : public TActorBootstrapped<TKikhouseDescribeTab
             WaitingResolveReply = true;
         } else {
             TAutoPtr<NSchemeCache::TSchemeCacheNavigate> request(new NSchemeCache::TSchemeCacheNavigate());
+            request->DatabaseName = CanonizePath(Request->GetDatabaseName().GetOrElse(""));
+
             NSchemeCache::TSchemeCacheNavigate::TEntry entry;
             entry.Path = std::move(path);
             if (entry.Path.empty()) {

ydb/core/grpc_services/rpc_object_storage.cpp

@@ -133,6 +133,8 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
         // TODO: check all params;
 
         TAutoPtr<NSchemeCache::TSchemeCacheNavigate> request(new NSchemeCache::TSchemeCacheNavigate());
+        request->DatabaseName = CanonizePath(GrpcRequest->GetDatabaseName().GetOrElse(""));
+
         NSchemeCache::TSchemeCacheNavigate::TEntry entry;
         entry.Path = NKikimr::SplitPath(table);
         if (entry.Path.empty()) {
@@ -372,7 +374,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
             TString err;
 
             bool filterParsedOk = CellsFromTuple(&filterType, columnValues, typesRef, true, false, cells, err, owner);
-            
+
             if (!filterParsedOk) {
                 ReplyWithError(Ydb::StatusIds::BAD_REQUEST, Sprintf("Invalid filter: '%s'", err.data()), ctx);
                 return false;
@@ -545,7 +547,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
         ev->Record.SetPathColumnDelimiter(Request->Getpath_column_delimiter());
         ev->Record.SetSerializedStartAfterKeySuffix(StartAfterSuffixColumns.GetBuffer());
         ev->Record.SetMaxKeys(MaxKeys - ContentsRows.size() - CommonPrefixesRows.size());
-        
+
         if (!CommonPrefixesRows.empty()) {
             // Next shard might have the same common prefix, need to skip it
             ev->Record.SetLastCommonPrefix(CommonPrefixesRows.back());
@@ -564,7 +566,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
 
         if (!Filter.ColumnIds.empty()) {
             auto* filter = ev->Record.mutable_filter();
-            
+
             for (const auto& colId : Filter.ColumnIds) {
                 filter->add_columns(colId);
             }
@@ -624,7 +626,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
         afterLastFolderPrefix.Parse(prefixColumns);
 
         auto& partitions = KeyRange->GetPartitions();
-        
+
         for (CurrentShardIdx++; CurrentShardIdx < partitions.size(); CurrentShardIdx++) {
             auto& partition = KeyRange->GetPartitions()[CurrentShardIdx];
 
@@ -693,7 +695,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
             shardResponse.GetMoreRows()) {
             if (!FindNextShard()) {
                 ReplySuccess(ctx, (hasMoreShards && shardResponse.GetMoreRows()) || maxKeysExhausted);
-                return;    
+                return;
             }
             MakeShardRequest(CurrentShardIdx, ctx);
         } else {
@@ -714,7 +716,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
                 return NYdb::TTypeBuilder().Pg(getPgTypeFromColMeta(colMeta)).Build();
             case NScheme::NTypeIds::Decimal:
                 return NYdb::TTypeBuilder().Decimal(NYdb::TDecimalType(
-                        typeInfo.GetDecimalType().GetPrecision(), 
+                        typeInfo.GetDecimalType().GetPrecision(),
                         typeInfo.GetDecimalType().GetScale()))
                     .Build();
             default:
@@ -727,7 +729,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
         for (const auto& colMeta : columns) {
             const auto type = getTypeFromColMeta(colMeta);
             auto* col = resultSet.Addcolumns();
-            
+
             *col->mutable_type()->mutable_optional_type()->mutable_item() = NYdb::TProtoAccessor::GetProto(type);
             *col->mutable_name() = colMeta.Name;
         }
@@ -754,7 +756,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
                     Ydb::Value valueProto;
                     valueProto.set_low_128(loHi.first);
                     valueProto.set_high_128(loHi.second);
-                    const NYdb::TDecimalValue decimal(valueProto, 
+                    const NYdb::TDecimalValue decimal(valueProto,
                         {static_cast<ui8>(colMeta.PType.GetDecimalType().GetPrecision()), static_cast<ui8>(colMeta.PType.GetDecimalType().GetScale())});
                     vb.Decimal(decimal);
                 } else {
@@ -801,10 +803,10 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
         if (CommonPrefixesRows.size() > 0) {
             lastDirectory = CommonPrefixesRows[CommonPrefixesRows.size() - 1];
         }
-        
+
         if (isTruncated && (lastDirectory || lastFile)) {
             NKikimrTxDataShard::TObjectStorageListingContinuationToken token;
-            
+
             if (lastDirectory > lastFile) {
                 token.set_last_path(lastDirectory);
                 token.set_is_folder(true);
@@ -814,7 +816,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
             }
 
             TString serializedToken = token.SerializeAsString();
-            
+
             resp->set_next_continuation_token(serializedToken);
         }
 
@@ -824,7 +826,7 @@ class TObjectStorageListingRequestGrpc : public TActorBootstrapped<TObjectStorag
             GrpcRequest->RaiseIssue(NYql::ExceptionToIssue(ex));
             GrpcRequest->ReplyWithYdbStatus(Ydb::StatusIds::INTERNAL_ERROR);
         }
-        
+
         Finished = true;
         Die(ctx);
     }

ydb/core/grpc_services/rpc_rate_limiter_api.cpp

@@ -169,6 +169,7 @@ class TRateLimiterControlRequest : public TRateLimiterRequest<TRateLimiterContro
         }
 
         auto req = MakeHolder<NSchemeCache::TSchemeCacheNavigate>();
+        req->DatabaseName = CanonizePath(this->Request_->GetDatabaseName().GetOrElse(DatabaseFromDomain(AppData())));
         req->ResultSet.emplace_back();
         req->ResultSet.back().Path.swap(path);
         req->ResultSet.back().Operation = NSchemeCache::TSchemeCacheNavigate::OpPath;

ydb/core/grpc_services/rpc_read_columns.cpp

@@ -154,6 +154,8 @@ class TReadColumnsRPC : public TActorBootstrapped<TReadColumnsRPC> {
             WaitingResolveReply = true;
         } else {
             TAutoPtr<NSchemeCache::TSchemeCacheNavigate> request(new NSchemeCache::TSchemeCacheNavigate());
+            request->DatabaseName = CanonizePath(Request->GetDatabaseName().GetOrElse(""));
+
             NSchemeCache::TSchemeCacheNavigate::TEntry entry;
             entry.Path = std::move(path);
             if (entry.Path.empty()) {

ydb/core/grpc_services/rpc_read_rows.cpp

@@ -360,6 +360,7 @@ class TReadRowsRPC : public TActorBootstrapped<TReadRowsRPC> {
         entry.SyncVersion = false;
         entry.ShowPrivatePath = false;
         auto request = std::make_unique<NSchemeCache::TSchemeCacheNavigate>();
+        request->DatabaseName = CanonizePath(GetDatabase());
         request->ResultSet.emplace_back(entry);
         Send(MakeSchemeCacheID(), new TEvTxProxySchemeCache::TEvNavigateKeySet(request.release()), 0, 0, Span.GetTraceId());
         return true;

ydb/core/health_check/health_check.cpp

@@ -1067,6 +1067,7 @@ class TSelfCheckRequest : public TActorBootstrapped<TSelfCheckRequest> {
     [[nodiscard]] TRequestResponse<TEvTxProxySchemeCache::TEvNavigateKeySetResult> MakeRequestSchemeCacheNavigate(const TString& path, ui64 cookie) {
         THolder<TSchemeCacheNavigate> request = MakeHolder<TSchemeCacheNavigate>();
         request->Cookie = cookie;
+        request->DatabaseName = CanonizePath(AppData()->DomainsInfo->GetDomain()->Name);
         TSchemeCacheNavigate::TEntry& entry = request->ResultSet.emplace_back();
         entry.Path = NKikimr::SplitPath(path);
         entry.Operation = TSchemeCacheNavigate::EOp::OpPath;
@@ -1080,6 +1081,7 @@ class TSelfCheckRequest : public TActorBootstrapped<TSelfCheckRequest> {
     [[nodiscard]] TRequestResponse<TEvTxProxySchemeCache::TEvNavigateKeySetResult> MakeRequestSchemeCacheNavigate(const TPathId& pathId, ui64 cookie) {
         THolder<TSchemeCacheNavigate> request = MakeHolder<TSchemeCacheNavigate>();
         request->Cookie = cookie;
+        request->DatabaseName = CanonizePath(AppData()->DomainsInfo->GetDomain()->Name);
         TSchemeCacheNavigate::TEntry& entry = request->ResultSet.emplace_back();
         entry.TableId.PathId = pathId;
         entry.RequestType = TSchemeCacheNavigate::TEntry::ERequestType::ByTableId;

ydb/core/kafka_proxy/actors/kafka_sasl_auth_actor.cpp

@@ -189,6 +189,8 @@ void TKafkaSaslAuthActor::SendDescribeRequest(const TActorContext& ctx) {
 
 void TKafkaSaslAuthActor::GetPathByPathId(const TPathId& pathId, const TActorContext& ctx) {
     auto schemeCacheRequest = std::make_unique<NKikimr::NSchemeCache::TSchemeCacheNavigate>();
+    schemeCacheRequest->DatabaseName = DatabasePath;
+
     NKikimr::NSchemeCache::TSchemeCacheNavigate::TEntry entry;
     entry.TableId.PathId = pathId;
     entry.RequestType = NKikimr::NSchemeCache::TSchemeCacheNavigate::TEntry::ERequestType::ByTableId;

ydb/core/kqp/executer_actor/kqp_scheme_executer.cpp

@@ -200,6 +200,7 @@ class TKqpSchemeExecuter : public TActorBootstrapped<TKqpSchemeExecuter> {
         AFL_ENSURE(dirPath.size() >= 2);
 
         auto request = MakeHolder<NSchemeCache::TSchemeCacheNavigate>();
+        request->DatabaseName = Database;
         TVector<TString> path;
 
         for (const auto& part : dirPath) {
@@ -564,7 +565,7 @@ class TKqpSchemeExecuter : public TActorBootstrapped<TKqpSchemeExecuter> {
                 auto analyzePromise = NewPromise<IKqpGateway::TGenericResult>();
 
                 TVector<TString> columns{analyzeOperation.columns().begin(), analyzeOperation.columns().end()};
-                IActor* analyzeActor = new TAnalyzeActor(analyzeOperation.GetTablePath(), columns, analyzePromise);
+                IActor* analyzeActor = new TAnalyzeActor(Database, analyzeOperation.GetTablePath(), columns, analyzePromise);
 
                 auto actorSystem = TActivationContext::ActorSystem();
                 RegisterWithSameMailbox(analyzeActor);