Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/grafana/xk6-sql

k6 extension to load test RDBMSs (PostgreSQL, MySQL, MS SQL and SQLite3)
https://github.com/grafana/xk6-sql

k6 sql xk6

Last synced: about 1 month ago
JSON representation

k6 extension to load test RDBMSs (PostgreSQL, MySQL, MS SQL and SQLite3)

Awesome Lists containing this project

README

        

# xk6-sql

This is a [k6](https://github.com/grafana/k6) extension using the
[xk6](https://github.com/grafana/xk6) system.

Supported RDBMSs: `mysql`, `postgres`, `sqlite3`, `sqlserver`, `azuresql`, `clickhouse`. See the [examples](examples)
directory for usage. Other RDBMSs are not supported, see
[details below](#support-for-other-rdbmss).

## Table of Contents
- [Build](#build)
- [Development](#development)
- [Example](#example)
- [TLS support](#tls-support)
- [Support for other RDBMSs](#support-for-other-rdbmss)
- [Docker](#docker)

## Build

To build a `k6` binary with this plugin, first ensure you have the prerequisites:

- [Go toolchain](https://go101.org/article/go-toolchain.html)
- If you're using SQLite, a build toolchain for your system that includes `gcc` or
another C compiler. On Debian and derivatives install the `build-essential`
package. On Windows you can use [tdm-gcc](https://jmeubank.github.io/tdm-gcc/).
Make sure that `gcc` is in your `PATH`.
- Git

Then:

1. Install `xk6`:
```shell
go install go.k6.io/xk6/cmd/xk6@latest
```

2. Build the binary:
```shell
xk6 build --with github.com/grafana/xk6-sql
```

If you're using SQLite, ensure you have a C compiler installed (see the
prerequisites note) and set `CGO_ENABLED=1` in the environment:
```shell
CGO_ENABLED=1 xk6 build --with github.com/grafana/xk6-sql
```

On Windows this is done slightly differently:
```shell
set CGO_ENABLED=1
xk6 build --with github.com/grafana/xk6-sql
```

## Development
To make development a little smoother, use the `Makefile` in the root folder. The default target will format your code, run tests, and create a `k6` binary with your local code rather than from GitHub.

```bash
make
```
Once built, you can run your newly extended `k6` using:
```shell
./k6 run examples/sqlite3_test.js
```

## Example

```javascript
// script.js
import sql from 'k6/x/sql';

const db = sql.open("sqlite3", "./test.db");

export function setup() {
db.exec(`CREATE TABLE IF NOT EXISTS keyvalues (
id integer PRIMARY KEY AUTOINCREMENT,
key varchar NOT NULL,
value varchar);`);
}

export function teardown() {
db.close();
}

export default function () {
db.exec("INSERT INTO keyvalues (key, value) VALUES('plugin-name', 'k6-plugin-sql');");

let results = sql.query(db, "SELECT * FROM keyvalues;");
for (const row of results) {
console.log(`key: ${row.key}, value: ${row.value}`);
}
}
```

Result output:

```shell
$ ./k6 run script.js

/\ |‾‾| /‾‾/ /‾‾/
/\ / \ | |/ / / /
/ \/ \ | ( / ‾‾\
/ \ | |\ \ | (‾) |
/ __________ \ |__| \__\ \_____/ .io

execution: local
script: /tmp/script.js
output: -

scenarios: (100.00%) 1 scenario, 1 max VUs, 10m30s max duration (incl. graceful stop):
* default: 1 iterations for each of 1 VUs (maxDuration: 10m0s, gracefulStop: 30s)

INFO[0000] key: plugin-name, value: k6-plugin-sql source=console

running (00m00.1s), 0/1 VUs, 1 complete and 0 interrupted iterations
default ✓ [======================================] 1 VUs 00m00.0s/10m0s 1/1 iters, 1 per VU

█ setup

█ teardown

data_received........: 0 B 0 B/s
data_sent............: 0 B 0 B/s
iteration_duration...: avg=9.22ms min=19.39µs med=8.86ms max=18.8ms p(90)=16.81ms p(95)=17.8ms
iterations...........: 1 15.292228/s
```

#### See also

- [Load Testing SQL Databases with k6](https://k6.io/blog/load-testing-sql-databases-with-k6/)

### TLS Support
Presently, TLS support is available only for the MySQL driver.

To enable TLS support, call `sql.loadTLS` from the script, before calling `sql.open`. [mysql_secure_test.js](examples/mysql_secure_test.js) is an example.

`sql.loadTLS` accepts the following options:

```javascript
sql.loadTLS({
enableTLS: true,
insecureSkipTLSverify: true,
minVersion: sql.TLS_1_2,
// Possible values: sql.TLS_1_0, sql.TLS_1_1, sql.TLS_1_2, sql.TLS_1_3
caCertFile: '/filepath/to/ca.pem',
clientCertFile: '/filepath/to/client-cert.pem',
clientKeyFile: '/filepath/to/client-key.pem',
});
```

### Support for other RDBMSs

Note that this project is not accepting support for additional SQL implementations
and RDBMSs. See the discussion in [issue #17](https://github.com/grafana/xk6-sql/issues/17)
for the reasoning.

There are however forks of this project that add additional support for:
- [Oracle](https://github.com/stefnedelchevbrady/xk6-sql-with-oracle)
- [Snowflake](https://github.com/libertymutual/xk6-sql)

You can build k6 binaries by simply specifying these project URLs in `xk6 build`.
E.g. `CGO_ENABLED=1 xk6 build --with github.com/stefnedelchevbrady/xk6-sql-with-oracle`.
Please report any issues with these extensions in their respective GitHub issue trackers,
and not in grafana/xk6-sql.

## Docker

For those who do not have a Go development environment available, or simply want
to run an extended version of `k6` as a container, Docker is an option to build
and run the application.

The following command will build a custom `k6` image incorporating the `xk6-sql` extension
built from the local source files.
```shell
docker build -t grafana/k6-for-sql:latest .
```
Using this image, you may then execute the [examples/sqlite3_test.js](examples/sqlite3_test.js) script
by running the following command:
```shell
docker run -v $PWD:/scripts -it --rm grafana/k6-for-sql:latest run /scripts/examples/sqlite3_test.js
```
For those on Mac or Linux, the `docker-run.sh` script simplifies the command:
```shell
./docker-run.sh examples/sqlite3_test.js
```