eric/duckdb-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use get_md_token to handle request

Browse Source
This commit is contained in:
Yves
2025-02-18 11:25:55 +01:00
parent d03b87dfce
commit e9d9c1ceac
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/http_server.cpp
View File

@@ -166,10 +166,24 @@ void HttpServer::HandleGetLocalEvents(const httplib::Request &req, httplib::Resp
} }
void HttpServer::HandleGetLocalToken(const httplib::Request &req, httplib::Response &res) { void HttpServer::HandleGetLocalToken(const httplib::Request &req, httplib::Response &res) {
throw new std::runtime_error("Not implemented"); // FIXME if (!ddb_instance_->ExtensionIsLoaded("motherduck")) {
// auto md_config = ClientExtension::GetState(*ddb_instance_).GetConfig(); res.status = 500;
// std::string md_token = md_config ? md_config->token : ""; res.set_content("MotherDuck extension is not loaded", "text/plain");
// res.set_content(md_token, "text/plain"); return;
}
Connection connection(*ddb_instance_);
auto query_res = connection.Query("CALL get_md_token()");
if (query_res->HasError()) {
res.status = 500;
res.set_content("Could not get token: " + query_res->GetError(), "text/plain");
return;
}
auto chunk = query_res->Fetch();
auto token = chunk->GetValue(0, 0).GetValue<std::string>();
res.status = 200;
res.set_content(token, "text/plain");
} }
void HttpServer::HandleGet(const httplib::Request &req, httplib::Response &res) { void HttpServer::HandleGet(const httplib::Request &req, httplib::Response &res) {