YugabyteDB (2.13.1.0-b60, 21121d69985fbf76aa6958d8f04a9bfa936293b5)

// Copyright (c) YugaByte, Inc.

//

// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except

// in compliance with the License.  You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software distributed under the License

// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express

// or implied.  See the License for the specific language governing permissions and limitations

// under the License.

#ifndef ENT_SRC_YB_MASTER_CATALOG_MANAGER_H

#define ENT_SRC_YB_MASTER_CATALOG_MANAGER_H

#include "../../../../src/yb/master/catalog_manager.h"

#include "yb/master/master_snapshot_coordinator.h"

#include "yb/master/snapshot_coordinator_context.h"

namespace yb {

class UniverseKeyRegistryPB;

namespace master {

namespace enterprise {

YB_DEFINE_ENUM(CreateObjects, (kOnlyTables)(kOnlyIndexes));

class CatalogManager : public yb::master::CatalogManager, SnapshotCoordinatorContext {

  typedef yb::master::CatalogManager super;

 public:

  explicit CatalogManager(yb::master::Master* master)

      : super(master), snapshot_coordinator_(this) {}

  virtual ~CatalogManager();

  void CompleteShutdown();

  CHECKED_STATUS RunLoaders(int64_t term) override REQUIRES(mutex_);

  // API to start a snapshot creation.

  CHECKED_STATUS CreateSnapshot(const CreateSnapshotRequestPB* req,

                                CreateSnapshotResponsePB* resp,

                                rpc::RpcContext* rpc);

  // API to list all available snapshots.

  CHECKED_STATUS ListSnapshots(const ListSnapshotsRequestPB* req,

                               ListSnapshotsResponsePB* resp);

  CHECKED_STATUS ListSnapshotRestorations(const ListSnapshotRestorationsRequestPB* req,

                                          ListSnapshotRestorationsResponsePB* resp);

  // API to restore a snapshot.

  CHECKED_STATUS RestoreSnapshot(const RestoreSnapshotRequestPB* req,

                                 RestoreSnapshotResponsePB* resp);

  // API to delete a snapshot.

  CHECKED_STATUS DeleteSnapshot(const DeleteSnapshotRequestPB* req,

                                DeleteSnapshotResponsePB* resp,

                                rpc::RpcContext* rpc);

  CHECKED_STATUS ImportSnapshotMeta(const ImportSnapshotMetaRequestPB* req,

                                    ImportSnapshotMetaResponsePB* resp,

                                    rpc::RpcContext* rpc);

  CHECKED_STATUS CreateSnapshotSchedule(const CreateSnapshotScheduleRequestPB* req,

                                        CreateSnapshotScheduleResponsePB* resp,

                                        rpc::RpcContext* rpc);

  CHECKED_STATUS ListSnapshotSchedules(const ListSnapshotSchedulesRequestPB* req,

                                       ListSnapshotSchedulesResponsePB* resp,

                                       rpc::RpcContext* rpc);

  CHECKED_STATUS DeleteSnapshotSchedule(const DeleteSnapshotScheduleRequestPB* req,

                                        DeleteSnapshotScheduleResponsePB* resp,

                                        rpc::RpcContext* rpc);

  CHECKED_STATUS ChangeEncryptionInfo(const ChangeEncryptionInfoRequestPB* req,

                                      ChangeEncryptionInfoResponsePB* resp) override;

  CHECKED_STATUS UpdateXClusterConsumerOnTabletSplit(

      const TableId& consumer_table_id, const SplitTabletIds& split_tablet_ids) override;

  CHECKED_STATUS UpdateXClusterProducerOnTabletSplit(

      const TableId& producer_table_id, const SplitTabletIds& split_tablet_ids) override;

  CHECKED_STATUS InitCDCConsumer(const std::vector<CDCConsumerStreamInfo>& consumer_info,

                                 const std::string& master_addrs,

                                 const std::string& producer_universe_uuid,

                                 std::shared_ptr<CDCRpcTasks> cdc_rpc_tasks);

