• 14.4 用户认证
    • HTTP Basic和 HTTP Digest认证
    • oauth和oauth2的认证
    • links

    14.4 用户认证

    在开发Web应用过程中，用户认证是开发者经常遇到的问题，用户登录、注册、登出等操作，而一般认证也分为三个方面的认证

    • HTTP Basic和 HTTP Digest认证
    • 第三方集成认证：QQ、微博、豆瓣、OPENID、google、github、facebook和twitter等
    • 自定义的用户登录、注册、登出，一般都是基于session、cookie认证

    beego目前没有针对这三种方式进行任何形式的集成，但是可以充分的利用第三方开源库来实现上面的三种方式的用户认证，不过后续beego会对前面两种认证逐步集成。

    HTTP Basic和 HTTP Digest认证

    这两个认证是一些应用采用的比较简单的认证，目前已经有开源的第三方库支持这两个认证：

    2. github.com/abbot/go-http-auth

    下面代码演示了如何把这个库引入beego中从而实现认证：

    2. package controllers
    4. import (
    5.     "github.com/abbot/go-http-auth"
    6.     "github.com/astaxie/beego"
    7. )
    9. func Secret(user, realm string) string {
    10.     if user == "john" {
    11.         // password is "hello"
    12.         return "$1$dlPL2MqE$oQmn16q49SqdmhenQuNgs1"
    13.     }
    14.     return ""
    15. }
    17. type MainController struct {
    18.     beego.Controller
    19. }
    21. func (this *MainController) Prepare() {
    22.     a := auth.NewBasicAuthenticator("example.com", Secret)
    23.     if username := a.CheckAuth(this.Ctx.Request); username == "" {
    24.         a.RequireAuth(this.Ctx.ResponseWriter, this.Ctx.Request)
    25.     }
    26. }
    28. func (this *MainController) Get() {
    29.     this.Data["Username"] = "astaxie"
    30.     this.Data["Email"] = "astaxie@gmail.com"
    31.     this.TplNames = "index.tpl"
    32. }

    上面代码利用了beego的prepare函数，在执行正常逻辑之前调用了认证函数，这样就非常简单的实现了http auth，digest的认证也是同样的原理。

    oauth和oauth2的认证

    oauth和oauth2是目前比较流行的两种认证方式，还好第三方有一个库实现了这个认证，但是是国外实现的，并没有QQ、微博之类的国内应用认证集成：

    2. github.com/bradrydzewski/go.auth

    下面代码演示了如何把该库引入beego中从而实现oauth的认证，这里以github为例演示：

    1. 添加两条路由
      ```Go

    beego.RegisterController(“/auth/login”, &controllers.GithubController{})
    beego.RegisterController(“/mainpage”, &controllers.PageController{})

    1. 2. 然后我们处理GithubController登陆的页面：
    2. ```Go
    3. package controllers
    5. import (
    6.     "github.com/astaxie/beego"
    7.     "github.com/bradrydzewski/go.auth"
    8. )
    10. const (
    11.     githubClientKey = "a0864ea791ce7e7bd0df"
    12.     githubSecretKey = "a0ec09a647a688a64a28f6190b5a0d2705df56ca"
    13. )
    15. type GithubController struct {
    16.     beego.Controller
    17. }
    19. func (this *GithubController) Get() {
    20.     // set the auth parameters
    21.     auth.Config.CookieSecret = []byte("7H9xiimk2QdTdYI7rDddfJeV")
    22.     auth.Config.LoginSuccessRedirect = "/mainpage"
    23.     auth.Config.CookieSecure = false
    25.     githubHandler := auth.Github(githubClientKey, githubSecretKey)
    27.     githubHandler.ServeHTTP(this.Ctx.ResponseWriter, this.Ctx.Request)
    28. }
    1. 处理登陆成功之后的页面
      ```Go
      package controllers

    import (
    “github.com/astaxie/beego”
    “github.com/bradrydzewski/go.auth”
    “net/http”
    “net/url”
    )

    type PageController struct {
    beego.Controller
    }

    func (this *PageController) Get() {
    // set the auth parameters
    auth.Config.CookieSecret = []byte(“7H9xiimk2QdTdYI7rDddfJeV”)
    auth.Config.LoginSuccessRedirect = “/mainpage”
    auth.Config.CookieSecure = false

    1. user, err := auth.GetUserCookie(this.Ctx.Request)
    3. //if no active user session then authorize user
    4. if err != nil || user.Id() == "" {
    5.     http.Redirect(this.Ctx.ResponseWriter, this.Ctx.Request, auth.Config.LoginRedirect, http.StatusSeeOther)
    6.     return
    7. }
    9. //else, add the user to the URL and continue
    10. this.Ctx.Request.URL.User = url.User(user.Id())
    11. this.Data["pic"] = user.Picture()
    12. this.Data["id"] = user.Id()
    13. this.Data["name"] = user.Name()
    14. this.TplNames = "home.tpl"

    }

    1. 整个的流程如下，首先打开浏览器输入地址：
    3. ![](images/14.4.github.png?raw=true)
    5. 图14.4 显示带有登录按钮的首页
    7. 然后点击链接出现如下界面：
    9. ![](images/14.4.github2.png?raw=true)
    11. 图14.5 点击登录按钮后显示github的授权页
    13. 然后点击Authorize app就出现如下界面：
    15. ![](images/14.4.github3.png?raw=true)
    17. 图14.6 授权登录之后显示的获取到的github信息页
    19. ## 自定义认证
    20. 自定义的认证一般都是和session结合验证的，如下代码来源于一个基于beego的开源博客：
    21. ```Go
    23. //登陆处理
    24. func (this *LoginController) Post() {
    25.     this.TplNames = "login.tpl"
    26.     this.Ctx.Request.ParseForm()
    27.     username := this.Ctx.Request.Form.Get("username")
    28.     password := this.Ctx.Request.Form.Get("password")
    29.     md5Password := md5.New()
    30.     io.WriteString(md5Password, password)
    31.     buffer := bytes.NewBuffer(nil)
    32.     fmt.Fprintf(buffer, "%x", md5Password.Sum(nil))
    33.     newPass := buffer.String()
    35.     now := time.Now().Format("2006-01-02 15:04:05")
    37.     userInfo := models.GetUserInfo(username)
    38.     if userInfo.Password == newPass {
    39.         var users models.User
    40.         users.Last_logintime = now
    41.         models.UpdateUserInfo(users)
    43.         //登录成功设置session
    44.         sess := globalSessions.SessionStart(this.Ctx.ResponseWriter, this.Ctx.Request)
    45.         sess.Set("uid", userInfo.Id)
    46.         sess.Set("uname", userInfo.Username)
    48.         this.Ctx.Redirect(302, "/")
    49.     }    
    50. }
    52. //注册处理
    53. func (this *RegController) Post() {
    54.     this.TplNames = "reg.tpl"
    55.     this.Ctx.Request.ParseForm()
    56.     username := this.Ctx.Request.Form.Get("username")
    57.     password := this.Ctx.Request.Form.Get("password")
    58.     usererr := checkUsername(username)
    59.     fmt.Println(usererr)
    60.     if usererr == false {
    61.         this.Data["UsernameErr"] = "Username error, Please to again"
    62.         return
    63.     }
    65.     passerr := checkPassword(password)
    66.     if passerr == false {
    67.         this.Data["PasswordErr"] = "Password error, Please to again"
    68.         return
    69.     }
    71.     md5Password := md5.New()
    72.     io.WriteString(md5Password, password)
    73.     buffer := bytes.NewBuffer(nil)
    74.     fmt.Fprintf(buffer, "%x", md5Password.Sum(nil))
    75.     newPass := buffer.String()
    77.     now := time.Now().Format("2006-01-02 15:04:05")
    79.     userInfo := models.GetUserInfo(username)
    81.     if userInfo.Username == "" {
    82.         var users models.User
    83.         users.Username = username
    84.         users.Password = newPass
    85.         users.Created = now
    86.         users.Last_logintime = now
    87.         models.AddUser(users)
    89.         //登录成功设置session
    90.         sess := globalSessions.SessionStart(this.Ctx.ResponseWriter, this.Ctx.Request)
    91.         sess.Set("uid", userInfo.Id)
    92.         sess.Set("uname", userInfo.Username)
    93.         this.Ctx.Redirect(302, "/")
    94.     } else {
    95.         this.Data["UsernameErr"] = "User already exists"
    96.     }
    98. }
    100. func checkPassword(password string) (b bool) {
    101.     if ok, _ := regexp.MatchString("^[a-zA-Z0-9]{4,16}$", password); !ok {
    102.         return false
    103.     }
    104.     return true
    105. }
    107. func checkUsername(username string) (b bool) {
    108.     if ok, _ := regexp.MatchString("^[a-zA-Z0-9]{4,16}$", username); !ok {
    109.         return false
    110.     }
    111.     return true
    112. }

    有了用户登陆和注册之后，其他模块的地方可以增加如下这样的用户是否登陆的判断：

    2. func (this *AddBlogController) Prepare() {
    3.     sess := globalSessions.SessionStart(this.Ctx.ResponseWriter, this.Ctx.Request)
    4.     sess_uid := sess.Get("userid")
    5.     sess_username := sess.Get("username")
    6.     if sess_uid == nil {
    7.         this.Ctx.Redirect(302, "/admin/login")
    8.         return
    9.     }
    10.     this.Data["Username"] = sess_username
    11. }
    • 目录
    • 上一节: 表单及验证支持
    • 下一节: 多语言支持