rbd: refactor mirroring to work with volume and volumegroup

internal/rbd/types/mirror.go

@@ -0,0 +1,95 @@
+/*
+Copyright 2024 The Ceph-CSI Authors.
+
+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.
+*/
+
+package types
+
+import (
+	"context"
+	"time"
+
+	"github.com/ceph/ceph-csi/internal/util"
+
+	librbd "github.com/ceph/go-ceph/rbd"
+	"github.com/ceph/go-ceph/rbd/admin"
+)
+
+// FlattenMode is used to indicate the flatten mode for an RBD image.
+type FlattenMode string
+
+const (
+	// FlattenModeNever indicates that the image should never be flattened.
+	FlattenModeNever FlattenMode = "never"
+	// FlattenModeForce indicates that the image with the parent must be flattened.
+	FlattenModeForce FlattenMode = "force"
+)
+
+// Mirror is the interface for managing mirroring on an RBD image or a group.
+type Mirror interface {
+	// EnableMirroring enables mirroring on the resource with the specified mode.
+	EnableMirroring(mode librbd.ImageMirrorMode) error
+	// DisableMirroring disables mirroring on the resource with the option to force the operation
+	DisableMirroring(force bool) error
+	// Promote promotes the resource to primary status with the option to force the operation
+	Promote(force bool) error
+	// ForcePromote promotes the resource to primary status with a timeout
+	ForcePromote(cr *util.Credentials) error
+	// Demote demotes the resource to secondary status
+	Demote() error
+	// Resync resynchronizes the resource
+	Resync() error
+	// GetMirroringInfo returns the mirroring information of the resource
+	GetMirroringInfo() (MirrorInfo, error)
+	// GetMirroringInfo returns the mirroring information of the resource
+	GetGlobalMirroringStatus() (GlobalStatus, error)
+	// AddSnapshotScheduling adds a snapshot scheduling to the resource
+	AddSnapshotScheduling(interval admin.Interval, startTime admin.StartTime) error
+}
+
+// MirrorImage is the interface for managing mirroring on an RBD image or group of images.
+// This will be used to get the state of resource and it is primary or secondary.
+type MirrorInfo interface {
+	// IsPrimary returns true if the resource is primary
+	IsPrimary() bool
+	// GetState returns the state of the resource
+	GetState() string
+}
+
+// GlobalStatus is the interface for fetching the global status of the mirroring.
+// This will be used to get the status of the local site and remote site or all sites.
+type GlobalStatus interface {
+	MirrorInfo
+	// GetLocalSiteStatus returns the local site status
+	GetLocalSiteStatus() (SiteStatus, error)
+	// GetAllSitesStatus returns the status of all sites
+	GetAllSitesStatus() []SiteStatus
+	// GetRemoteSiteStatus returns the status of the remote site
+	GetRemoteSiteStatus(ctx context.Context) (SiteStatus, error)
+}
+
+// SiteStatus is the interface for fetching the status of a site.
+// This will be used to get the status of the local site and remote site.
+type SiteStatus interface {
+	// GetMirrorUUID returns the mirror UUID
+	GetMirrorUUID() string
+	// IsUP returns true if the site is up
+	IsUP() bool
+	// GetState returns the state of the site
+	GetState() string
+	// GetDescription returns the description of the site
+	GetDescription() string
+	// GetLastUpdate returns the last update time
+	GetLastUpdate() time.Time
+}