scheme connection has been supported (#8232)

Author: dorooleg

ydb/core/fq/libs/compute/common/config.h

@@ -108,6 +108,34 @@ class TComputeConfig {
         }
     }
 
+    NFq::NConfig::TYdbStorageConfig GetSchemeConnection(const TString& scope) const {
+        const auto& controlPlane = ComputeConfig.GetYdb().GetControlPlane();
+        switch (controlPlane.type_case()) {
+            case NConfig::TYdbComputeControlPlane::TYPE_NOT_SET:
+                return {};
+            case NConfig::TYdbComputeControlPlane::kSingle:
+                return controlPlane.GetSingle().GetConnection();
+            case NConfig::TYdbComputeControlPlane::kCms:
+                return GetSchemeConnection(scope, controlPlane.GetCms().GetDatabaseMapping());
+            case NConfig::TYdbComputeControlPlane::kYdbcp:
+                return GetSchemeConnection(scope, controlPlane.GetYdbcp().GetDatabaseMapping());
+        }
+    }
+
+    NFq::NConfig::TYdbStorageConfig GetSchemeConnection(const TString& scope, const ::NFq::NConfig::TDatabaseMapping& databaseMapping) const {
+        auto it = databaseMapping.GetScopeToComputeDatabase().find(scope);
+        if (it != databaseMapping.GetScopeToComputeDatabase().end()) {
+            return it->second.HasSchemeConnection() ? it->second.GetSchemeConnection() : it->second.GetExecutionConnection(); // TODO: for backward compatibility, cleanup it after migration
+        }
+
+        if (databaseMapping.GetCommon().empty()) {
+            return NFq::NConfig::TYdbStorageConfig{};
+        }
+
+        auto config = databaseMapping.GetCommon(MultiHash(scope) % databaseMapping.GetCommon().size());
+        return config.HasSchemeConnection() ? config.GetSchemeConnection() : config.GetExecutionConnection(); // TODO: for backward compatibility, cleanup it after migration
+    }
+
     NFq::NConfig::TYdbStorageConfig GetExecutionConnection(const TString& scope, const ::NFq::NConfig::TDatabaseMapping& databaseMapping) const {
         auto it = databaseMapping.GetScopeToComputeDatabase().find(scope);
         if (it != databaseMapping.GetScopeToComputeDatabase().end()) {

ydb/core/fq/libs/compute/common/utils.h

@@ -18,8 +18,7 @@ inline std::shared_ptr<NYdb::NTable::TTableClient> CreateNewTableClient(const TS
                                                                  const ::NFq::NConfig::TYdbStorageConfig& connection,
                                                                  const TYqSharedResources::TPtr& yqSharedResources,
                                                                  const NKikimr::TYdbCredentialsProviderFactory& credentialsProviderFactory) {
-    
-    ::NFq::NConfig::TYdbStorageConfig computeConnection = computeConfig.GetExecutionConnection(scope);
+    ::NFq::NConfig::TYdbStorageConfig computeConnection = computeConfig.GetSchemeConnection(scope);
     computeConnection.set_endpoint(connection.endpoint());
     computeConnection.set_database(connection.database());
     computeConnection.set_usessl(connection.usessl());

ydb/core/fq/libs/config/protos/compute.proto

@@ -44,6 +44,7 @@ message TWorkloadManagerConfig {
 
 message TComputeDatabaseConfig {
     TYdbStorageConfig ControlPlaneConnection = 1;
+    TYdbStorageConfig SchemeConnection = 6;
     TYdbStorageConfig ExecutionConnection = 3;
     string Tenant = 2;
     TLoadControlConfig LoadControlConfig = 4;