  void HandleCreateTabletSnapshotResponse(TabletInfo *tablet, bool error) override;

  void HandleRestoreTabletSnapshotResponse(TabletInfo *tablet, bool error) override;

  void HandleDeleteTabletSnapshotResponse(

      const SnapshotId& snapshot_id, TabletInfo *tablet, bool error) override;

  void DumpState(std::ostream* out, bool on_disk_dump = false) const override;

  // Fills the heartbeat response with the decrypted universe key registry.

  CHECKED_STATUS FillHeartbeatResponse(const TSHeartbeatRequestPB* req,

                                       TSHeartbeatResponsePB* resp) override;

  // Is encryption at rest enabled for this cluster.

  CHECKED_STATUS IsEncryptionEnabled(const IsEncryptionEnabledRequestPB* req,

                                     IsEncryptionEnabledResponsePB* resp);

  // Create a new CDC stream with the specified attributes.

  CHECKED_STATUS CreateCDCStream(const CreateCDCStreamRequestPB* req,

                                 CreateCDCStreamResponsePB* resp,

                                 rpc::RpcContext* rpc);

  // Delete the specified CDCStream.

      CHECKED_STATUS DeleteCDCStream(const DeleteCDCStreamRequestPB* req,

                                 DeleteCDCStreamResponsePB* resp,

                                 rpc::RpcContext* rpc);

  // List CDC streams (optionally, for a given table).

  CHECKED_STATUS ListCDCStreams(const ListCDCStreamsRequestPB* req,

                                ListCDCStreamsResponsePB* resp) override;

  // Fetch CDC stream info corresponding to a db stream id

  CHECKED_STATUS GetCDCDBStreamInfo(const GetCDCDBStreamInfoRequestPB* req,

                                    GetCDCDBStreamInfoResponsePB* resp) override;

  // Get CDC stream.

  CHECKED_STATUS GetCDCStream(const GetCDCStreamRequestPB* req,

                              GetCDCStreamResponsePB* resp,

                              rpc::RpcContext* rpc);

  // Update a CDC stream.

  CHECKED_STATUS UpdateCDCStream(const UpdateCDCStreamRequestPB* req,

                                 UpdateCDCStreamResponsePB* resp,

                                 rpc::RpcContext* rpc);

  // Delete CDC streams for a table.

  CHECKED_STATUS DeleteCDCStreamsForTable(const TableId& table_id) override;

  CHECKED_STATUS DeleteCDCStreamsForTables(const vector<TableId>& table_ids) override;

  // Setup Universe Replication to consume data from another YB universe.

  CHECKED_STATUS SetupUniverseReplication(const SetupUniverseReplicationRequestPB* req,

                                          SetupUniverseReplicationResponsePB* resp,

                                          rpc::RpcContext* rpc);

  // Delete Universe Replication.

  CHECKED_STATUS DeleteUniverseReplication(const DeleteUniverseReplicationRequestPB* req,

                                           DeleteUniverseReplicationResponsePB* resp,

                                           rpc::RpcContext* rpc);

  // Alter Universe Replication.

  CHECKED_STATUS AlterUniverseReplication(const AlterUniverseReplicationRequestPB* req,

                                          AlterUniverseReplicationResponsePB* resp,

                                          rpc::RpcContext* rpc);

  // Rename an existing Universe Replication.

  CHECKED_STATUS RenameUniverseReplication(scoped_refptr<UniverseReplicationInfo> universe,

                                           const AlterUniverseReplicationRequestPB* req,

                                           AlterUniverseReplicationResponsePB* resp,

                                           rpc::RpcContext* rpc);

  // Enable/Disable an Existing Universe Replication.

  CHECKED_STATUS SetUniverseReplicationEnabled(const SetUniverseReplicationEnabledRequestPB* req,

                                               SetUniverseReplicationEnabledResponsePB* resp,

                                               rpc::RpcContext* rpc);

  // Get Universe Replication.

  CHECKED_STATUS GetUniverseReplication(const GetUniverseReplicationRequestPB* req,

                                        GetUniverseReplicationResponsePB* resp,

                                        rpc::RpcContext* rpc);

  // Checks if the universe is in an active state or has failed during setup.

  CHECKED_STATUS IsSetupUniverseReplicationDone(const IsSetupUniverseReplicationDoneRequestPB* req,

                                                IsSetupUniverseReplicationDoneResponsePB* resp,

                                                rpc::RpcContext* rpc);

  // On a producer side split, creates new pollers on the consumer for the new tablet children.

  CHECKED_STATUS UpdateConsumerOnProducerSplit(const UpdateConsumerOnProducerSplitRequestPB* req,

                                               UpdateConsumerOnProducerSplitResponsePB* resp,

                                               rpc::RpcContext* rpc);

  // Find all the CDC streams that have been marked as DELETED.

  CHECKED_STATUS FindCDCStreamsMarkedAsDeleting(std::vector<scoped_refptr<CDCStreamInfo>>* streams);

  // Delete specified CDC streams.

  CHECKED_STATUS CleanUpDeletedCDCStreams(const std::vector<scoped_refptr<CDCStreamInfo>>& streams);

  bool IsCdcEnabled(const TableInfo& table_info) const override;

  tablet::SnapshotCoordinator& snapshot_coordinator() override {

    return snapshot_coordinator_;

  }

  Result<size_t> GetNumLiveTServersForActiveCluster() override;

 private:

  friend class SnapshotLoader;

  friend class yb::master::ClusterLoadBalancer;

  friend class CDCStreamLoader;

  friend class UniverseReplicationLoader;

  CHECKED_STATUS RestoreEntry(const SysRowEntry& entry, const SnapshotId& snapshot_id)

      REQUIRES(mutex_);

  // Per table structure for external cluster snapshot importing to this cluster.

  // Old IDs mean IDs on external cluster, new IDs - IDs on this cluster.

  struct ExternalTableSnapshotData {

    bool is_index() const {

      return !table_entry_pb.indexed_table_id().empty();

    }

    NamespaceId old_namespace_id;

    TableId old_table_id;

    TableId new_table_id;

    SysTablesEntryPB table_entry_pb;

    std::string pg_schema_name;

    size_t num_tablets = 0;

    typedef std::pair<std::string, std::string> PartitionKeys;

    typedef std::map<PartitionKeys, TabletId> PartitionToIdMap;

    typedef std::vector<PartitionPB> Partitions;

    Partitions partitions;

    PartitionToIdMap new_tablets_map;

    // Mapping: Old tablet ID -> New tablet ID.

    google::protobuf::RepeatedPtrField<IdPairPB>* tablet_id_map = nullptr;

    ImportSnapshotMetaResponsePB_TableMetaPB* table_meta = nullptr;

  };

  // Map: old_namespace_id (key) -> new_namespace_id (value) + db_type.

  typedef std::pair<NamespaceId, YQLDatabase> NamespaceData;

  typedef std::map<NamespaceId, NamespaceData> NamespaceMap;

  typedef std::map<TableId, ExternalTableSnapshotData> ExternalTableSnapshotDataMap;

  CHECKED_STATUS ImportSnapshotPreprocess(const SnapshotInfoPB& snapshot_pb,

                                          ImportSnapshotMetaResponsePB* resp,

                                          NamespaceMap* namespace_map,

                                          ExternalTableSnapshotDataMap* tables_data);

  CHECKED_STATUS ImportSnapshotCreateObject(const SnapshotInfoPB& snapshot_pb,

                                            ImportSnapshotMetaResponsePB* resp,

                                            NamespaceMap* namespace_map,

                                            ExternalTableSnapshotDataMap* tables_data,

                                            CreateObjects create_objects);

