feat: enhance websocket connection and support stmt2 on Websocket

.codecov.yml

@@ -1,4 +1,5 @@
 ignore:
   - "bench"
   - "benchmark"
-  - "examples"
+  - "examples"
+  - "ws/unified/tests/cmd"

.github/actions/setup-tdengine-base/action.yml

@@ -0,0 +1,147 @@
+name: 'Setup TDengine Base'
+description: 'Base action for installing TDengine from local package (internal use)'
+
+inputs:
+  install-package:
+    description: 'Path to TDengine package or directory'
+    required: true
+  package-type:
+    description: 'Type of package: tar.gz or dir'
+    required: false
+    default: 'tar.gz'
+  start-services:
+    description: 'Whether to start taosd and taosadapter'
+    required: false
+    default: 'true'
+  config-files:
+    description: 'Comma-separated list of config files to copy'
+    required: false
+    default: 'taos.cfg,taosadapter.toml'
+  version:
+    description: 'TDengine version for logging'
+    required: false
+    default: 'unknown'
+  install-args:
+    description: 'Extra arguments passed to install.sh'
+    required: false
+    default: ''
+
+runs:
+  using: "composite"
+  steps:
+    - name: Install TDengine
+      shell: bash
+      run: |
+        set +o pipefail  # Ignore pipe errors (e.g., from head/tail closing pipes early)
+        echo "Installing TDengine ${{ inputs.version }}..."
+
+        case "${{ inputs.package-type }}" in
+          tar.gz)
+            if [ ! -f "${{ inputs.install-package }}" ]; then
+              echo "Error: Package file not found: ${{ inputs.install-package }}"
+              exit 1
+            fi
+            echo "Extracting tarball..."
+            tar -zxf "${{ inputs.install-package }}" 2>&1 | tail -20 || true
+
+            # Find extracted directory
+            echo "Looking for extracted directory..."
+            if [ -d "release" ]; then
+              cd "release"
+            else
+              # Try to find TDengine directory
+              for dir in TDengine-server-* tsdb-nightly-*; do
+                if [ -d "$dir" ]; then
+                  echo "Found directory: $dir"
+                  cd "$dir"
+                  break
+                fi
+              done
+            fi
+            ;;
+          dir)
+            if [ ! -d "${{ inputs.install-package }}" ]; then
+              echo "Error: Package directory not found: ${{ inputs.install-package }}"
+              exit 1
+            fi
+            cd "${{ inputs.install-package }}"
+            ;;
+          *)
+            echo "Error: Unknown package type: ${{ inputs.package-type }}"
+            exit 1
+            ;;
+        esac
+
+        echo "Running install.sh..."
+        if [ -x ./install.sh ] && head -n 1 ./install.sh | grep -q '^#!'; then
+          install_cmd=(sudo ./install.sh)
+        else
+          install_cmd=(sudo sh ./install.sh)
+        fi
+
+        if [ -n "${{ inputs.install-args }}" ]; then
+          # shellcheck disable=SC2206
+          install_args=(${{ inputs.install-args }})
+          "${install_cmd[@]}" "${install_args[@]}"
+        else
+          "${install_cmd[@]}"
+        fi
+        echo "TDengine installed successfully"
+
+    - name: Configure TDengine
+      if: inputs.config-files != ''
+      shell: bash
+      run: |
+        echo "Configuring TDengine..."
+        sudo mkdir -p /etc/taos
+        IFS=',' read -ra configs < <(echo "${{ inputs.config-files }}")
+        for config in "${configs[@]}"; do
+          config_file=".github/workflows/$config"
+          if [ -f "$config_file" ]; then
+            echo "Copying $config..."
+            sudo cp "$config_file" "/etc/taos/$config"
+          else
+            echo "Warning: Config file not found: $config_file"
+          fi
+        done
+        echo "Configuration complete"
+
+    - name: Start TDengine services
+      if: inputs.start-services == 'true'
+      shell: bash
+      run: |
+        echo "Starting TDengine services..."
+
+        # Create start script for taosd
+        cat > start.sh << 'EOF'
+        ulimit -n 65535 && TAOS_SUPPORT_VNODES=256 taosd
+        EOF
+        chmod +x start.sh
+
+        # Start taosd
+        echo "Starting taosd..."
+        nohup sudo sh ./start.sh >taosd.out 2>&1 &
+        sleep 2
+
+        # Start taosadapter
+        echo "Starting taosadapter..."
+        sudo taosadapter >taosadapter.out 2>&1 &
+        sleep 2
+
+        echo "Services started"
+
+    - name: Wait for services ready
+      if: inputs.start-services == 'true'
+      shell: bash
+      run: |
+        echo "Waiting for taosadapter..."
+        for i in $(seq 1 30); do
+          if curl -sf "http://127.0.0.1:6041/-/ping" >/dev/null 2>&1; then
+            echo "taosadapter is ready"
+            exit 0
+          fi
+          sleep 1
+        done
+        echo "Warning: taosadapter may not be ready"
+        echo "Last 20 lines of taosd.log:"
+        tail -20 taosd.out 2>/dev/null || true

.github/actions/setup-tdengine/action.yml

