feat(sys): add taos wrapper [TD-34596]

.github/workflows/build.yml

@@ -44,14 +44,14 @@ jobs:
           path: target
           key: ${{ runner.os }}-taos-target-${{ steps.setup-rust.outputs.cachekey }}
 
+      - name: Run Clippy
+        run: cargo clippy -- -D warnings
+
       - name: Cargo Build
         run: |
           cargo build -p taos --no-default-features --features ws-rustls
           cargo build -p taos-ws-sys
 
-      - name: Run Clippy
-        run: cargo clippy -- -D warnings
-
   test:
     runs-on: ubuntu-latest
     steps:

.github/workflows/coverage.yml

@@ -151,8 +151,8 @@ jobs:
         with:
           toolchain: 1.78.0
 
-      - name: Install cargo-llvm-cov
-        uses: taiki-e/install-action@cargo-llvm-cov
+      - uses: taiki-e/install-action@cargo-llvm-cov
+      - uses: taiki-e/install-action@cargo-nextest
 
       - uses: actions/cache@v4
         with:
@@ -188,7 +188,91 @@ jobs:
           sleep 1
 
       - name: Generate code coverage
-        run: RUST_BACKTRACE=1 cargo llvm-cov -j 2 --workspace --lcov --output-path lcov.info
+        run: RUST_BACKTRACE=1 cargo llvm-cov nextest -j 2 --workspace --lcov --output-path lcov.info --no-fail-fast
+
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: lcov.info
+          fail_ci_if_error: true
+          verbose: true
+
+  coverage-sys-native:
+    runs-on: ubuntu-latest
+    needs: build-tdengine
+    steps:
+      - name: Get cache server by push
+        if: github.event_name == 'push'
+        id: get-cache-server-push
+        uses: actions/cache@v4
+        with:
+          path: server.tar.gz
+          key: ${{ runner.os }}-build-main-${{ needs.build-tdengine.outputs.commit_id }}
+          restore-keys: ${{ runner.os }}-build-main-
+
+      - name: Get cache server manually
+        if: github.event_name == 'workflow_dispatch'
+        id: get-cache-server-manually
+        uses: actions/cache@v4
+        with:
+          path: server.tar.gz
+          key: ${{ runner.os }}-build-${{ inputs.tdengine_branch }}-${{ needs.build-tdengine.outputs.commit_id }}
+          restore-keys: ${{ runner.os }}-build-${{ inputs.tdengine_branch }}-
+
+      - name: Install TDengine
+        run: |
+          tar -zxvf server.tar.gz
+          cd release && sudo sh install.sh
+
+      - uses: actions/checkout@v4
+
+      - name: Set up Rust
+        id: setup-rust
+        uses: dtolnay/rust-toolchain@stable
+        with:
+          toolchain: 1.78.0
+
+      - uses: taiki-e/install-action@cargo-llvm-cov
+      - uses: taiki-e/install-action@cargo-nextest
+
+      - uses: actions/cache@v4
+        with:
+          path: |
+            ~/.cargo/bin/
+            ~/.cargo/registry/index/
+            ~/.cargo/registry/cache/
+            ~/.cargo/git/db/
+          key: ${{ runner.os }}-cargo-${{ steps.setup-rust.outputs.rustc_hash }}
+
+      - uses: actions/cache@v4
+        with:
+          path: target
+          key: ${{ runner.os }}-taos-target-${{ steps.setup-rust.outputs.rustc_hash }}
+
+      - name: Prepare development tools
+        run: |
+          sudo apt update
+          sudo apt install libjansson-dev libsnappy-dev liblzma-dev libz-dev pkg-config libgflags2.2 libgflags-dev libgeos-dev -y
+          sudo timedatectl set-timezone Asia/Shanghai
+          sudo date
+
+      - name: Copy taos.cfg
+        run: |
+          sudo mkdir -p /etc/taos
+          sudo cp ./.github/workflows/taos.cfg /etc/taos/taos.cfg
+
+      - name: Start TDengine
+        run: |
+          mkdir -p /tmp/taos/v3/log /tmp/taos/v3/data
+          taosd &
+          TAOS_ADAPTER_LOG_PATH=/tmp/taos/v3/log taosadapter &
+          sleep 1
+
+      - name: Generate code coverage
+        run: |
+          export TAOS_DRIVER=native
+          RUST_BACKTRACE=1 cargo llvm-cov nextest -j 2 --workspace --lcov --output-path lcov.info --no-fail-fast
 
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v5

