Also send linearizable reads to leader

logbn · Jan 5, 2025 · f0d5f88 · f0d5f88
1 parent c28afcc
commit f0d5f88
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/shard_client.go b/shard_client.go
@@ -120,6 +120,11 @@ func (c *client) Read(ctx context.Context, query []byte, stale bool) (value uint
 			return
 		}
 	}
+	if c.writeToLeader {
+		if client, ok := c.manager.clientLeader[c.shardID]; ok {
+			return client.Read(ctx, c.shardID, query, stale)
+		}
+	}
 	c.manager.mutex.RLock()
 	list, ok := c.manager.clientMember[c.shardID]
 	c.manager.mutex.RUnlock()
@@ -159,6 +164,11 @@ func (c *client) Watch(ctx context.Context, query []byte, results chan<- *Result
 			return
 		}
 	}
+	if c.writeToLeader {
+		if client, ok := c.manager.clientLeader[c.shardID]; ok {
+			return client.Watch(ctx, c.shardID, query, results, stale)
+		}
+	}
 	c.manager.mutex.RLock()
 	list, ok := c.manager.clientMember[c.shardID]
 	c.manager.mutex.RUnlock()