Examples with Param com.usthe.common.entity.manager.Param used on opensource projects

Example 1 with Param

use of com.usthe.common.entity.manager.Param in project hertzbeat by dromara.

the class MonitorServiceImpl method detectMonitor.

@Override
@Transactional(readOnly = true)
public void detectMonitor(Monitor monitor, List<Param> params) throws MonitorDetectException {
    Long monitorId = monitor.getId();
    if (monitorId == null || monitorId == 0) {
        monitorId = MONITOR_ID_TMP;
    }
    Job appDefine = appService.getAppDefine(monitor.getApp());
    appDefine.setMonitorId(monitorId);
    appDefine.setCyclic(false);
    appDefine.setTimestamp(System.currentTimeMillis());
    List<Configmap> configmaps = params.stream().map(param -> new Configmap(param.getField(), param.getValue(), param.getType())).collect(Collectors.toList());
    appDefine.setConfigmap(configmaps);
    // To detect availability, you only need to collect the set of availability indicators with a priority of 0.
    // 探测可用性只需要采集优先级为0的可用性指标集合
    List<Metrics> availableMetrics = appDefine.getMetrics().stream().filter(item -> item.getPriority() == 0).collect(Collectors.toList());
    appDefine.setMetrics(availableMetrics);
    List<CollectRep.MetricsData> collectRep = collectJobService.collectSyncJobData(appDefine);
    // 判断探测结果 失败则抛出探测异常
    if (collectRep == null || collectRep.isEmpty()) {
        throw new MonitorDetectException("No collector response");
    }
    if (collectRep.get(0).getCode() != CollectRep.Code.SUCCESS) {
        throw new MonitorDetectException(collectRep.get(0).getMsg());
    }
}

Example 2 with Param

use of com.usthe.common.entity.manager.Param in project hertzbeat by dromara.

the class MonitorServiceImpl method validate.

@Override
@Transactional(readOnly = true)
public void validate(MonitorDto monitorDto, Boolean isModify) throws IllegalArgumentException {
    // The request monitoring parameter matches the monitoring parameter definition mapping check
    // 请求监控参数与监控参数定义映射校验匹配
    Monitor monitor = monitorDto.getMonitor();
    monitor.setHost(monitor.getHost().trim());
    monitor.setName(monitor.getName().trim());
    Map<String, Param> paramMap = monitorDto.getParams().stream().peek(param -> {
        param.setMonitorId(monitor.getId());
        String value = param.getValue() == null ? null : param.getValue().trim();
        param.setValue(value);
    }).collect(Collectors.toMap(Param::getField, param -> param));
    // Check name uniqueness    校验名称唯一性
    if (isModify != null) {
        Optional<Monitor> monitorOptional = monitorDao.findMonitorByNameEquals(monitor.getName());
        if (monitorOptional.isPresent()) {
            Monitor existMonitor = monitorOptional.get();
            if (isModify) {
                if (!existMonitor.getId().equals(monitor.getId())) {
                    throw new IllegalArgumentException("监控名称不能重复!");
                }
            } else {
                throw new IllegalArgumentException("监控名称不能重复!");
            }
        }
    }
    // todo 校验标签
    if (monitor.getTags() != null) {
        monitor.setTags(monitor.getTags().stream().distinct().collect(Collectors.toList()));
    }
    // Parameter definition structure verification  参数定义结构校验
    List<ParamDefine> paramDefines = appService.getAppParamDefines(monitorDto.getMonitor().getApp());
    if (paramDefines != null) {
        for (ParamDefine paramDefine : paramDefines) {
            String field = paramDefine.getField();
            Param param = paramMap.get(field);
            if (paramDefine.isRequired() && (param == null || param.getValue() == null)) {
                throw new IllegalArgumentException("Params field " + field + " is required.");
            }
            if (param != null && param.getValue() != null && !"".equals(param.getValue())) {
                switch(paramDefine.getType()) {
                    case "number":
                        double doubleValue;
                        try {
                            doubleValue = Double.parseDouble(param.getValue());
                        } catch (Exception e) {
                            throw new IllegalArgumentException("Params field " + field + " type " + paramDefine.getType() + " is invalid.");
                        }
                        if (paramDefine.getRange() != null) {
                            if (!IntervalExpressionUtil.validNumberIntervalExpress(doubleValue, paramDefine.getRange())) {
                                throw new IllegalArgumentException("Params field " + field + " type " + paramDefine.getType() + " over range " + paramDefine.getRange());
                            }
                        }
                        param.setType(CommonConstants.PARAM_TYPE_NUMBER);
                        break;
                    case "textarea":
                    case "text":
                        Short limit = paramDefine.getLimit();
                        if (limit != null) {
                            if (param.getValue() != null && param.getValue().length() > limit) {
                                throw new IllegalArgumentException("Params field " + field + " type " + paramDefine.getType() + " over limit " + limit);
                            }
                        }
                        break;
                    case "host":
                        String hostValue = param.getValue();
                        if (!IpDomainUtil.validateIpDomain(hostValue)) {
                            throw new IllegalArgumentException("Params field " + field + " value " + hostValue + " is invalid host value.");
                        }
                        break;
                    case "password":
                        // The plaintext password needs to be encrypted for transmission and storage
                        // 明文密码需加密传输存储
                        String passwordValue = param.getValue();
                        if (!AesUtil.isCiphertext(passwordValue)) {
                            passwordValue = AesUtil.aesEncode(passwordValue);
                            param.setValue(passwordValue);
                        }
                        param.setType(CommonConstants.PARAM_TYPE_PASSWORD);
                        break;
                    case "boolean":
                        // boolean check
                        String booleanValue = param.getValue();
                        try {
                            Boolean.parseBoolean(booleanValue);
                        } catch (Exception e) {
                            throw new IllegalArgumentException("Params field " + field + " value " + booleanValue + " is invalid boolean value.");
                        }
                        break;
                    case "radio":
                        // radio single value check  radio单选值校验
                        List<ParamDefine.Option> options = paramDefine.getOptions();
                        boolean invalid = true;
                        if (options != null) {
                            for (ParamDefine.Option option : options) {
                                if (param.getValue().equalsIgnoreCase(option.getValue())) {
                                    invalid = false;
                                    break;
                                }
                            }
                        }
                        if (invalid) {
                            throw new IllegalArgumentException("Params field " + field + " value " + param.getValue() + " is invalid option value");
                        }
                        break;
                    case "checkbox":
                        // todo checkbox校验
                        break;
                    case "key-value":
                        // todo key-value校验
                        break;
                    // 更多参数定义与实际值格式校验
                    default:
                        throw new IllegalArgumentException("ParamDefine type " + paramDefine.getType() + " is invalid.");
                }
            }
        }
    }
}

