implement Close() to gracefully stop all servers

tferdous17 · tferdous17 · commit c37f9379f086 · 2024-12-10T22:45:28.000-05:00
diff --git a/http/service.go b/http/service.go
@@ -25,6 +25,7 @@ type Cluster interface {
 	Delete(key string) error
 	AddNode()
 	RemoveNode(addr string)
+	Close()
 }
 
 type Service struct {
diff --git a/store/cluster.go b/store/cluster.go
@@ -114,6 +114,13 @@ func (c *Cluster) Open() {
 	}
 }
 
+func (c *Cluster) Close() {
+	fmt.Println("Closing entire cluster..")
+	for _, node := range c.nodes {
+		node.server.GracefulStop()
+	}
+}
+
 func (c *Cluster) Put(key, value string) error {
 	nodeAddr, _ := c.hashRing.GetNode(key) // get which node this key should be on
 	fmt.Printf("key = %s\t", key)

Original file line number	Diff line number	Diff line change
`@@ -114,6 +114,13 @@ func (c *Cluster) Open() {`
`114`	`114`	`}`
`115`	`115`	`}`
`116`	`116`
	`117`	`+func (c *Cluster) Close() {`
	`118`	`+ fmt.Println("Closing entire cluster..")`
	`119`	`+ for _, node := range c.nodes {`
	`120`	`+ node.server.GracefulStop()`
	`121`	`+ }`
	`122`	`+}`
	`123`	`+`
`117`	`124`	`func (c *Cluster) Put(key, value string) error {`
`118`	`125`	`nodeAddr, _ := c.hashRing.GetNode(key) // get which node this key should be on`
`119`	`126`	`fmt.Printf("key = %s\t", key)`