fix: Add common client inheritence to RhelHosts (#11)

project-kessel · Sep 30, 2024 · b54c5cd · b54c5cd
1 parent 21e8fec
commit b54c5cd
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/main/java/org/project_kessel/inventory/client/RhelHostClient.java b/src/main/java/org/project_kessel/inventory/client/RhelHostClient.java
@@ -7,20 +7,16 @@
 import org.project_kessel.api.inventory.v1beta1.resources.CreateRhelHostRequest;
 import org.project_kessel.api.inventory.v1beta1.resources.CreateRhelHostResponse;
 import org.project_kessel.api.inventory.v1beta1.resources.KesselRhelHostServiceGrpc;
-
+import org.project_kessel.clients.KesselClient;
 
 
 import java.util.logging.Logger;
 
-public class RhelHostClient {
+public class RhelHostClient extends KesselClient<KesselRhelHostServiceGrpc.KesselRhelHostServiceStub, KesselRhelHostServiceGrpc.KesselRhelHostServiceBlockingStub> {
     private static final Logger logger = Logger.getLogger(RhelHostClient.class.getName());
 
-    private final KesselRhelHostServiceGrpc.KesselRhelHostServiceStub asyncStub;
-    private final KesselRhelHostServiceGrpc.KesselRhelHostServiceBlockingStub blockingStub;
-
     RhelHostClient(Channel channel){
-        asyncStub = KesselRhelHostServiceGrpc.newStub(channel);
-        blockingStub = KesselRhelHostServiceGrpc.newBlockingStub(channel);
+        super(KesselRhelHostServiceGrpc.newStub(channel), KesselRhelHostServiceGrpc.newBlockingStub(channel));
     }
 
     public void CreateRhelHost(CreateRhelHostRequest request, StreamObserver<CreateRhelHostResponse> responseStreamObserver) {