Update online-websocket-server.cc

Author: manickavela29

sherpa-onnx/csrc/online-websocket-server.cc

@@ -30,80 +30,92 @@ Please refer to
 for a list of pre-trained models to download.
 )";
 
-int32_t main(int32_t argc, char *argv[]) {
-  sherpa_onnx::ParseOptions po(kUsageMessage);
+class OnlineWebsocketServerApp {
+public:
+    OnlineWebsocketServerApp(int32_t argc, char *argv[]) : argc_(argc), argv_(argv) {}
+
+    void Run() {
+        sherpa_onnx::ParseOptions po(kUsageMessage);
+        sherpa_onnx::OnlineWebsocketServerConfig config;
 
-  sherpa_onnx::OnlineWebsocketServerConfig config;
+        // the server will listen on this port
+        int32_t port = 6006;
 
-  // the server will listen on this port
-  int32_t port = 6006;
+        // size of the thread pool for handling network connections
+        int32_t num_io_threads = 1;
 
-  // size of the thread pool for handling network connections
-  int32_t num_io_threads = 1;
+        // size of the thread pool for neural network computation and decoding
+        int32_t num_work_threads = 3;
 
-  // size of the thread pool for neural network computation and decoding
-  int32_t num_work_threads = 3;
+        po.Register("num-io-threads", &num_io_threads,
+                    "Thread pool size for network connections.");
 
-  po.Register("num-io-threads", &num_io_threads,
-              "Thread pool size for network connections.");
+        po.Register("num-work-threads", &num_work_threads,
+                    "Thread pool size for for neural network "
+                    "computation and decoding.");
 
-  po.Register("num-work-threads", &num_work_threads,
-              "Thread pool size for for neural network "
-              "computation and decoding.");
+        po.Register("port", &port, "The port on which the server will listen.");
 
-  po.Register("port", &port, "The port on which the server will listen.");
+        config.Register(&po);
 
-  config.Register(&po);
+        if (argc_ == 1) {
+            po.PrintUsage();
+            exit(EXIT_FAILURE);
+        }
 
-  if (argc == 1) {
-    po.PrintUsage();
-    exit(EXIT_FAILURE);
-  }
+        po.Read(argc_, argv_);
 
-  po.Read(argc, argv);
+        if (po.NumArgs() != 0) {
+            SHERPA_ONNX_LOGE("Unrecognized positional arguments!");
+            po.PrintUsage();
+            exit(EXIT_FAILURE);
+        }
 
-  if (po.NumArgs() != 0) {
-    SHERPA_ONNX_LOGE("Unrecognized positional arguments!");
-    po.PrintUsage();
-    exit(EXIT_FAILURE);
-  }
+        config.Validate();
 
-  config.Validate();
+        asio::io_context io_conn;  // for network connections
+        asio::io_context io_work;  // for neural network and decoding
 
-  asio::io_context io_conn;  // for network connections
-  asio::io_context io_work;  // for neural network and decoding
+        sherpa_onnx::OnlineWebsocketServer server(io_conn, io_work, config);
+        server.Run(port);
 
-  sherpa_onnx::OnlineWebsocketServer server(io_conn, io_work, config);
-  server.Run(port);
+        SHERPA_ONNX_LOGE("Started!");
+        SHERPA_ONNX_LOGE("Listening on: %d", port);
+        SHERPA_ONNX_LOGE("Number of work threads: %d", num_work_threads);
 
-  SHERPA_ONNX_LOGE("Started!");
-  SHERPA_ONNX_LOGE("Listening on: %d", port);
-  SHERPA_ONNX_LOGE("Number of work threads: %d", num_work_threads);
+        // give some work to do for the io_work pool
+        auto work_guard = asio::make_work_guard(io_work);
 
-  // give some work to do for the io_work pool
-  auto work_guard = asio::make_work_guard(io_work);
+        std::vector<std::thread> io_threads;
 
-  std::vector<std::thread> io_threads;
+        // decrement since the main thread is also used for network communications
+        for (int32_t i = 0; i < num_io_threads - 1; ++i) {
+            io_threads.emplace_back([&io_conn]() { io_conn.run(); });
+        }
 
-  // decrement since the main thread is also used for network communications
-  for (int32_t i = 0; i < num_io_threads - 1; ++i) {
-    io_threads.emplace_back([&io_conn]() { io_conn.run(); });
-  }
+        std::vector<std::thread> work_threads;
+        for (int32_t i = 0; i < num_work_threads; ++i) {
+            work_threads.emplace_back([&io_work]() { io_work.run(); });
+        }
 
-  std::vector<std::thread> work_threads;
-  for (int32_t i = 0; i < num_work_threads; ++i) {
-    work_threads.emplace_back([&io_work]() { io_work.run(); });
-  }
+        io_conn.run();
 
-  io_conn.run();
+        for (auto &t : io_threads) {
+            t.join();
+        }
 
-  for (auto &t : io_threads) {
-    t.join();
-  }
+        for (auto &t : work_threads) {
+            t.join();
+        }
+    }
 
-  for (auto &t : work_threads) {
-    t.join();
-  }
+private:
+    int32_t argc_;
+    char **argv_;
+};
 
-  return 0;
+int32_t main(int32_t argc, char *argv[]) {
+    OnlineWebsocketServerApp app(argc, argv);
+    app.Run();
+    return 0;
 }