org.dromara.jpom.service.dblog.DbBuildHistoryLogService 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.service.dblog;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.db.Entity;
import cn.hutool.db.Page;
import cn.hutool.db.sql.Direction;
import cn.hutool.db.sql.Order;
import cn.keepbx.jpom.IJsonMessage;
import cn.keepbx.jpom.event.ISystemTask;
import cn.keepbx.jpom.model.JsonMessage;
import lombok.extern.slf4j.Slf4j;
import org.dromara.jpom.build.BuildExtraModule;
import org.dromara.jpom.build.BuildUtil;
import org.dromara.jpom.common.i18n.I18nMessageUtil;
import org.dromara.jpom.configuration.BuildExtConfig;
import org.dromara.jpom.model.BaseDbModel;
import org.dromara.jpom.model.PageResultDto;
import org.dromara.jpom.model.data.BuildInfoModel;
import org.dromara.jpom.model.enums.BuildStatus;
import org.dromara.jpom.model.log.BuildHistoryLog;
import org.dromara.jpom.service.h2db.BaseWorkspaceService;
import org.springframework.stereotype.Service;
import java.io.File;
import java.util.List;
import java.util.Optional;
import java.util.function.Predicate;
import java.util.stream.Collectors;
/**
* 构建历史db
*
* @author bwcx_jzy
* @since 2019/7/20
*/
@Service
@Slf4j
public class DbBuildHistoryLogService extends BaseWorkspaceService implements ISystemTask {
private final BuildInfoService buildService;
private final BuildExtConfig buildExtConfig;
public DbBuildHistoryLogService(BuildInfoService buildService,
BuildExtConfig buildExtConfig) {
this.buildService = buildService;
this.buildExtConfig = buildExtConfig;
}
/**
* 更新状态
*
* @param logId 记录id
* @param resultDirFile 构建产物目录
*/
public void updateResultDirFile(String logId, String resultDirFile) {
if (logId == null || StrUtil.isEmpty(resultDirFile)) {
return;
}
BuildHistoryLog buildHistoryLog = new BuildHistoryLog();
buildHistoryLog.setId(logId);
buildHistoryLog.setResultDirFile(resultDirFile);
this.updateById(buildHistoryLog);
}
/**
* 清理文件并删除记录
*
* @param buildHistoryLog 构建记录
* @return json
*/
public IJsonMessage deleteLogAndFile(BuildHistoryLog buildHistoryLog) {
if (buildHistoryLog == null) {
return JsonMessage.success(I18nMessageUtil.get("i18n.no_corresponding_build_record_ignore_deletion.86a0"));
}
BuildInfoModel item = buildService.getByKey(buildHistoryLog.getBuildDataId());
if (item != null) {
File logFile = BuildUtil.getLogFile(item.getId(), buildHistoryLog.getBuildNumberId());
if (logFile != null) {
File dataFile = logFile.getParentFile();
if (dataFile.exists()) {
boolean s = FileUtil.del(dataFile);
if (!s) {
return new JsonMessage<>(500, I18nMessageUtil.get("i18n.file_cleanup_failed.511e"));
}
}
}
}
int count = this.delByKey(buildHistoryLog.getId());
return new JsonMessage<>(200, I18nMessageUtil.get("i18n.delete_success.0007"), String.valueOf(count));
}
@Override
protected void fillSelectResult(BuildHistoryLog data) {
super.fillSelectResult(data);
Optional.ofNullable(data).ifPresent(buildHistoryLog -> {
// 不能返回环境变量的信息（存在隐私字段）
buildHistoryLog.setBuildEnvCache(null);
});
}
@Override
public int insert(BuildHistoryLog buildHistoryLog) {
int count = super.insert(buildHistoryLog);
// 清理单个
BuildExtraModule build = BuildExtraModule.build(buildHistoryLog);
int resultKeepCount = ObjectUtil.defaultIfNull(build.getResultKeepCount(), 0);
int buildItemMaxHistoryCount = buildExtConfig.getItemMaxHistoryCount();
if (resultKeepCount > 0 || buildItemMaxHistoryCount > 0) {
// 至少有一个配置
int useCount;
if (resultKeepCount > 0 && buildItemMaxHistoryCount > 0) {
// 都配置过，使用最小值
useCount = Math.min(resultKeepCount, buildItemMaxHistoryCount);
} else {
// 只配置了一处，使用最大值
useCount = Math.max(resultKeepCount, buildItemMaxHistoryCount);
}
super.autoLoopClear("startTime", useCount, entity -> this.fillClearWhere(entity, buildHistoryLog.getBuildDataId()), this.predicate());
}
return count;
}
private Predicate predicate() {
return buildHistoryLog1 -> {
IJsonMessage jsonMessage = this.deleteLogAndFile(buildHistoryLog1);
if (!jsonMessage.success()) {
log.warn("{} {} {}", buildHistoryLog1.getBuildName(), buildHistoryLog1.getBuildNumberId(), jsonMessage);
return false;
}
return true;
};
}
private void fillClearWhere(Entity entity, String buildDataId) {
entity.set("buildDataId", buildDataId);
// 清理单项构建历史保留个数只判断（构建结束、发布中、发布失败、发布失败）有效构建状态，避免无法保留有效构建历史
entity.set("status", CollUtil.newArrayList(
BuildStatus.Success.getCode(),
BuildStatus.PubIng.getCode(),
BuildStatus.PubSuccess.getCode(),
BuildStatus.PubError.getCode()));
}
@Override
protected void executeClearImpl(int count) {
// 清理总数据
int buildMaxHistoryCount = buildExtConfig.getMaxHistoryCount();
int saveCount = Math.min(count, buildMaxHistoryCount);
if (saveCount <= 0) {
// 不清除
return;
}
super.autoLoopClear("startTime", saveCount,
null,
buildHistoryLog1 -> {
IJsonMessage jsonMessage = this.deleteLogAndFile(buildHistoryLog1);
if (!jsonMessage.success()) {
log.warn("{} {} {}", buildHistoryLog1.getBuildName(), buildHistoryLog1.getBuildNumberId(), jsonMessage);
return false;
}
return true;
});
}
@Override
protected String[] clearTimeColumns() {
return super.clearTimeColumns();
}
@Override
public void executeTask() {
List buildInfoModels = buildService.hasResultKeep();
if (CollUtil.isEmpty(buildInfoModels)) {
return;
}
for (BuildInfoModel buildInfoModel : buildInfoModels) {
Integer resultKeepDay = buildInfoModel.getResultKeepDay();
if (resultKeepDay == null || resultKeepDay <= 0) {
continue;
}
log.debug(I18nMessageUtil.get("i18n.auto_delete_expired_build_history_files.723b"), buildInfoModel.getName(), resultKeepDay);
Entity entity = Entity.create();
this.fillClearWhere(entity, buildInfoModel.getId());
DateTime date = DateTime.now();
date = DateUtil.offsetDay(date, -resultKeepDay);
date = DateUtil.beginOfDay(date);
entity.set("startTime", "< " + date.getTime());
while (true) {
Page page = new Page(1, 50);
page.addOrder(new Order("startTime", Direction.DESC));
PageResultDto pageResult = this.listPage(entity, page);
if (pageResult.isEmpty()) {
break;
}
List ids = pageResult.getResult()
.stream()
.filter(this.predicate())
.map(BaseDbModel::getId)
.collect(Collectors.toList());
//
this.delByKey(ids, null);
}
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy