From 31f505bfc16ccfef1353b4ca4fbd51be0a5eb444 Mon Sep 17 00:00:00 2001
From: astaxie <xiemengjun@gmail.com>
Date: Thu, 14 Mar 2013 21:17:37 +0800
Subject: [PATCH] add  RenderBytes function

---
 controller.go | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/controller.go b/controller.go
index 682aedf0..44440adb 100644
--- a/controller.go
+++ b/controller.go
@@ -83,6 +83,20 @@ func (c *Controller) Options() {
 }
 
 func (c *Controller) Render() error {
+	rb, err := c.RenderBytes()
+
+	if err != nil {
+		return err
+	} else {
+		c.Ctx.SetHeader("Content-Length", strconv.Itoa(len(rb)), true)
+		c.Ctx.ContentType("text/html")
+		c.Ctx.ResponseWriter.Write(rb)
+		return nil
+	}
+	return nil
+}
+
+func (c *Controller) RenderBytes() ([]byte, error) {
 	//if the controller has set layout, then first get the tplname's content set the content to the layout
 	if c.Layout != "" {
 		if c.TplNames == "" {
@@ -95,22 +109,28 @@ func (c *Controller) Render() error {
 		tplcontent, _ := ioutil.ReadAll(newbytes)
 		c.Data["LayoutContent"] = template.HTML(string(tplcontent))
 		_, file = path.Split(c.Layout)
-		err := BeeTemplates[subdir].ExecuteTemplate(c.Ctx.ResponseWriter, file, c.Data)
+		ibytes := bytes.NewBufferString("")
+		err := BeeTemplates[subdir].ExecuteTemplate(ibytes, file, c.Data)
 		if err != nil {
 			Trace("template Execute err:", err)
 		}
+		icontent, _ := ioutil.ReadAll(ibytes)
+		return icontent, nil
 	} else {
 		if c.TplNames == "" {
 			c.TplNames = c.ChildName + "/" + c.Ctx.Request.Method + "." + c.TplExt
 		}
 		_, file := path.Split(c.TplNames)
 		subdir := path.Dir(c.TplNames)
-		err := BeeTemplates[subdir].ExecuteTemplate(c.Ctx.ResponseWriter, file, c.Data)
+		ibytes := bytes.NewBufferString("")
+		err := BeeTemplates[subdir].ExecuteTemplate(ibytes, file, c.Data)
 		if err != nil {
 			Trace("template Execute err:", err)
 		}
+		icontent, _ := ioutil.ReadAll(ibytes)
+		return icontent, nil
 	}
-	return nil
+	return []byte{}, nil
 }
 
 func (c *Controller) Redirect(url string, code int) {