123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- package utils
- import (
- "github.com/dgrijalva/jwt-go"
- "github.com/gin-gonic/gin"
- "strconv"
- "time"
- )
- var jwtSecret = []byte("martin")
- type Claims struct {
- Username string `json:"username"`
- Rolename string `json:"rolename"`
- UserId string `json:"userid"`
- Id int `json:"id"`
- jwt.StandardClaims
- }
- func GenerateToken(username, rolename, userid string, id int) (string, error) {
- nowTime := time.Now()
- expireTime := nowTime.Add(3 * time.Hour)
- claims := Claims{
- username,
- rolename,
- userid,
- id,
- jwt.StandardClaims{
- ExpiresAt: expireTime.Unix(),
- Issuer: "gin-blog",
- },
- }
- tokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
- token, err := tokenClaims.SignedString(jwtSecret)
- return token, err
- }
- func ParseToken(token string) (*Claims, error) {
- tokenClaims, err := jwt.ParseWithClaims(token, &Claims{}, func(token *jwt.Token) (interface{}, error) {
- return jwtSecret, nil
- })
- if tokenClaims != nil {
- if claims, ok := tokenClaims.Claims.(*Claims); ok && tokenClaims.Valid {
- return claims, nil
- }
- }
- return nil, err
- }
- func GetId(c *gin.Context) int64 {
- valueString := c.GetString("CONTEXT_ID")
- val, _ := strconv.ParseInt(valueString, 10, 64)
- return val
- }
- func GetUserId(c *gin.Context) int64 {
- valueString := c.GetString("CONTEXT_USER_ID")
- val, _ := strconv.ParseInt(valueString, 10, 64)
- return val
- }
- func GetRoleName(c *gin.Context) string {
- return c.GetString("CONTEXT_ROLE_NAME")
- }
|