Beego学习——实现用户登录登出

controller包

  user.go

import (
	"demo03.cn/models"
	"encoding/json"
	"fmt"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/orm"
)

type UserController struct {
	beego.Controller
}

func (c *UserController) Reg()  {

	resp :=make(map[string]interface{})
	defer c.RetData(resp)

	//获取前端传过来的json的数据
	_ = json.Unmarshal(c.Ctx.Input.RequestBody,&resp)
	//封装成结构体
	o := orm.NewOrm()
	user := models.User{}
	user.Name = resp["name"].(string)
	user.Password = resp["password"].(string)
	id,err :=o.Insert(&user)
	if err != nil{
		resp["errmsg"]="注册失败"
		return
	}
	resp["errmsg"]=fmt.Sprintf("注册成功,id:%d",id)
	c.SetSession("name",user.Name)
}

// 格式化数据 返回json格式
func (c *UserController) RetData(resp map[string]interface{}) {
	c.Data["json"] =resp
	c.ServeJSON()
}

  session.go

controller session.go

import (
	"demo03.cn/models"
	"encoding/json"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/logs"
	"github.com/astaxie/beego/orm"
)

type SessionController struct {
	beego.Controller
}

//TODO 获取Session
func (c *SessionController) GetSessionData()  {
	logs.Info("connect success")
	resp :=make(map[string]interface{})
	defer c.RetData(resp)

	user := models.User{}

	session := c.GetSession("name")
	if session!=nil {
		user.Name = session.(string)
		resp["msg"] = "获取成功"
		resp["data"] = user
	}else {
		resp["msg"] = "获取失败"
		logs.Info(user)
	}
}

//删除对应的Session
func (c *SessionController) DeleteSessionData()  {
	resp := make(map[string]interface{})
	defer c.RetData(resp)
	c.DelSession("name")
	resp["msg"]="删除成功"
}

// TODO 登录
func (c *SessionController) Login()  {
	resp := make(map[string]interface{})
	defer c.RetData(resp)

	//得到用户信息获取前端传递过来的json数据
	_ = json.Unmarshal(c.Ctx.Input.RequestBody,&resp)
	logs.Info(&resp)

	//判断是否合法
	if resp["name"] == nil || resp["password"] ==nil{
		resp["msg"]="用户名和密码不可以为空！"
		return
	}
	//与数据库匹配账号密码是否正确
	o := orm.NewOrm()
	user := models.User{Name:resp["name"].(string)}
	name := resp["name"].(string)
	qs := o.QueryTable("user")
	err := qs.Filter("name",name).One(&user)
	if err !=nil {
		resp["msg"]="数据错误"
		logs.Info("2222name=",resp["name"],"password:",resp["password"])
		return
	}
	if user.Password != resp["password"] {
		resp["msg"]="用户名或密码错误"
		logs.Info("3333name=",resp["name"],"password:",resp["password"])
		return
	}
	//添加Session
	c.SetSession("name",resp["name"])
	c.SetSession("user_id",user.Id)
	//返回json数据给前端
	resp["msg"]="登录成功"
}
func (c *SessionController) RetData(resp map[string]interface{}) {
	c.Data["json"] =resp
	c.ServeJSON()
}

models包

  userModel.go

import (
	"github.com/astaxie/beego/orm"
	_ "github.com/go-sql-driver/mysql"
)

type User struct {
	Id       int    `json:"id"`
	Name string `json:"name"`
	Password string `json:"password"`
}

func init() {
	// 映射model数据
	orm.RegisterModel(new(User))
}

router包

  router.go

func init() {
	beego.Router("/user/session",&controllers.SessionController{},"get:GetSessionData;delete:DeleteSessionData")
	beego.Router("/user/sessions",&controllers.SessionController{},"post:Login")
	beego.Router("/user/users",&controllers.UserController{},"post:Reg")
}

main.go

import (
	_ "demo03.cn/routers"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/context"
	"github.com/astaxie/beego/logs"
	"github.com/astaxie/beego/orm"
	"net/http"
	"strings"
)
//

func main() {
	ignoreStaticPath()
	beego.Run()
}

func ignoreStaticPath()  {
	beego.InsertFilter("/",beego.BeforeRouter,TransparentStatic)
	beego.InsertFilter("/*",beego.BeforeRouter,TransparentStatic)
}

func TransparentStatic(ctx *context.Context){
	// 获取请求的url
	orpath := ctx.Request.URL.Path
	logs.Info("request url:",orpath)
	//如果请求url包含api字段，说明指令应该取消静态资源重定向
	if strings.Index(orpath,"user") >= 0{
		return
	}
	// 静态资源重定向
	//http.ServeFile(ctx.ResponseWriter,ctx.Request,"static/html"+ctx.Request.URL.Path)
	http.ServeFile(ctx.ResponseWriter,ctx.Request,"static/img/img")
}

func init() {
	// 设置数据库基本信息，相当于连接数据库
	_ = orm.RegisterDataBase("default", "mysql", beego.AppConfig.String("dataSource"), 30)
	// 生成表
	_ = orm.RunSyncdb("default", false, true)
}