From d0d7f3b074053e07a0af9449082f0b0448df89da Mon Sep 17 00:00:00 2001
From: Kari Makinen <karimak123@gmail.com>
Date: Mon, 3 Oct 2022 11:53:09 +0300
Subject: [PATCH] add ClusterQuorumResolver

---
 network-segmentation/pom.xml                  |  3 +-
 .../segmentation/IgSegmentationProcessor.java | 14 ++--
 .../segment/ClusterQuorumResolver.java        | 64 +++++++++++++++++++
 pom.xml                                       | 16 ++++-
 4 files changed, 88 insertions(+), 9 deletions(-)
 create mode 100644 network-segmentation/src/main/java/com/ig/segmentation/network/segment/ClusterQuorumResolver.java

diff --git a/network-segmentation/pom.xml b/network-segmentation/pom.xml
index 343dfcc..f65c5e4 100644
--- a/network-segmentation/pom.xml
+++ b/network-segmentation/pom.xml
@@ -18,6 +18,7 @@
         <dependency>
             <groupId>org.apache.ignite</groupId>
             <artifactId>ignite-core</artifactId>
+            <version>${apache-ignite.version}</version>
         </dependency>
 
         <dependency>
@@ -27,4 +28,4 @@
         </dependency>
     </dependencies>
 
-</project>
\ No newline at end of file
+</project>
diff --git a/network-segmentation/src/main/java/com/ig/segmentation/internal/processors/segmentation/IgSegmentationProcessor.java b/network-segmentation/src/main/java/com/ig/segmentation/internal/processors/segmentation/IgSegmentationProcessor.java
index afc0cf3..334ed3f 100644
--- a/network-segmentation/src/main/java/com/ig/segmentation/internal/processors/segmentation/IgSegmentationProcessor.java
+++ b/network-segmentation/src/main/java/com/ig/segmentation/internal/processors/segmentation/IgSegmentationProcessor.java
@@ -1,9 +1,7 @@
 package com.ig.segmentation.internal.processors.segmentation;
 
-import java.util.List;
-import java.util.Map;
-import java.util.TreeMap;
-
+import com.ig.segmentation.network.segment.ClusterQuorumResolver;
+import com.ig.segmentation.plugin.IgSegmentationPlugin;
 import org.apache.ignite.IgniteCheckedException;
 import org.apache.ignite.configuration.IgniteConfiguration;
 import org.apache.ignite.internal.GridKernalContext;
@@ -15,7 +13,9 @@
 import org.apache.ignite.lang.IgniteBiTuple;
 import org.apache.ignite.plugin.segmentation.SegmentationResolver;
 
-import com.ig.segmentation.plugin.IgSegmentationPlugin;
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
 
 /**
  * This class defines {@link GridSegmentationProcessor} implementation which can be replaced with
@@ -91,7 +91,9 @@ private boolean hasValidSegments(List<SegmentationResolver> resolvers, Map<Strin
                     if (isLogDebugModeEnabled())
                         this.log.debug( String.format(SEG_RESOLVER_START_MSG, resolverName, i) );
 
-                    validSegment = resolver.isValidSegment();
+                    if (resolver instanceof ClusterQuorumResolver)
+                        validSegment = ((ClusterQuorumResolver) resolver).isQuorum(this.ctx);
+                    else validSegment = resolver.isValidSegment();
 
                     if (isLogDebugModeEnabled())
                         this.log.debug( String.format(SEG_RESOLVER_END_MSG, resolverName, i, validSegment) );
diff --git a/network-segmentation/src/main/java/com/ig/segmentation/network/segment/ClusterQuorumResolver.java b/network-segmentation/src/main/java/com/ig/segmentation/network/segment/ClusterQuorumResolver.java
new file mode 100644
index 0000000..3567a5d
--- /dev/null
+++ b/network-segmentation/src/main/java/com/ig/segmentation/network/segment/ClusterQuorumResolver.java
@@ -0,0 +1,64 @@
+package com.ig.segmentation.network.segment;
+
+import com.ig.segmentation.network.exception.IgSegmentationException;
+import com.ig.segmentation.plugin.segmentation.IgSegmentationResolver;
+import org.apache.ignite.IgniteLogger;
+import org.apache.ignite.cluster.ClusterNode;
+import org.apache.ignite.internal.GridKernalContext;
+import org.apache.ignite.internal.cluster.IgniteClusterEx;
+
+public class ClusterQuorumResolver implements IgSegmentationResolver {
+
+    private int maxServerNodesSeen = 0;
+
+    private int quorumAttempts = 0;
+
+    private int quorumNodes = 0;
+
+    private int maxWaitForQuorumAttempts = 5;
+
+
+    public boolean isValidSegment() throws IgSegmentationException {
+        return true;
+    }
+
+    public ClusterQuorumResolver setQuorumNodes(int nodes) throws IgSegmentationException {
+        this.quorumNodes = nodes;
+        return this;
+    }
+
+    public ClusterQuorumResolver setMaxWaitForQuorumAttempts(int attempts) throws IgSegmentationException {
+        this.maxWaitForQuorumAttempts = attempts;
+        return this;
+    }
+
+    public boolean isQuorum(GridKernalContext ctx) throws IgSegmentationException {
+        if (ctx == null)
+            return true;
+
+        IgniteClusterEx cluster = ctx.grid().cluster();
+        if (cluster == null)
+            return true;
+
+        IgniteLogger log = ctx.log(getClass());
+
+        var serverNodes = 0;
+        try {
+            for (ClusterNode node : cluster.nodes()) {
+                if (!node.isClient()) serverNodes += 1;
+            }
+            if (serverNodes > maxServerNodesSeen) maxServerNodesSeen = serverNodes;
+
+            log.info("quorumNodes=" + quorumNodes + " serverNodes=" +  serverNodes + " maxServerNodesSeen=" + maxServerNodesSeen + " attempts=" + quorumAttempts + " maxWaitForQuorumAttempts=" + maxWaitForQuorumAttempts);
+            if (serverNodes < quorumNodes) {
+                if (quorumAttempts > maxWaitForQuorumAttempts) {
+                    return false;
+                }
+                quorumAttempts += 1;
+            }
+        } catch (NullPointerException e) {
+            log.warning("null", e);
+        }
+        return true;
+    }
+}
diff --git a/pom.xml b/pom.xml
index 6f3837f..cf5fe3c 100644
--- a/pom.xml
+++ b/pom.xml
@@ -16,10 +16,22 @@
 
     <properties>
         <ignite-plugins.version>0.0.1</ignite-plugins.version>
-        <apache-ignite.version>2.6.0</apache-ignite.version>
+        <apache-ignite.version>2.13.0</apache-ignite.version>
         <junit.version>4.12</junit.version>
     </properties>
 
+    <build>
+        <plugins>
+            <plugin>
+              <artifactId>maven-compiler-plugin</artifactId>
+              <version>3.8.0</version>
+              <configuration>
+                <release>11</release>
+              </configuration>
+          </plugin>
+      </plugins>
+    </build>
+
     <dependencyManagement>
         <dependencies>
             <dependency>
@@ -38,4 +50,4 @@
         </dependencies>
     </dependencyManagement>
 
-</project>
\ No newline at end of file
+</project>