package controller
import ( "fmt" "go_openapi/app/config" "go_openapi/app/model" "net/http" "time"
"github.com/gin-gonic/gin" "github.com/golang-jwt/jwt")
// 生成一个 AccessTokenfunc AccessToken(c *gin.Context) { // 获取 AppKey 和 appSecret 参数 appKey := c.PostForm("app_key") if len(appKey) == 0 { c.JSON(http.StatusOK, gin.H{ "code": 400, "msg": "AppKey参数为空", }) return }
appSecret := c.PostForm("app_secret") if len(appSecret) == 0 { c.JSON(http.StatusOK, gin.H{ "code": 400, "msg": "appSecret参数为空", }) return }
// 在数据库中判断 AppKey 和 appSecret 是否存在 var tenant *model.Tenant dbRes := config.DemoDB.Model(&model.Tenant{}). Where("app_key = ?", appKey). Where("app_secret = ?", appSecret). First(&tenant) if dbRes.Error != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return }
// 生成一个 AccessToken expiresIn := int64(7 * 24 * 3600) // 7 天内有效 nowTime := time.Now().Unix()
jwtToken := jwt.New(jwt.SigningMethodHS256) claims := jwtToken.Claims.(jwt.MapClaims) claims["iss"] = "manongsen" // 签发者 可以为空 claims["aud"] = "tenant" // 面向的用户，可以为空 claims["iat"] = nowTime // 签发时间 claims["nbf"] = nowTime // 生效时间 claims["exp"] = nowTime + expiresIn // AccessToken 过期时间 accessToken, err := jwtToken.SignedString([]byte(tenant.AppSecret)) if err != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return }
scope := tenant.Scope data := map[string]interface{}{ "access_token": accessToken, // 访问令牌 "token_type": "bearer", // 令牌类型 "expires_in": expiresIn, // 过期时间，单位为秒 "scope": scope, // 权限范围 }
// 存储 AccessToken 到 Redis config.RedisConn.Set(fmt.Sprintf("%s.%s", config.ACCESS_TOKEN_PREFIX, accessToken), tenant.AppKey, time.Second*time.Duration(expiresIn)).Result() c.JSON(http.StatusOK, gin.H{ "code": 200, "msg": "ok", "data": data, })}
// 通过 AccessToken 换取 RefreshTokenfunc ExchangeToken(c *gin.Context) { // 获取 AccessToken 参数 accessToken := c.PostForm("access_token") if len(accessToken) == 0 { c.JSON(http.StatusOK, gin.H{ "code": 400, "msg": "AccessToken参数为空", }) return }
// 校验 AccessToken appKey, err := config.RedisConn.Get(fmt.Sprintf("%s.%s", config.ACCESS_TOKEN_PREFIX, accessToken)).Result() if err != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return } if len(appKey) == 0 { c.JSON(http.StatusOK, gin.H{ "code": 400, "msg": "AccessToken参数失效", }) return }
var tenant *model.Tenant dbRes := config.DemoDB.Model(&model.Tenant{}). Where("app_key = ?", appKey). First(&tenant) if dbRes.Error != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return }
expiresIn := int64(2 * 3600) // 2 小时内有效 nowTime := time.Now().Unix()
jwtToken := jwt.New(jwt.SigningMethodHS256) claims := jwtToken.Claims.(jwt.MapClaims) claims["iss"] = "manongsen" // 签发者 可以为空 claims["aud"] = "tenant" // 面向的用户，可以为空 claims["iat"] = nowTime // 签发时间 claims["nbf"] = nowTime // 生效时间 claims["exp"] = nowTime + expiresIn // RefreshToken 过期时间 refreshToken, err := jwtToken.SignedString([]byte(tenant.AppSecret)) if err != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return }
// 颁发 RefreshToken data := map[string]interface{}{ "refresh_token": refreshToken, // 刷新令牌 "expires_in": expiresIn, // 过期时间，单位为秒 }
// 存储到 Redis config.RedisConn.Set(fmt.Sprintf("%s.%s", config.REFRESH_TOKEN_PREFIX, refreshToken), appKey, time.Second*time.Duration(expiresIn)) c.JSON(http.StatusOK, gin.H{ "code": 200, "msg": "ok", "data": data, })}
// 刷新 RefreshTokenfunc RefreshToken(c *gin.Context) { // 获取 RefreshToken 参数 refreshToken := c.PostForm("refresh_token") if len(refreshToken) == 0 { c.JSON(http.StatusOK, gin.H{ "code": 400, "msg": "RefreshToken参数为空", }) return }
// 校验 RefreshToken appKey, err := config.RedisConn.Get(fmt.Sprintf("%s.%s", config.REFRESH_TOKEN_PREFIX, refreshToken)).Result() if err != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) } if len(appKey) == 0 { c.JSON(http.StatusOK, gin.H{ "code": 400, "msg": "AccessToken参数失效", }) return }
var tenant *model.Tenant dbRes := config.DemoDB.Model(&model.Tenant{}). Where("app_key = ?", appKey). First(&tenant) if dbRes.Error != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return }
// 颁发一个新的 RefreshToken expiresIn := int64(2 * 3600) // 2 小时内有效 nowTime := time.Now().Unix()
jwtToken := jwt.New(jwt.SigningMethodHS256) claims := jwtToken.Claims.(jwt.MapClaims) claims["iss"] = "manongsen" // 签发者 可以为空 claims["aud"] = "tenant" // 面向的用户，可以为空 claims["iat"] = nowTime // 签发时间 claims["nbf"] = nowTime // 生效时间 claims["exp"] = nowTime + expiresIn // RefreshToken 过期时间 newRefreshToken, err := jwtToken.SignedString([]byte(tenant.AppSecret)) if err != nil { c.JSON(http.StatusOK, gin.H{ "code": 500, "msg": "内部服务错误", }) return }
data := map[string]interface{}{ "refresh_token": newRefreshToken, // 新的刷新令牌 "expires_in": expiresIn, // 过期时间，单位为秒 }
// 将新的 RefreshToken 存储到 Redis config.RedisConn.Set(fmt.Sprintf("%s.%s", config.REFRESH_TOKEN_PREFIX, newRefreshToken), appKey, time.Second*time.Duration(expiresIn))
// 删除旧的 RefreshToken config.RedisConn.Del(fmt.Sprintf("%s.%s", config.REFRESH_TOKEN_PREFIX, refreshToken)) c.JSON(http.StatusOK, gin.H{ "code": 200, "msg": "ok", "data": data, })}
评论