Update objecttree with tryclose

This commit is contained in:
mcrakhman 2023-03-09 23:09:26 +01:00 committed by Mikhail Iudin
parent afd134c7a1
commit 520f2cf470
No known key found for this signature in database
GPG Key ID: FAAAA8BAABDFF1C0
4 changed files with 39 additions and 0 deletions

View File

@ -350,6 +350,21 @@ func (mr *MockObjectTreeMockRecorder) Storage() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Storage", reflect.TypeOf((*MockObjectTree)(nil).Storage)) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Storage", reflect.TypeOf((*MockObjectTree)(nil).Storage))
} }
// TryClose mocks base method.
func (m *MockObjectTree) TryClose() (bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "TryClose")
ret0, _ := ret[0].(bool)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// TryClose indicates an expected call of TryClose.
func (mr *MockObjectTreeMockRecorder) TryClose() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TryClose", reflect.TypeOf((*MockObjectTree)(nil).TryClose))
}
// TryLock mocks base method. // TryLock mocks base method.
func (m *MockObjectTree) TryLock() bool { func (m *MockObjectTree) TryLock() bool {
m.ctrl.T.Helper() m.ctrl.T.Helper()

View File

@ -82,6 +82,7 @@ type ObjectTree interface {
Delete() error Delete() error
Close() error Close() error
TryClose() (bool, error)
} }
type objectTree struct { type objectTree struct {
@ -560,6 +561,10 @@ func (ot *objectTree) Root() *Change {
return ot.tree.Root() return ot.tree.Root()
} }
func (ot *objectTree) TryClose() (bool, error) {
return true, ot.Close()
}
func (ot *objectTree) Close() error { func (ot *objectTree) Close() error {
return nil return nil
} }

View File

@ -501,6 +501,21 @@ func (mr *MockSyncTreeMockRecorder) SyncWithPeer(arg0, arg1 interface{}) *gomock
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncWithPeer", reflect.TypeOf((*MockSyncTree)(nil).SyncWithPeer), arg0, arg1) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncWithPeer", reflect.TypeOf((*MockSyncTree)(nil).SyncWithPeer), arg0, arg1)
} }
// TryClose mocks base method.
func (m *MockSyncTree) TryClose() (bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "TryClose")
ret0, _ := ret[0].(bool)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// TryClose indicates an expected call of TryClose.
func (mr *MockSyncTreeMockRecorder) TryClose() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TryClose", reflect.TypeOf((*MockSyncTree)(nil).TryClose))
}
// TryLock mocks base method. // TryLock mocks base method.
func (m *MockSyncTree) TryLock() bool { func (m *MockSyncTree) TryLock() bool {
m.ctrl.T.Helper() m.ctrl.T.Helper()

View File

@ -209,6 +209,10 @@ func (s *syncTree) Delete() (err error) {
return return
} }
func (s *syncTree) TryClose() (bool, error) {
return true, s.Close()
}
func (s *syncTree) Close() (err error) { func (s *syncTree) Close() (err error) {
log.Debug("closing sync tree", zap.String("id", s.Id())) log.Debug("closing sync tree", zap.String("id", s.Id()))
defer func() { defer func() {