From 5a9bff2000701fd565027c184cfa010c7c820592 Mon Sep 17 00:00:00 2001
From: JessonChan <chinachendongpo@gmail.com>
Date: Thu, 17 Mar 2016 19:09:38 +0800
Subject: [PATCH] init gzip level

---
 beego.go | 8 +++++---
 hooks.go | 8 ++++++++
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/beego.go b/beego.go
index 8f82cdcf..89e4f671 100644
--- a/beego.go
+++ b/beego.go
@@ -22,12 +22,12 @@ import (
 )
 
 const (
-	// VERSION represent beego web framework version.
+// VERSION represent beego web framework version.
 	VERSION = "1.6.1"
 
-	// DEV is for develop
+// DEV is for develop
 	DEV = "dev"
-	// PROD is for production
+// PROD is for production
 	PROD = "prod"
 )
 
@@ -51,6 +51,7 @@ func AddAPPStartHook(hf hookfunc) {
 // beego.Run(":8089")
 // beego.Run("127.0.0.1:8089")
 func Run(params ...string) {
+
 	initBeforeHTTPRun()
 
 	if len(params) > 0 && params[0] != "" {
@@ -74,6 +75,7 @@ func initBeforeHTTPRun() {
 	AddAPPStartHook(registerDocs)
 	AddAPPStartHook(registerTemplate)
 	AddAPPStartHook(registerAdmin)
+	AddAPPStartHook(registerGzip)
 
 	for _, hk := range hooks {
 		if err := hk(); err != nil {
diff --git a/hooks.go b/hooks.go
index 59b10b32..1a7937b5 100644
--- a/hooks.go
+++ b/hooks.go
@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"path/filepath"
 
+	"github.com/astaxie/beego/context"
 	"github.com/astaxie/beego/session"
 )
 
@@ -91,3 +92,10 @@ func registerAdmin() error {
 	}
 	return nil
 }
+
+func registerGzip() error {
+	if BConfig.EnableGzip {
+		context.InitGzip(AppConfig)
+	}
+	return nil
+}