Merge pull request #217 from fatalem0/fatalem0/add-slo-tests

feat: add slo tests

.github/workflows/rust-tests.yml

@@ -11,7 +11,7 @@ on:
 env:
   CARGO_TERM_COLOR: always
   CARGO_REGISTRIES_CRATES_IO_PROTOCOL: sparse
-  RUST_VERSION_OLD: "1.60.0"
+  RUST_VERSION_OLD: "1.68.0"
   RUST_VERSION_NEW: "1.81.0"
 
 jobs:

.github/workflows/slo.yml

@@ -0,0 +1,112 @@
+name: SLO tests
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+    branches: [ master ]
+  workflow_dispatch:
+    inputs:
+      github_pull_request_number:
+        required: true
+        description: ''
+      slo_workload_duration_seconds:
+        default: '600'
+        required: false
+        description: ''
+      slo_workload_read_max_rps:
+        default: '1000'
+        required: false
+        description: ''
+      slo_workload_write_max_rps:
+        default: '100'
+        required: false
+        description: ''
+
+env:
+  CARGO_TERM_COLOR: always
+  CARGO_REGISTRIES_CRATES_IO_PROTOCOL: sparse
+  RUST_VERSION_OLD: "1.68.0"
+  RUST_VERSION_NEW: "1.81.0"
+
+jobs:
+  ydb-slo-action-init:
+    if: (!contains(github.event.pull_request.labels.*.name, 'no slo'))
+
+    name: Run YDB SLO Tests
+    runs-on: ubuntu-latest
+
+    strategy:
+      matrix:
+        example:
+          - native
+        rust_version:
+          - "RUST_VERSION_OLD"
+          - "RUST_VERSION_NEW"
+
+    concurrency:
+      group: slo-${{ github.ref }}-${{ matrix.example }}-${{ matrix.rust_version }}
+      cancel-in-progress: true
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+        with:
+          submodules: true
+
+      - name: Install rust
+        uses: dtolnay/rust-toolchain@v1
+        with:
+          toolchain: ${{ env[matrix.rust_version] }}
+          components: clippy
+
+      - name: Rust version
+        id: rust_version_step
+        run: |
+          rustc --version
+          cargo --version
+          echo "CARGO_INCREMENTAL=$CARGO_INCREMENTAL"
+          echo "::set-output name=version::$(rustc --version | cut -d ' ' -f 2)"
+
+      - name: Rust cache
+        uses: Swatinem/rust-cache@v2
+
+      - name: Initialize YDB SLO
+        uses: ydb-platform/ydb-slo-action/init@main
+        with:
+          github_pull_request_number: ${{ github.event.inputs.github_pull_request_number }}
+          github_token: ${{ secrets.GITHUB_TOKEN }}
+          workload_name: ${{ matrix.example }}-${{ matrix.rust_version }}
+          ydb_database_node_count: 5
+
+      - name: Prepare SLO Database
+        run: |
+          cargo run --example ${{ matrix.example }} grpc://localhost:2135 /Root/testdb tableName create
+
+      - name: Run SLO Tests
+        env:
+          REF: '${{ github.head_ref || github.ref }}'
+        run: |
+          cargo run --example ${{ matrix.example }} grpc://localhost:2135 /Root/testdb tableName run \
+            --time ${{ inputs.slo_workload_duration_seconds || 600}} \
+            --read-rps ${{ inputs.slo_workload_read_max_rps || 1000}} \
+            --write-rps ${{ inputs.slo_workload_write_max_rps || 100}} \
+            --read-timeout 10000 \
+            --write-timeout 10000 || true
+
+      - if: always()
+        name: Cleanup SLO Database
+        run: |
+          cargo run --example ${{ matrix.example }} grpc://localhost:2135 /Root/testdb tableName cleanup
+
+      - if: always()
+        name: Store ydb chaos testing logs
+        run: |
+          docker logs ydb-chaos > chaos-ydb.log
+
+      - if: always()
+        uses: actions/upload-artifact@v4
+        with:
+          name: ${{ matrix.example}}-${{ matrix.rust_version }}-chaos-ydb.log
+          path: ./chaos-ydb.log
+          retention-days: 1