118 lines
2.2 KiB
Go
118 lines
2.2 KiB
Go
package middleware
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/gin-gonic/gin"
|
||
"io"
|
||
"net/http"
|
||
"time"
|
||
"xiawan/wx/db"
|
||
)
|
||
|
||
/*
|
||
**
|
||
认证
|
||
*/
|
||
func BasicAuth() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
auth := c.Request.Header.Get("Authorization")
|
||
if auth == "" {
|
||
RsqK(c, "授权码不能为空")
|
||
return
|
||
}
|
||
modelAuth := &Auth{}
|
||
var key = fmt.Sprintf("api:auth:%s", auth)
|
||
modelAuth = GetCacheAuth(key)
|
||
if modelAuth == nil {
|
||
rsq := Get("http://119.45.28.143:8000/api/v1/auth?key=" + auth)
|
||
if err := json.Unmarshal([]byte(rsq), &modelAuth); err == nil {
|
||
if modelAuth.Data == 200 {
|
||
CacheAuthAdd(key, modelAuth)
|
||
c.Next()
|
||
return
|
||
}
|
||
RsqK(c, modelAuth.Msg)
|
||
return
|
||
}
|
||
} else {
|
||
if modelAuth.Data == 200 {
|
||
CacheAuthAdd(key, modelAuth)
|
||
c.Next()
|
||
return
|
||
}
|
||
RsqK(c, modelAuth.Msg)
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
func RsqK(c *gin.Context, msg string) {
|
||
c.AbortWithStatus(http.StatusCreated)
|
||
respVo := &RespVo{
|
||
Code: http.StatusOK,
|
||
Message: msg,
|
||
Data: http.StatusCreated,
|
||
}
|
||
c.JSON(http.StatusOK, respVo)
|
||
}
|
||
|
||
type Auth struct {
|
||
Code int32
|
||
Data int32
|
||
Msg string
|
||
}
|
||
|
||
// {"code":0,"message":"","data":""}
|
||
type RespVo struct {
|
||
Code int `json:"code"`
|
||
Message string `json:"message"`
|
||
Data interface{} `json:"data"`
|
||
}
|
||
|
||
// 发送GET请求
|
||
// url： 请求地址
|
||
// response： 请求返回的内容
|
||
func Get(url string) string {
|
||
// 超时时间：15秒
|
||
client := &http.Client{Timeout: 15 * time.Second}
|
||
resp, err := client.Get(url)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
defer resp.Body.Close()
|
||
var buffer [512]byte
|
||
result := bytes.NewBuffer(nil)
|
||
for {
|
||
n, err := resp.Body.Read(buffer[0:])
|
||
result.Write(buffer[0:n])
|
||
if err != nil && err == io.EOF {
|
||
break
|
||
} else if err != nil {
|
||
panic(err)
|
||
}
|
||
}
|
||
return result.String()
|
||
}
|
||
|
||
// 添加授权缓存
|
||
func CacheAuthAdd(key string, val *Auth) {
|
||
cache := db.GetCaChe()
|
||
isContains := cache.Contains(key)
|
||
if !isContains {
|
||
cache.Set(key, val, time.Hour*1)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 查询授权缓存
|
||
func GetCacheAuth(key string) *Auth {
|
||
cache := db.GetCaChe()
|
||
isContains := cache.Contains(key)
|
||
if !isContains {
|
||
return nil
|
||
}
|
||
return cache.Get(key).(*Auth)
|
||
}
|