beego入門之登陸程序（sql數據庫操作、session操作、前後端數據交互）

原創

cwnu-abin

2018-12-09 20:58

此次更新內容爲go語言框架beego的入門程序，該程序能夠實現簡單的登陸功能，包括的內容有sql數據庫操作，session操作、數據交互，適合beego入門時琢磨。

前提是go語言環境已經安裝好，集成開發環境我用的GoLand

1、安裝需要用到的插件：

go get github.com/astaxie/beego    //安裝beego
go get github.com/beego/bee        //安裝bee工具
go get github.com/astaxie/beedb    //安裝beedb
go get github.com/Go-SQL-Driver/MySQL    //安裝mysql驅動文件

【1】beego框架，詳情https://beego.me/

【2】bee工具是一個爲了協助快速開發beego項目而創建的項目，可以通過bee快速創建項目、實現熱編譯、開發測試以及開發完之後打包發佈的一整套從創建、開發到部署的方案。

【3】beedb框架，用於操作數據庫，詳情https://github.com/astaxie/beedb

【4】mysql數據庫驅動，如果使用其他數據庫參考https://github.com/astaxie/beedb

2、使用bee工具新建一個項目：

bee new Test

使用GoLand打開項目，大概目錄如下（修改後）：

項目目錄如下：

3、創建數據庫、插入測試數據

DROP TABLE IF EXISTS `admin_user`;
CREATE TABLE `admin_user` (
  `id` int(3) unsigned NOT NULL AUTO_INCREMENT,
  `username` varchar(32) NOT NULL,
  `password` varchar(32) NOT NULL,
  `email` varchar(32) DEFAULT NULL,
  `author` int(1) NOT NULL DEFAULT '1',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;

INSERT INTO `admin_user` VALUES ('1', 'admin', 'admin', '[email protected]', '1');

4、開發項目

下面是源文件代碼：

（1）app.config（先在數據庫創建test數據庫，根據需要修改）

appname = Test
httpport = 8080
runmode = dev

mysqluser = "root"
mysqlpass = "root"
mysqlurl = "127.0.0.1"
mysqlport = "3306"
mysqldbname   = "test"

sessionon = true

（2）dao.go（用於連接數據庫）

package models

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

var(
	user = beego.AppConfig.String("mysqluser")
	pass = beego.AppConfig.String("mysqlpass")
	url = beego.AppConfig.String("mysqlurl")
	port = beego.AppConfig.String("mysqlport")
	dbname = beego.AppConfig.String("mysqldbname")
)
var db_info = user+":"+pass+"@tcp("+url+":"+port+")/"+dbname+"?charset=utf8"

func getLink() beedb.Model {
	db, err := sql.Open("mysql", db_info)
	if err != nil {
		panic(err)
	}
	orm := beedb.New(db)
	return orm
}

（3）admin_user.go（用戶模型，包括操作數據庫）

package models

import "errors"

type AdminUser struct {
	Id int
	UserName string
	Password string
	Email string
	Author int
}

func (user AdminUser)ValidateUser() error {
	orm := getLink()    //獲得用於操作數據庫的orm
	var u AdminUser
	orm.Where("username=? and password=?", user.UserName, user.Password).Find(&u)
	if u.UserName == "" {
		return errors.New("用戶名或密碼錯誤！")
	}
	return nil
}

（4）router.go（路由設置）

package routers

import (
	"Test/controllers"
	"github.com/astaxie/beego"
)

func init() {
    beego.Router("/", &controllers.MainController{})
	beego.Router("/login", &controllers.MainController{},"get:Login;post:LoginCheck")
	beego.Router("/out", &controllers.MainController{},"post:Out")

}

beego.Router("/login", &controllers.MainController{},"get:Login;post:LoginCheck")的意思是，當請求的連接爲“/login”時，如果請求方式爲Get，則調用controllers包裏面的Login方法，如果請求方式爲Post，則調用controllers包裏面的LoginCheck方法。

（5）default.go（用於判斷用戶是否已經登錄，如果已經登錄 [存在對應session]，則跳轉到主頁，如果沒有登錄，則跳轉到登錄界面）

package controllers

import (
	"Test/models"
	"github.com/astaxie/beego"
)

type MainController struct {
	beego.Controller
}


func (this *MainController) Get() {
	sess := this.GetSession("name")    //判斷此次會話的session是否已經存在
	if sess == nil{
		this.Redirect("/login",301)    //跳轉到登錄邏輯
	} else {
		user := sess.(models.AdminUser)
		this.Data["user"] = user.UserName    //用於向前端頁面傳送數據
		this.Data["pass"] = user.Password
		this.TplName = "succeed.html"    //渲染succeed.html頁面
	}
}

this.Data["user"] = user.UserName 用於向前端頁面傳輸數據，這樣在前端界面中可以通過 {{.user}} 的方式引用，注意有一個點。

（6）admin_user.go（用於實現與用戶有關的邏輯，包括登錄，退出等）

package controllers

import (
	"Test/models"
	"fmt"
)

func (c *MainController) Login() {
	c.TplName = "login.html"
}

func (c *MainController) LoginCheck() {
	var user models.AdminUser
	inputs := c.Input()
	user.UserName = inputs.Get("username")
	user.Password = inputs.Get("password")
	//fmt.Println(inputs)
	err := user.ValidateUser()
	if err == nil {
		c.SetSession("name", user)
		c.Redirect("/",301)
	} else {
		fmt.Println(err)
		c.Data["info"] = err
		c.TplName = "error.html"
	}
}

func (c *MainController) Out() {
	c.DelSession("name")
	c.Redirect("/",301)
}

（7）html頁面

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Login</title>
</head>
<body>
    <form name="login" action="/login" method="post">
        賬號：<input type="text" name="username"/><br>
        密碼：<input type="text" name="password"/><br>
        <input type="submit" value="登陸" />
    </form>
</body>
</html>

succeed.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Succeed</title>
</head>
<body>
    Login Succeed !

    Welcome !
    user:{{.user}},password:{{.pass}}
    <form name="logout" action="/out" method="post">
        <input type="submit" value="退出">
    </form>
</body>
</html>

error.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Error</title>
</head>
<body>
    Error!!!{{.info}}
    <form name="login" action="/login" method="get">
        <input type="submit" value="返回登錄界面">
    </form>
</body>
</html>

4、運行、測試

輸入以下命令運行：