package com.tourguide.bootstrap;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.tourguide.config.properties.BootstrapAdminProperties;
|
import com.tourguide.modules.user.domain.UserRole;
|
import com.tourguide.modules.user.entity.AdminUser;
|
import com.tourguide.modules.user.mapper.AdminUserMapper;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.boot.ApplicationArguments;
|
import org.springframework.boot.ApplicationRunner;
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
import org.springframework.stereotype.Component;
|
|
@Component
|
public class AdminBootstrapRunner implements ApplicationRunner {
|
private static final Logger LOGGER = LoggerFactory.getLogger(AdminBootstrapRunner.class);
|
private final BootstrapAdminProperties properties;
|
private final AdminUserMapper userMapper;
|
private final PasswordEncoder passwordEncoder;
|
|
public AdminBootstrapRunner(BootstrapAdminProperties properties, AdminUserMapper userMapper, PasswordEncoder passwordEncoder) {
|
this.properties = properties;
|
this.userMapper = userMapper;
|
this.passwordEncoder = passwordEncoder;
|
}
|
|
@Override
|
public void run(ApplicationArguments args) {
|
if (!properties.isEnabled()) { return; }
|
Long count = userMapper.selectCount(new LambdaQueryWrapper<AdminUser>().eq(AdminUser::getUsername, properties.getUsername()));
|
if (count > 0) { return; }
|
AdminUser admin = new AdminUser();
|
admin.setUsername(properties.getUsername());
|
admin.setPassword(passwordEncoder.encode(properties.getPassword()));
|
admin.setNickname(properties.getNickname());
|
admin.setRoleCode(UserRole.ADMIN.name());
|
admin.setStatus(1);
|
admin.setDeleted(0);
|
userMapper.insert(admin);
|
LOGGER.warn("已创建本地初始化管理员账号：{}，部署前请修改初始化密码", properties.getUsername());
|
}
|
}
|