  CHECKED_STATUS ImportSnapshotWaitForTables(const SnapshotInfoPB& snapshot_pb,

                                             ImportSnapshotMetaResponsePB* resp,

                                             ExternalTableSnapshotDataMap* tables_data,

                                             CoarseTimePoint deadline);

  CHECKED_STATUS ImportSnapshotProcessTablets(const SnapshotInfoPB& snapshot_pb,

                                              ImportSnapshotMetaResponsePB* resp,

                                              ExternalTableSnapshotDataMap* tables_data);

  void DeleteNewSnapshotObjects(const NamespaceMap& namespace_map,

                                const ExternalTableSnapshotDataMap& tables_data);

  // Helper function for ImportTableEntry.

  Result<bool> CheckTableForImport(

      scoped_refptr<TableInfo> table,

      ExternalTableSnapshotData* snapshot_data) REQUIRES_SHARED(mutex_);

  CHECKED_STATUS ImportNamespaceEntry(const SysRowEntry& entry,

                                      NamespaceMap* namespace_map);

  CHECKED_STATUS RecreateTable(const NamespaceId& new_namespace_id,

                               const ExternalTableSnapshotDataMap& table_map,

                               ExternalTableSnapshotData* table_data);

  CHECKED_STATUS RepartitionTable(scoped_refptr<TableInfo> table,

                                  const ExternalTableSnapshotData* table_data);

  CHECKED_STATUS ImportTableEntry(const NamespaceMap& namespace_map,

                                  const ExternalTableSnapshotDataMap& table_map,

                                  ExternalTableSnapshotData* s_data);

  CHECKED_STATUS PreprocessTabletEntry(const SysRowEntry& entry,

                                       ExternalTableSnapshotDataMap* table_map);

  CHECKED_STATUS ImportTabletEntry(const SysRowEntry& entry,

                                   ExternalTableSnapshotDataMap* table_map);

  TabletInfos GetTabletInfos(const std::vector<TabletId>& ids) override;

  Result<SysRowEntries> CollectEntries(

      const google::protobuf::RepeatedPtrField<TableIdentifierPB>& tables,

      CollectFlags flags);

  Result<SysRowEntries> CollectEntriesForSnapshot(

      const google::protobuf::RepeatedPtrField<TableIdentifierPB>& tables) override;

  server::Clock* Clock() override;

  const Schema& schema() override;

  void Submit(std::unique_ptr<tablet::Operation> operation, int64_t leader_term) override;

  AsyncTabletSnapshotOpPtr CreateAsyncTabletSnapshotOp(

      const TabletInfoPtr& tablet, const std::string& snapshot_id,

      tserver::TabletSnapshotOpRequestPB::Operation operation,

      TabletSnapshotOperationCallback callback) override;

  void ScheduleTabletSnapshotOp(const AsyncTabletSnapshotOpPtr& operation) override;

  CHECKED_STATUS RestoreSysCatalog(

      SnapshotScheduleRestoration* restoration, tablet::Tablet* tablet,

      Status* complete_status) override;

  CHECKED_STATUS VerifyRestoredObjects(const SnapshotScheduleRestoration& restoration) override;

  void CleanupHiddenObjects(const ScheduleMinRestoreTime& schedule_min_restore_time) override;

  void CleanupHiddenTablets(

      const std::vector<TabletInfoPtr>& hidden_tablets,

      const ScheduleMinRestoreTime& schedule_min_restore_time);

  // Will filter tables content, so pass it by value here.

  void CleanupHiddenTables(std::vector<TableInfoPtr> tables);

  rpc::Scheduler& Scheduler() override;

  int64_t LeaderTerm() override;

  Result<bool> IsTablePartOfSomeSnapshotSchedule(const TableInfo& table_info) override;

  Result<SnapshotSchedulesToObjectIdsMap> MakeSnapshotSchedulesToObjectIdsMap(

      SysRowEntryType type) override;

  static void SetTabletSnapshotsState(SysSnapshotEntryPB::State state,

                                      SysSnapshotEntryPB* snapshot_pb);

