hd/controllers/author.go

package controllers

import (
	"hudongzhuanjia/libs/jwt"
	"hudongzhuanjia/utils/code"
	"hudongzhuanjia/utils/define"

	jwt2 "github.com/dgrijalva/jwt-go"
)

//执行路由方法前校验登陆态，并且解析page、pageSize
type AuthorCtl struct {
	BaseCtl
	claims *jwt.Claims
}

func (t *AuthorCtl) Prepare() {
	t.BaseCtl.Prepare()
	skip, _ := t.GetInt("skip")
	if skip != 0 {
		t.claims = &jwt.Claims{
			AccountType:    "customer",
			AccountId:      skip,
			CustomerId:     1,
			CustomerPid:    0,
			ActivityId:     1,
			AreaId:         1,
			StandardClaims: jwt2.StandardClaims{},
		}
		return
	} else {
		token := ""
		if tokenStr, ok := t.Request.SESSION[define.TOKEN]; ok {
			token = tokenStr
		} else if tokenStr, ok = t.Request.REQUEST[define.TOKEN]; ok {
			token = tokenStr
		} else if tokenStr, ok = t.Request.HEADER[define.TOKEN]; ok {
			token = tokenStr
		} else {
			var param = make(map[string]interface{}, 0)
			err := t.RequestToStruct(&param)
			t.CheckErr(err)
			if tokenStr, ok := param[define.TOKEN]; ok {
				token = tokenStr.(string)
			}
		}
		claims, err := jwt.ParseAccessToken(token)
		if err != nil {
			t.ERROR("token 失效", code.MSG_ERR_Authority)
		}
		t.claims = claims
		// 最后多地区：子账号的area_id = area_id， 但是主账号的area_id 需要通过activity_id 进行获取
	}
}

func (t *AuthorCtl) GetAccountId() int {
	return t.claims.AccountId
}

func (t *AuthorCtl) GetAccountType() string {
	return t.claims.AccountType
}