Add runningSlots tracking to Host and display available slots in GUI

- Add runningSlots field to Host protobuf and HostEntity
- Update Whiteboard DAO queries to include runningSlots
- Show available slots (concurrentSlotsLimit - runningSlots) in CueGUI host monitor tree

Author: DiegoTavares

cuebot/src/main/java/com/imageworks/spcue/HostEntity.java

@@ -64,6 +64,7 @@ public HostEntity(Host grpcHost) {
         this.gpuMemory = grpcHost.getGpuMemory();
         this.idleGpuMemory = grpcHost.getIdleGpuMemory();
         this.concurrentSlotsLimit = grpcHost.getConcurrentSlotsLimit();
+        this.runningSlots = grpcHost.getRunningSlots();
     }
 
     public String getHostId() {

cuebot/src/main/java/com/imageworks/spcue/dao/postgres/NestedWhiteboardDaoJdbc.java

@@ -283,7 +283,8 @@ private static final NestedJob mapResultSetToJob(ResultSet rs) throws SQLExcepti
             + "host.int_gpus, " + "host.int_gpus_idle, " + "host.int_gpu_mem, "
             + "host.int_gpu_mem_idle, " + "host.int_mem, " + "host.int_mem_idle, "
             + "host.str_lock_state, " + "host.str_tags, " + "host.b_comment, "
-            + "host.int_thread_mode, " + "host_stat.str_os, " + "host_stat.int_mem_total, "
+            + "host.int_thread_mode, " + "host_stat.int_running_slots, " + "host_stat.str_os, "
+            + "host_stat.int_mem_total, "
             + "host_stat.int_mem_free, " + "host_stat.int_swap_total, "
             + "host_stat.int_swap_free, " + "host_stat.int_mcp_total, " + "host_stat.int_mcp_free, "
             + "host_stat.int_gpu_mem_total, " + "host_stat.int_gpu_mem_free, "

cuebot/src/main/java/com/imageworks/spcue/dao/postgres/WhiteboardDaoJdbc.java

@@ -962,6 +962,7 @@ public static NestedHost.Builder mapNestedHostBuilder(ResultSet rs) throws SQLEx
                 .setHasComment(rs.getBoolean("b_comment"))
                 .setThreadMode(ThreadMode.values()[rs.getInt("int_thread_mode")])
                 .setConcurrentSlotsLimit(rs.getInt("int_concurrent_slots_limit"))
+                .setRunningSlots(rs.getInt("int_running_slots"))
                 .setOs(SqlUtil.getString(rs, "str_os"));
 
         String tags = SqlUtil.getString(rs, "str_tags");
@@ -1000,6 +1001,7 @@ public static Host.Builder mapHostBuilder(ResultSet rs) throws SQLException {
         builder.setHasComment(rs.getBoolean("b_comment"));
         builder.setThreadMode(ThreadMode.values()[rs.getInt("int_thread_mode")]);
         builder.setConcurrentSlotsLimit(rs.getInt("int_concurrent_slots_limit"));
+        builder.setRunningSlots(rs.getInt("int_running_slots"));
         builder.setOs(SqlUtil.getString(rs, "str_os"));
 
         String tags = SqlUtil.getString(rs, "str_tags");
@@ -1723,7 +1725,8 @@ public Show mapRow(ResultSet rs, int rowNum) throws SQLException {
             + "host_stat.int_mem_total," + "host_stat.int_mem_free," + "host_stat.int_swap_total,"
             + "host_stat.int_swap_free," + "host_stat.int_mcp_total," + "host_stat.int_mcp_free,"
             + "host_stat.int_gpu_mem_total," + "host_stat.int_gpu_mem_free,"
-            + "host_stat.int_load, " + "alloc.str_name AS alloc_name " + "FROM " + "alloc,"
+            + "host_stat.int_load, " + "host_stat.int_running_slots, "
+            + "alloc.str_name AS alloc_name " + "FROM " + "alloc,"
             + "facility, " + "host_stat," + "host " + "WHERE " + "host.pk_alloc = alloc.pk_alloc "
             + "AND " + "facility.pk_facility = alloc.pk_facility " + "AND "
             + "host.pk_host = host_stat.pk_host ";

cuegui/cuegui/HostMonitorTree.py

@@ -168,6 +168,12 @@ def __init__(self, parent):
                            "The host can only run this amount of slots at the same time "
                            "(Usually: 1 frame = 1 slot)\n\n"
                            "This host will only run layers with a slots_required field configured.")
+        self.addColumn("Available Slots", 50, id=24,
+                       data=lambda host: (
+                           host.data.concurrent_slots_limit - host.data.running_slots
+                       ) if host.data.concurrent_slots_limit >= 0 else "-",
+                       tip="The number of slots still available on the host.\n"
+                           "Computed as Concurrent Slots minus currently running slots.")
 
         self.hostSearch = opencue.search.HostSearch()
 

proto/src/host.proto

@@ -282,6 +282,7 @@ message Host {
     // 0: slot-based booking is disabled (default behavior using cores/memory/gpu)
     // >0: slot-based booking is enabled with specified limit (only layers with slots_required > 0)
     int32 concurrent_slots_limit = 30;
+    int32 running_slots = 31;
 }
 
 message HostSearchCriteria {
@@ -330,6 +331,7 @@ message NestedHost {
     float gpus = 29;
     float idle_gpus = 30;
     int32 concurrent_slots_limit = 31;
+    int32 running_slots = 32;
 }
 
 message NestedHostSeq {