feat(shulker-server-agent): add watchGameServer method to api

packages/google-agones-sdk/bindings/java/src/main/java/com/agones/dev/sdk/AgonesSDK.java

@@ -2,6 +2,7 @@
 
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
+import java.util.function.Consumer;
 
 public interface AgonesSDK {
     void destroy();
@@ -12,6 +13,7 @@ public interface AgonesSDK {
     CompletableFuture<Void> setReady();
     CompletableFuture<Void> setAllocated();
     CompletableFuture<Void> setReserved(long seconds);
+    void watchGameServer(Consumer<GameServer> consumer);
     void askShutdown();
 
     void sendHealthcheck();

packages/google-agones-sdk/bindings/java/src/main/java/com/agones/dev/sdk/AgonesSDKImpl.java

@@ -11,6 +11,7 @@
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
+import java.util.function.Consumer;
 
 public final class AgonesSDKImpl implements AgonesSDK {
     private static final Empty EMPTY_PAYLOAD = Empty.getDefaultInstance();
@@ -75,6 +76,33 @@ public CompletableFuture<Void> setReserved(long seconds) {
         ).thenAccept((reply) -> {});
     }
 
+    @Override
+    public void watchGameServer(Consumer<GameServer> consumer) {
+        // Create a StreamObserver that forwards each GameServer to the Consumer
+        StreamObserver<GameServer> responseObserver = new StreamObserver<>() {
+            @Override
+            public void onNext(GameServer value) {
+                // Forward the GameServer to the Consumer for handling
+                consumer.accept(value);
+            }
+
+            @Override
+            public void onError(Throwable t) {
+                // Optionally handle errors
+                System.err.println("Error during stream: " + t.getMessage());
+            }
+
+            @Override
+            public void onCompleted() {
+                // Optionally handle completion of the stream
+                System.out.println("Stream completed.");
+            }
+        };
+
+        // Make the gRPC call to start watching the GameServer stream
+        this.stub.watchGameServer(EMPTY_PAYLOAD, responseObserver);
+    }
+
     @Override
     public void askShutdown() {
         try {

packages/shulker-server-agent/build.gradle.kts

@@ -5,6 +5,8 @@ dependencies {
 
     // Agones
     commonImplementation(project(":packages:google-agones-sdk"))
+
+    commonImplementation(libs.grpc.stub)
 }
 
 setOf("processPaperResources").forEach { taskName ->

packages/shulker-server-agent/src/common/kotlin/io/shulkermc/serveragent/api/ShulkerServerAPIImpl.kt

@@ -1,12 +1,17 @@
 package io.shulkermc.serveragent.api
 
+import com.agones.dev.sdk.GameServer
 import io.shulkermc.serveragent.ShulkerServerAgentCommon
 import java.util.concurrent.CompletableFuture
+import java.util.function.Consumer
 
 class ShulkerServerAPIImpl(private val agent: ShulkerServerAgentCommon) : ShulkerServerAPI() {
     override fun askShutdown() = this.agent.shutdown()
 
     override fun setReady(): CompletableFuture<Void> = this.agent.agonesGateway.setReady().thenAccept {}
+    override fun watchGameServer(consumer: Consumer<GameServer>) {
+        return this.agent.agonesGateway.watchGameServer(consumer)
+    }
 
     override fun setAllocated(): CompletableFuture<Void> = this.agent.agonesGateway.setAllocated().thenAccept {}
 

packages/shulker-server-api/build.gradle.kts

@@ -1,3 +1,6 @@
+dependencies {
+    implementation(project(":packages:google-agones-sdk"))
+}
 configure<JavaPluginExtension> {
     withJavadocJar()
     withSourcesJar()

packages/shulker-server-api/src/main/java/io/shulkermc/serveragent/api/ShulkerServerAPI.java

@@ -1,12 +1,16 @@
 package io.shulkermc.serveragent.api;
 
+import com.agones.dev.sdk.GameServer;
+
 import java.util.concurrent.CompletableFuture;
+import java.util.function.Consumer;
 
 public abstract class ShulkerServerAPI {
     public static ShulkerServerAPI INSTANCE;
 
     abstract public void askShutdown();
     abstract public CompletableFuture<Void> setReady();
+    abstract public void watchGameServer(Consumer<GameServer> consumer);
     abstract public CompletableFuture<Void> setAllocated();
     abstract public CompletableFuture<Void> setReserved(long seconds);
 }