org.dromara.jpom.build.BuildExecuteService Maven / Gradle / Ivy
/*
* Copyright (c) 2019 Of Him Code Technology Studio
* Jpom is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
* http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
*/
package org.dromara.jpom.build;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.date.SystemClock;
import cn.hutool.core.lang.Opt;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.keepbx.jpom.IJsonMessage;
import cn.keepbx.jpom.model.JsonMessage;
import lombok.extern.slf4j.Slf4j;
import org.dromara.jpom.common.i18n.I18nMessageUtil;
import org.dromara.jpom.model.BaseEnum;
import org.dromara.jpom.model.EnvironmentMapBuilder;
import org.dromara.jpom.model.data.BuildInfoModel;
import org.dromara.jpom.model.data.RepositoryModel;
import org.dromara.jpom.model.enums.BuildStatus;
import org.dromara.jpom.model.log.BuildHistoryLog;
import org.dromara.jpom.model.user.UserModel;
import org.dromara.jpom.service.dblog.BuildInfoService;
import org.dromara.jpom.service.dblog.DbBuildHistoryLogService;
import org.dromara.jpom.service.dblog.RepositoryService;
import org.dromara.jpom.service.system.WorkspaceEnvVarService;
import org.dromara.jpom.util.LogRecorder;
import org.dromara.jpom.util.StringUtil;
import org.springframework.stereotype.Service;
import org.springframework.util.Assert;
import java.io.File;
import java.util.Objects;
/**
* @author bwcx_jzy
* @since 2022/1/26
*/
@Service
@Slf4j
public class BuildExecuteService {
private final BuildInfoService buildService;
private final DbBuildHistoryLogService dbBuildHistoryLogService;
private final RepositoryService repositoryService;
private final WorkspaceEnvVarService workspaceEnvVarService;
private final BuildExecutorPoolService buildExecutorPoolService;
public BuildExecuteService(BuildInfoService buildService,
DbBuildHistoryLogService dbBuildHistoryLogService,
RepositoryService repositoryService,
WorkspaceEnvVarService workspaceEnvVarService,
BuildExecutorPoolService buildExecutorPoolService) {
this.buildService = buildService;
this.dbBuildHistoryLogService = dbBuildHistoryLogService;
this.repositoryService = repositoryService;
this.workspaceEnvVarService = workspaceEnvVarService;
this.buildExecutorPoolService = buildExecutorPoolService;
}
/**
* check status
*
* @param buildInfoModel 构建信息
* @return 错误消息
*/
public String checkStatus(BuildInfoModel buildInfoModel) {
if (buildInfoModel == null) {
return I18nMessageUtil.get("i18n.build_info_not_exist.4470");
}
Integer status = buildInfoModel.getStatus();
if (status == null) {
return null;
}
BuildStatus nowStatus = BaseEnum.getEnum(BuildStatus.class, status);
Objects.requireNonNull(nowStatus);
if (nowStatus.isProgress()) {
return buildInfoModel.getName() + I18nMessageUtil.get("i18n.current_status.81c0") + nowStatus.getDesc();
}
return null;
}
/**
* start build
*
* @param buildInfoId 构建Id
* @param userModel 用户信息
* @param delay 延迟的时间
* @param triggerBuildType 触发构建类型
* @param buildRemark 构建备注
* @param parametersEnv 外部环境变量
* @return json
*/
public IJsonMessage start(String buildInfoId, UserModel userModel, Integer delay, int triggerBuildType, String buildRemark, Object... parametersEnv) {
return this.start(buildInfoId, userModel, delay, triggerBuildType, buildRemark, null, parametersEnv);
}
/**
* start build
*
* @param buildInfoId 构建Id
* @param userModel 用户信息
* @param delay 延迟的时间
* @param triggerBuildType 触发构建类型
* @param buildRemark 构建备注
* @param checkRepositoryDiff 差异构建
* @param parametersEnv 外部环境变量
* @return json
*/
public IJsonMessage start(String buildInfoId, UserModel userModel, Integer delay,
int triggerBuildType, String buildRemark, String checkRepositoryDiff,
Object... parametersEnv) {
synchronized (buildInfoId.intern()) {
BuildInfoModel buildInfoModel = buildService.getByKey(buildInfoId);
String e = this.checkStatus(buildInfoModel);
Assert.isNull(e, () -> e);
//
boolean containsKey = BuildExecuteManage.BUILD_MANAGE_MAP.containsKey(buildInfoModel.getId());
Assert.state(!containsKey, I18nMessageUtil.get("i18n.build_in_progress.4d33"));
//
BuildExtraModule buildExtraModule = buildInfoModel.extraData();
Assert.notNull(buildExtraModule, I18nMessageUtil.get("i18n.build_info_missing.0ab0"));
// load repository
RepositoryModel repositoryModel = repositoryService.getByKey(buildInfoModel.getRepositoryId(), false);
Assert.notNull(repositoryModel, I18nMessageUtil.get("i18n.repository_info_does_not_exist.4142"));
EnvironmentMapBuilder environmentMapBuilder = workspaceEnvVarService.getEnv(buildInfoModel.getWorkspaceId());
// 解析外部变量
environmentMapBuilder.putObjectArray(parametersEnv).putStr(StringUtil.parseEnvStr(buildInfoModel.getBuildEnvParameter()));
// set buildId field
buildInfoModel.setBuildId(this.nextBuildId(buildInfoModel));
//
TaskData.TaskDataBuilder taskBuilder = TaskData.builder()
.buildInfoModel(buildInfoModel)
.repositoryModel(repositoryModel)
.userModel(userModel)
.buildRemark(buildRemark)
.delay(delay)
.environmentMapBuilder(environmentMapBuilder)
.triggerBuildType(triggerBuildType);
//
Opt.ofBlankAble(checkRepositoryDiff).map(Convert::toBool).ifPresent(taskBuilder::checkRepositoryDiff);
this.runTask(taskBuilder.build(), buildExtraModule);
String startMsg = I18nMessageUtil.get("i18n.start_building.1039");
String delayMsg = StrUtil.format(I18nMessageUtil.get("i18n.delay_build.7d62"), delay);
String msg = (delay == null || delay <= 0) ? startMsg : delayMsg;
return JsonMessage.success(msg, buildInfoModel.getBuildId());
}
}
/**
* 回滚
*
* @param oldLog 构建历史
* @param item 构建项
* @param userModel 用户信息
*/
public int rollback(BuildHistoryLog oldLog, BuildInfoModel item, UserModel userModel) {
synchronized (item.getId().intern()) {
String e = this.checkStatus(item);
Assert.isNull(e, () -> e);
Integer fromBuildNumberId = ObjectUtil.defaultIfNull(oldLog.getFromBuildNumberId(), oldLog.getBuildNumberId());
int buildId = this.nextBuildId(item);
item.setBuildId(buildId);
// 创建新的构建记录
BuildHistoryLog buildHistoryLog = oldLog.toJson().to(BuildHistoryLog.class);
buildHistoryLog.setId(null);
buildHistoryLog.setCreateUser(null);
buildHistoryLog.setCreateTimeMillis(null);
buildHistoryLog.setModifyUser(null);
buildHistoryLog.setModifyTimeMillis(null);
buildHistoryLog.setResultFileSize(null);
BuildStatus pubIng = BuildStatus.PubIng;
buildHistoryLog.setStatus(pubIng.getCode());
buildHistoryLog.setTriggerBuildType(3);
buildHistoryLog.setBuildNumberId(buildId);
buildHistoryLog.setFromBuildNumberId(fromBuildNumberId);
buildHistoryLog.setStartTime(SystemClock.now());
buildHistoryLog.setEndTime(null);
dbBuildHistoryLogService.insert(buildHistoryLog);
//
buildService.updateStatus(buildHistoryLog.getBuildDataId(), pubIng, I18nMessageUtil.get("i18n.start_rolling_back_execution.a019"));
BuildExtraModule buildExtraModule = BuildExtraModule.build(buildHistoryLog);
//
EnvironmentMapBuilder environmentMapBuilder = buildHistoryLog.toEnvironmentMapBuilder();
//
File logFile = BuildUtil.getLogFile(item.getId(), buildId);
LogRecorder logRecorder = LogRecorder.builder().file(logFile).build();
ReleaseManage manage = ReleaseManage.builder()
.buildExtraModule(buildExtraModule)
.logId(buildHistoryLog.getId())
.userModel(userModel)
.buildNumberId(buildHistoryLog.getBuildNumberId())
.fromBuildNumberId(fromBuildNumberId)
.logRecorder(logRecorder)
.buildEnv(environmentMapBuilder)
.build();
//
logRecorder.system(I18nMessageUtil.get("i18n.prepare_rollback.dba6"), fromBuildNumberId, buildId);
//
buildExecutorPoolService.execute(() -> manage.rollback(item));
return buildId;
}
}
private int nextBuildId(BuildInfoModel buildInfoModel) {
// set buildId field
int buildId = ObjectUtil.defaultIfNull(buildInfoModel.getBuildId(), 0);
BuildInfoModel update = new BuildInfoModel();
update.setBuildId(buildId + 1);
update.setId(buildInfoModel.getId());
buildService.updateById(update);
return update.getBuildId();
}
/**
* 创建构建
*
* @param taskData 任务
* @param buildExtraModule 构建更多配置信息
*/
private void runTask(TaskData taskData, BuildExtraModule buildExtraModule) {
String logId = this.insertLog(buildExtraModule, taskData);
//
BuildExecuteManage.BuildExecuteManageBuilder builder = BuildExecuteManage.builder()
.taskData(taskData)
.logId(logId)
.buildExtraModule(buildExtraModule);
builder.build().submitTask();
}
/**
* 插入记录
*/
private String insertLog(BuildExtraModule buildExtraModule, TaskData taskData) {
BuildInfoModel buildInfoModel = taskData.buildInfoModel;
buildExtraModule.updateValue(buildInfoModel);
BuildHistoryLog buildHistoryLog = new BuildHistoryLog();
// 更新其他配置字段
//buildHistoryLog.fillLogValue(buildExtraModule);
buildHistoryLog.setTriggerBuildType(taskData.triggerBuildType);
//
buildHistoryLog.setBuildNumberId(buildInfoModel.getBuildId());
buildHistoryLog.setBuildName(buildInfoModel.getName());
buildHistoryLog.setBuildDataId(buildInfoModel.getId());
buildHistoryLog.setWorkspaceId(buildInfoModel.getWorkspaceId());
buildHistoryLog.setResultDirFile(buildInfoModel.getResultDirFile());
buildHistoryLog.setReleaseMethod(buildExtraModule.getReleaseMethod());
//
BuildStatus waitExec = BuildStatus.WaitExec;
buildHistoryLog.setStatus(waitExec.getCode());
buildHistoryLog.setStartTime(SystemClock.now());
buildHistoryLog.setBuildRemark(taskData.buildRemark);
// 缓存数据 - 保证数据一直
buildHistoryLog.setExtraData(buildExtraModule.toJson().toString());
dbBuildHistoryLogService.insert(buildHistoryLog);
//
buildService.updateStatus(buildHistoryLog.getBuildDataId(), waitExec, I18nMessageUtil.get("i18n.start_queuing_for_execution.7417"));
return buildHistoryLog.getId();
}
/**
* 更新状态
*
* @param buildId 构建ID
* @param logId 记录ID
* @param buildStatus to status
*/
public void updateStatus(String buildId, String logId, int buildNumberId, BuildStatus buildStatus, String msg) {
BuildHistoryLog buildHistoryLog = new BuildHistoryLog();
buildHistoryLog.setId(logId);
buildHistoryLog.setStatusMsg(msg);
buildHistoryLog.setStatus(buildStatus.getCode());
if (!buildStatus.isProgress()) {
// 结束
buildHistoryLog.setEndTime(SystemClock.now());
}
dbBuildHistoryLogService.updateById(buildHistoryLog);
buildService.updateStatus(buildId, buildNumberId, buildStatus, msg);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy