Update ais-stats to use new API.

Author: zbyrne

tools/ais-stats/ais-stats.cpp

@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: MIT
  */
 
-#include "stats.h"
+#include "include_internal/hipfile-stats.h"
 #include <iostream>
 #include <unistd.h>
 #include <cstring>
@@ -47,17 +47,37 @@ main(int argc, char *argv[])
             return 1;
         }
     }
-    hipFile::StatsClient client{pid};
-    if (!client.connectServer()) {
-        std::cerr << "Failed to collect info from target process.\n";
+
+    hipFileStatsContext_t *context;
+    if (createContext(&context, pid) != hipFileStatsSuccess) {
+        std::cerr << "Failed to create context for target process.\n";
+        return 1;
+    }
+    if (connectToTargetProcess(context) != hipFileStatsSuccess) {
+        std::cerr << "Failed to connect to target process.\n";
+        closeContext(context);
         return 1;
     }
     if (!imm) {
-        client.pollProcess(-1);
+        if (pollTargetProcess(context, -1) != hipFileStatsSuccess) {
+            std::cerr << "Failed to poll target process.\n";
+            closeContext(context);
+            return 1;
+        }
+    }
+    int fd{dup(STDOUT_FILENO)};
+    if (fd < 0) {
+        std::cerr << "Failed to duplicate stdout file descriptor.\n";
+        closeContext(context);
+        return 1;
     }
-    if (!client.generateReport(std::cout)) {
-        std::cerr << "No stats could be collected from target process.\n";
+    if (generateReport(context, fd) != hipFileStatsSuccess) {
+        std::cerr << "Failed to generate report from target process.\n";
+        close(fd);
+        closeContext(context);
         return 1;
     }
+    close(fd);
+    closeContext(context);
     return 0;
 }