Search in sources :

Example 1 with NotificationType

use of com.epam.pipeline.entity.notification.NotificationSettings.NotificationType in project cloud-pipeline by epam.

the class NotificationManager method notifyIdleRuns.

/**
 * Issues a notification of an idle Pipeline Run for multiple runs.
 *
 * @param pipelineCpuRatePairs a list of pairs of PipelineRun and Double cpu usage rate value
 * @param notificationType a type of notification to be issued. Supported types are IDLE_RUN, IDLE_RUN_PAUSED,
 *                         IDLE_RUN_STOPPED
 * @throws IllegalArgumentException if notificationType is not from IDLE_RUN group
 */
@Transactional(propagation = Propagation.REQUIRED)
public void notifyIdleRuns(List<Pair<PipelineRun, Double>> pipelineCpuRatePairs, NotificationType notificationType) {
    if (CollectionUtils.isEmpty(pipelineCpuRatePairs)) {
        return;
    }
    Assert.isTrue(NotificationSettings.NotificationGroup.IDLE_RUN == notificationType.getGroup(), "Only IDLE_RUN group notification types are allowed");
    NotificationSettings idleRunSettings = notificationSettingsManager.load(notificationType);
    if (idleRunSettings == null || !idleRunSettings.isEnabled()) {
        LOGGER.info("No template configured for idle pipeline run notifications or it was disabled!");
        return;
    }
    List<Long> ccUserIds = getKeepInformedUserIds(idleRunSettings);
    if (idleRunSettings.isKeepInformedAdmins()) {
        ExtendedRole extendedRole = roleManager.loadRoleWithUsers(DefaultRoles.ROLE_ADMIN.getId());
        ccUserIds.addAll(extendedRole.getUsers().stream().map(PipelineUser::getId).collect(Collectors.toList()));
    }
    double idleCpuLevel = preferenceManager.getPreference(SystemPreferences.SYSTEM_IDLE_CPU_THRESHOLD_PERCENT);
    Map<String, PipelineUser> pipelineOwners = userManager.loadUsersByNames(pipelineCpuRatePairs.stream().map(p -> p.getLeft().getOwner()).collect(Collectors.toList())).stream().collect(Collectors.toMap(PipelineUser::getUserName, user -> user));
    List<NotificationMessage> messages = pipelineCpuRatePairs.stream().map(pair -> {
        NotificationMessage message = new NotificationMessage();
        message.setTemplate(new NotificationTemplate(idleRunSettings.getTemplateId()));
        message.setTemplateParameters(PipelineRunMapper.map(pair.getLeft(), null));
        message.getTemplateParameters().put("idleCpuLevel", idleCpuLevel);
        message.getTemplateParameters().put("cpuRate", pair.getRight() * PERCENT);
        message.setToUserId(pipelineOwners.getOrDefault(pair.getLeft().getOwner(), new PipelineUser()).getId());
        message.setCopyUserIds(ccUserIds);
        return message;
    }).collect(Collectors.toList());
    monitoringNotificationDao.createMonitoringNotifications(messages);
}
Also used : PipelineRunMapper(com.epam.pipeline.mapper.PipelineRunMapper) Arrays(java.util.Arrays) MessageConstants(com.epam.pipeline.common.MessageConstants) LoggerFactory(org.slf4j.LoggerFactory) Autowired(org.springframework.beans.factory.annotation.Autowired) SystemPreferences(com.epam.pipeline.manager.preference.SystemPreferences) ArrayList(java.util.ArrayList) PipelineRun(com.epam.pipeline.entity.pipeline.PipelineRun) Matcher(java.util.regex.Matcher) Pair(org.apache.commons.lang3.tuple.Pair) MessageHelper(com.epam.pipeline.common.MessageHelper) Propagation(org.springframework.transaction.annotation.Propagation) ExtendedRole(com.epam.pipeline.entity.user.ExtendedRole) Service(org.springframework.stereotype.Service) Map(java.util.Map) NotificationType(com.epam.pipeline.entity.notification.NotificationSettings.NotificationType) IssueComment(com.epam.pipeline.entity.issue.IssueComment) TypeReference(com.fasterxml.jackson.core.type.TypeReference) PipelineUser(com.epam.pipeline.entity.user.PipelineUser) EntityManager(com.epam.pipeline.manager.EntityManager) NotificationMessage(com.epam.pipeline.entity.notification.NotificationMessage) Logger(org.slf4j.Logger) AbstractSecuredEntity(com.epam.pipeline.entity.AbstractSecuredEntity) PreferenceManager(com.epam.pipeline.manager.preference.PreferenceManager) RoleManager(com.epam.pipeline.manager.user.RoleManager) DefaultRoles(com.epam.pipeline.entity.user.DefaultRoles) JsonMapper(com.epam.pipeline.config.JsonMapper) NotificationTemplate(com.epam.pipeline.entity.notification.NotificationTemplate) NotificationMessageVO(com.epam.pipeline.controller.vo.notification.NotificationMessageVO) Collectors(java.util.stream.Collectors) List(java.util.List) UserManager(com.epam.pipeline.manager.user.UserManager) MonitoringNotificationDao(com.epam.pipeline.dao.notification.MonitoringNotificationDao) NotificationSettings(com.epam.pipeline.entity.notification.NotificationSettings) CollectionUtils(org.springframework.util.CollectionUtils) Optional(java.util.Optional) Pattern(java.util.regex.Pattern) Issue(com.epam.pipeline.entity.issue.Issue) Collections(java.util.Collections) Transactional(org.springframework.transaction.annotation.Transactional) Assert(org.springframework.util.Assert) ExtendedRole(com.epam.pipeline.entity.user.ExtendedRole) PipelineUser(com.epam.pipeline.entity.user.PipelineUser) NotificationMessage(com.epam.pipeline.entity.notification.NotificationMessage) NotificationSettings(com.epam.pipeline.entity.notification.NotificationSettings) NotificationTemplate(com.epam.pipeline.entity.notification.NotificationTemplate) Transactional(org.springframework.transaction.annotation.Transactional)

