support auto get session from input fix #211

2013-09-22 11:43:22 +08:00 · 2013-09-22 11:43:22 +08:00 · 4a3d32dc1f
parent 048be29fcd
commit 4a3d32dc1f
2 changed files with 5 additions and 1 deletions
--- a/controller.go
+++ b/controller.go
@ -270,7 +270,7 @@ func (c *Controller) SaveToFile(fromfile, tofile string) error {

 func (c *Controller) StartSession() session.SessionStore {
 	if c.CruSession == nil {
-		c.CruSession = GlobalSessions.SessionStart(c.Ctx.ResponseWriter, c.Ctx.Request)
+		c.CruSession = c.Ctx.Input.CruSession
 	}
 	return c.CruSession
 }
--- a/router.go
+++ b/router.go
@ -306,6 +306,10 @@ func (p *ControllerRegistor) ServeHTTP(rw http.ResponseWriter, r *http.Request)
 		context.Output = beecontext.NewOutput(rw)
 	}

+	if SessionOn {
+		context.Input.CruSession = GlobalSessions.SessionStart(w, r)
+	}
+
 	var runrouter *controllerInfo
 	var findrouter bool