-
Notifications
You must be signed in to change notification settings - Fork 39
/
api.go
41 lines (33 loc) · 1.16 KB
/
api.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package piece
import (
"context"
"io"
"github.com/bnb-chain/greenfield-storage-provider/store/piecestore/storage"
)
// PieceAPI for mock use
//
//go:generate mockgen -source=./api.go -destination=./api_mock.go -package=piece
type PieceAPI interface {
Get(ctx context.Context, key string, offset, limit int64) (io.ReadCloser, error)
Put(ctx context.Context, key string, reader io.Reader) error
Delete(ctx context.Context, key string) error
}
type PieceStore struct {
storeAPI storage.ObjectStorage
}
// Get one piece from PieceStore
func (p *PieceStore) Get(ctx context.Context, key string, offset, limit int64) (io.ReadCloser, error) {
return p.storeAPI.GetObject(ctx, key, offset, limit)
}
// Put one piece to PieceStore
func (p *PieceStore) Put(ctx context.Context, key string, reader io.Reader) error {
return p.storeAPI.PutObject(ctx, key, reader)
}
// Delete one piece in PieceStore
func (p *PieceStore) Delete(ctx context.Context, key string) error {
return p.storeAPI.DeleteObject(ctx, key)
}
// Head returns piece info in PieceStore
func (p *PieceStore) Head(ctx context.Context, key string) (storage.Object, error) {
return p.storeAPI.HeadObject(ctx, key)
}