Golang中使用Osin实现OAuth2认证服务

Osin是一个用Go语言实现的OAuth2服务器库，它提供了完整的OAuth2授权框架实现。下面我将介绍如何使用Osin来构建一个OAuth2认证服务。

安装Osin

首先安装Osin库：

go get github.com/openshift/osin

基本配置

首先我们需要配置OAuth2服务器：

import (
    "github.com/openshift/osin"
    "github.com/openshift/osin/storage/memory"
)

func NewOAuthServer() *osin.Server {
    // 使用内存存储(生产环境应该使用数据库存储)
    storage := memory.NewMemoryStorage()
    
    // 配置OAuth服务器
    config := osin.NewServerConfig()
    config.AllowedAuthorizeTypes = osin.AllowedAuthorizeType{
        osin.CODE,
        osin.TOKEN,
    }
    config.AllowedAccessTypes = osin.AllowedAccessType{
        osin.AUTHORIZATION_CODE,
        osin.REFRESH_TOKEN,
        osin.PASSWORD,
        osin.CLIENT_CREDENTIALS,
    }
    config.AllowGetAccessRequest = true
    config.AllowClientSecretInParams = true
    
    server := osin.NewServer(config, storage)
    
    return server
}

添加客户端

在内存存储中添加一个测试客户端：

func AddTestClient(storage *memory.MemoryStorage) {
    client := &osin.DefaultClient{
        Id:          "1234",
        Secret:      "aabbccdd",
        RedirectUri: "http://localhost:14000/appauth",
    }
    storage.SetClient(client.Id, client)
}

实现授权端点

func AuthorizeHandler(w http.ResponseWriter, r *http.Request) {
    server := NewOAuthServer()
    
    // 解析授权请求
    resp := server.NewResponse()
    defer resp.Close()
    
    if ar := server.HandleAuthorizeRequest(resp, r); ar != nil {
        // 这里通常需要用户登录和授权确认
        // 示例中我们直接授权
        ar.Authorized = true
        server.FinishAuthorizeRequest(resp, r, ar)
    }
    
    if resp.IsError && resp.InternalError != nil {
        fmt.Printf("ERROR: %s\n", resp.InternalError)
    }
    
    osin.OutputJSON(resp, w, r)
}

实现令牌端点

func TokenHandler(w http.ResponseWriter, r *http.Request) {
    server := NewOAuthServer()
    
    resp := server.NewResponse()
    defer resp.Close()
    
    if ar := server.HandleAccessRequest(resp, r); ar != nil {
        ar.Authorized = true
        server.FinishAccessRequest(resp, r, ar)
    }
    
    if resp.IsError && resp.InternalError != nil {
        fmt.Printf("ERROR: %s\n", resp.InternalError)
    }
    
    osin.OutputJSON(resp, w, r)
}

完整服务器示例

package main

import (
    "fmt"
    "net/http"
    "github.com/openshift/osin"
    "github.com/openshift/osin/storage/memory"
)

func main() {
    // 初始化服务器
    srv := NewOAuthServer()
    AddTestClient(srv.Storage.(*memory.MemoryStorage))
    
    // 设置路由
    http.HandleFunc("/authorize", func(w http.ResponseWriter, r *http.Request) {
        AuthorizeHandler(w, r, srv)
    })
    http.HandleFunc("/token", func(w http.ResponseWriter, r *http.Request) {
        TokenHandler(w, r, srv)
    })
    
    // 启动服务器
    fmt.Println("Server is running at 14000 port.")
    fmt.Println("Authorize url: http://localhost:14000/authorize")
    fmt.Println("Token url: http://localhost:14000/token")
    http.ListenAndServe(":14000", nil)
}

使用密码授权示例

func PasswordHandler(w http.ResponseWriter, r *http.Request) {
    server := NewOAuthServer()
    
    resp := server.NewResponse()
    defer resp.Close()
    
    if r.Method == "POST" {
        if ar := server.HandleAccessRequest(resp, r); ar != nil {
            // 验证用户名和密码
            username := r.FormValue("username")
            password := r.FormValue("password")
            
            // 这里应该查询数据库验证用户
            if username == "test" && password == "test" {
                ar.Authorized = true
                ar.UserData = struct{ UserId string }{UserId: "123"}
                server.FinishAccessRequest(resp, r, ar)
            }
        }
    }
    
    osin.OutputJSON(resp, w, r)
}

刷新令牌

Osin自动处理刷新令牌，当使用refresh_token授权类型时，只需在令牌端点处理即可。

生产环境建议

1. 使用数据库存储替代内存存储
2. 实现自定义存储接口以持久化数据
3. 添加HTTPS支持
4. 实现更严格的客户端验证
5. 添加日志记录和监控

Osin提供了灵活的接口，可以轻松扩展以满足各种OAuth2需求。通过实现自定义存储接口，可以将其与任何数据库系统集成。

希望这个示例能帮助你开始使用Osin构建OAuth2认证服务！