Fixed scripts

Author: GrigoriyPA

ydb/tests/tools/fqrun/.gitignore

@@ -5,3 +5,4 @@ sync_dir
 *.conf
 *.parquet
 *.json
+*.yaml

ydb/tests/tools/kqprun/README.md

@@ -4,6 +4,23 @@ Tool can be used to execute queries by using kikimr provider.
 
 For profiling memory allocations build kqprun with ya make flag `-D PROFILE_MEMORY_ALLOCATIONS`.
 
+## Scripts
+
+* `flame_graph.sh` - script for collecting flame graphs in svg format, usage:
+    ```(bash)
+    ./flame_graph.sh [graph collection time in seconds]
+    ```
+
+* `start_prometheus.sh` - script for starting prometheus web UI, can be used for counters visualisation (kqprun should be runned with flag `-M <monitoring port>`), usage:
+    ```(bash)
+    ./start_prometheus.sh <monitoring port> <web UI port> [config path]
+    ```
+
+* `start_connector.sh` - script for starting local FQ connector, usage:
+    ```(bash)
+    ./start_connector.sh <connector port> [config path]
+    ```
+
 ## Examples
 
 ### Queries

ydb/tests/tools/kqprun/flame_graph.sh

@@ -2,6 +2,12 @@
 
 set -eux
 
+function cleanup {
+    sudo rm ./profdata
+    rm ./profdata.txt
+}
+trap cleanup EXIT
+
 if [ $# -gt 1 ]; then
     echo "Too many arguments"
     exit -1
@@ -12,6 +18,8 @@ kqprun_pid=$(pgrep -u $USER kqprun)
 sudo perf record -F 50 --call-graph dwarf -g --proc-map-timeout=10000 --pid $kqprun_pid -v -o profdata -- sleep ${1:-'30'}
 sudo perf script -i profdata > profdata.txt
 
-flame_graph_tool="../../../../contrib/tools/flame-graph/"
+SCRIPT_DIR=$(cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd)
+
+flame_graph_tool="$SCRIPT_DIR/../../../../contrib/tools/flame-graph/"
 
 ${flame_graph_tool}/stackcollapse-perf.pl profdata.txt | ${flame_graph_tool}/flamegraph.pl > profdata.svg

ydb/tests/tools/kqprun/start_connector.sh

@@ -2,12 +2,17 @@
 
 set -eux
 
+function cleanup {
+    rm ./connector_config.yaml
+}
+trap cleanup EXIT
+
 if [ $# -le 0 ]; then
     echo "Please provide connector port (value of Generic::Connector::Endpoint::Port in config)"
     exit -1
 fi
 
-if [ $# -gt 1 ]; then
+if [ $# -gt 2 ]; then
     echo "Too many arguments"
     exit -1
 fi
@@ -17,7 +22,9 @@ if netstat -tuln | grep ":$1"; then
     exit -1
 fi
 
-cp ${3:-'./configuration/connector_config.yaml'} ./connector_config.yaml
+SCRIPT_DIR=$(cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd)
+
+cp ${2:-"$SCRIPT_DIR/configuration/connector_config.yaml"} ./connector_config.yaml
 sed -i "s/\${TARGET_PORT}/$1/g" ./connector_config.yaml
 
 docker run --rm --name=fq-connector-go-$1 --network host -v ./connector_config.yaml:/opt/ydb/cfg/fq-connector-go.yaml ghcr.io/ydb-platform/fq-connector-go:latest

ydb/tests/tools/kqprun/start_prometheus.sh

@@ -2,6 +2,11 @@
 
 set -eux
 
+function cleanup {
+    rm ./prometheus_config.yaml
+}
+trap cleanup EXIT
+
 if [ $# -le 0 ]; then
     echo "Please provide monitoring port to listen (value of -M argument)"
     exit -1
@@ -25,7 +30,9 @@ if netstat -tuln | grep ":$2"; then
     exit -1
 fi
 
-cp ${3:-'./configuration/prometheus_config.yaml'} ./prometheus_config.yaml
+SCRIPT_DIR=$(cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd)
+
+cp ${3:-"$SCRIPT_DIR/configuration/prometheus_config.yaml"} ./prometheus_config.yaml
 sed -i "s/\${TARGET_PORT}/$1/g" ./prometheus_config.yaml
 
 docker run --rm --name=prometeus-$1 --network host -v ./prometheus_config.yaml:/etc/prometheus/prometheus.yml prom/prometheus --config.file=/etc/prometheus/prometheus.yml --web.listen-address=:$2