Example 2 with NotificationType

use of com.epam.pipeline.entity.notification.NotificationSettings.NotificationType in project cloud-pipeline by epam.

the class NotificationSettingsManager method loadAll.

public List<NotificationSettings> loadAll() {
    List<NotificationSettings> settings = notificationSettingsDao.loadAllNotificationsSettings();
    Set<NotificationType> existingTemplateTypes = settings.stream().map(NotificationSettings::getType).collect(Collectors.toSet());
    for (NotificationType type : NotificationType.values()) {
        if (!existingTemplateTypes.contains(type)) {
            NotificationSettings defaultSetting = NotificationSettings.getDefault(type);
            settings.add(defaultSetting);
        }
    }
    return settings;
}
Also used : NotificationType(com.epam.pipeline.entity.notification.NotificationSettings.NotificationType) NotificationSettings(com.epam.pipeline.entity.notification.NotificationSettings)

Example 3 with NotificationType

use of com.epam.pipeline.entity.notification.NotificationSettings.NotificationType in project cloud-pipeline by epam.

the class NotificationTemplateManager method loadAll.

public List<NotificationTemplate> loadAll() {
    List<NotificationTemplate> templates = notificationTemplateDao.loadAllNotificationTemplates();
    Set<Long> existingTemplateTypes = templates.stream().map(NotificationTemplate::getId).collect(Collectors.toSet());
    for (NotificationType type : NotificationType.values()) {
        if (!existingTemplateTypes.contains(type.getId())) {
            NotificationTemplate defaultTemplate = NotificationTemplate.getDefault(type);
            templates.add(defaultTemplate);
        }
    }
    return templates;
}
Also used : NotificationType(com.epam.pipeline.entity.notification.NotificationSettings.NotificationType) NotificationTemplate(com.epam.pipeline.entity.notification.NotificationTemplate)

Aggregations

NotificationType (com.epam.pipeline.entity.notification.NotificationSettings.NotificationType)3 NotificationSettings (com.epam.pipeline.entity.notification.NotificationSettings)2 NotificationTemplate (com.epam.pipeline.entity.notification.NotificationTemplate)2 MessageConstants (com.epam.pipeline.common.MessageConstants)1 MessageHelper (com.epam.pipeline.common.MessageHelper)1 JsonMapper (com.epam.pipeline.config.JsonMapper)1 NotificationMessageVO (com.epam.pipeline.controller.vo.notification.NotificationMessageVO)1 MonitoringNotificationDao (com.epam.pipeline.dao.notification.MonitoringNotificationDao)1 AbstractSecuredEntity (com.epam.pipeline.entity.AbstractSecuredEntity)1 Issue (com.epam.pipeline.entity.issue.Issue)1 IssueComment (com.epam.pipeline.entity.issue.IssueComment)1 NotificationMessage (com.epam.pipeline.entity.notification.NotificationMessage)1 PipelineRun (com.epam.pipeline.entity.pipeline.PipelineRun)1 DefaultRoles (com.epam.pipeline.entity.user.DefaultRoles)1 ExtendedRole (com.epam.pipeline.entity.user.ExtendedRole)1 PipelineUser (com.epam.pipeline.entity.user.PipelineUser)1 EntityManager (com.epam.pipeline.manager.EntityManager)1 PreferenceManager (com.epam.pipeline.manager.preference.PreferenceManager)1 SystemPreferences (com.epam.pipeline.manager.preference.SystemPreferences)1 RoleManager (com.epam.pipeline.manager.user.RoleManager)1