io.github.ye17186.myhelper.token.MhTokenService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of myhelper-token-spring-boot-starter Show documentation

There is a newer version: 0.0.6

Show newest version

package io.github.ye17186.myhelper.token;

import cn.dev33.satoken.stp.SaTokenInfo;
import cn.dev33.satoken.stp.StpUtil;
import io.github.ye17186.myhelper.token.model.LoginKey;
import io.github.ye17186.myhelper.token.model.TokenInfo;

import java.util.List;

/**
 * @author ye17186
 * @since 2023-03-03
 */
public class MhTokenService {

    /**
     * 用户登录
     *
     * @param key 登录唯一标识
     */
    public TokenInfo login(LoginKey key) {

        StpUtil.login(key.format());
        return toToken(StpUtil.getTokenInfo(), key);
    }

    /**
     * 用户登出
     */
    public void logout() {

        StpUtil.logout();
    }

    /**
     * 指定用户登出
     */
    public void logout(LoginKey key) {

        StpUtil.logout(key.format());
    }

    /**
     * 获取当前登录用户的登录标识
     *
     * @return 登录标识
     */
    public LoginKey getLoginKey() {

        return LoginKey.decode(StpUtil.getLoginIdAsString());
    }

    /**
     * 当前状态是否已登录
     */
    public boolean isLogin() {

        return StpUtil.isLogin();
    }

    /**
     * 获取当前用户的权限码集合
     *
     * @return 权限码集合
     */
    public List permissions() {

        return StpUtil.getPermissionList();
    }

    /**
     * 获取当前用户的角色码集合
     *
     * @return
     */
    public List roles() {

        return StpUtil.getRoleList();
    }

    public boolean hasPermission(String permission) {

        return StpUtil.hasPermission(permission);
    }

    public boolean hasRole(String role) {

        return StpUtil.hasRole(role);
    }

    private TokenInfo toToken(SaTokenInfo info, LoginKey key) {

        TokenInfo token = new TokenInfo();
        token.setLoginType(key.getLoginType());
        token.setTokenName(info.getTokenName());
        token.setTokenValue(info.getTokenValue());
        token.setExpiredAt(System.currentTimeMillis() + info.getTokenTimeout() * 1000);
        return token;
    }
}