@@ -0,0 +1,149 @@
+name: 'Setup TDengine'
+description: 'Universal TDengine setup action supporting multiple modes'
+
+inputs:
+  mode:
+    description: 'Setup mode: nightly, build, or fixed'
+    required: true
+    default: 'nightly'
+  branch:
+    description: 'TDengine branch (for nightly mode)'
+    required: false
+    default: 'main'
+  commit-id:
+    description: 'TDengine commit ID (for build mode)'
+    required: false
+    default: ''
+  event-key:
+    description: 'Event key for cache (for build mode)'
+    required: false
+    default: ''
+  package-path:
+    description: 'Path to TDengine package (for fixed mode)'
+    required: false
+    default: ''
+  version:
+    description: 'Version string for logging'
+    required: false
+    default: 'unknown'
+  install-args:
+    description: 'Extra arguments passed to install.sh'
+    required: false
+    default: ''
+  start-services:
+    description: 'Whether to start taosd and taosadapter'
+    required: false
+    default: 'true'
+  config-files:
+    description: 'Comma-separated list of config files'
+    required: false
+    default: 'taos.cfg,taosadapter.toml'
+  nightly-url:
+    description: 'Base URL for nightly builds (required for nightly mode)'
+    required: false
+    default: ''
+
+runs:
+  using: "composite"
+  steps:
+    - name: Validate inputs
+      shell: bash
+      run: |
+        set +o pipefail  # Ignore pipe errors
+        case "${{ inputs.mode }}" in
+          nightly)
+            if [ -z "${{ inputs.branch }}" ] || [ -z "${{ inputs.nightly-url }}" ]; then
+              echo "Error: branch and nightly-url are required for nightly mode"
+              exit 1
+            fi
+            ;;
+          build)
+            if [ -z "${{ inputs.commit-id }}" ] || [ -z "${{ inputs.event-key }}" ]; then
+              echo "Error: commit-id and event-key are required for build mode"
+              exit 1
+            fi
+            ;;
+          fixed)
+            if [ -z "${{ inputs.package-path }}" ]; then
+              echo "Error: package-path is required for fixed mode"
+              exit 1
+            fi
+            ;;
+          *)
+            echo "Error: mode must be one of [nightly, build, fixed]"
+            exit 1
+            ;;
+        esac
+
+    - name: Setup nightly build
+      if: inputs.mode == 'nightly'
+      id: setup-nightly
+      shell: bash
+      run: |
+        echo "Downloading TDengine nightly for branch: ${{ inputs.branch }}"
+        set -o pipefail
+        if ! wget "${{ inputs.nightly-url }}/tsdb-nightly-${{ inputs.branch }}.tar.gz?v=$(date +%s)" \
+          -O "tdengine-nightly.tar.gz" 2>&1 | tail -n5; then
+          echo "Error: failed to download nightly package"
+          exit 1
+        fi
+        echo "package_path=tdengine-nightly.tar.gz" >> $GITHUB_OUTPUT
+        echo "package_type=tar.gz" >> $GITHUB_OUTPUT
+
+    - name: Restore build cache
+      if: inputs.mode == 'build'
+      id: cache-server
+      uses: actions/cache@v5
+      with:
+        path: server.tar.gz
+        key: ${{ runner.os }}-build-${{ inputs.event-key }}-${{ inputs.commit-id }}
+
+    - name: Setup build mode paths
+      if: inputs.mode == 'build'
+      id: setup-build
+      shell: bash
+      run: |
+        echo "package_path=server.tar.gz" >> $GITHUB_OUTPUT
+        echo "package_type=tar.gz" >> $GITHUB_OUTPUT
+
+    - name: Setup fixed mode paths
+      if: inputs.mode == 'fixed'
+      id: setup-fixed
+      shell: bash
+      run: |
+        package_path="${{ inputs.package-path }}"
+        if [ -d "$package_path" ]; then
+          package_type="dir"
+        elif [ -f "$package_path" ]; then
+          package_type="tar.gz"
+        else
+          echo "Error: package-path not found or inaccessible: $package_path"
+          exit 1
+        fi
+        echo "package_path=$package_path" >> $GITHUB_OUTPUT
+        echo "package_type=$package_type" >> $GITHUB_OUTPUT
+
+    - name: Determine package info
+      id: package-info
+      shell: bash
+      run: |
+        if [ "${{ inputs.mode }}" == "nightly" ]; then
+          echo "path=${{ steps.setup-nightly.outputs.package_path }}" >> $GITHUB_OUTPUT
+          echo "type=${{ steps.setup-nightly.outputs.package_type }}" >> $GITHUB_OUTPUT
+        elif [ "${{ inputs.mode }}" == "build" ]; then
+          echo "path=${{ steps.setup-build.outputs.package_path }}" >> $GITHUB_OUTPUT
+          echo "type=${{ steps.setup-build.outputs.package_type }}" >> $GITHUB_OUTPUT
+        else
+          echo "path=${{ steps.setup-fixed.outputs.package_path }}" >> $GITHUB_OUTPUT
+          echo "type=${{ steps.setup-fixed.outputs.package_type }}" >> $GITHUB_OUTPUT
+        fi
+
+    - name: Install and setup TDengine
+      uses: ./.github/actions/setup-tdengine-base
+      with:
+        install-package: ${{ steps.package-info.outputs.path }}
+        package-type: ${{ steps.package-info.outputs.type }}
+        install-args: ${{ inputs.install-args }}
+        start-services: ${{ inputs.start-services }}
+        config-files: ${{ inputs.config-files }}
+        version: ${{ inputs.version }}