41 lines
969 B
Go
41 lines
969 B
Go
package helpers
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/goravel/framework/contracts/http"
|
|
|
|
"goravel/app/models"
|
|
)
|
|
|
|
// GetAdminFromContext 从 context 中获取 admin 对象
|
|
// 如果 context 中没有 admin 或类型不匹配，返回错误
|
|
func GetAdminFromContext(ctx http.Context) (*models.Admin, error) {
|
|
adminValue := ctx.Value("admin")
|
|
if adminValue == nil {
|
|
return nil, errors.New("admin not found in context")
|
|
}
|
|
|
|
// 尝试值类型
|
|
if admin, ok := adminValue.(models.Admin); ok {
|
|
return &admin, nil
|
|
}
|
|
|
|
// 尝试指针类型
|
|
if adminPtr, ok := adminValue.(*models.Admin); ok {
|
|
return adminPtr, nil
|
|
}
|
|
|
|
return nil, errors.New("invalid admin type in context")
|
|
}
|
|
|
|
// GetAdminIDFromContext 从 context 中获取 admin ID
|
|
// 如果 context 中没有 admin 或类型不匹配，返回 0 和错误
|
|
func GetAdminIDFromContext(ctx http.Context) (uint, error) {
|
|
admin, err := GetAdminFromContext(ctx)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return admin.ID, nil
|
|
}
|