/
kv.go
152 lines (127 loc) · 4.65 KB
/
kv.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
// Code generated by MockGen. DO NOT EDIT.
// Source: github.com/pingcap/br/pkg/lightning/backend/kv (interfaces: Encoder,Rows,Row)
// $ mockgen -package mock github.com/pingcap/br/pkg/lightning/backend/kv Encoder,Rows,Row
// Package mock is a generated GoMock package.
package mock
import (
gomock "github.com/golang/mock/gomock"
kv "github.com/pingcap/br/pkg/lightning/backend/kv"
log "github.com/pingcap/br/pkg/lightning/log"
verification "github.com/pingcap/br/pkg/lightning/verification"
types "github.com/pingcap/tidb/types"
reflect "reflect"
)
// MockEncoder is a mock of Encoder interface
type MockEncoder struct {
ctrl *gomock.Controller
recorder *MockEncoderMockRecorder
}
// MockEncoderMockRecorder is the mock recorder for MockEncoder
type MockEncoderMockRecorder struct {
mock *MockEncoder
}
// NewMockEncoder creates a new mock instance
func NewMockEncoder(ctrl *gomock.Controller) *MockEncoder {
mock := &MockEncoder{ctrl: ctrl}
mock.recorder = &MockEncoderMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockEncoder) EXPECT() *MockEncoderMockRecorder {
return m.recorder
}
// Close mocks base method
func (m *MockEncoder) Close() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Close")
}
// Close indicates an expected call of Close
func (mr *MockEncoderMockRecorder) Close() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockEncoder)(nil).Close))
}
// Encode mocks base method
func (m *MockEncoder) Encode(arg0 log.Logger, arg1 []types.Datum, arg2 int64, arg3 []int) (kv.Row, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Encode", arg0, arg1, arg2, arg3)
ret0, _ := ret[0].(kv.Row)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Encode indicates an expected call of Encode
func (mr *MockEncoderMockRecorder) Encode(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Encode", reflect.TypeOf((*MockEncoder)(nil).Encode), arg0, arg1, arg2, arg3)
}
// MockRows is a mock of Rows interface
type MockRows struct {
ctrl *gomock.Controller
recorder *MockRowsMockRecorder
}
// MockRowsMockRecorder is the mock recorder for MockRows
type MockRowsMockRecorder struct {
mock *MockRows
}
// NewMockRows creates a new mock instance
func NewMockRows(ctrl *gomock.Controller) *MockRows {
mock := &MockRows{ctrl: ctrl}
mock.recorder = &MockRowsMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockRows) EXPECT() *MockRowsMockRecorder {
return m.recorder
}
// Clear mocks base method
func (m *MockRows) Clear() kv.Rows {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Clear")
ret0, _ := ret[0].(kv.Rows)
return ret0
}
// Clear indicates an expected call of Clear
func (mr *MockRowsMockRecorder) Clear() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Clear", reflect.TypeOf((*MockRows)(nil).Clear))
}
// SplitIntoChunks mocks base method
func (m *MockRows) SplitIntoChunks(arg0 int) []kv.Rows {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "SplitIntoChunks", arg0)
ret0, _ := ret[0].([]kv.Rows)
return ret0
}
// SplitIntoChunks indicates an expected call of SplitIntoChunks
func (mr *MockRowsMockRecorder) SplitIntoChunks(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SplitIntoChunks", reflect.TypeOf((*MockRows)(nil).SplitIntoChunks), arg0)
}
// MockRow is a mock of Row interface
type MockRow struct {
ctrl *gomock.Controller
recorder *MockRowMockRecorder
}
// MockRowMockRecorder is the mock recorder for MockRow
type MockRowMockRecorder struct {
mock *MockRow
}
// NewMockRow creates a new mock instance
func NewMockRow(ctrl *gomock.Controller) *MockRow {
mock := &MockRow{ctrl: ctrl}
mock.recorder = &MockRowMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockRow) EXPECT() *MockRowMockRecorder {
return m.recorder
}
// ClassifyAndAppend mocks base method
func (m *MockRow) ClassifyAndAppend(arg0 *kv.Rows, arg1 *verification.KVChecksum, arg2 *kv.Rows, arg3 *verification.KVChecksum) {
m.ctrl.T.Helper()
m.ctrl.Call(m, "ClassifyAndAppend", arg0, arg1, arg2, arg3)
}
// ClassifyAndAppend indicates an expected call of ClassifyAndAppend
func (mr *MockRowMockRecorder) ClassifyAndAppend(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClassifyAndAppend", reflect.TypeOf((*MockRow)(nil).ClassifyAndAppend), arg0, arg1, arg2, arg3)
}