java项目接口太多如何做接口权限

原创

mob64ca12d7c9ee 2023-11-23 04:18:23 ©著作权

文章标签 java 权限管理 ide 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d7c9ee的原创作品，请联系作者获取转载授权，否则将追究法律责任

项目方案：Java项目接口权限管理

背景

在Java项目开发中，随着项目规模的扩大和功能的增加，项目中会存在大量的接口。为了保证系统的安全性和数据的完整性，需要对接口进行权限管理。本文将介绍一种基于角色的接口权限管理方案。

方案概述

我们将使用Spring Security框架来实现接口权限管理。通过定义角色和权限，然后将角色与用户关联起来，来控制用户对接口的访问权限。

实施步骤

步骤一：定义接口权限

在项目中，我们需要定义接口权限。可以根据项目需求，将接口权限分为不同的角色，例如管理员、普通用户等。

public enum Permission {
    READ("read"),
    WRITE("write"),
    DELETE("delete");

    private final String permission;

    Permission(String permission) {
        this.permission = permission;
    }

    public String getPermission() {
        return permission;
    }
}

步骤二：定义用户角色

在项目中，我们需要定义用户角色，例如管理员、普通用户等。可以使用Spring Security提供的GrantedAuthority接口来表示角色。

public enum Role implements GrantedAuthority {
    ADMIN("ROLE_ADMIN"),
    USER("ROLE_USER");

    private final String role;

    Role(String role) {
        this.role = role;
    }

    @Override
    public String getAuthority() {
        return role;
    }
}

步骤三：配置Spring Security

在项目的配置文件中，配置Spring Security，定义权限控制规则。

@Configuration
@EnableWebSecurity
public class SecurityConfig extends WebSecurityConfigurerAdapter {

    @Autowired
    private UserDetailsService userDetailsService;

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http.authorizeRequests()
                .antMatchers("/admin/**").hasRole(Role.ADMIN.getAuthority())
                .antMatchers("/user/**").hasRole(Role.USER.getAuthority())
                .anyRequest().authenticated()
                .and()
                .formLogin()
                .and()
                .httpBasic();
    }

    @Override
    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
        auth.userDetailsService(userDetailsService);
    }
}

步骤四：实现用户认证和授权

在项目中，我们需要实现用户认证和授权的逻辑。可以通过继承UserDetailsService接口，并实现loadUserByUsername方法来查询用户信息和角色信息。

@Service
public class UserDetailsServiceImpl implements UserDetailsService {

    @Autowired
    private UserRepository userRepository;

    @Override
    public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
        User user = userRepository.findByUsername(username);
        if (user == null) {
            throw new UsernameNotFoundException("User not found");
        }
        return new org.springframework.security.core.userdetails.User(user.getUsername(), user.getPassword(),
                AuthorityUtils.createAuthorityList(user.getRoles().stream()
                        .map(Role::getAuthority)
                        .toArray(String[]::new)));
    }
}

步骤五：应用接口权限

在项目中，我们可以通过@PreAuthorize注解来应用接口权限。根据接口的访问权限要求，可以在方法或类级别上使用该注解。

@RestController
@RequestMapping("/admin")
@PreAuthorize("hasRole('ROLE_ADMIN')")
public class AdminController {
    // 管理员接口，需要ROLE_ADMIN角色才能访问

    @GetMapping("/users")
    public List<User> getUsers() {
        return userRepository.findAll();
    }
}

@RestController
@RequestMapping("/user")
@PreAuthorize("hasRole('ROLE_USER')")
public class UserController {
    // 普通用户接口，需要ROLE_USER角色才能访问

    @GetMapping("/profile")
    public User getProfile() {
        String username = SecurityContextHolder.getContext().getAuthentication().getName();
        return userRepository.findByUsername(username);
    }
}