fix: extra preflight check on write operations to ensure leadership

webmeshproj · Jul 29, 2023 · eced44f · eced44f
1 parent cce00a5
commit eced44f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/raft/storage.go b/pkg/raft/storage.go
@@ -145,6 +145,9 @@ type raftStorage struct {
 
 // Put sets the value of a key.
 func (rs *raftStorage) Put(ctx context.Context, key, value string) error {
+	if !rs.raft.IsLeader() {
+		return ErrNotLeader
+	}
 	logEntry := &v1.RaftLogEntry{
 		Type:  v1.RaftCommandType_PUT,
 		Key:   key,
@@ -155,6 +158,9 @@ func (rs *raftStorage) Put(ctx context.Context, key, value string) error {
 
 // Delete removes a key.
 func (rs *raftStorage) Delete(ctx context.Context, key string) error {
+	if !rs.raft.IsLeader() {
+		return ErrNotLeader
+	}
 	logEntry := &v1.RaftLogEntry{
 		Type: v1.RaftCommandType_DELETE,
 		Key:  key,