[Go] Add SetLockManager (#60)

Signed-off-by: sunby <[email protected]>
milvus-io · Sep 22, 2023 · 0f44a36 · 0f44a36
1 parent 6c10834
commit 0f44a36
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/go/storage/options/options.go b/go/storage/options/options.go
@@ -29,8 +29,8 @@ func (b *SpaceOptionsBuilder) SetVersion(version int64) *SpaceOptionsBuilder {
 	return b
 }
 
-func (b *SpaceOptionsBuilder) SetLockHandler(lockHandler lock.LockManager) *SpaceOptionsBuilder {
-	b.options.LockManager = lockHandler
+func (b *SpaceOptionsBuilder) SetLockManager(lockManager lock.LockManager) *SpaceOptionsBuilder {
+	b.options.LockManager = lockManager
 	return b
 }
 

diff --git a/go/storage/space.go b/go/storage/space.go
@@ -193,3 +193,7 @@ func (s *Space) Manifest() *manifest.Manifest {
 func (s *Space) LockManager() lock.LockManager {
 	return s.lockManager
 }
+
+func (s *Space) SetLockManager(lockManager lock.LockManager) {
+	s.lockManager = lockManager
+}