搭建框架

controllers/base.go

@@ -1,7 +1,85 @@
 package controllers
 
-import "github.com/astaxie/beego"
+import (
+
+	"bytes"
+
+	"github.com/lifei6671/godoc/models"
+	"github.com/lifei6671/godoc/conf"
+	"github.com/astaxie/beego"
+)
+
 
 type BaseController struct {
 	beego.Controller
+	Member *models.Member
 }
+
+// Prepare 预处理.
+func (c *BaseController) Prepare (){
+	c.Data["SiteName"] = "MinDoc"
+	c.Data["Member"] = models.Member{}
+
+	if member,ok := c.GetSession(conf.LoginSessionName).(models.Member); ok && member.MemberId > 0{
+		c.Member = &member
+		c.Data["Member"] = c.Member
+	}else{
+		c.Member = models.NewMember()
+		c.Member.Find(1)
+		c.Data["Member"] = *c.Member
+	}
+	c.Data["BaseUrl"] = c.Ctx.Input.Scheme() + "://" + c.Ctx.Request.Host
+
+	if options,err := models.NewOption().All();err == nil {
+		for _,item := range options {
+			c.Data[item.OptionName] = item.OptionValue
+		}
+	}
+}
+
+// SetMember 获取或设置当前登录用户信息,如果 MemberId 小于 0 则标识删除 Session
+func (c *BaseController) SetMember(member models.Member) {
+
+	if member.MemberId <= 0 {
+		c.DelSession(conf.LoginSessionName)
+		c.DelSession("uid")
+		c.DestroySession()
+	} else {
+		c.SetSession(conf.LoginSessionName, member)
+		c.SetSession("uid", member.MemberId)
+	}
+}
+
+// JsonResult 响应 json 结果
+func (c *BaseController) JsonResult(errCode int,errMsg string,data ...interface{}){
+	json := make(map[string]interface{},3)
+
+	json["errcode"] = errCode
+	json["message"] = errMsg
+
+	if len(data) > 0 && data[0] != nil{
+		json["data"] = data[0]
+	}
+
+	c.Data["json"] = json
+	c.ServeJSON(true)
+	c.StopRun()
+}
+
+// ExecuteViewPathTemplate 执行指定的模板并返回执行结果.
+func (c *BaseController) ExecuteViewPathTemplate(tplName string,data interface{}) (string,error){
+	var buf bytes.Buffer
+
+	viewPath := c.ViewPath
+
+	if c.ViewPath == "" {
+		viewPath = beego.BConfig.WebConfig.ViewsPath
+
+	}
+
+	if err := beego.ExecuteViewPathTemplate(&buf,tplName,viewPath,data); err != nil {
+		return "",err
+	}
+	return buf.String(),nil
+}
+