  // Create the cdc_state table if needed (i.e. if it does not exist already).

  //

  // This is called at the end of CreateCDCStream.

  CHECKED_STATUS CreateCdcStateTableIfNeeded(rpc::RpcContext *rpc);

  // Check if cdc_state table creation is done.

  CHECKED_STATUS IsCdcStateTableCreated(IsCreateTableDoneResponsePB* resp);

  // Return all CDC streams.

  void GetAllCDCStreams(std::vector<scoped_refptr<CDCStreamInfo>>* streams);

  // Mark specified CDC streams as DELETING so they can be removed later.

  CHECKED_STATUS MarkCDCStreamsAsDeleting(const std::vector<scoped_refptr<CDCStreamInfo>>& streams);

  // Find CDC streams for a table.

  std::vector<scoped_refptr<CDCStreamInfo>> FindCDCStreamsForTable(const TableId& table_id) const;

  bool CDCStreamExistsUnlocked(const CDCStreamId& stream_id) override REQUIRES_SHARED(mutex_);

  CHECKED_STATUS FillHeartbeatResponseEncryption(const SysClusterConfigEntryPB& cluster_config,

                                                 const TSHeartbeatRequestPB* req,

                                                 TSHeartbeatResponsePB* resp);

  CHECKED_STATUS FillHeartbeatResponseCDC(const SysClusterConfigEntryPB& cluster_config,

                                          const TSHeartbeatRequestPB* req,

                                          TSHeartbeatResponsePB* resp);

  // Helper functions for GetTableSchemaCallback, GetTablegroupSchemaCallback

  // and GetColocatedTabletSchemaCallback.

  // Validates a single table's schema with the corresponding table on the consumer side, and

  // updates consumer_table_id with the new table id. Return the consumer table schema if the

  // validation is successful.

  CHECKED_STATUS ValidateTableSchema(

      const std::shared_ptr<client::YBTableInfo>& info,

      const std::unordered_map<TableId, std::string>& table_bootstrap_ids,

      GetTableSchemaResponsePB* resp);

  // Adds a validated table to the sys catalog table map for the given universe, and if all tables

  // have been validated, creates a CDC stream for each table.

  CHECKED_STATUS AddValidatedTableAndCreateCdcStreams(

      scoped_refptr<UniverseReplicationInfo> universe,

      const std::unordered_map<TableId, std::string>& table_bootstrap_ids,

      const TableId& producer_table,

      const TableId& consumer_table);

  void GetTableSchemaCallback(

      const std::string& universe_id, const std::shared_ptr<client::YBTableInfo>& info,

      const std::unordered_map<TableId, std::string>& producer_bootstrap_ids, const Status& s);

  void GetTablegroupSchemaCallback(

      const std::string& universe_id, const std::shared_ptr<std::vector<client::YBTableInfo>>& info,

      const TablegroupId& producer_tablegroup_id,

      const std::unordered_map<TableId, std::string>& producer_bootstrap_ids, const Status& s);

  void GetColocatedTabletSchemaCallback(

      const std::string& universe_id, const std::shared_ptr<std::vector<client::YBTableInfo>>& info,

      const std::unordered_map<TableId, std::string>& producer_bootstrap_ids, const Status& s);

  void GetCDCStreamCallback(const CDCStreamId& bootstrap_id,

                            std::shared_ptr<TableId> table_id,

                            std::shared_ptr<std::unordered_map<std::string, std::string>> options,

                            const std::string& universe_id,

                            const TableId& table,

                            std::shared_ptr<CDCRpcTasks> cdc_rpc,

                            const Status& s);

  void AddCDCStreamToUniverseAndInitConsumer(const std::string& universe_id, const TableId& table,

                                             const Result<CDCStreamId>& stream_id,

                                             std::function<void()> on_success_cb = nullptr);

  void MergeUniverseReplication(scoped_refptr<UniverseReplicationInfo> info);

