gian/cubefs
1
0
Fork 0
mirror of https://github.com/cubefs/cubefs.git synced 2026-08-02 02:00:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(shardnode): modify the naming of api parameters

Browse Source 
with #22357426

Signed-off-by: xiejian <xiejian3@oppo.com>
This commit is contained in:
xiejian 2024-08-14 15:31:57 +08:00 committed by slasher
parent 3ddbdba340
commit 4dc732792f
11 changed files with 1462 additions and 758 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
blobstore/api/shardnode/blob.go
View File

@ -16,22 +16,22 @@ package shardnode
import "context"
func (c *Client) CreateBlob(ctx context.Context, args CreateBlobRequest) (CreateBlobResponse, error) {
return CreateBlobResponse{}, nil
func (c *Client) CreateBlob(ctx context.Context, args CreateBlobArgs) (CreateBlobRet, error) {
return CreateBlobRet{}, nil
}
func (c *Client) DeleteBlob(ctx context.Context, args DeleteBlobRequest) error {
func (c *Client) DeleteBlob(ctx context.Context, args DeleteBlobArgs) error {
return nil
}
func (c *Client) GetBlob(ctx context.Context, args GetBlobRequest) (GetBlobResponse, error) {
return GetBlobResponse{}, nil
func (c *Client) GetBlob(ctx context.Context, args GetBlobArgs) (GetBlobRet, error) {
return GetBlobRet{}, nil
}
func (c *Client) SealBlob(ctx context.Context, args SealBlobRequest) error {
func (c *Client) SealBlob(ctx context.Context, args SealBlobArgs) error {
return nil
}
func (c *Client) ListBlob(ctx context.Context, args ListBlobRequest) (ListBlobResponse, error) {
return ListBlobResponse{}, nil
func (c *Client) ListBlob(ctx context.Context, args ListBlobArgs) (ListBlobRet, error) {
return ListBlobRet{}, nil
}

12
blobstore/api/shardnode/item.go
View File

@ -16,24 +16,24 @@ package shardnode
import "context"
func (c *Client) AddItem(ctx context.Context, args InsertItemRequest) error {
func (c *Client) AddItem(ctx context.Context, args InsertItemArgs) error {
return nil
}
func (c *Client) UpdateItem(ctx context.Context, args UpdateItemRequest) error {
func (c *Client) UpdateItem(ctx context.Context, args UpdateItemArgs) error {
return nil
}
func (c *Client) DeleteItem(ctx context.Context, args DeleteItemRequest) error {
func (c *Client) DeleteItem(ctx context.Context, args DeleteItemArgs) error {
return nil
}
func (c *Client) GetItem(ctx context.Context, args GetItemRequest) (Item, error) {
func (c *Client) GetItem(ctx context.Context, args GetItemArgs) (Item, error) {
return Item{}, nil
}
func (c *Client) ListItem(ctx context.Context, args ListItemRequest) (ListItemResponse, error) {
return ListItemResponse{}, nil
func (c *Client) ListItem(ctx context.Context, args ListItemArgs) (ListItemRet, error) {
return ListItemRet{}, nil
}
func (i *Item) Read(data []byte) (int, error) {

6
blobstore/api/shardnode/shard.go
View File

@ -20,14 +20,14 @@ import (
"github.com/cubefs/cubefs/blobstore/api/clustermgr"
)
func (c *Client) AddShard(ctx context.Context, host string, args AddShardRequest) error {
func (c *Client) AddShard(ctx context.Context, host string, args AddShardArgs) error {
return nil
}
func (c *Client) UpdateShard(ctx context.Context, host string, args UpdateShardRequest) error {
func (c *Client) UpdateShard(ctx context.Context, host string, args UpdateShardArgs) error {
return nil
}
func (c *Client) GetShardUintInfo(ctx context.Context, host string, args GetShardRequest) (ret clustermgr.ShardUnitInfo, err error) {
func (c *Client) GetShardUintInfo(ctx context.Context, host string, args GetShardArgs) (ret clustermgr.ShardUnitInfo, err error) {
return clustermgr.ShardUnitInfo{}, nil
}

1761
blobstore/api/shardnode/shardnode.pb.go
View File

File diff suppressed because it is too large Load Diff

58
blobstore/api/shardnode/shardnode.proto
View File

@ -45,49 +45,49 @@ message ShardOpHeader {
repeated bytes shard_keys = 6;
}
message InsertItemRequest {
message InsertItemArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
Item item = 2 [(gogoproto.nullable) = false];
}
message InsertItemResponse {}
message InsertItemRet {}
message UpdateItemRequest {
message UpdateItemArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
Item item = 2 [(gogoproto.nullable) = false];
}
message UpdateItemResponse {}
message UpdateItemRet {}
message DeleteItemRequest {
message DeleteItemArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes id = 2 [(gogoproto.customname) = "ID"];
}
message DeleteItemResponse {}
message DeleteItemRet {}
message GetItemRequest {
message GetItemArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes id = 2 [(gogoproto.customname) = "ID"];
}
message GetItemResponse {
message GetItemRet {
Item item = 1 [(gogoproto.nullable) = false];
}
message ListItemRequest {
message ListItemArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes prefix = 2;
bytes marker = 3;
uint64 count = 4;
}
message ListItemResponse {
message ListItemRet {
repeated Item items = 1 [(gogoproto.nullable) = false];
bytes nextMarker = 2;
}
message AddShardRequest {
message AddShardArgs {
uint32 disk_id = 1 [(gogoproto.customname) = "DiskID", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.DiskID"];
uint64 suid = 2 [(gogoproto.customname) = "Suid", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.Suid"];
cubefs.blobstore.common.sharding.Range range = 3 [(gogoproto.nullable) = false];
@ -95,28 +95,28 @@ message AddShardRequest {
uint64 route_version = 5 [(gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.RouteVersion"];
}
message AddShardResponse {}
message AddShardRet {}
message UpdateShardRequest {
message UpdateShardArgs {
uint32 disk_id = 1 [(gogoproto.customname) = "DiskID", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.DiskID"];
uint64 suid = 2 [(gogoproto.customname) = "Suid", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.Suid"];
uint32 ShardUpdateType = 3 [(gogoproto.customname) = "ShardUpdateType", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.ShardUpdateType"];
cubefs.blobstore.api.clustermgr.ShardUnit unit = 4 [(gogoproto.nullable) = false];
}
message UpdateShardResponse {
message UpdateShardRet {
}
message GetShardRequest {
message GetShardArgs {
uint32 disk_id = 1 [(gogoproto.customname) = "DiskID", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.DiskID"];
uint64 suid = 2 [(gogoproto.customname) = "Suid", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/proto.Suid"];
}
message GetShardResponse {
message GetShardRet {
cubefs.blobstore.api.clustermgr.ShardUnitInfo shard = 1 [(gogoproto.nullable) = false];
}
message CreateBlobRequest {
message CreateBlobArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes name = 2;
uint32 codemode = 3 [(gogoproto.customname) = "CodeMode", (gogoproto.casttype) = "github.com/cubefs/cubefs/blobstore/common/codemode.CodeMode"];
@ -124,44 +124,50 @@ message CreateBlobRequest {
uint32 slice_size = 5;
}
message CreateBlobResponse {
message CreateBlobRet {
cubefs.blobstore.common.proto.Blob blob = 1 [(gogoproto.nullable) = false];
}
message GetBlobRequest {
message GetBlobArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes name = 2;
}
message GetBlobResponse {
message GetBlobRet {
cubefs.blobstore.common.proto.Blob blob = 1 [(gogoproto.nullable) = false];
}
message ListBlobRequest {
message ListBlobArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes prefix = 2;
bytes marker = 3;
uint64 count = 4;
}
message ListBlobResponse {
message ListBlobRet {
repeated cubefs.blobstore.common.proto.Blob blobs = 1 [(gogoproto.nullable) = false];
bytes nextMarker = 2;
}
message DeleteBlobRequest {
message DeleteBlobArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes name = 2;
}
message RetainBlobRequest {
message DeleteBlobRet {}
message RetainBlobArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
bytes name = 2;
uint64 lease = 3;
}
message SealBlobRequest {
message RetainBlobRet {}
message SealBlobArgs {
ShardOpHeader header = 1 [(gogoproto.nullable) = false];
uint64 size = 2;
bytes name = 3;
}
}
message SealBlobRet {}

14
blobstore/shardnode/blob.go
View File

@ -21,7 +21,7 @@ import (
"github.com/cubefs/cubefs/blobstore/api/shardnode"
)
func (s *service) CreateBlob(ctx context.Context, req *shardnode.CreateBlobRequest) (resp shardnode.CreateBlobResponse, err error) {
func (s *service) CreateBlob(ctx context.Context, req *shardnode.CreateBlobArgs) (resp shardnode.CreateBlobRet, err error) {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -30,7 +30,7 @@ func (s *service) CreateBlob(ctx context.Context, req *shardnode.CreateBlobReque
return space.CreateBlob(ctx, req)
}
func (s *service) DeleteBlob(ctx context.Context, req *shardnode.DeleteBlobRequest) error {
func (s *service) DeleteBlob(ctx context.Context, req *shardnode.DeleteBlobArgs) error {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -39,7 +39,7 @@ func (s *service) DeleteBlob(ctx context.Context, req *shardnode.DeleteBlobReque
return space.DeleteBlob(ctx, req)
}
func (s *service) SealBlob(ctx context.Context, req *shardnode.SealBlobRequest) error {
func (s *service) SealBlob(ctx context.Context, req *shardnode.SealBlobArgs) error {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -48,20 +48,20 @@ func (s *service) SealBlob(ctx context.Context, req *shardnode.SealBlobRequest)
return space.SealBlob(ctx, req)
}
func (s *service) GetBlob(ctx context.Context, req *shardnode.GetBlobRequest) (blob shardnode.GetBlobResponse, err error) {
func (s *service) GetBlob(ctx context.Context, req *shardnode.GetBlobArgs) (blob shardnode.GetBlobRet, err error) {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
return shardnode.GetBlobResponse{}, err
return shardnode.GetBlobRet{}, err
}
return space.GetBlob(ctx, req)
}
func (s *service) ListBlob(ctx context.Context, req *shardnode.ListBlobRequest) (resp shardnode.ListBlobResponse, err error) {
func (s *service) ListBlob(ctx context.Context, req *shardnode.ListBlobArgs) (resp shardnode.ListBlobRet, err error) {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
return shardnode.ListBlobResponse{}, err
return shardnode.ListBlobRet{}, err
}
blobs, nextMarker, err := space.ListBlob(ctx, req.GetHeader(), req.GetPrefix(), req.GetMarker(), req.GetCount())
if err != nil {

8
blobstore/shardnode/catalog/space.go
View File

@ -157,7 +157,7 @@ func (s *Space) ListItem(ctx context.Context, h shardnode.ShardOpHeader, prefix,
return items, s.decodeSpaceKey(nextMarker), nil
}
func (s *Space) CreateBlob(ctx context.Context, req *shardnode.CreateBlobRequest) (resp shardnode.CreateBlobResponse, err error) {
func (s *Space) CreateBlob(ctx context.Context, req *shardnode.CreateBlobArgs) (resp shardnode.CreateBlobRet, err error) {
h := req.Header
sd, err := s.shardGetter.GetShard(h.DiskID, h.Suid)
if err != nil {
@ -202,7 +202,7 @@ func (s *Space) CreateBlob(ctx context.Context, req *shardnode.CreateBlobRequest
return
}
func (s *Space) GetBlob(ctx context.Context, req *shardnode.GetBlobRequest) (resp shardnode.GetBlobResponse, err error) {
func (s *Space) GetBlob(ctx context.Context, req *shardnode.GetBlobArgs) (resp shardnode.GetBlobRet, err error) {
h := req.Header
sd, err := s.shardGetter.GetShard(h.DiskID, h.Suid)
if err != nil {
@ -229,7 +229,7 @@ func (s *Space) GetBlob(ctx context.Context, req *shardnode.GetBlobRequest) (res
return
}
func (s *Space) DeleteBlob(ctx context.Context, req *shardnode.DeleteBlobRequest) error {
func (s *Space) DeleteBlob(ctx context.Context, req *shardnode.DeleteBlobArgs) error {
h := req.Header
sd, err := s.shardGetter.GetShard(h.DiskID, h.Suid)
if err != nil {
@ -242,7 +242,7 @@ func (s *Space) DeleteBlob(ctx context.Context, req *shardnode.DeleteBlobRequest
}, key)
}
func (s *Space) SealBlob(ctx context.Context, req *shardnode.SealBlobRequest) error {
func (s *Space) SealBlob(ctx context.Context, req *shardnode.SealBlobArgs) error {
h := req.Header
sd, err := s.shardGetter.GetShard(h.DiskID, h.Suid)
if err != nil {

8
blobstore/shardnode/catalog/space_test.go
View File

@ -157,7 +157,7 @@ func TestSpace_Blob(t *testing.T) {
name := []byte("blob")
oph := shardnode.ShardOpHeader{}
args := &shardnode.CreateBlobRequest{
args := &shardnode.CreateBlobArgs{
Header: oph,
Name: name,
CodeMode: codemode.EC6P6,
@ -174,7 +174,7 @@ func TestSpace_Blob(t *testing.T) {
// get
gomock.InOrder(mockSpace.mockHandler.EXPECT().Get(A, A, A).Return(newMockValGetter(blobBytes), nil))
ret1, err := mockSpace.space.GetBlob(ctx, &shardnode.GetBlobRequest{Header: oph, Name: name})
ret1, err := mockSpace.space.GetBlob(ctx, &shardnode.GetBlobArgs{Header: oph, Name: name})
require.Nil(t, err)
require.Equal(t, ret.Blob, ret1.Blob)
@ -182,12 +182,12 @@ func TestSpace_Blob(t *testing.T) {
gomock.InOrder(mockSpace.mockHandler.EXPECT().Get(A, A, A).Return(newMockValGetter(blobBytes), nil))
gomock.InOrder(mockSpace.mockHandler.EXPECT().Update(A, A, A).Return(nil))
err = mockSpace.space.SealBlob(ctx, &shardnode.SealBlobRequest{Header: oph, Name: name})
err = mockSpace.space.SealBlob(ctx, &shardnode.SealBlobArgs{Header: oph, Name: name})
require.Nil(t, err)
// delete
gomock.InOrder(mockSpace.mockHandler.EXPECT().Delete(A, A, A).Return(nil))
err = mockSpace.space.DeleteBlob(ctx, &shardnode.DeleteBlobRequest{Header: oph, Name: name})
err = mockSpace.space.DeleteBlob(ctx, &shardnode.DeleteBlobArgs{Header: oph, Name: name})
require.Nil(t, err)
}

10
blobstore/shardnode/item.go
View File

@ -21,7 +21,7 @@ import (
"github.com/cubefs/cubefs/blobstore/api/shardnode"
)
func (s *service) InsertItem(ctx context.Context, req *shardnode.InsertItemRequest) error {
func (s *service) InsertItem(ctx context.Context, req *shardnode.InsertItemArgs) error {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -30,7 +30,7 @@ func (s *service) InsertItem(ctx context.Context, req *shardnode.InsertItemReque
return space.InsertItem(ctx, req.GetHeader(), req.GetItem())
}
func (s *service) UpdateItem(ctx context.Context, req *shardnode.UpdateItemRequest) error {
func (s *service) UpdateItem(ctx context.Context, req *shardnode.UpdateItemArgs) error {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -39,7 +39,7 @@ func (s *service) UpdateItem(ctx context.Context, req *shardnode.UpdateItemReque
return space.UpdateItem(ctx, req.GetHeader(), req.GetItem())
}
func (s *service) DeleteItem(ctx context.Context, req *shardnode.DeleteItemRequest) error {
func (s *service) DeleteItem(ctx context.Context, req *shardnode.DeleteItemArgs) error {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -48,7 +48,7 @@ func (s *service) DeleteItem(ctx context.Context, req *shardnode.DeleteItemReque
return space.DeleteItem(ctx, req.GetHeader(), req.GetID())
}
func (s *service) GetItem(ctx context.Context, req *shardnode.GetItemRequest) (item shardnode.Item, err error) {
func (s *service) GetItem(ctx context.Context, req *shardnode.GetItemArgs) (item shardnode.Item, err error) {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {
@ -58,7 +58,7 @@ func (s *service) GetItem(ctx context.Context, req *shardnode.GetItemRequest) (i
return
}
func (s *service) ListItem(ctx context.Context, req *shardnode.ListItemRequest) (resp shardnode.ListItemResponse, err error) {
func (s *service) ListItem(ctx context.Context, req *shardnode.ListItemArgs) (resp shardnode.ListItemRet, err error) {
sid := req.Header.SpaceID
space, err := s.catalog.GetSpace(ctx, sid)
if err != nil {

4
blobstore/shardnode/shard.go
View File

@ -28,7 +28,7 @@ import (
"github.com/cubefs/cubefs/blobstore/util/errors"
)
func (s *service) AddShard(ctx context.Context, req *shardnode.AddShardRequest) error {
func (s *service) AddShard(ctx context.Context, req *shardnode.AddShardArgs) error {
disk, err := s.getDisk(req.DiskID)
if err != nil {
return err
@ -38,7 +38,7 @@ func (s *service) AddShard(ctx context.Context, req *shardnode.AddShardRequest)
}
// UpdateShard update shard info
func (s *service) UpdateShard(ctx context.Context, req *shardnode.UpdateShardRequest) error {
func (s *service) UpdateShard(ctx context.Context, req *shardnode.UpdateShardArgs) error {
disk, err := s.getDisk(req.DiskID)
if err != nil {
return err

323
blobstore/shardnode/storage/mock_shard.go
View File

@ -1,17 +1,3 @@
// Copyright 2022 The CubeFS 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.
// Code generated by MockGen. DO NOT EDIT.
// Source: ../storage/shard.go
@ -27,6 +13,247 @@ import (
gomock "github.com/golang/mock/gomock"
)
// MockValGetter is a mock of ValGetter interface.
type MockValGetter struct {
ctrl *gomock.Controller
recorder *MockValGetterMockRecorder
}
// MockValGetterMockRecorder is the mock recorder for MockValGetter.
type MockValGetterMockRecorder struct {
mock *MockValGetter
}
// NewMockValGetter creates a new mock instance.
func NewMockValGetter(ctrl *gomock.Controller) *MockValGetter {
mock := &MockValGetter{ctrl: ctrl}
mock.recorder = &MockValGetterMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockValGetter) EXPECT() *MockValGetterMockRecorder {
return m.recorder
}
// Close mocks base method.
func (m *MockValGetter) Close() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Close")
}
// Close indicates an expected call of Close.
func (mr *MockValGetterMockRecorder) Close() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockValGetter)(nil).Close))
}
// Read mocks base method.
func (m *MockValGetter) Read(p []byte) (int, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Read", p)
ret0, _ := ret[0].(int)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Read indicates an expected call of Read.
func (mr *MockValGetterMockRecorder) Read(p interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockValGetter)(nil).Read), p)
}
// Size mocks base method.
func (m *MockValGetter) Size() int {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Size")
ret0, _ := ret[0].(int)
return ret0
}
// Size indicates an expected call of Size.
func (mr *MockValGetterMockRecorder) Size() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Size", reflect.TypeOf((*MockValGetter)(nil).Size))
}
// Value mocks base method.
func (m *MockValGetter) Value() []byte {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Value")
ret0, _ := ret[0].([]byte)
return ret0
}
// Value indicates an expected call of Value.
func (mr *MockValGetterMockRecorder) Value() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Value", reflect.TypeOf((*MockValGetter)(nil).Value))
}
// MockShardKVHandler is a mock of ShardKVHandler interface.
type MockShardKVHandler struct {
ctrl *gomock.Controller
recorder *MockShardKVHandlerMockRecorder
}
// MockShardKVHandlerMockRecorder is the mock recorder for MockShardKVHandler.
type MockShardKVHandlerMockRecorder struct {
mock *MockShardKVHandler
}
// NewMockShardKVHandler creates a new mock instance.
func NewMockShardKVHandler(ctrl *gomock.Controller) *MockShardKVHandler {
mock := &MockShardKVHandler{ctrl: ctrl}
mock.recorder = &MockShardKVHandlerMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockShardKVHandler) EXPECT() *MockShardKVHandlerMockRecorder {
return m.recorder
}
// Delete mocks base method.
func (m *MockShardKVHandler) Delete(ctx context.Context, h OpHeader, key []byte) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Delete", ctx, h, key)
ret0, _ := ret[0].(error)
return ret0
}
// Delete indicates an expected call of Delete.
func (mr *MockShardKVHandlerMockRecorder) Delete(ctx, h, key interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockShardKVHandler)(nil).Delete), ctx, h, key)
}
// Get mocks base method.
func (m *MockShardKVHandler) Get(ctx context.Context, h OpHeader, key []byte) (ValGetter, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Get", ctx, h, key)
ret0, _ := ret[0].(ValGetter)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Get indicates an expected call of Get.
func (mr *MockShardKVHandlerMockRecorder) Get(ctx, h, key interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockShardKVHandler)(nil).Get), ctx, h, key)
}
// Insert mocks base method.
func (m *MockShardKVHandler) Insert(ctx context.Context, h OpHeader, kv *KV) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Insert", ctx, h, kv)
ret0, _ := ret[0].(error)
return ret0
}
// Insert indicates an expected call of Insert.
func (mr *MockShardKVHandlerMockRecorder) Insert(ctx, h, kv interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockShardKVHandler)(nil).Insert), ctx, h, kv)
}
// List mocks base method.
func (m *MockShardKVHandler) List(ctx context.Context, h OpHeader, prefix, marker []byte, count uint64, rangeFunc func([]byte) error) ([]byte, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "List", ctx, h, prefix, marker, count, rangeFunc)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// List indicates an expected call of List.
func (mr *MockShardKVHandlerMockRecorder) List(ctx, h, prefix, marker, count, rangeFunc interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockShardKVHandler)(nil).List), ctx, h, prefix, marker, count, rangeFunc)
}
// Update mocks base method.
func (m *MockShardKVHandler) Update(ctx context.Context, h OpHeader, kv *KV) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Update", ctx, h, kv)
ret0, _ := ret[0].(error)
return ret0
}
// Update indicates an expected call of Update.
func (mr *MockShardKVHandlerMockRecorder) Update(ctx, h, kv interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockShardKVHandler)(nil).Update), ctx, h, kv)
}
// MockShardItemHandler is a mock of ShardItemHandler interface.
type MockShardItemHandler struct {
ctrl *gomock.Controller
recorder *MockShardItemHandlerMockRecorder
}
// MockShardItemHandlerMockRecorder is the mock recorder for MockShardItemHandler.
type MockShardItemHandlerMockRecorder struct {
mock *MockShardItemHandler
}
// NewMockShardItemHandler creates a new mock instance.
func NewMockShardItemHandler(ctrl *gomock.Controller) *MockShardItemHandler {
mock := &MockShardItemHandler{ctrl: ctrl}
mock.recorder = &MockShardItemHandlerMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockShardItemHandler) EXPECT() *MockShardItemHandlerMockRecorder {
return m.recorder
}
// GetItem mocks base method.
func (m *MockShardItemHandler) GetItem(ctx context.Context, h OpHeader, id []byte) (shardnode.Item, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetItem", ctx, h, id)
ret0, _ := ret[0].(shardnode.Item)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetItem indicates an expected call of GetItem.
func (mr *MockShardItemHandlerMockRecorder) GetItem(ctx, h, id interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItem", reflect.TypeOf((*MockShardItemHandler)(nil).GetItem), ctx, h, id)
}
// ListItem mocks base method.
func (m *MockShardItemHandler) ListItem(ctx context.Context, h OpHeader, prefix, id []byte, count uint64) ([]shardnode.Item, []byte, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListItem", ctx, h, prefix, id, count)
ret0, _ := ret[0].([]shardnode.Item)
ret1, _ := ret[1].([]byte)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// ListItem indicates an expected call of ListItem.
func (mr *MockShardItemHandlerMockRecorder) ListItem(ctx, h, prefix, id, count interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListItem", reflect.TypeOf((*MockShardItemHandler)(nil).ListItem), ctx, h, prefix, id, count)
}
// UpdateItem mocks base method.
func (m *MockShardItemHandler) UpdateItem(ctx context.Context, h OpHeader, i shardnode.Item) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateItem", ctx, h, i)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateItem indicates an expected call of UpdateItem.
func (mr *MockShardItemHandlerMockRecorder) UpdateItem(ctx, h, i interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateItem", reflect.TypeOf((*MockShardItemHandler)(nil).UpdateItem), ctx, h, i)
}
// MockSpaceShardHandler is a mock of ShardHandler interface.
type MockSpaceShardHandler struct {
ctrl *gomock.Controller
@ -64,18 +291,33 @@ func (mr *MockSpaceShardHandlerMockRecorder) Checkpoint(ctx interface{}) *gomock
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Checkpoint", reflect.TypeOf((*MockSpaceShardHandler)(nil).Checkpoint), ctx)
}
// DeleteItem mocks base method.
func (m *MockSpaceShardHandler) DeleteItem(ctx context.Context, h OpHeader, id []byte) error {
// Delete mocks base method.
func (m *MockSpaceShardHandler) Delete(ctx context.Context, h OpHeader, key []byte) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteItem", ctx, h, id)
ret := m.ctrl.Call(m, "Delete", ctx, h, key)
ret0, _ := ret[0].(error)
return ret0
}
// DeleteItem indicates an expected call of DeleteItem.
func (mr *MockSpaceShardHandlerMockRecorder) DeleteItem(ctx, h, id interface{}) *gomock.Call {
// Delete indicates an expected call of Delete.
func (mr *MockSpaceShardHandlerMockRecorder) Delete(ctx, h, key interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteItem", reflect.TypeOf((*MockSpaceShardHandler)(nil).DeleteItem), ctx, h, id)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSpaceShardHandler)(nil).Delete), ctx, h, key)
}
// Get mocks base method.
func (m *MockSpaceShardHandler) Get(ctx context.Context, h OpHeader, key []byte) (ValGetter, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Get", ctx, h, key)
ret0, _ := ret[0].(ValGetter)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Get indicates an expected call of Get.
func (mr *MockSpaceShardHandlerMockRecorder) Get(ctx, h, key interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockSpaceShardHandler)(nil).Get), ctx, h, key)
}
// GetItem mocks base method.
@ -107,18 +349,33 @@ func (mr *MockSpaceShardHandlerMockRecorder) GetRouteVersion() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRouteVersion", reflect.TypeOf((*MockSpaceShardHandler)(nil).GetRouteVersion))
}
// InsertItem mocks base method.
func (m *MockSpaceShardHandler) InsertItem(ctx context.Context, h OpHeader, i shardnode.Item) error {
// Insert mocks base method.
func (m *MockSpaceShardHandler) Insert(ctx context.Context, h OpHeader, kv *KV) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "InsertItem", ctx, h, i)
ret := m.ctrl.Call(m, "Insert", ctx, h, kv)
ret0, _ := ret[0].(error)
return ret0
}
// InsertItem indicates an expected call of InsertItem.
func (mr *MockSpaceShardHandlerMockRecorder) InsertItem(ctx, h, i interface{}) *gomock.Call {
// Insert indicates an expected call of Insert.
func (mr *MockSpaceShardHandlerMockRecorder) Insert(ctx, h, kv interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertItem", reflect.TypeOf((*MockSpaceShardHandler)(nil).InsertItem), ctx, h, i)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockSpaceShardHandler)(nil).Insert), ctx, h, kv)
}
// List mocks base method.
func (m *MockSpaceShardHandler) List(ctx context.Context, h OpHeader, prefix, marker []byte, count uint64, rangeFunc func([]byte) error) ([]byte, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "List", ctx, h, prefix, marker, count, rangeFunc)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// List indicates an expected call of List.
func (mr *MockSpaceShardHandlerMockRecorder) List(ctx, h, prefix, marker, count, rangeFunc interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockSpaceShardHandler)(nil).List), ctx, h, prefix, marker, count, rangeFunc)
}
// ListItem mocks base method.
@ -151,6 +408,20 @@ func (mr *MockSpaceShardHandlerMockRecorder) Stats() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stats", reflect.TypeOf((*MockSpaceShardHandler)(nil).Stats))
}
// Update mocks base method.
func (m *MockSpaceShardHandler) Update(ctx context.Context, h OpHeader, kv *KV) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Update", ctx, h, kv)
ret0, _ := ret[0].(error)
return ret0
}
// Update indicates an expected call of Update.
func (mr *MockSpaceShardHandlerMockRecorder) Update(ctx, h, kv interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockSpaceShardHandler)(nil).Update), ctx, h, kv)
}
// UpdateItem mocks base method.
func (m *MockSpaceShardHandler) UpdateItem(ctx context.Context, h OpHeader, i shardnode.Item) error {
m.ctrl.T.Helper()