taos-optin/tests/query.rs

@@ -13,7 +13,7 @@ fn ws_sync_json() -> anyhow::Result<()> {
     let dsn = std::env::var("TEST_DSN").unwrap_or("taos://localhost:6030".to_string());
     let dsn = Dsn::from_str(&dsn)?;
     let client = TaosBuilder::from_dsn(dsn)?.build()?;
-    let db = "ws_sync_json";
+    let db = "ws_sync_json_1745828338";
     assert_eq!(client.exec(format!("drop database if exists {db}"))?, 0);
     assert_eq!(client.exec(format!("create database {db} keep 36500"))?, 0);
     assert_eq!(

taos-optin/tests/query_with_req_id.rs

@@ -10,7 +10,7 @@ fn sync_json_with_req_id() -> anyhow::Result<()> {
     let dsn = std::env::var("TEST_DSN").unwrap_or("taos://localhost:6030".to_string());
     let dsn = Dsn::from_str(&dsn)?;
     let client = TaosBuilder::from_dsn(dsn)?.build()?;
-    let db = "ws_sync_json";
+    let db = "ws_sync_json_1745828307";
     assert_eq!(client.exec(format!("drop database if exists {db}"))?, 0);
     assert_eq!(client.exec(format!("create database {db} keep 36500"))?, 0);
     assert_eq!(

taos-query/src/common/raw/views/mod.rs

@@ -602,7 +602,6 @@ impl ColumnView {
     }
 
     /// Get pointer to value.
-    /// FIXME: for Decimal/Decimal64 type, we can not get complete data from data ptr, we also need precision/scale in schema
     #[inline]
     pub(super) unsafe fn get_raw_value_unchecked(&self, row: usize) -> (Ty, u32, *const c_void) {
         match self {

taos-query/src/util/inline_str.rs

@@ -88,17 +88,15 @@ macro_rules! _impl_inline_str {
                 }
 
                 #[inline]
-                #[rustversion::attr(nightly, const)]
-                pub fn as_bytes(&self) -> &[u8] {
+                pub const fn as_bytes(&self) -> &[u8] {
                     if self.len() == 0 {
                         return &[]
                     }
                     unsafe { std::slice::from_raw_parts(self.data.as_ptr(), self.len()) }
                 }
 
                 #[inline]
-                #[rustversion::attr(nightly, const)]
-                pub fn as_str(&self) -> &str {
+                pub const fn as_str(&self) -> &str {
                     unsafe { std::str::from_utf8_unchecked(self.as_bytes()) }
                 }
 

taos-ws-sys/Cargo.toml

@@ -5,6 +5,7 @@ edition = "2021"
 publish = false
 
 [lib]
+# FIXME: rename taos
 name = "taosws"
 crate-type = ["cdylib", "staticlib"]
 

taos-ws-sys/build.rs

@@ -6,7 +6,7 @@ use syn::visit::Visit;
 use syn::{ItemConst, ItemFn, Visibility};
 
 fn main() -> Result<(), Box<dyn Error>> {
-    write_header("taos.h", "src/ws", "cbindgen_ws.toml")?;
+    write_header("taos.h", "src/taos", "cbindgen.toml")?;
     write_header("taosws.h", "src/old_ws", "cbindgen_old_ws.toml")?;
     Ok(())
 }

taos-ws-sys/cbindgen_ws.toml → taos-ws-sys/cbindgen.toml

@@ -1,6 +1,5 @@
 language = "C"
 cpp_compat = true
-include_guard = "TDENGINE_TAOS_H"
 header = """
 /*
  * Copyright (c) 2024 TAOS Data, Inc. <[email protected]>
@@ -16,3 +15,15 @@ header = """
  * You should have received a copy of the GNU Affero General Public License
  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  */"""
+
+include_guard = "TDENGINE_TAOS_H"
+after_includes = """
+
+#if defined(_WIN32) || defined(_WIN64)
+#define DLL_EXPORT __declspec(dllexport)
+#else
+#define DLL_EXPORT
+#endif"""
+
+[fn]
+prefix = "DLL_EXPORT"

taos-ws-sys/src/lib.rs

@@ -1,4 +1,6 @@
 #![feature(c_variadic)]
 
+pub mod native;
 pub mod old_ws;
+pub mod taos;
 pub mod ws;