Examples with NotificationService org.graylog2.notifications.NotificationService used on opensource projects

Example 6 with NotificationService

use of org.graylog2.notifications.NotificationService in project graylog2-server by Graylog2.

the class IndexRotationThreadTest method testPerformRotation.

@Test
public void testPerformRotation() throws NoTargetIndexException {
    final Provider<RotationStrategy> provider = new RotationStrategyProvider() {

        @Override
        public void doRotate(IndexSet indexSet) {
            indexSet.cycle();
        }
    };
    final IndexRotationThread rotationThread = new IndexRotationThread(notificationService, indices, indexSetRegistry, cluster, new NullActivityWriter(), nodeId, ImmutableMap.<String, Provider<RotationStrategy>>builder().put("strategy", provider).build());
    when(indexSetConfig.rotationStrategyClass()).thenReturn("strategy");
    rotationThread.checkForRotation(indexSet);
    verify(indexSet, times(1)).cycle();
}

Example 7 with NotificationService

use of org.graylog2.notifications.NotificationService in project graylog2-server by Graylog2.

the class IndexerClusterCheckerThreadTest method ignoresNonDataNodes.

@Test
public void ignoresNonDataNodes() throws Exception {
    final Set<NodeDiskUsageStats> nodeDiskUsageStats = mockNodeDiskUsageStats(EnumSet.of(MASTER_ELIGIBLE, TRANSFORM));
    when(cluster.getDiskUsageStats()).thenReturn(nodeDiskUsageStats);
    when(cluster.getClusterAllocationDiskSettings()).thenReturn(buildThresholdTriggeredClusterAllocationDiskSettings(ES_NODE_DISK_WATERMARK_LOW, WatermarkSettings.SettingsType.ABSOLUTE));
    Notification notification = new NotificationImpl();
    when(notificationService.buildNow()).thenReturn(notification);
    when(notificationService.isFirst(ES_NODE_DISK_WATERMARK_LOW)).thenReturn(true);
    indexerClusterCheckerThread.checkDiskUsage();
    verify(notificationService, never()).publishIfFirst(any());
}

Example 8 with NotificationService

use of org.graylog2.notifications.NotificationService in project graylog2-server by Graylog2.

the class IndexerClusterCheckerThreadTest method notificationCreated.

private void notificationCreated(Notification.Type notificationType, WatermarkSettings.SettingsType watermarkSettingsType) throws Exception {
    Set<NodeDiskUsageStats> nodeDiskUsageStats = mockNodeDiskUsageStats();
    when(cluster.getDiskUsageStats()).thenReturn(nodeDiskUsageStats);
    if (watermarkSettingsType == WatermarkSettings.SettingsType.ABSOLUTE) {
        when(cluster.getClusterAllocationDiskSettings()).thenReturn(buildThresholdTriggeredClusterAllocationDiskSettings(notificationType, WatermarkSettings.SettingsType.ABSOLUTE));
    } else {
        when(cluster.getClusterAllocationDiskSettings()).thenReturn(buildThresholdTriggeredClusterAllocationDiskSettings(notificationType, WatermarkSettings.SettingsType.PERCENTAGE));
    }
    when(notificationService.isFirst(notificationType)).thenReturn(true);
    Notification notification = new NotificationImpl();
    when(notificationService.buildNow()).thenReturn(notification);
    indexerClusterCheckerThread.checkDiskUsage();
    ArgumentCaptor<Notification> argument = ArgumentCaptor.forClass(Notification.class);
    verify(notificationService, times(1)).publishIfFirst(argument.capture());
    Notification publishedNotification = argument.getValue();
    assertThat(publishedNotification.getType()).isEqualTo(notificationType);
}

Example 9 with NotificationService

use of org.graylog2.notifications.NotificationService in project graylog2-server by Graylog2.

the class Server method startNodeRegistration.

@Override
protected void startNodeRegistration(Injector injector) {
    // Register this node.
    final NodeService nodeService = injector.getInstance(NodeService.class);
    final ServerStatus serverStatus = injector.getInstance(ServerStatus.class);
    final ActivityWriter activityWriter = injector.getInstance(ActivityWriter.class);
    final LeaderElectionService leaderElectionService = injector.getInstance(LeaderElectionService.class);
    nodeService.registerServer(serverStatus.getNodeId().toString(), leaderElectionService.isLeader(), httpConfiguration.getHttpPublishUri(), Tools.getLocalCanonicalHostname());
    serverStatus.setLocalMode(isLocal());
    if (leaderElectionService.isLeader() && !nodeService.isOnlyLeader(serverStatus.getNodeId())) {
        LOG.warn("Detected another leader in the cluster. Retrying in {} seconds to make sure it is not " + "an old stale instance.", TimeUnit.MILLISECONDS.toSeconds(configuration.getStaleLeaderTimeout()));
        try {
            Thread.sleep(configuration.getStaleLeaderTimeout());
        } catch (InterruptedException e) {
        /* nope */
        }
        if (!nodeService.isOnlyLeader(serverStatus.getNodeId())) {
            // All devils here.
            String what = "Detected other leader node in the cluster! Starting as non-leader! " + "This is a mis-configuration you should fix.";
            LOG.warn(what);
            activityWriter.write(new Activity(what, Server.class));
            final NotificationService notificationService = injector.getInstance(NotificationService.class);
            // remove legacy notification, if present
            // noinspection deprecation
            notificationService.fixed(notificationService.build().addType(Notification.Type.MULTI_MASTER));
            // Write a notification.
            Notification notification = notificationService.buildNow().addType(Notification.Type.MULTI_LEADER).addSeverity(Notification.Severity.URGENT);
            notificationService.publishIfFirst(notification);
            configuration.setIsLeader(false);
        } else {
            LOG.warn("Stale leader has gone. Starting as leader.");
        }
    }
}