64 lines
2.2 KiB
Java
64 lines
2.2 KiB
Java
package com.designhub.service;
|
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
import com.designhub.entity.Role;
|
|
import com.designhub.mapper.DepartmentMapper;
|
|
import com.designhub.mapper.RoleMapper;
|
|
import com.designhub.mapper.TaskMapper;
|
|
import com.designhub.mapper.UserMapper;
|
|
import com.designhub.vo.SystemHealthVO;
|
|
import lombok.RequiredArgsConstructor;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import java.util.List;
|
|
|
|
/**
|
|
* 系统健康检查服务
|
|
*/
|
|
@Service
|
|
@RequiredArgsConstructor
|
|
public class SystemHealthService {
|
|
|
|
private final UserMapper userMapper;
|
|
private final TaskMapper taskMapper;
|
|
private final RoleMapper roleMapper;
|
|
private final DepartmentMapper departmentMapper;
|
|
|
|
@Value("${spring.profiles.active:default}")
|
|
private String activeProfile;
|
|
|
|
/**
|
|
* 执行健康检查
|
|
*/
|
|
public SystemHealthVO check() {
|
|
SystemHealthVO health = new SystemHealthVO();
|
|
health.setProfile(activeProfile);
|
|
|
|
try {
|
|
health.setUserCount(userMapper.selectCount(null));
|
|
health.setTaskCount(taskMapper.selectCount(null));
|
|
health.setDepartmentCount(departmentMapper.selectCount(null));
|
|
health.setRoleCount(roleMapper.selectCount(new LambdaQueryWrapper<Role>().select(Role::getId)));
|
|
health.setDatabaseUp(true);
|
|
} catch (Exception ex) {
|
|
health.setDatabaseUp(false);
|
|
health.getChecks().add(new SystemHealthVO.CheckItem("database", false, ex.getMessage()));
|
|
return health;
|
|
}
|
|
|
|
// 校验 sys_role 表结构是否与实体一致（只查询存在的列）
|
|
try {
|
|
roleMapper.selectList(new LambdaQueryWrapper<Role>()
|
|
.select(Role::getId, Role::getRoleCode, Role::getRoleName, Role::getStatus)
|
|
.last("LIMIT 1"));
|
|
health.getChecks().add(new SystemHealthVO.CheckItem("sys_role_schema", true, "Role columns verified"));
|
|
} catch (Exception ex) {
|
|
health.getChecks().add(new SystemHealthVO.CheckItem("sys_role_schema", false, ex.getMessage()));
|
|
}
|
|
|
|
return health;
|
|
}
|
|
}
|
|
|