Example 3 with Param

use of com.usthe.common.entity.manager.Param in project hertzbeat by dromara.

the class MonitorServiceImpl method modifyMonitor.

@Override
@Transactional(rollbackFor = Exception.class)
public void modifyMonitor(Monitor monitor, List<Param> params) throws RuntimeException {
    long monitorId = monitor.getId();
    // Check to determine whether the monitor corresponding to the monitor id exists
    // 查判断monitorId对应的此监控是否存在
    Optional<Monitor> queryOption = monitorDao.findById(monitorId);
    if (!queryOption.isPresent()) {
        throw new IllegalArgumentException("The Monitor " + monitorId + " not exists");
    }
    Monitor preMonitor = queryOption.get();
    if (!preMonitor.getApp().equals(monitor.getApp())) {
        // 监控的类型不能修改
        throw new IllegalArgumentException("Can not modify monitor's app type");
    }
    // Auto Update Default Tags: monitorName
    List<Tag> tags = monitor.getTags();
    if (tags == null) {
        tags = new LinkedList<>();
        monitor.setTags(tags);
    }
    for (Tag tag : tags) {
        if (CommonConstants.TAG_MONITOR_NAME.equals(tag.getName())) {
            tag.setValue(monitor.getName());
        }
    }
    // Construct the collection task Job entity
    // 构造采集任务Job实体
    Job appDefine = appService.getAppDefine(monitor.getApp());
    appDefine.setId(preMonitor.getJobId());
    appDefine.setMonitorId(monitorId);
    appDefine.setInterval(monitor.getIntervals());
    appDefine.setCyclic(true);
    appDefine.setTimestamp(System.currentTimeMillis());
    List<Configmap> configmaps = params.stream().map(param -> new Configmap(param.getField(), param.getValue(), param.getType())).collect(Collectors.toList());
    appDefine.setConfigmap(configmaps);
    // 下发更新成功后刷库
    try {
        monitor.setJobId(preMonitor.getJobId());
        monitor.setStatus(preMonitor.getStatus());
        monitorDao.save(monitor);
        paramDao.saveAll(params);
        // Update the collection task after the storage is completed
        // 入库完成后更新采集任务
        collectJobService.updateAsyncCollectJob(appDefine);
    } catch (Exception e) {
        log.error(e.getMessage(), e);
        throw new MonitorDatabaseException(e.getMessage());
    }
}