  CHECKED_STATUS DeleteUniverseReplicationUnlocked(scoped_refptr<UniverseReplicationInfo> info);

  void MarkUniverseReplicationFailed(scoped_refptr<UniverseReplicationInfo> universe,

                                     const Status& failure_status);

  // Checks if table has at least one cdc stream (includes producers for xCluster replication).

  bool IsTableCdcProducer(const TableInfo& table_info) const override REQUIRES_SHARED(mutex_);

  // Checks if the table is a consumer in an xCluster replication universe.

  bool IsTableCdcConsumer(const TableInfo& table_info) const REQUIRES_SHARED(mutex_);

  // Maps producer universe id to the corresponding cdc stream for that table.

  typedef std::unordered_map<std::string, CDCStreamId> XClusterConsumerTableStreamInfoMap;

  // Gets the set of CDC stream info for an xCluster consumer table.

  XClusterConsumerTableStreamInfoMap GetXClusterStreamInfoForConsumerTable(const TableId& table_id)

      const;

  CHECKED_STATUS CreateTransactionAwareSnapshot(

      const CreateSnapshotRequestPB& req, CreateSnapshotResponsePB* resp, rpc::RpcContext* rpc);

  CHECKED_STATUS CreateNonTransactionAwareSnapshot(

      const CreateSnapshotRequestPB* req, CreateSnapshotResponsePB* resp, rpc::RpcContext* rpc);

  CHECKED_STATUS RestoreNonTransactionAwareSnapshot(const SnapshotId& snapshot_id);

  CHECKED_STATUS DeleteNonTransactionAwareSnapshot(const SnapshotId& snapshot_id);

  void Started() override;

  void SysCatalogLoaded(int64_t term) override;

  // Snapshot map: snapshot-id -> SnapshotInfo.

  typedef std::unordered_map<SnapshotId, scoped_refptr<SnapshotInfo>> SnapshotInfoMap;

  SnapshotInfoMap non_txn_snapshot_ids_map_;

  SnapshotId current_snapshot_id_;

  // mutex on should_send_universe_key_registry_mutex_.

  mutable simple_spinlock should_send_universe_key_registry_mutex_;

  // Should catalog manager resend latest universe key registry to tserver.

  std::unordered_map<TabletServerId, bool> should_send_universe_key_registry_

  GUARDED_BY(should_send_universe_key_registry_mutex_);

  // CDC Stream map: CDCStreamId -> CDCStreamInfo.

  typedef std::unordered_map<CDCStreamId, scoped_refptr<CDCStreamInfo>> CDCStreamInfoMap;

  CDCStreamInfoMap cdc_stream_map_ GUARDED_BY(mutex_);

  // Map of tables -> number of cdc streams they are producers for.

  std::unordered_map<TableId, int> cdc_stream_tables_count_map_ GUARDED_BY(mutex_);

  // Map of all consumer tables that are part of xcluster replication, to a map of the stream infos.

  std::unordered_map<TableId, XClusterConsumerTableStreamInfoMap>

      xcluster_consumer_tables_to_stream_map_ GUARDED_BY(mutex_);

  typedef std::unordered_map<std::string, scoped_refptr<UniverseReplicationInfo>>

      UniverseReplicationInfoMap;

  UniverseReplicationInfoMap universe_replication_map_ GUARDED_BY(mutex_);

  // mutex on should_send_consumer_registry_mutex_.

  mutable simple_spinlock should_send_consumer_registry_mutex_;

  // Should catalog manager resend latest consumer registry to tserver.

  std::unordered_map<TabletServerId, bool> should_send_consumer_registry_

  GUARDED_BY(should_send_consumer_registry_mutex_);

  MasterSnapshotCoordinator snapshot_coordinator_;

  DISALLOW_COPY_AND_ASSIGN(CatalogManager);

};

} // namespace enterprise

} // namespace master

} // namespace yb

#endif // ENT_SRC_YB_MASTER_CATALOG_MANAGER_H