package common
|
|
import (
|
"context"
|
"github.com/gin-gonic/gin"
|
"github.com/micro/go-micro/metadata"
|
"github.com/micro/go-micro/server"
|
"video-analysis-shop/config"
|
"video-analysis-shop/proto/user"
|
)
|
|
var (
|
userCli user.UserService
|
)
|
func GetUserClient() user.UserService {
|
if userCli == nil {
|
uSrvC := GetMicroClient(config.ServiceConf.User)
|
userCli = user.NewUserService(config.ServiceConf.User, uSrvC.Client())
|
}
|
return userCli
|
}
|
|
func AuthWrapper(fn server.HandlerFunc) server.HandlerFunc {
|
return func(ctx context.Context, req server.Request, resp interface{}) error {
|
//meta, ok := metadata.FromContext(ctx)
|
//if !ok {
|
// fmt.Println("no auth meta-data found in request")
|
// return errors.New("no auth meta-data found in request")
|
//}
|
|
//fmt.Println("meta:", meta)
|
|
//loginUserId := ""
|
|
//token := meta["token"]
|
//if token != "" {
|
// authResp, err := GetUserClient().ValidateToken(context.Background(), &user.ValidateTokenReq{
|
// Token: token,
|
// })
|
// fmt.Println("Auth Resp:", authResp, "err:", err)
|
//
|
// if err !=nil {
|
// return err
|
// }
|
// loginUserId = authResp.UserId
|
//} else {
|
// fmt.Println("metadata token is nil")
|
//}
|
|
//loginUserId = meta[CurrentUserIdKey]
|
//fmt.Println("meta CurrentUserId:", loginUserId)
|
|
//ctx = context.WithValue(ctx, "current_user_id", loginUserId)
|
|
return fn(ctx, req, resp)
|
}
|
}
|
|
const (
|
CurrentUserIdKey = "Current_user_id"
|
)
|
|
// ContextWithSpan 返回context
|
func ContextWithMeta(c *gin.Context) context.Context {
|
ctx := context.TODO()
|
md := make(map[string]string)
|
v, ok := c.Get(CurrentUserIdKey)
|
if ok {
|
md[CurrentUserIdKey] = v.(string)
|
} else {
|
md[CurrentUserIdKey] = ""
|
}
|
return metadata.NewContext(ctx, md)
|
}
|
|
func GetUserIdFromCtx(ctx context.Context) string {
|
meta, ok := metadata.FromContext(ctx)
|
if !ok {
|
return ""
|
}
|
return meta[CurrentUserIdKey]
|
}
|
