소스 검색

raft: make memory storage set method thread safe

杜军 9 년 전
부모
커밋
8151d4d0bc
1개의 변경된 파일과 2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      raft/storage.go

+ 2 - 0
raft/storage.go

@@ -98,6 +98,8 @@ func (ms *MemoryStorage) InitialState() (pb.HardState, pb.ConfState, error) {
 
 
 // SetHardState saves the current HardState.
 // SetHardState saves the current HardState.
 func (ms *MemoryStorage) SetHardState(st pb.HardState) error {
 func (ms *MemoryStorage) SetHardState(st pb.HardState) error {
+	ms.Lock()
+	defer ms.Unlock()
 	ms.hardState = st
 	ms.hardState = st
 	return nil
 	return nil
 }
 }