Added transparent compression support to helloworldserver example

2 files changed, 22 insertions, 3 deletions

diff --git a/examples/helloworldserver/README.md b/examples/helloworldserver/README.md
index 37bf022..80e801e 100644
--- a/examples/helloworldserver/README.md
+++ b/examples/helloworldserver/README.md
@@ -1,6 +1,8 @@
 # HelloWorld server example
 
-Displays various request info.
+* Displays various request info.
+* Sets response headers and cookies.
+* Supports transparent compression.
 
 # How to build
 
diff --git a/examples/helloworldserver/helloworldserver.go b/examples/helloworldserver/helloworldserver.go
index f4c9c3e..22b518a 100644
--- a/examples/helloworldserver/helloworldserver.go
+++ b/examples/helloworldserver/helloworldserver.go
@@ -8,12 +8,20 @@ import (
 	"github.com/valyala/fasthttp"
 )
 
-var addr = flag.String("addr", ":8080", "TCP address to listen to")
+var (
+	addr     = flag.String("addr", ":8080", "TCP address to listen to")
+	compress = flag.Bool("compress", false, "Whether to enable transparent response compression")
+)
 
 func main() {
 	flag.Parse()
 
-	if err := fasthttp.ListenAndServe(*addr, requestHandler); err != nil {
+	h := requestHandler
+	if *compress {
+		h = fasthttp.CompressHandler(h)
+	}
+
+	if err := fasthttp.ListenAndServe(*addr, h); err != nil {
 		log.Fatalf("Error in ListenAndServe: %s", err)
 	}
 }
@@ -35,4 +43,13 @@ func requestHandler(ctx *fasthttp.RequestCtx) {
 	fmt.Fprintf(ctx, "Raw request is:\n---CUT---\n%s\n---CUT---", &ctx.Request)
 
 	ctx.SetContentType("text/plain; charset=utf8")
+
+	// Set arbitrary headers
+	ctx.Response.Header.Set("X-My-Header", "my-header-value")
+
+	// Set cookies
+	var c fasthttp.Cookie
+	c.SetKey("cookie-name")
+	c.SetValue("cookie-value")
+	ctx.Response.Header.SetCookie(&c)
 }