如何在java主方法中使用安全框架

在Java主方法中使用安全框架，首先需要将安全框架添加到项目的依赖中。这里以使用Apache Shiro为例，介绍如何在Java主方法中使用安全框架。

1. 添加Apache Shiro依赖

在Maven项目的pom.xml文件中添加以下依赖：

<dependencies>
    <!-- Apache Shiro Core -->
    <dependency>
        <groupId>org.apache.shiro</groupId>
        <artifactId>shiro-core</artifactId>
        <version>1.7.1</version>
    </dependency>
    <!-- Apache Shiro Spring -->
    <dependency>
        <groupId>org.apache.shiro</groupId>
        <artifactId>shiro-spring</artifactId>
        <version>1.7.1</version>
    </dependency>
    <!-- Apache Shiro Spring Boot Starter -->
    <dependency>
        <groupId>org.apache.shiro</groupId>
        <artifactId>shiro-spring-boot-starter</artifactId>
        <version>1.7.1</version>
    </dependency>
</dependencies>

2. 配置Shiro

在项目的resources目录下创建一个名为shiro.ini的配置文件，并添加以下内容：

[main]
# 配置安全管理器
securityManager = org.apache.shiro.mgt.DefaultSecurityManager

# 配置Realm
myRealm = com.example.shiro.MyRealm
securityManager.realms = $myRealm

# 配置缓存管理器
cacheManager = org.apache.shiro.cache.MemoryConstrainedCacheManager
securityManager.cacheManager = $cacheManager

[urls]
# 配置URL权限控制
/admin/** = authc
/user/** = authc, roles[admin]

3. 创建自定义Realm

在项目中创建一个名为MyRealm的类，继承org.apache.shiro.realm.AuthorizingRealm，并实现doGetAuthenticationInfo和doGetAuthorizationInfo方法：

package com.example.shiro;

import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;

public class MyRealm extends AuthorizingRealm {

    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
        // 获取用户名和密码
        String username = (String) token.getPrincipal();
        String password = new String((char[]) token.getCredentials());

        // 根据用户名和密码查询用户信息
        User user = getUserByUsernameAndPassword(username, password);

        if (user == null) {
            throw new UnknownAccountException("用户名或密码错误");
        }

        return new SimpleAuthenticationInfo(user.getUsername(), user.getPassword(), getName());
    }

    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
        // 获取用户角色和权限
        String username = (String) principals.getPrimaryPrincipal();
        User user = getUserByUsername(username);

        if (user == null) {
            return null;
        }

        SimpleAuthorizationInfo authorizationInfo = new SimpleAuthorizationInfo();
        authorizationInfo.addRole("admin");
        authorizationInfo.addStringPermission("user:view");

        return authorizationInfo;
    }

    private User getUserByUsernameAndPassword(String username, String password) {
        // 根据用户名和密码查询用户信息，这里需要实现具体的查询逻辑
        return new User(username, password);
    }

    private User getUserByUsername(String username) {
        // 根据用户名查询用户信息，这里需要实现具体的查询逻辑
        return new User(username);
    }
}

4. 创建User类

创建一个名为User的类，包含用户名、密码、角色和权限等属性：

public class User {
    private String username;
    private String password;
    private String role;
    private String permission;

    public User(String username) {
        this.username = username;
    }

    public User(String username, String password) {
        this.username = username;
        this.password = password;
    }

    // getter和setter方法
}

5. 编写Java主方法

在主方法中，使用Shiro的SecurityUtils获取当前用户，并根据用户角色和权限进行相应的操作：

package com.example;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.subject.Subject;

public class Main {
    public static void main(String[] args) {
        // 获取当前用户
        Subject currentUser = SecurityUtils.getSubject();

        // 登录
        currentUser.login(new UsernamePasswordToken("admin", "123456"));

        // 获取当前用户的角色和权限
        String username = (String) currentUser.getPrincipal();
        User user = getUserByUsername(username);

        if (user.getRole().equals("admin")) {
            System.out.println("你是管理员，拥有查看用户权限");
        } else if (user.getPermission().equals("user:view")) {
            System.out.println("你拥有查看用户权限");
        } else {
            System.out.println("你没有查看用户权限");
        }

        // 登出
        currentUser.logout();
    }

    private static User getUserByUsername(String username) {
        // 根据用户名查询用户信息，这里需要实现具体的查询逻辑
        return new User(username);
    }
}

这样，在Java主方法中就可以使用Apache Shiro安全框架进行用户认证和授权了。