leader election

Author: wu8685

examples/src/main/java/io/kubernetes/client/examples/LeaderElectionExample.java

@@ -0,0 +1,38 @@
+package io.kubernetes.client.examples;
+
+import io.kubernetes.client.ApiClient;
+import io.kubernetes.client.Configuration;
+import io.kubernetes.client.extended.leaderelection.LeaderElectionConfig;
+import io.kubernetes.client.extended.leaderelection.LeaderElector;
+import io.kubernetes.client.extended.leaderelection.resourcelock.EndpointsLock;
+import io.kubernetes.client.util.Config;
+import java.time.Duration;
+
+/**
+ * A simple example of how to use the Java API
+ *
+ * <p>Easiest way to run this: mvn exec:java
+ * -Dexec.mainClass="io.kubernetes.client.examples.LeaderElectionExample"
+ *
+ * <p>From inside $REPO_DIR/examples
+ */
+public class LeaderElectionExample {
+  public static void main(String[] args) throws Exception {
+    ApiClient client = Config.defaultClient();
+    Configuration.setDefaultApiClient(client);
+
+    EndpointsLock lock = new EndpointsLock("kube-system", "leader-election", "foo");
+
+    LeaderElectionConfig leaderElectionConfig =
+        new LeaderElectionConfig(lock, Duration.ofMillis(10000), null, Duration.ofMillis(5000));
+    LeaderElector leaderElector = new LeaderElector(leaderElectionConfig);
+
+    leaderElector.run(
+        () -> {
+          System.out.println("Do something when getting leadership.");
+        },
+        () -> {
+          System.out.println("Do something when losing leadership.");
+        });
+  }
+}

extended/src/main/java/io/kubernetes/client/extended/leaderelection/LeaderElectionConfig.java

@@ -0,0 +1,56 @@
+package io.kubernetes.client.extended.leaderelection;
+
+import java.time.Duration;
+
+public class LeaderElectionConfig {
+
+  private Lock lock;
+
+  private Duration leaseDuration;
+
+  private Duration renewDeadline;
+
+  private Duration retryPeriod;
+
+  public LeaderElectionConfig() {}
+
+  public LeaderElectionConfig(
+      Lock lock, Duration leaseDuration, Duration renewDeadline, Duration retryPeriod) {
+    this.lock = lock;
+    this.leaseDuration = leaseDuration;
+    this.renewDeadline = renewDeadline;
+    this.retryPeriod = retryPeriod;
+  }
+
+  public Lock getLock() {
+    return lock;
+  }
+
+  public void setLock(Lock lock) {
+    this.lock = lock;
+  }
+
+  public Duration getLeaseDuration() {
+    return leaseDuration;
+  }
+
+  public void setLeaseDuration(Duration leaseDuration) {
+    this.leaseDuration = leaseDuration;
+  }
+
+  public Duration getRenewDeadline() {
+    return renewDeadline;
+  }
+
+  public void setRenewDeadline(Duration renewDeadline) {
+    this.renewDeadline = renewDeadline;
+  }
+
+  public Duration getRetryPeriod() {
+    return retryPeriod;
+  }
+
+  public void setRetryPeriod(Duration retryPeriod) {
+    this.retryPeriod = retryPeriod;
+  }
+}

extended/src/main/java/io/kubernetes/client/extended/leaderelection/LeaderElectionRecord.java

@@ -0,0 +1,96 @@
+package io.kubernetes.client.extended.leaderelection;
+
+import java.util.Date;
+
+public class LeaderElectionRecord {
+
+  private String holderIdentity;
+
+  private int leaseDurationSeconds;
+
+  private Date acquireTime;
+
+  private Date renewTime;
+
+  private int leaderTransitions;
+
+  public LeaderElectionRecord() {}
+
+  public LeaderElectionRecord(
+      String holderIdentity,
+      int leaseDurationSeconds,
+      Date acquireTime,
+      Date renewTime,
+      int leaderTransitions) {
+    this.holderIdentity = holderIdentity;
+    this.leaseDurationSeconds = leaseDurationSeconds;
+    this.acquireTime = acquireTime;
+    this.renewTime = renewTime;
+    this.leaderTransitions = leaderTransitions;
+  }
+
+  public String getHolderIdentity() {
+    return holderIdentity;
+  }
+
+  public void setHolderIdentity(String holderIdentity) {
+    this.holderIdentity = holderIdentity;
+  }
+
+  public int getLeaseDurationSeconds() {
+    return leaseDurationSeconds;
+  }
+
+  public void setLeaseDurationSeconds(int leaseDurationSeconds) {
+    this.leaseDurationSeconds = leaseDurationSeconds;
+  }
+
+  public Date getAcquireTime() {
+    return acquireTime;
+  }
+
+  public void setAcquireTime(Date acquireTime) {
+    this.acquireTime = acquireTime;
+  }
+
+  public Date getRenewTime() {
+    return renewTime;
+  }
+
+  public void setRenewTime(Date renewTime) {
+    this.renewTime = renewTime;
+  }
+
+  public int getLeaderTransitions() {
+    return leaderTransitions;
+  }
+
+  public void setLeaderTransitions(int leaderTransitions) {
+    this.leaderTransitions = leaderTransitions;
+  }
+
+  @Override
+  public boolean equals(Object o) {
+    if (this == o) return true;
+    if (!(o instanceof LeaderElectionRecord)) return false;
+
+    LeaderElectionRecord that = (LeaderElectionRecord) o;
+
+    if (leaseDurationSeconds != that.leaseDurationSeconds) return false;
+    if (leaderTransitions != that.leaderTransitions) return false;
+    if (holderIdentity == null) {
+      if (that.holderIdentity != null) return false;
+    } else {
+      if (!holderIdentity.equals(that.holderIdentity)) return false;
+    }
+    if (acquireTime == null) {
+      if (that.acquireTime != null) return false;
+    } else {
+      if (!acquireTime.equals(that.acquireTime)) return false;
+    }
+    if (renewTime == null) {
+      return that.holderIdentity == null;
+    }
+    return renewTime.equals(that.renewTime);
+  }
+}