Add API endpoint that returns the local datacenter

k8ssandra · Jul 6, 2020 · b82d6e0 · b82d6e0
1 parent 2d289d8
commit b82d6e0
Show file tree

Hide file tree

Showing 6 changed files with 36 additions and 0 deletions.
diff --git a/management-api-agent/src/main/java/com/datastax/mgmtapi/NodeOpsProvider.java b/management-api-agent/src/main/java/com/datastax/mgmtapi/NodeOpsProvider.java
@@ -332,4 +332,10 @@ public void createKeyspace(@RpcParam(name="keyspaceName") String keyspaceName, @
                         .asCql(),
                 ConsistencyLevel.ONE);
     }
+
+    @Rpc(name = "getLocalDataCenter")
+    public String getLocalDataCenter()
+    {
+        return ShimLoader.instance.get().getLocalDataCenter();
+    }
 }
diff --git a/management-api-common/src/main/java/com/datastax/mgmtapi/shims/CassandraAPI.java b/management-api-common/src/main/java/com/datastax/mgmtapi/shims/CassandraAPI.java
@@ -15,6 +15,7 @@
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelInitializer;
 import org.apache.cassandra.auth.IRoleManager;
+import org.apache.cassandra.config.DatabaseDescriptor;
 import org.apache.cassandra.cql3.QueryProcessor;
 import org.apache.cassandra.cql3.UntypedResultSet;
 import org.apache.cassandra.db.ConsistencyLevel;
@@ -63,4 +64,6 @@ default Object handleRpcResult(Callable<Object> rpcResult) throws Exception
     {
         return rpcResult.call();
     }
+
+    String getLocalDataCenter();
 }
diff --git a/management-api-server/src/main/java/com/datastax/mgmtapi/resources/MetadataResources.java b/management-api-server/src/main/java/com/datastax/mgmtapi/resources/MetadataResources.java
@@ -54,6 +54,15 @@ public Response getEndpointStates()
         return executeWithJSONResponse("CALL NodeOps.getEndpointStates()");
     }
 
+    @GET
+    @Path("/localdc")
+    @Operation(summary = "Returns the DataCenter the local node belongs to")
+    @Produces(MediaType.TEXT_PLAIN)
+    public Response getLocalDataCenter()
+    {
+        return executeWithStringResponse("CALL NodeOps.getLocalDataCenter()");
+    }
+
     /**
      * Executes a CQL query with the expectation that there will be a single row returned with type String
      *

diff --git a/management-api-shim-3.x/src/main/java/com/datastax/mgmtapi/shim/CassandraAPI3x.java b/management-api-shim-3.x/src/main/java/com/datastax/mgmtapi/shim/CassandraAPI3x.java
@@ -291,4 +291,10 @@ public Gossiper getGossiper()
     {
         return Gossiper.instance;
     }
+
+    @Override
+    public String getLocalDataCenter()
+    {
+        return DatabaseDescriptor.getLocalDataCenter();
+    }
 }
diff --git a/management-api-shim-4.x/src/main/java/com/datastax/mgmtapi/shim/CassandraAPI4x.java b/management-api-shim-4.x/src/main/java/com/datastax/mgmtapi/shim/CassandraAPI4x.java
@@ -295,4 +295,10 @@ public Gossiper getGossiper()
     {
         return Gossiper.instance;
     }
+
+    @Override
+    public String getLocalDataCenter()
+    {
+        return DatabaseDescriptor.getLocalDataCenter();
+    }
 }
diff --git a/management-api-shim-dse-6.8/src/main/java/com/datastax/mgmtapi/shim/DseAPI68.java b/management-api-shim-dse-6.8/src/main/java/com/datastax/mgmtapi/shim/DseAPI68.java
@@ -309,4 +309,10 @@ public Object handleRpcResult(Callable<Object> rpcResult)
     {
         return RxThreads.subscribeOnIo(Single.defer(() -> Single.fromCallable(rpcResult)), TPCTaskType.UNKNOWN);
     }
+
+    @Override
+    public String getLocalDataCenter()
+    {
+        return DatabaseDescriptor.getLocalDataCenter();
+    }
 }