Implement toString() for model classes

Author: tadayosi

src/main/java/com/github/tadayosi/torchserve/client/model/Api.java

@@ -53,4 +53,13 @@ public Map<String, Object> getPaths() {
     public void setPaths(Map<String, Object> paths) {
         this.paths = paths;
     }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {\n" +
+            "    openapi: " + openapi + "\n" +
+            "    info: " + info + "\n" +
+            "    paths: " + paths + "\n" +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/JobQueueStatus.java

@@ -11,6 +11,10 @@ public JobQueueStatus() {
     }
 
     public static JobQueueStatus from(ModelsmodelNameJobQueueStatus src) {
+        if (src == null) {
+            return null;
+        }
+
         JobQueueStatus status = new JobQueueStatus();
         status.setRemainingCapacity(src.getRemainingCapacity());
         status.setPendingRequests(src.getPendingRequests());
@@ -32,4 +36,12 @@ public Integer getPendingRequests() {
     public void setPendingRequests(Integer pendingRequests) {
         this.pendingRequests = pendingRequests;
     }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {" +
+            " remainingCapacity: " + remainingCapacity + "," +
+            " pendingRequests: " + pendingRequests + " " +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/Metrics.java

@@ -1,14 +1,59 @@
 package com.github.tadayosi.torchserve.client.model;
 
 import com.github.tadayosi.torchserve.client.management.model.ModelsmodelNameMetrics;
+import com.google.gson.annotations.SerializedName;
 
 public class Metrics {
 
+    private Integer rejectedRequests = null;
+    private Integer waitingQueueSize = null;
+    private Integer requests = null;
+
     public Metrics() {
     }
 
     public static Metrics from(ModelsmodelNameMetrics src) {
+        if (src == null) {
+            return null;
+        }
+
         Metrics metrics = new Metrics();
+        metrics.setRejectedRequests(src.getRejectedRequests());
+        metrics.setWaitingQueueSize(src.getWaitingQueueSize());
+        metrics.setRequests(src.getRequests());
         return metrics;
     }
+
+    public Integer getRejectedRequests() {
+        return rejectedRequests;
+    }
+
+    public void setRejectedRequests(Integer rejectedRequests) {
+        this.rejectedRequests = rejectedRequests;
+    }
+
+    public Integer getWaitingQueueSize() {
+        return waitingQueueSize;
+    }
+
+    public void setWaitingQueueSize(Integer waitingQueueSize) {
+        this.waitingQueueSize = waitingQueueSize;
+    }
+
+    public Integer getRequests() {
+        return requests;
+    }
+
+    public void setRequests(Integer requests) {
+        this.requests = requests;
+    }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {" +
+            " rejectedRequests: " + rejectedRequests + "," +
+            " waitingQueueSize: " + waitingQueueSize + "," +
+            " requests: " + requests + " " +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/Model.java

@@ -11,8 +11,8 @@ public class Model {
 
     private static final Logger LOG = LoggerFactory.getLogger(Model.class);
 
-    private String modelName = null;
-    private String modelUrl = null;
+    protected String modelName = null;
+    protected String modelUrl = null;
 
     public Model() {
     }
@@ -49,4 +49,12 @@ public String getModelUrl() {
     public void setModelUrl(String modelUrl) {
         this.modelUrl = modelUrl;
     }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {\n" +
+            "    modelName: " + modelName + "\n" +
+            "    modelUrl: " + modelUrl + "\n" +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/ModelDetail.java

@@ -110,4 +110,21 @@ public JobQueueStatus getJobQueueStatus() {
     public void setJobQueueStatus(JobQueueStatus jobQueueStatus) {
         this.jobQueueStatus = jobQueueStatus;
     }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {\n" +
+            "    modelName: " + modelName + "\n" +
+            "    modelVersion: " + modelVersion + "\n" +
+            "    modelUrl: " + modelUrl + "\n" +
+            "    minWorkers: " + minWorkers + "\n" +
+            "    maxWorkers: " + maxWorkers + "\n" +
+            "    batchSize: " + batchSize + "\n" +
+            "    maxBatchDelay: " + maxBatchDelay + "\n" +
+            "    status: " + status + "\n" +
+            "    workers: " + workers + "\n" +
+            "    metrics: " + metrics + "\n" +
+            "    jobQueueStatus: " + jobQueueStatus + "\n" +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/ModelList.java

@@ -35,4 +35,12 @@ public List<Model> getModels() {
     public void setModels(List<Model> models) {
         this.models = models;
     }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {\n" +
+            "    nextPageToken: " + nextPageToken + "\n" +
+            "    models: " + models + "\n" +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/Response.java

@@ -26,4 +26,11 @@ public String getStatus() {
     public void setStatus(String status) {
         this.status = status;
     }
+
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {" +
+            "    status: " + status + "\n" +
+            "}";
+    }
 }

src/main/java/com/github/tadayosi/torchserve/client/model/Worker.java

@@ -16,6 +16,10 @@ public Worker() {
     }
 
     public static Worker from(ModelsmodelNameWorkers src) {
+        if (src == null) {
+            return null;
+        }
+
         Worker worker = new Worker();
         worker.setId(src.getId());
         worker.setStartTime(src.getStartTime());
@@ -60,6 +64,16 @@ public void setStatus(Status status) {
         this.status = status;
     }
 
+    @Override
+    public String toString() {
+        return getClass().getSimpleName() + " {" +
+            " id: " + id + "," +
+            " startTime: " + startTime + "," +
+            " gpu: " + gpu + "," +
+            " status: " + status + " " +
+            "}";
+    }
+
     public enum Status {
         READY,
         LOADING,