Fix session usage

Author: auxten

lib/README.md

@@ -0,0 +1,11 @@
+## libchdb `Query` function binding
+
+```bash
+npm install
+
+# install dependencies
+./update_libchdb.sh
+
+# build the dynamic library
+node-gyp build
+```

lib/chdb.h

@@ -0,0 +1,28 @@
+#pragma once
+
+#ifdef __cplusplus
+#    include <cstddef>
+#    include <cstdint>
+extern "C" {
+#else
+#    include <stddef.h>
+#    include <stdint.h>
+#endif
+
+#define CHDB_EXPORT __attribute__((visibility("default")))
+struct CHDB_EXPORT local_result
+{
+    char * buf;
+    size_t len;
+    void * _vec; // std::vector<char> *, for freeing
+    double elapsed;
+    uint64_t rows_read;
+    uint64_t bytes_read;
+};
+
+CHDB_EXPORT struct local_result * query_stable(int argc, char ** argv);
+CHDB_EXPORT void free_result(struct local_result * result);
+
+#ifdef __cplusplus
+}
+#endif

lib/chdb_node.cpp

@@ -0,0 +1,137 @@
+#include "chdb.h"
+#include "chdb_node.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <iostream>
+#include <napi.h>
+
+#define MAX_FORMAT_LENGTH 64
+#define MAX_PATH_LENGTH 4096
+#define MAX_ARG_COUNT 6
+
+// Utility function to construct argument string
+void construct_arg(char *dest, const char *prefix, const char *value,
+                   size_t dest_size) {
+  snprintf(dest, dest_size, "%s%s", prefix, value);
+}
+
+// Generalized query function
+char *general_query(int argc, char *args[]) {
+  struct local_result *result = query_stable(argc, args);
+
+  if (result == NULL) {
+    return NULL;
+  } else {
+    return result->buf;
+  }
+}
+
+// Query function without session
+char *Query(const char *query, const char *format) {
+  char dataFormat[MAX_FORMAT_LENGTH];
+  char *dataQuery;
+  char *args[MAX_ARG_COUNT] = {"clickhouse", "--multiquery", NULL, NULL};
+  int argc = 4;
+
+  construct_arg(dataFormat, "--output-format=", format, MAX_FORMAT_LENGTH);
+  args[2] = dataFormat;
+
+  dataQuery = (char *)malloc(strlen(query) + strlen("--query=") + 1);
+  if (dataQuery == NULL) {
+    return NULL;
+  }
+  construct_arg(dataQuery, "--query=", query,
+                strlen(query) + strlen("--query=") + 1);
+  args[3] = dataQuery;
+
+  char *result = general_query(argc, args);
+  free(dataQuery);
+  return result;
+}
+
+// QuerySession function will save the session to the path
+// queries with same path will use the same session
+char *QuerySession(const char *query, const char *format, const char *path) {
+  char dataFormat[MAX_FORMAT_LENGTH];
+  char dataPath[MAX_PATH_LENGTH];
+  char *dataQuery;
+  char *args[MAX_ARG_COUNT] = {"clickhouse", "--multiquery", NULL, NULL, NULL};
+  int argc = 5;
+
+  construct_arg(dataFormat, "--output-format=", format, MAX_FORMAT_LENGTH);
+  args[2] = dataFormat;
+
+  dataQuery = (char *)malloc(strlen(query) + strlen("--query=") + 1);
+  if (dataQuery == NULL) {
+    return NULL;
+  }
+  construct_arg(dataQuery, "--query=", query,
+                strlen(query) + strlen("--query=") + 1);
+  args[3] = dataQuery;
+
+  construct_arg(dataPath, "--path=", path, MAX_PATH_LENGTH);
+  args[4] = dataPath;
+
+  char *result = general_query(argc, args);
+  free(dataQuery);
+  return result;
+}
+
+Napi::String QueryWrapper(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+
+  // Check argument types and count
+  if (info.Length() < 2 || !info[0].IsString() || !info[1].IsString()) {
+    Napi::TypeError::New(env, "String expected").ThrowAsJavaScriptException();
+    return Napi::String::New(env, "");
+  }
+
+  // Get the arguments
+  std::string query = info[0].As<Napi::String>().Utf8Value();
+  std::string format = info[1].As<Napi::String>().Utf8Value();
+
+  // Call the native function
+  char *result = Query(query.c_str(), format.c_str());
+
+  // Return the result
+  return Napi::String::New(env, result);
+}
+
+Napi::String QuerySessionWrapper(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+
+  // Check argument types and count
+  if (info.Length() < 3 || !info[0].IsString() || !info[1].IsString() ||
+      !info[2].IsString()) {
+    Napi::TypeError::New(env, "String expected").ThrowAsJavaScriptException();
+    return Napi::String::New(env, "");
+  }
+
+  // Get the arguments
+  std::string query = info[0].As<Napi::String>().Utf8Value();
+  std::string format = info[1].As<Napi::String>().Utf8Value();
+  std::string path = info[2].As<Napi::String>().Utf8Value();
+
+  // std::cerr << query << std::endl;
+  // std::cerr << format << std::endl;
+  // std::cerr << path << std::endl;
+  // Call the native function
+  char *result = QuerySession(query.c_str(), format.c_str(), path.c_str());
+  if (result == NULL) {
+    // std::cerr << "result is null" << std::endl;
+    return Napi::String::New(env, "");
+  }
+
+  // Return the result
+  return Napi::String::New(env, result);
+}
+
+Napi::Object Init(Napi::Env env, Napi::Object exports) {
+  // Export the functions
+  exports.Set("Query", Napi::Function::New(env, QueryWrapper));
+  exports.Set("QuerySession", Napi::Function::New(env, QuerySessionWrapper));
+  return exports;
+}
+
+NODE_API_MODULE(NODE_GYP_MODULE_NAME, Init)

lib/chdb_node.h

@@ -0,0 +1,4 @@
+#pragma once
+
+char *Query(const char *query, const char *format);
+char *QuerySession(const char *query, const char *format, const char *path);

lib/update_libchdb.sh

@@ -0,0 +1,43 @@
+
+#!/bin/bash
+
+# Get the newest release version
+LATEST_RELEASE=$(curl --silent "https://api.github.com/repos/chdb-io/chdb/releases/latest" | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
+
+# Download the correct version based on the platform
+case "$(uname -s)" in
+    Linux)
+        if [[ $(uname -m) == "aarch64" ]]; then
+            PLATFORM="linux-aarch64-libchdb.tar.gz"
+        else
+            PLATFORM="linux-x86_64-libchdb.tar.gz"
+        fi
+        ;;
+    Darwin)
+        if [[ $(uname -m) == "arm64" ]]; then
+            PLATFORM="macos-arm64-libchdb.tar.gz"
+        else
+            PLATFORM="macos-x86_64-libchdb.tar.gz"
+        fi
+        ;;
+    *)
+        echo "Unsupported platform"
+        exit 1
+        ;;
+esac
+
+DOWNLOAD_URL="https://github.com/chdb-io/chdb/releases/download/$LATEST_RELEASE/$PLATFORM"
+
+echo "Downloading $PLATFORM from $DOWNLOAD_URL"
+
+# Download the file
+curl -L -o libchdb.tar.gz $DOWNLOAD_URL
+
+# Untar the file
+tar -xzf libchdb.tar.gz
+
+# Set execute permission for libchdb.so
+chmod +x libchdb.so
+
+# Clean up
+rm -f libchdb.tar.gz