package tech.powerjob.server.initializer;
|
|
import com.google.common.collect.Maps;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.RandomStringUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import tech.powerjob.common.serialize.JsonUtils;
|
import tech.powerjob.server.auth.PowerJobUser;
|
import tech.powerjob.server.auth.Role;
|
import tech.powerjob.server.auth.RoleScope;
|
import tech.powerjob.server.auth.common.AuthConstants;
|
import tech.powerjob.server.auth.service.login.LoginRequest;
|
import tech.powerjob.server.auth.service.login.PowerJobLoginService;
|
import tech.powerjob.server.auth.service.permission.PowerJobPermissionService;
|
import tech.powerjob.server.persistence.remote.model.NamespaceDO;
|
import tech.powerjob.server.persistence.remote.model.PwjbUserInfoDO;
|
import tech.powerjob.server.web.request.ModifyNamespaceRequest;
|
import tech.powerjob.server.web.request.ModifyUserInfoRequest;
|
import tech.powerjob.server.web.service.NamespaceWebService;
|
import tech.powerjob.server.web.service.PwjbUserWebService;
|
|
import javax.annotation.Resource;
|
import javax.transaction.Transactional;
|
import java.util.Map;
|
|
/**
|
* 初始化 PowerJob 首次部署相关的内容
|
* 为了可维护性足够高，统一使用 WEB 请求进行初始化，不直接操作底层，防止后续内部逻辑变更后出现问题
|
*
|
* @author tjq
|
* @since 2024/2/15
|
*/
|
@Slf4j
|
@Service
|
public class SystemInitializeServiceImpl implements SystemInitializeService {
|
|
@Value("${oms.auth.initiliaze.admin.password:#{null}}")
|
private String defaultAdminPassword;
|
@Resource
|
private PwjbUserWebService pwjbUserWebService;
|
@Resource
|
private NamespaceWebService namespaceWebService;
|
@Resource
|
private PowerJobLoginService powerJobLoginService;
|
@Resource
|
private PowerJobPermissionService powerJobPermissionService;
|
|
private static final String SYSTEM_ADMIN_NAME = "ADMIN";
|
|
|
private static final String SYSTEM_DEFAULT_NAMESPACE = "default_namespace";
|
|
@Override
|
@Transactional(rollbackOn = Exception.class)
|
public void initAdmin() {
|
|
String username = SYSTEM_ADMIN_NAME;
|
String password = StringUtils.isEmpty(defaultAdminPassword) ? RandomStringUtils.randomAlphabetic(8) : defaultAdminPassword;
|
|
// STEP1: 创建 PWJB 用户
|
ModifyUserInfoRequest createUser = new ModifyUserInfoRequest();
|
createUser.setUsername(username);
|
createUser.setNick(username);
|
createUser.setPassword(password);
|
|
log.info("[SystemInitializeService] [S1] create default PWJB user by request: {}", createUser);
|
PwjbUserInfoDO savedPwjbUser = pwjbUserWebService.save(createUser);
|
log.info("[SystemInitializeService] [S1] create default PWJB user successfully: {}", savedPwjbUser);
|
|
Map<String, Object> params = Maps.newHashMap();
|
params.put(AuthConstants.PARAM_KEY_USERNAME, username);
|
params.put(AuthConstants.PARAM_KEY_PASSWORD, password);
|
|
// STEP2: 创建 USER 对象
|
LoginRequest loginRequest = new LoginRequest()
|
.setLoginType(AuthConstants.ACCOUNT_TYPE_POWER_JOB)
|
.setOriginParams(JsonUtils.toJSONString(params));
|
log.info("[SystemInitializeService] [S2] createPowerJobUser user by request: {}", loginRequest);
|
PowerJobUser powerJobUser = powerJobLoginService.doLogin(loginRequest);
|
log.info("[SystemInitializeService] [S2] createPowerJobUser successfully: {}", powerJobUser);
|
|
// STEP3: 授予全局管理员权限
|
powerJobPermissionService.grantRole(RoleScope.GLOBAL, AuthConstants.GLOBAL_ADMIN_TARGET_ID, powerJobUser.getId(), Role.ADMIN, null);
|
log.info("[SystemInitializeService] [S3] GRANT ADMIN successfully!");
|
|
// 循环10遍，强提醒用户，第一次使用必须更改 admin 密码
|
for (int i = 0; i < 10; i++) {
|
log.warn("[SystemInitializeService] The system has automatically created a super administrator account[username={},password={}], please log in and change the password immediately!", username, password);
|
}
|
}
|
|
@Override
|
@Transactional(rollbackOn = Exception.class)
|
public void initNamespace() {
|
ModifyNamespaceRequest saveNamespaceReq = new ModifyNamespaceRequest();
|
saveNamespaceReq.setName(SYSTEM_DEFAULT_NAMESPACE);
|
saveNamespaceReq.setCode(SYSTEM_DEFAULT_NAMESPACE);
|
|
log.info("[SystemInitializeService] create default namespace by request: {}", saveNamespaceReq);
|
NamespaceDO savedNamespaceDO = namespaceWebService.save(saveNamespaceReq);
|
log.info("[SystemInitializeService] create default namespace successfully: {}", savedNamespaceDO);
|
}
|
}
|