Example 4 with Param

use of com.usthe.common.entity.manager.Param in project hertzbeat by dromara.

the class MonitorServiceImpl method addMonitor.

@Override
@Transactional(rollbackFor = Exception.class)
public void addMonitor(Monitor monitor, List<Param> params) throws RuntimeException {
    // Apply for monitor id         申请 monitor id
    long monitorId = SnowFlakeIdGenerator.generateId();
    // Init Set Default Tags: monitorId monitorName app
    List<Tag> tags = monitor.getTags();
    if (tags == null) {
        tags = new LinkedList<>();
        monitor.setTags(tags);
    }
    tags.add(Tag.builder().name(CommonConstants.TAG_MONITOR_ID).value(String.valueOf(monitorId)).type((byte) 0).build());
    tags.add(Tag.builder().name(CommonConstants.TAG_MONITOR_NAME).value(String.valueOf(monitor.getName())).type((byte) 0).build());
    // Construct the collection task Job entity     构造采集任务Job实体
    Job appDefine = appService.getAppDefine(monitor.getApp());
    appDefine.setMonitorId(monitorId);
    appDefine.setInterval(monitor.getIntervals());
    appDefine.setCyclic(true);
    appDefine.setTimestamp(System.currentTimeMillis());
    List<Configmap> configmaps = params.stream().map(param -> {
        param.setMonitorId(monitorId);
        return new Configmap(param.getField(), param.getValue(), param.getType());
    }).collect(Collectors.toList());
    appDefine.setConfigmap(configmaps);
    // Send the collection task to get the job ID
    // 下发采集任务得到jobId
    long jobId = collectJobService.addAsyncCollectJob(appDefine);
    // 下发成功后刷库
    try {
        monitor.setId(monitorId);
        monitor.setJobId(jobId);
        monitor.setStatus(CommonConstants.AVAILABLE_CODE);
        monitorDao.save(monitor);
        paramDao.saveAll(params);
    } catch (Exception e) {
        log.error(e.getMessage(), e);
        // Repository brushing abnormally cancels the previously delivered task
        // 刷库异常取消之前的下发任务
        collectJobService.cancelAsyncCollectJob(jobId);
        throw new MonitorDatabaseException(e.getMessage());
    }
}

Example 5 with Param

use of com.usthe.common.entity.manager.Param in project hertzbeat by dromara.

the class MonitorServiceImpl method enableManageMonitors.

@Override
public void enableManageMonitors(HashSet<Long> ids) {
    // Update monitoring status Add corresponding monitoring periodic task
    // 更新监控状态 新增对应的监控周期性任务
    List<Monitor> unManagedMonitors = monitorDao.findMonitorsByIdIn(ids).stream().filter(monitor -> monitor.getStatus() == CommonConstants.UN_MANAGE_CODE && monitor.getJobId() != null).peek(monitor -> monitor.setStatus(CommonConstants.AVAILABLE_CODE)).collect(Collectors.toList());
    if (!unManagedMonitors.isEmpty()) {
        monitorDao.saveAll(unManagedMonitors);
        for (Monitor monitor : unManagedMonitors) {
            // Construct the collection task Job entity
            // 构造采集任务Job实体
            Job appDefine = appService.getAppDefine(monitor.getApp());
            appDefine.setMonitorId(monitor.getId());
            appDefine.setInterval(monitor.getIntervals());
            appDefine.setCyclic(true);
            appDefine.setTimestamp(System.currentTimeMillis());
            List<Param> params = paramDao.findParamsByMonitorId(monitor.getId());
            List<Configmap> configmaps = params.stream().map(param -> new Configmap(param.getField(), param.getValue(), param.getType())).collect(Collectors.toList());
            appDefine.setConfigmap(configmaps);
            // Issue collection tasks       下发采集任务
            collectJobService.addAsyncCollectJob(appDefine);
        }
    }
}