SunYF
4ec6784629
本次提交包含以下核心变更： 1. 新增按钮级权限控制框架与前端权限校验 2. 新增工位管理、工艺流程、巡检终端等核心业务模块 3. 完善用户体系，支持工位终端专属登录权限 4. 新增文件上传下载、报表查询等配套功能 5. 修复多系统兼容性与交互体验问题 6. 完成所有文档与配置项的规范化更新
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package tokenx
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/golang-jwt/jwt/v4"
|
|
)
|
|
|
|
// Claims JWT 自定义负载
|
|
type Claims struct {
|
|
UserId int `json:"userId"`
|
|
Username string `json:"username"`
|
|
Name string `json:"name"`
|
|
RoleId int `json:"roleId"`
|
|
RoleCode string `json:"roleCode"`
|
|
RoleName string `json:"roleName"`
|
|
StationNo int `json:"stationNo,omitempty"` // 工位终端登录时的工位号
|
|
jwt.RegisteredClaims
|
|
}
|
|
|
|
// Issue 签发 JWT
|
|
func Issue(secret string, expire time.Duration, c Claims) (string, error) {
|
|
c.IssuedAt = jwt.NewNumericDate(time.Now())
|
|
c.ExpiresAt = jwt.NewNumericDate(time.Now().Add(expire))
|
|
c.Subject = fmt.Sprintf("%d", c.UserId)
|
|
t := jwt.NewWithClaims(jwt.SigningMethodHS256, c)
|
|
return t.SignedString([]byte(secret))
|
|
}
|
|
|
|
// Parse 校验 JWT 并返回负载
|
|
func Parse(secret, tokenStr string) (*Claims, error) {
|
|
token, err := jwt.ParseWithClaims(tokenStr, &Claims{}, func(t *jwt.Token) (any, error) {
|
|
if _, ok := t.Method.(*jwt.SigningMethodHMAC); !ok {
|
|
return nil, errors.New("unexpected signing method")
|
|
}
|
|
return []byte(secret), nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if claims, ok := token.Claims.(*Claims); ok && token.Valid {
|
|
return claims, nil
|
|
}
|
|
return nil, errors.New("invalid token")
|
|
} |