Skip to content

Commit cbcd437

Browse files
committed
add cbt interfaces
Signed-off-by: Lyndon-Li <lyonghui@vmware.com>
1 parent 87b5445 commit cbcd437

4 files changed

Lines changed: 66 additions & 2 deletions

File tree

pkg/cbtservice/service.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,15 @@ type Range struct {
2424
Length int64
2525
}
2626

27+
// SourceInfo is the information provided to the uploader, the uploader calls CBT service with this information
28+
type SourceInfo struct {
29+
// Snapshot is the identifier of the current snapshot
30+
Snapshot string
31+
32+
// ChangeID is the identifier associated to the current snapshot that is used as changeID for following backups
33+
ChangeID string
34+
}
35+
2736
// Service defines the methods for CBT service which could be implemented by Kubernetes SnapshotMetadataService or other customized services
2837
type Service interface {
2938
// GetAllocatedBlocks enumerates the allocated blocks of the snapshot and call the record callback

pkg/uploader/cbt/bitmap.go

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,15 @@ type Bitmap interface {
2323
// Set sets bits within the provided range
2424
Set(cbtservice.Range)
2525

26+
// SetFull sets all bits to the bitmap
27+
SetFull()
28+
29+
// Snapshot returns snapshot of the bitmap
30+
SourceID() string
31+
32+
// ChangeID returns the changeID of the bitmap
33+
ChangeID() string
34+
2635
// Iterator returns the iterator for the CBT Bitmap
2736
Iterator() Iterator
2837
}
@@ -32,8 +41,8 @@ type Iterator interface {
3241
// ChangeID returns the changeID of the bitmap
3342
ChangeID() string
3443

35-
// SnapshotID returns snapshotID of the bitmap
36-
SnapshotID() string
44+
// Snapshot returns snapshot of the bitmap
45+
Snapshot() string
3746

3847
// BlockSize returns the granularity of the bitmap
3948
BlockSize() int

pkg/uploader/cbt/set.go

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,46 @@
1+
/*
2+
Copyright The Velero Contributors.
3+
Licensed under the Apache License, Version 2.0 (the "License");
4+
you may not use this file except in compliance with the License.
5+
You may obtain a copy of the License at
6+
http://www.apache.org/licenses/LICENSE-2.0
7+
Unless required by applicable law or agreed to in writing, software
8+
distributed under the License is distributed on an "AS IS" BASIS,
9+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10+
See the License for the specific language governing permissions and
11+
limitations under the License.
12+
*/
13+
14+
package cbt
15+
16+
import (
17+
"context"
18+
19+
"github.com/vmware-tanzu/velero/pkg/cbtservice"
20+
)
21+
22+
// SetBitmapOrFull translates the allocated/changed blocks from CBT service to the given bitmap or set the bitmap to full when error happens
23+
func SetBitmapOrFull(ctx context.Context, service cbtservice.Service, bitmap Bitmap) error {
24+
var err error
25+
if bitmap.ChangeID() == "" {
26+
err = setFromAllocatedBlocks(ctx, service, bitmap)
27+
} else {
28+
err = setFromChangedBlocks(ctx, service, bitmap)
29+
}
30+
31+
if err != nil {
32+
bitmap.SetFull()
33+
}
34+
35+
return err
36+
}
37+
38+
// TODO implement in following PRs
39+
func setFromAllocatedBlocks(_ context.Context, _ cbtservice.Service, _ Bitmap) error {
40+
return nil
41+
}
42+
43+
// TODO implement in following PRs
44+
func setFromChangedBlocks(_ context.Context, _ cbtservice.Service, _ Bitmap) error {
45+
return nil
46+
}

0 commit comments

Comments
 (0)