graceful shutdown: some improvements

rvflash · Sep 3, 2019 · 7dceed7 · 7dceed7
1 parent d2ebeb4
commit 7dceed7
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/example/graceful_server/main.go b/example/graceful_server/main.go
@@ -0,0 +1,28 @@
+package main
+
+import (
+	"log"
+
+	"github.com/rvflash/tcp"
+)
+
+func main() {
+	r := tcp.Default()
+	r.ACK(func(c *tcp.Context) {
+		// new message received
+		body, err := c.ReadAll()
+		if err != nil {
+			c.Error(err)
+			return
+		}
+		log.Println(string(body))
+		c.String("read")
+	})
+	r.SYN(func(c *tcp.Context) {
+		c.String("hello")
+	})
+	r.FIN(func(c *tcp.Context) {
+		log.Println("bye")
+	})
+	log.Fatal(r.Run(":9090"))
+}