add new handler

Signed-off-by: vsian <[email protected]>
infiniflow · Oct 11, 2024 · e1b2da9 · e1b2da9
1 parent d0d683a
commit e1b2da9
Showing 1 changed file with 113 additions and 0 deletions.
diff --git a/src/network/http_server.cpp b/src/network/http_server.cpp
@@ -3590,6 +3590,116 @@ class ShowFullCheckpointHandler final : public HttpRequestHandler {
     }
 };
 
+
+class ShowQueryHandler final : public HttpRequestHandler {
+public:
+SharedPtr<OutgoingResponse> handle(const SharedPtr<IncomingRequest> &request) final {
+        auto infinity = Infinity::RemoteConnect();
+        DeferFn defer_fn([&]() { infinity->RemoteDisconnect(); });
+
+        nlohmann::json json_response;
+        nlohmann::json json_table;
+        HTTPStatus http_status;
+        String query_id = request->getPathVariable("query_id");
+        QueryResult result = infinity->Query(fmt::format("show query {}", query_id)); 
+
+        if (result.IsOk()) {
+            DataBlock *data_block = result.result_table_->GetDataBlockById(0).get();
+            auto column_cnt = result.result_table_->ColumnCount();
+            for (SizeT col = 0; col < column_cnt; ++col) {
+                const String &column_name = result.result_table_->GetColumnNameById(col);
+                Value value = data_block->GetValue(col, 0);
+                const String &column_value = value.ToString();
+                json_table[column_name] = column_value;
+            }
+            json_response["query"] = json_table;
+            json_response["error_code"] = 0;
+            http_status = HTTPStatus::CODE_200;
+        } else {
+            json_response["error_code"] = result.ErrorCode();
+            json_response["error_message"] = result.ErrorMsg();
+            http_status = HTTPStatus::CODE_500;
+        }
+
+        return ResponseFactory::createResponse(http_status, json_response.dump());
+    }
+};
+
+class ShowTransactionsHandler final : public HttpRequestHandler {
+public:
+SharedPtr<OutgoingResponse> handle(const SharedPtr<IncomingRequest> &request) final {
+        auto infinity = Infinity::RemoteConnect();
+        DeferFn defer_fn([&]() { infinity->RemoteDisconnect(); });
+
+        nlohmann::json json_response;
+        HTTPStatus http_status;
+        QueryResult result = infinity->Query("show transactions");
+
+        if (result.IsOk()) {
+            SizeT block_rows = result.result_table_->DataBlockCount();
+            for (SizeT block_id = 0; block_id < block_rows; ++block_id) {
+                DataBlock *data_block = result.result_table_->GetDataBlockById(block_id).get();
+                auto row_count = data_block->row_count();
+                auto column_cnt = result.result_table_->ColumnCount();
+                for (int row = 0; row < row_count; ++row) {
+                    nlohmann::json json_table;
+                    for (SizeT col = 0; col < column_cnt; ++col) {
+                        const String &column_name = result.result_table_->GetColumnNameById(col);
+                        Value value = data_block->GetValue(col, row);
+                        const String &column_value = value.ToString();
+                        json_table[column_name] = column_value;
+                    }
+                    json_response["transactions"].push_back(json_table);
+                }
+            }
+            json_response["error_code"] = 0;
+            http_status = HTTPStatus::CODE_200;
+        } else {
+            json_response["error_code"] = result.ErrorCode();
+            json_response["error_message"] = result.ErrorMsg();
+            http_status = HTTPStatus::CODE_500;
+        }
+
+        return ResponseFactory::createResponse(http_status, json_response.dump());
+    }
+};
+
+class ShowTransactionHandler final : public HttpRequestHandler {
+public:
+SharedPtr<OutgoingResponse> handle(const SharedPtr<IncomingRequest> &request) final {
+        auto infinity = Infinity::RemoteConnect();
+        DeferFn defer_fn([&]() { infinity->RemoteDisconnect(); });
+
+        nlohmann::json json_response;
+        nlohmann::json json_table;
+        HTTPStatus http_status;
+        String transaction_id = request->getPathVariable("transaction_id");
+        QueryResult result = infinity->Query(fmt::format("show transaction {}", transaction_id));
+
+        if (result.IsOk()) {
+            DataBlock *data_block = result.result_table_->GetDataBlockById(0).get();
+            auto column_cnt = result.result_table_->ColumnCount();
+            for (SizeT col = 0; col < column_cnt; ++col) {
+                const String &column_name = result.result_table_->GetColumnNameById(col);
+                Value value = data_block->GetValue(col, 0);
+                const String &column_value = value.ToString();
+                json_table[column_name] = column_value;
+            }
+            json_response["error_code"] = 0;
+            json_response["transaction"]= json_table;
+            http_status = HTTPStatus::CODE_200;
+        } else {
+            json_response["error_code"] = result.ErrorCode();
+            json_response["error_message"] = result.ErrorMsg();
+            http_status = HTTPStatus::CODE_500;
+        }
+
+        return ResponseFactory::createResponse(http_status, json_response.dump());
+    }
+};
+
+
+
 class AdminShowCurrentNodeHandler final : public HttpRequestHandler {
 public:
     SharedPtr<OutgoingResponse> handle(const SharedPtr<IncomingRequest> &request) final {
@@ -3926,6 +4036,9 @@ void HTTPServer::Start(const String &ip_address, u16 port) {
     router->route("GET", "/instance/logs", MakeShared<ShowLogsHandler>());
     router->route("GET", "/instance/delta_checkpoint", MakeShared<ShowDeltaCheckpointHandler>());
     router->route("GET", "/instance/global_checkpoint", MakeShared<ShowFullCheckpointHandler>()); 
+    router->route("GET", "/instance/queries/{query_id}", MakeShared<ShowQueryHandler>());
+    router->route("GET", "/instance/transactions", MakeShared<ShowTransactionsHandler>());
+    router->route("GET", "/instance/transactions/{transaction_id}", MakeShared<ShowTransactionHandler>());
 
     // variable
     router->route("GET", "/variables/global", MakeShared<ShowGlobalVariablesHandler>());